fix: better main test

jeremad · Nov 22, 2023 · f745427 · f745427
1 parent 5d462f3
commit f745427
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 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) -> Looper:
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "cmd",
@@ -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,6 @@ 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.max_tries == 7