def test_handle_assign_task_help(self, activate_session_mock, stderr,
stdout):
arguments = []
options = mock.MagicMock()
progname = os.path.basename(sys.argv[0]) or 'koji'
# Mock out the xmlrpc server
session = mock.MagicMock()
# Run it and check immediate output
with self.assertRaises(SystemExit) as cm:
handle_assign_task(options, session, arguments)
actual_stdout = stdout.getvalue()
actual_stderr = stderr.getvalue()
expected_stdout = ''
expected_stderr = """Usage: %s assign-task task_id hostname
(Specify the --help global option for a list of other help options)
%s: error: please specify a task id and a hostname
""" % (progname, progname)
self.assertMultiLineEqual(actual_stdout, expected_stdout)
self.assertMultiLineEqual(actual_stderr, expected_stderr)
# Finally, assert that things were called as we expected.
activate_session_mock.assert_not_called()
session.hasHost.assert_not_called()
session.addHost.assert_not_called()
self.assertEqual(cm.exception.code, 2)
def test_handle_assign_task_help(
self, activate_session_mock, stderr, stdout):
arguments = []
options = mock.MagicMock()
progname = os.path.basename(sys.argv[0]) or 'koji'
# Mock out the xmlrpc server
session = mock.MagicMock()
# Run it and check immediate output
with self.assertRaises(SystemExit) as cm:
handle_assign_task(options, session, arguments)
actual_stdout = stdout.getvalue()
actual_stderr = stderr.getvalue()
expected_stdout = ''
expected_stderr = """Usage: %s assign-task task_id hostname
(Specify the --help global option for a list of other help options)
%s: error: please specify a task id and a hostname
""" % (progname, progname)
self.assertMultiLineEqual(actual_stdout, expected_stdout)
self.assertMultiLineEqual(actual_stderr, expected_stderr)
# Finally, assert that things were called as we expected.
activate_session_mock.assert_not_called()
session.hasHost.assert_not_called()
session.addHost.assert_not_called()
if isinstance(cm.exception, int):
self.assertEqual(cm.exception, 2)
else:
self.assertEqual(cm.exception.code, 2)
def test_handle_assign_task(
self, activate_session_mock, stdout):
hostname = "host"
task_id = "1"
arguments = [task_id, hostname]
options = mock.MagicMock()
# Mock out the xmlrpc server
session = mock.MagicMock()
session.getTaskInfo.return_value = None
with six.assertRaisesRegex(self, koji.GenericError,
"No such task: %s" % task_id):
handle_assign_task(options, session, arguments)
session.getTaskInfo.return_value = "task_info"
session.getHost.return_value = None
with six.assertRaisesRegex(self, koji.GenericError,
"No such host: %s" % hostname):
handle_assign_task(options, session, arguments)
arguments.append("--force")
session.getHost.return_value = hostname
session.hasPerm.return_value = False
handle_assign_task(options, session, arguments)
actual = stdout.getvalue()
expected = "This action requires admin privileges\n"
self.assertMultiLineEqual(actual, expected)
# Clean stdout buffer
stdout.truncate(0)
stdout.seek(0)
session.hasPerm.return_value = True
session.assignTask.return_value = True
handle_assign_task(options, session, arguments)
actual = stdout.getvalue()
expected = 'assigned task %s to host %s\n' % \
(task_id, hostname)
self.assertMultiLineEqual(actual, expected)
# Clean stdout buffer
stdout.truncate(0)
stdout.seek(0)
session.assignTask.return_value = False
handle_assign_task(options, session, arguments)
actual = stdout.getvalue()
expected = 'failed to assign task %s to host %s\n' % \
(task_id, hostname)
self.assertMultiLineEqual(actual, expected)
# Finally, assert that things were called as we expected.
activate_session_mock.assert_called_with(session, options)
session.assignTask.assert_called_with(int(task_id), hostname, True)
def test_handle_assign_task(self, activate_session_mock, stdout):
hostname = "host"
task_id = "1"
arguments = [task_id, hostname]
options = mock.MagicMock()
# Mock out the xmlrpc server
session = mock.MagicMock()
session.getTaskInfo.return_value = None
with six.assertRaisesRegex(self, koji.GenericError,
"No such task: %s" % task_id):
handle_assign_task(options, session, arguments)
session.getTaskInfo.return_value = "task_info"
session.getHost.return_value = None
with six.assertRaisesRegex(self, koji.GenericError,
"No such host: %s" % hostname):
handle_assign_task(options, session, arguments)
arguments.append("--force")
session.getHost.return_value = hostname
session.hasPerm.return_value = False
handle_assign_task(options, session, arguments)
actual = stdout.getvalue()
expected = "This action requires admin privileges\n"
self.assertMultiLineEqual(actual, expected)
# Clean stdout buffer
stdout.truncate(0)
stdout.seek(0)
session.hasPerm.return_value = True
session.assignTask.return_value = True
handle_assign_task(options, session, arguments)
actual = stdout.getvalue()
expected = 'assigned task %s to host %s\n' % \
(task_id, hostname)
self.assertMultiLineEqual(actual, expected)
# Clean stdout buffer
stdout.truncate(0)
stdout.seek(0)
session.assignTask.return_value = False
handle_assign_task(options, session, arguments)
actual = stdout.getvalue()
expected = 'failed to assign task %s to host %s\n' % \
(task_id, hostname)
self.assertMultiLineEqual(actual, expected)
# Finally, assert that things were called as we expected.
activate_session_mock.assert_called_with(session, options)
session.assignTask.assert_called_with(int(task_id), hostname, True)