119 lines
4.2 KiB
Python
119 lines
4.2 KiB
Python
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 TestHello(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()
|