aboutsummaryrefslogtreecommitdiff
path: root/tests/test_dl.py
diff options
context:
space:
mode:
authorCody Logan <cody@lokken.dev>2023-11-01 13:11:35 -0700
committerCody Logan <cody@lokken.dev>2023-11-01 13:11:35 -0700
commitbf7d8c97620471fed7e9276462984b1f33b512d0 (patch)
tree006e0a99b4e9c06028e9192cf6da33a68b59329f /tests/test_dl.py
parent235b3e6a723e3e18962212c7d2c0f19619c2fa6f (diff)
downloadwikiget-bf7d8c97620471fed7e9276462984b1f33b512d0.tar.gz
wikiget-bf7d8c97620471fed7e9276462984b1f33b512d0.zip
Condense parser construction logic
Diffstat (limited to 'tests/test_dl.py')
-rw-r--r--tests/test_dl.py12
1 files changed, 6 insertions, 6 deletions
diff --git a/tests/test_dl.py b/tests/test_dl.py
index ba9ce32..cbda95c 100644
--- a/tests/test_dl.py
+++ b/tests/test_dl.py
@@ -32,7 +32,7 @@ class TestPrepDownload:
"""
The prep_download function should create a file object.
"""
- args = construct_parser().parse_args(["File:Example.jpg"])
+ args = construct_parser(["File:Example.jpg"])
file = prep_download(args.FILE, args)
assert file is not None
@@ -43,7 +43,7 @@ class TestPrepDownload:
"""
tmp_file = tmp_path / "File:Example.jpg"
tmp_file.write_text("nothing")
- args = construct_parser().parse_args(["File:Example.jpg", "-o", str(tmp_file)])
+ args = construct_parser(["File:Example.jpg", "-o", str(tmp_file)])
with pytest.raises(FileExistsError):
_ = prep_download(args.FILE, args)
@@ -54,7 +54,7 @@ class TestProcessDownload:
"""
A successful batch download should not return any errors.
"""
- args = construct_parser().parse_args(["-a", "batch.txt"])
+ args = construct_parser(["-a", "batch.txt"])
mock_batch_download.return_value = 0
exit_code = process_download(args)
assert mock_batch_download.called
@@ -68,7 +68,7 @@ class TestProcessDownload:
Any errors during batch download should create a log message containing the
number of errors and result in a non-zero exit code.
"""
- args = construct_parser().parse_args(["-a", "batch.txt"])
+ args = construct_parser(["-a", "batch.txt"])
mock_batch_download.return_value = 4
exit_code = process_download(args)
assert mock_batch_download.called
@@ -83,7 +83,7 @@ class TestProcessDownload:
"""
A successful download should not return any errors.
"""
- args = construct_parser().parse_args(["File:Example.jpg"])
+ args = construct_parser(["File:Example.jpg"])
mock_download.return_value = 0
mock_prep_download.return_value = MagicMock(File)
exit_code = process_download(args)
@@ -99,7 +99,7 @@ class TestProcessDownload:
"""
Any errors during download should result in a non-zero exit code.
"""
- args = construct_parser().parse_args(["File:Example.jpg"])
+ args = construct_parser(["File:Example.jpg"])
mock_download.return_value = 1
mock_prep_download.return_value = MagicMock(File)
exit_code = process_download(args)