Merge "Add some UT test cases and fix pep8 errors for cli and env_setup."

1 files changed, 20 insertions, 0 deletions

diff --git a/tests/cli_test.py b/tests/cli_test.py
new file mode 100644
index 00000000..f12e8fed
--- /dev/null
+++ b/tests/cli_test.py
@@ -0,0 +1,20 @@
+import pytest
+from func.cli import cli
+
+
+class TestClass:
+    @pytest.mark.parametrize("test_input, expected", [
+        (['-l',
+          'zte',
+          '-f',
+          'compute'], "You have specified a lab that is not present in test_cases"),
+        (['-l',
+          'zte-pod1',
+          '-f',
+          'test'], "Test File Does not exist in test_list")
+    ])
+    def test_cli_error(self, capfd, test_input, expected):
+        with pytest.raises(SystemExit):
+            cli(test_input)
+            resout, reserr = capfd.readouterr()
+            assert expected in resout