From f745427dcec48d704cb5ad117505c060d2d5c78b Mon Sep 17 00:00:00 2001
From: Jeremy Tellaa <jeremy.tellaa@metalgear.xyz>
Date: Wed, 22 Nov 2023 14:12:37 +0100
Subject: [PATCH] fix: better main test

---
 py_loop/main.py    | 3 ++-
 tests/test_main.py | 6 +++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/py_loop/main.py b/py_loop/main.py
index e299b24..1fbf4a1 100755
--- 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
index b912026..37fe985 100644
--- 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