added tests for ganbreeder login args

1 files changed, 58 insertions, 2 deletions

diff --git a/test/cli_test.py b/test/cli_test.py
index 392d5da..de66828 100644
--- a/test/cli_test.py
+++ b/test/cli_test.py
@@ -1,7 +1,63 @@
 import unittest
+import context
+from gantools import cli
+
+def validate_args(args):
+    # check if username, password, and keys are all set or all empty
+    if not (lambda l: (not any(l)) or all(l))(\
+            [e is not None and e is not [] for e in [args.username, args.password, args.keys]]):
+        print(args.username)
+        print(args.password)
+        print(args.keys)
+        raise Exception('ganbreeder login credentials are invalid')
 
 class TestCli(unittest.TestCase):
-    def test_main(self):
-        self.assertEqual(True,True,"Will always pass")
+    def test_handle_args_ganbreeder_login_correct(self):
+        argv = [
+                '-u', '[email protected]',# user name
+                '-p', 'password123',# password
+                '-k', 'aaaa', 'bbbb', 'cccc',# keys
+                ]
+        args = cli.handle_args(argv=argv)
+        validate_args(args)
+
+    def test_handle_args_ganbreeder_login_no_user(self):
+        argv = [
+                '-p', 'password123',# password
+                '-k', 'aaaa', 'bbbb', 'cccc',# keys
+                ]
+        try:
+            args = cli.handle_args(argv=argv)
+        except:
+            return # this is supposed to cause an exception
+        validate_args(args)
+
+    def test_handle_args_ganbreeder_login_no_pass(self):
+        argv = [
+                '-u', '[email protected]',# user name
+                '-k', 'aaaa', 'bbbb', 'cccc',# keys
+                ]
+        try:
+            args = cli.handle_args(argv=argv)
+        except:
+            return # this is supposed to cause an exception
+        validate_args(args)
+
+    def test_handle_args_ganbreeder_login_no_keys(self):
+        argv = [
+                '-u', '[email protected]',# user name
+                '-p', 'password123',# password
+                ]
+        try:
+            args = cli.handle_args(argv=argv)
+        except:
+            return # this is supposed to cause an exception
+        validate_args(args)
+
+    def test_handle_args_ganbreeder_login_empty(self):
+        argv = []
+        args = cli.handle_args(argv=argv)
+        validate_args(args)
+
 if __name__ == '__main__':
     unittest.main()