test handling of invalid CLI flags in argument parser

--version in .toml changed

Author: arilloid

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-mage"
-version = "0.1.0"
+version = "0.9.0"
 description = ""
 authors = ["gitdevjin <[email protected]>"]
 readme = "README.md"

tests/test_argsParser.py

@@ -113,3 +113,12 @@ def test_arg_parser_token_usage_option_short(mock_config):
 	with patch("sys.argv", ["code_mage.py", "-t"]):
 		args = arg_parser(mock_config)
 		assert args.token_usage is True
+
+
+# Test invalid flag argument option
+def test_arg_parser_invalid_option():
+    with patch("sys.argv", ["code_mage.py", "--invalid"]):
+        with pytest.raises(SystemExit) as excinfo:
+            arg_parser({})
+        assert excinfo.type is SystemExit
+        assert excinfo.value.code != 0