add unittests for koji command line:

- call
 - hello / moshimoshi
 - maven-chain
 - resubmit

tests/test_cli/test_hello.py

@@ -0,0 +1,119 @@
+from __future__ import absolute_import
+import mock
+import os
+import six
+import sys
+import unittest
+import koji
+
+from koji_cli.commands import handle_moshimoshi
+
+
+class TestResubmit(unittest.TestCase):
+
+    # Show long diffs in error output...
+    maxDiff = None
+
+    def setUp(self):
+        self.progname = os.path.basename(sys.argv[0]) or 'koji'
+        self.huburl = "https://%s.local/%shub" % (self.progname, self.progname)
+
+    def assert_console_output(self, device, expected, wipe=True, regex=False):
+        if not isinstance(device, six.StringIO):
+            raise TypeError('Not a StringIO object')
+
+        output = device.getvalue()
+        if not regex:
+            self.assertMultiLineEqual(output, expected)
+        else:
+            six.assertRegex(self, output, expected)
+        if wipe:
+            device.truncate(0)
+            device.seek(0)
+
+    @mock.patch('sys.stdout', new_callable=six.StringIO)
+    @mock.patch('sys.stderr', new_callable=six.StringIO)
+    @mock.patch('koji_cli.commands._printable_unicode')
+    @mock.patch('koji_cli.commands.activate_session')
+    def test_handle_moshimoshi(
+            self,
+            activate_session_mock,
+            print_unicode_mock,
+            stderr,
+            stdout):
+        """Test handle_moshimoshi function"""
+        user = {'name': self.progname,
+                'krb_principal': '%s@localhost' % self.progname}
+        cert = '/etc/pki/user.cert'
+        options = mock.MagicMock()
+
+        # Mock out the xmlrpc server
+        session = mock.MagicMock(baseurl=self.huburl, authtype=None)
+        session.getLoggedInUser.return_value = None
+        print_unicode_mock.return_value = "Hello"
+
+        with self.assertRaises(SystemExit) as cm:
+            handle_moshimoshi(options, session, ['argument'])
+        expect = """Usage: %s moshimoshi [options]
+
+%s: error: This command takes no arguments
+""" % (self.progname, self.progname)
+        self.assert_console_output(stderr, expect)
+        self.assertEqual(cm.exception.code, 2)
+
+        auth_tests = {
+            koji.AUTHTYPE_NORMAL: 'Authenticated via password',
+            koji.AUTHTYPE_GSSAPI: 'Authenticated via GSSAPI',
+            koji.AUTHTYPE_KERB: 'Authenticated via Kerberos principal %s' %
+                                user['krb_principal'],
+            koji.AUTHTYPE_SSL: 'Authenticated via client certificate %s' %
+                                cert
+        }
+
+        message = "Not authenticated\n" + "Hello, anonymous user!"
+        hubinfo = "You are using the hub at %s" % self.huburl
+        handle_moshimoshi(options, session, [])
+        self.assert_console_output(
+            stdout, "{0}\n\n{1}\n".format(message, hubinfo))
+
+        session.getLoggedInUser.return_value = user
+        message = "Hello, %s!" % self.progname
+        options.cert = cert
+        for authtype, authinfo in auth_tests.items():
+            session.authtype = authtype
+            print_unicode_mock.reset_mock()
+            print_unicode_mock.return_value = "Hello"
+            handle_moshimoshi(options, session, [])
+            print_unicode_mock.assert_called_once()
+            self.assert_console_output(
+                stdout, "{0}\n\n{1}\n{2}\n".format(message, hubinfo, authinfo))
+
+    @mock.patch('sys.stdout', new_callable=six.StringIO)
+    @mock.patch('sys.stderr', new_callable=six.StringIO)
+    @mock.patch('koji_cli.commands.activate_session')
+    def test_handle_moshimoshi_help(
+            self, activate_session_mock, stderr, stdout):
+        """Test  handle_moshimoshi help message full output"""
+        arguments = ['--help']
+        options = mock.MagicMock()
+
+        # Mock out the xmlrpc server
+        session = mock.MagicMock()
+
+        # Run it and check immediate output
+        with self.assertRaises(SystemExit) as cm:
+            handle_moshimoshi(options, session, arguments)
+        expected_stdout = """Usage: %s moshimoshi [options]
+
+Options:
+  -h, --help  show this help message and exit
+""" % (self.progname)
+        self.assert_console_output(stdout, expected_stdout)
+        self.assert_console_output(stderr, '')
+
+        # Finally, assert that things were called as we expected.
+        activate_session_mock.assert_not_called()
+        self.assertEqual(cm.exception.code, 0)
+
+if __name__ == '__main__':
+    unittest.main()