add unit test for cli commands, coverage(40%)

tests/test_cli/test_list_api.py

@@ -0,0 +1,85 @@
+from __future__ import absolute_import
+import mock
+import six
+import unittest
+
+from koji_cli.commands import anon_handle_list_api
+from . import utils
+
+
+class TestListApi(utils.CliTestCase):
+
+    # Show long diffs in error output...
+    maxDiff = None
+
+    def setUp(self):
+        self.error_format = """Usage: %s list-api [options]
+(Specify the --help global option for a list of other help options)
+
+%s: error: {message}
+""" % (self.progname, self.progname)
+
+    @mock.patch('sys.stdout', new_callable=six.StringIO)
+    @mock.patch('koji_cli.commands.activate_session')
+    def test_anon_handle_list_api(
+            self,
+            activate_session_mock,
+            stdout):
+        """Test anon_handle_list_api function"""
+        session = mock.MagicMock()
+        options = mock.MagicMock()
+
+        # Case 1. argument error
+        expected = self.format_error_message(
+            "This command takes no arguments")
+        self.assert_system_exit(
+            anon_handle_list_api,
+            options,
+            session,
+            ['arg'],
+            stderr=expected,
+            activate_session=None)
+
+        # Case 2.
+        session._listapi.return_value = [
+            {
+                'argdesc': '(name, *args, **kwargs)',
+                'doc': 'A debug function',
+                'argspec': [['name'], 'args', 'kwargs', None],
+                'args': ['name'],
+                'name': 'debugFunction'
+            },
+            {
+                'doc': 'Add user to group',
+                'argspec': [['group', 'user', 'strict'], None, None, [True]],
+                'args': ['group', 'user', ['strict', True]],
+                'name': 'addGroupMember'
+            },
+            {
+                'doc': None,
+                'argspec': [[], None, None, None],
+                'args': [],
+                'name': 'host.getID'
+            }
+        ]
+        expected = "addGroupMember(group, user, strict=True)\n"
+        expected += "  description: Add user to group\n"
+        expected += "debugFunction(name, *args, **kwargs)\n"
+        expected += "  description: A debug function\n"
+        expected += "host.getID()\n"
+        anon_handle_list_api(options, session, [])
+        self.assert_console_message(stdout, expected)
+
+    def test_anon_handle_list_api_help(self):
+        self.assert_help(
+            anon_handle_list_api,
+            """Usage: %s list-api [options]
+(Specify the --help global option for a list of other help options)
+
+Options:
+  -h, --help  show this help message and exit
+""" % self.progname)
+
+
+if __name__ == '__main__':
+    unittest.main()