fix: better main test

jeremad · Nov 22, 2023 · 42e78ea · 42e78ea
1 parent 06b9ccc
commit 42e78ea
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/py_loop/main.py b/py_loop/main.py
@@ -9,7 +9,7 @@
 ArgsList = Optional[List[str]]
 
 
-def main(args: ArgsList = None) -> None:
+def main(args: ArgsList = None) -> Optional[Looper]:
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "cmd",
@@ -46,7 +46,7 @@ def main(args: ArgsList = None) -> None:
     args_ns = parser.parse_args(args=args)
     if args_ns.version:
         ui.info_1(Looper.version())
-        return
+        return None
     if not args_ns.cmd or not args_ns.cmd[0]:
         ui.error("no command provided")
         sys.exit(1)
@@ -59,3 +59,4 @@ def main(args: ArgsList = None) -> None:
         total_time=args_ns.total_time,
     )
     looper.loop()
+    return looper
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -13,6 +13,7 @@ def test_no_cmd() -> None:
     assert e.value.code == 1
 
 
-def test_cmd() -> None:
-    looper_main.main(["ls", "--max-tries", "1"])
-    looper_main.main(["--max-tries", "1", "ls"])
+def test_parsing() -> None:
+    looper = looper_main.main(["--max-tries", "7", "ls"])
+    assert looper is not None
+    assert looper.max_tries == 7