Make process_download return an exit code instead of exiting directly

1 files changed, 8 insertions, 8 deletions

diff --git a/tests/test_dl.py b/tests/test_dl.py
index 61efbe0..ba9ce32 100644
--- a/tests/test_dl.py
+++ b/tests/test_dl.py
@@ -56,8 +56,9 @@ class TestProcessDownload:
         """
         args = construct_parser().parse_args(["-a", "batch.txt"])
         mock_batch_download.return_value = 0
-        process_download(args)
+        exit_code = process_download(args)
         assert mock_batch_download.called
+        assert exit_code == 0
 
     @patch("wikiget.dl.batch_download")
     def test_batch_download_with_errors(
@@ -69,10 +70,9 @@ class TestProcessDownload:
         """
         args = construct_parser().parse_args(["-a", "batch.txt"])
         mock_batch_download.return_value = 4
-        with pytest.raises(SystemExit) as e:
-            process_download(args)
+        exit_code = process_download(args)
         assert mock_batch_download.called
-        assert e.value.code == 1
+        assert exit_code == 1
         assert "4 problems encountered during batch processing" in caplog.text
 
     @patch("wikiget.dl.prep_download")
@@ -86,9 +86,10 @@ class TestProcessDownload:
         args = construct_parser().parse_args(["File:Example.jpg"])
         mock_download.return_value = 0
         mock_prep_download.return_value = MagicMock(File)
-        process_download(args)
+        exit_code = process_download(args)
         assert mock_prep_download.called
         assert mock_download.called
+        assert exit_code == 0
 
     @patch("wikiget.dl.prep_download")
     @patch("wikiget.dl.download")
@@ -101,8 +102,7 @@ class TestProcessDownload:
         args = construct_parser().parse_args(["File:Example.jpg"])
         mock_download.return_value = 1
         mock_prep_download.return_value = MagicMock(File)
-        with pytest.raises(SystemExit) as e:
-            process_download(args)
+        exit_code = process_download(args)
         assert mock_prep_download.called
         assert mock_download.called
-        assert e.value.code == 1
+        assert exit_code == 1