Add test_main.py #1

Author: shenxianpeng

clang_tools/install.py

@@ -77,7 +77,10 @@ def move_and_chmod_binary(old_file_name, new_file_name, directory) -> None:
         shutil.move(old_file_name, f"{install_dir}/{new_file_name}")
         os.chmod(os.path.join(install_dir, new_file_name), 0o755)
     except PermissionError:
-        raise SystemExit("You don't have permission. Try to run with the appropriate permissions.")
+        raise SystemExit(
+            f"Don't have permission to install {new_file_name} to {install_dir}. \
+            Try to run with the appropriate permissions."
+        )
 
 
 def install_clang_tools(version, directory) -> None:

clang_tools/main.py

@@ -1,8 +1,9 @@
+import sys
 import argparse
 from clang_tools.install import install_clang_tools
 
 
-def main() -> int:
+def parse_args(args):
     parser = argparse.ArgumentParser(prog='clang-tools')
 
     parser.add_argument(
@@ -18,13 +19,16 @@ def main() -> int:
         default="",
         help="The directory where is the clang-tools install.",
     )
-    args = parser.parse_args()
+    return parser.parse_args()
+
+
+def main():
+    args = parse_args(sys.argv[1:])
 
     version = args.install
     directory = args.directory
-
     install_clang_tools(version, directory)
 
 
 if __name__ == '__main__':
-    raise SystemExit(main())
+    main()

tests/test_main.py

@@ -0,0 +1,13 @@
+import pytest
+from unittest import mock
+from clang_tools.main import main
+
+
+@pytest.mark.parametrize("argv, called, response", [(['-i'], True, False), (['-d'], True, False)])
+def test_main_install(argv, called, response):
+    with mock.patch('sys.argv', [''] + argv):
+        if called and not response:
+            with pytest.raises(SystemExit):
+                main()
+        else:
+            main()