From 26b2bfea7434aeb9d3687397341e0e7ad3f4edfc Mon Sep 17 00:00:00 2001 From: Cody Logan Date: Thu, 30 Nov 2023 10:11:21 -0800 Subject: Have functions return an exit code instead of calling sys.exit --- tests/test_dl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'tests/test_dl.py') diff --git a/tests/test_dl.py b/tests/test_dl.py index 7117e50..63408eb 100644 --- a/tests/test_dl.py +++ b/tests/test_dl.py @@ -241,10 +241,10 @@ class TestBatchDownload: mock_read_batch_file.side_effect = OSError("error message") args = parse_args(["-a", "batch.txt"]) - with pytest.raises(SystemExit): - _ = batch_download(args) + errors = batch_download(args) assert mock_read_batch_file.called + assert errors == 1 assert caplog.record_tuples == [ ("wikiget.dl", logging.ERROR, "File could not be read: error message"), ] -- cgit v1.2.3