add tests

adbar · Nov 19, 2024 · 817b9e5 · 817b9e5
1 parent c2ef646
commit 817b9e5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/cli_tests.py b/tests/cli_tests.py
@@ -240,6 +240,10 @@ def test_sysoutput():
 
 def test_download():
     """test page download and command-line interface"""
+    assert cli_utils._define_exit_code([], 0) == 0
+    assert cli_utils._define_exit_code(["a"], 1) == 126
+    assert cli_utils._define_exit_code(["a"], 2) == 1
+
     testargs = ["", "-v"]
     with patch.object(sys, "argv", testargs):
         args = cli.parse_args(testargs)