Add some unit tests for CLI

Reorganize the cli.py code to ease the testing.

1 files changed, 95 insertions, 0 deletions

diff --git a/gitlab/tests/test_cli.py b/gitlab/tests/test_cli.py
new file mode 100644
index 0000000..c32ad50
--- /dev/null
+++ b/gitlab/tests/test_cli.py
@@ -0,0 +1,95 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2016 Gauvain Pocentek <gauvain@pocentek.net>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from __future__ import print_function
+from __future__ import absolute_import
+
+import argparse
+
+import six
+try:
+    import unittest
+except ImportError:
+    import unittest2 as unittest
+
+from gitlab import cli
+
+
+class TestCLI(unittest.TestCase):
+    def test_what_to_cls(self):
+        self.assertEqual("Foo", cli._what_to_cls("foo"))
+        self.assertEqual("FooBar", cli._what_to_cls("foo-bar"))
+
+    def test_cls_to_what(self):
+        class Class(object):
+            pass
+
+        class TestClass(object):
+            pass
+
+        self.assertEqual("test-class", cli._cls_to_what(TestClass))
+        self.assertEqual("class", cli._cls_to_what(Class))
+
+    def test_die(self):
+        with self.assertRaises(SystemExit) as test:
+            cli._die("foobar")
+
+        self.assertEqual(test.exception.code, 1)
+
+    def test_extra_actions(self):
+        for cls, data in six.iteritems(cli.EXTRA_ACTIONS):
+            for key in data:
+                self.assertIsInstance(data[key], dict)
+
+    def test_parsing(self):
+        args = cli._parse_args(['-v', '-g', 'gl_id',
+                                '-c', 'foo.cfg', '-c', 'bar.cfg',
+                                'project', 'list'])
+        self.assertTrue(args.verbose)
+        self.assertEqual(args.gitlab, 'gl_id')
+        self.assertEqual(args.config_file, ['foo.cfg', 'bar.cfg'])
+        self.assertEqual(args.what, 'project')
+        self.assertEqual(args.action, 'list')
+
+    def test_parser(self):
+        parser = cli._build_parser()
+        subparsers = None
+        for action in parser._actions:
+            if type(action) == argparse._SubParsersAction:
+                subparsers = action
+                break
+        self.assertIsNotNone(subparsers)
+        self.assertIn('user', subparsers.choices)
+
+        user_subparsers = None
+        for action in subparsers.choices['user']._actions:
+            if type(action) == argparse._SubParsersAction:
+                user_subparsers = action
+                break
+        self.assertIsNotNone(user_subparsers)
+        self.assertIn('list', user_subparsers.choices)
+        self.assertIn('get', user_subparsers.choices)
+        self.assertIn('delete', user_subparsers.choices)
+        self.assertIn('update', user_subparsers.choices)
+        self.assertIn('create', user_subparsers.choices)
+        self.assertIn('block', user_subparsers.choices)
+        self.assertIn('unblock', user_subparsers.choices)
+
+        actions = user_subparsers.choices['create']._option_string_actions
+        self.assertFalse(actions['--twitter'].required)
+        self.assertTrue(actions['--username'].required)