Add test for keyboard interrupt

1 files changed, 41 insertions, 16 deletions

diff --git a/tests/test_wikiget_cli.py b/tests/test_wikiget_cli.py
index 28c4399..15e2a7b 100644
--- a/tests/test_wikiget_cli.py
+++ b/tests/test_wikiget_cli.py
@@ -27,7 +27,7 @@ from wikiget.wikiget import cli
 
 
 @patch("wikiget.wikiget.process_download")
-class TestCli:
+class TestWikigetCli:
     """Define tests related to wikiget.wikiget.cli."""
 
     def test_cli_no_args(self, monkeypatch: pytest.MonkeyPatch) -> None:
@@ -38,7 +38,7 @@ class TestCli:
             with pytest.raises(SystemExit) as e:
                 cli()
 
-            assert e.value.code == 2
+        assert e.value.code == 2
 
     def test_cli_completed_successfully(
         self, mock_process_download: MagicMock, monkeypatch: pytest.MonkeyPatch
@@ -52,7 +52,7 @@ class TestCli:
             with pytest.raises(SystemExit) as e:
                 cli()
 
-            assert e.value.code == 0
+        assert e.value.code == 0
 
     def test_cli_completed_with_problems(
         self, mock_process_download: MagicMock, monkeypatch: pytest.MonkeyPatch
@@ -66,7 +66,7 @@ class TestCli:
             with pytest.raises(SystemExit) as e:
                 cli()
 
-            assert e.value.code == 1
+        assert e.value.code == 1
 
     def test_cli_logs(
         self,
@@ -87,15 +87,40 @@ class TestCli:
             with pytest.raises(SystemExit):
                 cli()
 
-            assert caplog.record_tuples == [
-                (
-                    "wikiget.wikiget",
-                    logging.INFO,
-                    f"Starting download session using wikiget {__version__}",
-                ),
-                (
-                    "wikiget.wikiget",
-                    logging.DEBUG,
-                    f"User agent: {USER_AGENT}",
-                ),
-            ]
+        assert caplog.record_tuples == [
+            (
+                "wikiget.wikiget",
+                logging.INFO,
+                f"Starting download session using wikiget {__version__}",
+            ),
+            (
+                "wikiget.wikiget",
+                logging.DEBUG,
+                f"User agent: {USER_AGENT}",
+            ),
+        ]
+
+    def test_cli_interrupt(
+        self,
+        mock_process_download: MagicMock,
+        monkeypatch: pytest.MonkeyPatch,
+        caplog: pytest.LogCaptureFixture,
+    ) -> None:
+        """Test what happens when KeyboardInterrupt is raised during processing.
+
+        A critical log message should be printed and the exit code should be 130.
+        """
+        with monkeypatch.context() as m:
+            mock_process_download.side_effect = KeyboardInterrupt
+            m.setattr("sys.argv", ["wikiget", "File:Example.jpg"])
+
+            with pytest.raises(SystemExit) as e:
+                cli()
+
+        assert e.value.code == 130
+        # ignore the first two messages, since they're tested elsewhere
+        assert caplog.record_tuples[2] == (
+            "wikiget.wikiget",
+            logging.CRITICAL,
+            "Interrupted by user",
+        )