def test_handle_restart_hosts_other_options(
self, activate_session_mock, running_in_bg_mock, watch_tasks_mock):
"""Test %s function with --force option""" % handle_restart_hosts.__name__
arguments = ['--nowait',
'--channel', 'createrepo',
'--arch', 'x86_64',
'--timeout', '10']
options = mock.MagicMock(quiet=None, poll_interval=3)
session = mock.MagicMock()
# --no-wait is specified, running_in_bg() should not matter.
running_in_bg_mock.return_value = True
session.getHost.return_value = None
session.restartHosts.return_value = 101
session.logout.return_value = None
# has other restart tasks are running case
session.listTasks.return_value = []
handle_restart_hosts(options, session, arguments)
activate_session_mock.assert_called_once()
session.listTasks.assert_called_once()
session.restartHosts.assert_called_with(
options={'arches': ['x86_64'], 'timeout': 10, 'channel': 'createrepo'})
session.logout.assert_not_called()
watch_tasks_mock.assert_not_called()
def test_handle_restart_hosts_force_options(
self, activate_session_mock, running_in_bg_mock, watch_tasks_mock):
"""Test %s function with --force option""" % handle_restart_hosts.__name__
arguments = ['--force']
options = mock.MagicMock(quiet=None, poll_interval=3)
session = mock.MagicMock()
# set running in foreground
running_in_bg_mock.return_value = False
session.getHost.return_value = None
session.restartHosts.return_value = self.task_id
session.logout.return_value = None
# has other restart tasks are running case
session.listTasks.return_value = [{'id': 1}, {'id': 2}, {'id': 3}]
handle_restart_hosts(options, session, arguments)
activate_session_mock.assert_called_once()
session.listTasks.assert_not_called()
session.restartHosts.assert_called_with()
session.logout.assert_called_once()
watch_tasks_mock.assert_called_with(
session, [self.task_id], quiet=None, poll_interval=3)
def test_handle_restart_hosts_force_options(self, activate_session_mock,
running_in_bg_mock,
watch_tasks_mock):
"""Test %s function with --force option""" % handle_restart_hosts.__name__
arguments = ['--force']
options = mock.MagicMock(quiet=None, poll_interval=3)
session = mock.MagicMock()
# set running in foreground
running_in_bg_mock.return_value = False
session.getHost.return_value = None
session.restartHosts.return_value = self.task_id
session.logout.return_value = None
# has other restart tasks are running case
session.listTasks.return_value = [{'id': 1}, {'id': 2}, {'id': 3}]
handle_restart_hosts(options, session, arguments)
activate_session_mock.assert_called_once()
session.listTasks.assert_not_called()
session.restartHosts.assert_called_with()
session.logout.assert_called_once()
watch_tasks_mock.assert_called_with(session, [self.task_id],
quiet=None,
poll_interval=3)
def test_handle_restart_hosts_wait_option(
self, activate_session_mock, running_in_bg_mock, watch_tasks_mock):
"""Test %s function with --force option""" % handle_restart_hosts.__name__
arguments = ['--wait']
options = mock.MagicMock(quiet=None, poll_interval=3)
session = mock.MagicMock()
# --wait is specified, running_in_bg() should not matter.
running_in_bg_mock.return_value = True
session.getHost.return_value = None
session.restartHosts.return_value = self.task_id
session.logout.return_value = None
# has other restart tasks are running case
session.listTasks.return_value = []
handle_restart_hosts(options, session, arguments)
activate_session_mock.assert_called_once()
session.listTasks.assert_called_once()
session.restartHosts.assert_called_with()
session.logout.assert_called_once()
watch_tasks_mock.assert_called_with(
session, [self.task_id], quiet=None, poll_interval=3)
def test_handle_restart_hosts_other_options(self, activate_session_mock,
running_in_bg_mock,
watch_tasks_mock):
"""Test %s function with --force option""" % handle_restart_hosts.__name__
arguments = [
'--nowait', '--channel', 'createrepo', '--arch', 'x86_64',
'--timeout', '10'
]
options = mock.MagicMock(quiet=None, poll_interval=3)
session = mock.MagicMock()
# --no-wait is specified, running_in_bg() should not matter.
running_in_bg_mock.return_value = True
session.getHost.return_value = None
session.restartHosts.return_value = 101
session.logout.return_value = None
# has other restart tasks are running case
session.listTasks.return_value = []
handle_restart_hosts(options, session, arguments)
activate_session_mock.assert_called_once()
session.listTasks.assert_called_once()
session.restartHosts.assert_called_with(options={
'arches': ['x86_64'],
'timeout': 10,
'channel': 'createrepo'
})
session.logout.assert_not_called()
watch_tasks_mock.assert_not_called()
def test_handle_restart_hosts_wait_option(self, activate_session_mock,
running_in_bg_mock,
watch_tasks_mock):
"""Test %s function with --force option""" % handle_restart_hosts.__name__
arguments = ['--wait']
options = mock.MagicMock(quiet=None, poll_interval=3)
session = mock.MagicMock()
# --wait is specified, running_in_bg() should not matter.
running_in_bg_mock.return_value = True
session.getHost.return_value = None
session.restartHosts.return_value = self.task_id
session.logout.return_value = None
# has other restart tasks are running case
session.listTasks.return_value = []
handle_restart_hosts(options, session, arguments)
activate_session_mock.assert_called_once()
session.listTasks.assert_called_once()
session.restartHosts.assert_called_with()
session.logout.assert_called_once()
watch_tasks_mock.assert_called_with(session, [self.task_id],
quiet=None,
poll_interval=3)
def test_handle_restart_hosts_has_other_tasks(self, stderr):
"""Test %s function when there has other restart tasks exist
""" % handle_restart_hosts.__name__
self.session.getHost.return_value = None
self.session.restartHosts.return_value = True
self.session.logout.return_value = None
#
# session.listTasks returns:
#
# [{'weight': 0.1,
# 'awaited': None,
# 'completion_time': None,
# 'create_time': '2017-11-02 18:30:08.933753',
# 'result': None,
# 'owner': 1,
# 'id': 11,
# 'state': 1,
# 'label': None,
# 'priority': 5,
# 'waiting': True,
# 'completion_ts': None,
# 'method': 'restartHosts',
# 'owner_name': 'kojiadmin',
# 'parent': None,
# 'start_time': '2017-11-02 18:30:28.028843',
# 'start_ts': 1509647428.02884,
# 'create_ts': 1509647408.93375,
# 'host_id': 1, 'arch': 'noarch',
# 'request': "<?xml version='1.0'?>\n<methodCall>\n<methodName>restartHosts</methodName>
# \n<params>\n</params>\n</methodCall>\n",
# 'channel_id': 1,
# 'owner_type': 0}]
#
# has other restart tasks are running case
self.session.listTasks.return_value = [{'id': 1}, {'id': 2}, {'id': 3}]
with self.assertRaises(SystemExit) as ex:
handle_restart_hosts(self.options, self.session, [])
self.assertExitCode(ex, 1)
self.activate_session_mock.assert_called_once()
query_opt = {
'method': 'restartHosts',
'state':
[koji.TASK_STATES[s] for s in ('FREE', 'OPEN', 'ASSIGNED')]
}
self.session.listTasks.assert_called_with(query_opt)
self.session.restartHosts.assert_not_called()
self.session.logout.assert_not_called()
expect = "Found other restartHosts tasks running.\n"
expect += "Task ids: %r\n" % \
[t['id'] for t in self.session.listTasks.return_value]
expect += "Use --force to run anyway\n"
self.assert_console_message(stderr, expect)
def test_handle_restart_hosts_force_options(self):
"""Test %s function with --force option""" % handle_restart_hosts.__name__
arguments = ['--force']
self.session.getHost.return_value = None
self.session.restartHosts.return_value = self.task_id
self.session.logout.return_value = None
# has other restart tasks are running case
self.session.listTasks.return_value = [{'id': 1}, {'id': 2}, {'id': 3}]
handle_restart_hosts(self.options, self.session, arguments)
self.activate_session_mock.assert_called_once()
self.session.listTasks.assert_not_called()
self.session.restartHosts.assert_called_with()
self.session.logout.assert_called_once()
self.watch_tasks_mock.assert_called_with(self.session, [self.task_id],
quiet=None,
poll_interval=3,
topurl=self.options.topurl)
def test_handle_restart_hosts_arguments(self, running_in_bg_mock, stderr):
"""Test %s function with --force option""" % handle_restart_hosts.__name__
# --no-wait is specified, running_in_bg() should not matter.
running_in_bg_mock.return_value = True
with self.assertRaises(SystemExit) as ex:
handle_restart_hosts(self.options, self.session, ['10'])
self.assertExitCode(ex, 2)
expected_msg = """Usage: %s restart-hosts [options]
(Specify the --help global option for a list of other help options)
%s: error: restart-hosts does not accept arguments
""" % (self.progname, self.progname)
self.assert_console_message(stderr, expected_msg)
self.session.listTasks.assert_not_called()
self.activate_session_mock.assert_not_called()
self.session.restartHosts.assert_not_called()
self.session.logout.assert_not_called()
self.watch_tasks_mock.assert_not_called()
def test_handle_restart_hosts_has_other_tasks(
self,
activate_session_mock,
running_in_bg_mock,
watch_tasks_mock,
stdout):
"""Test %s function when there has other restart tasks exist""" % handle_restart_hosts.__name__
options = mock.MagicMock()
session = mock.MagicMock()
# set running in foreground
running_in_bg_mock.return_value = False
session.getHost.return_value = None
session.restartHosts.return_value = True
session.logout.return_value = None
#
# session.listTasks returns:
#
# [{'weight': 0.1,
# 'awaited': None,
# 'completion_time': None,
# 'create_time': '2017-11-02 18:30:08.933753',
# 'result': None,
# 'owner': 1,
# 'id': 11,
# 'state': 1,
# 'label': None,
# 'priority': 5,
# 'waiting': True,
# 'completion_ts': None,
# 'method': 'restartHosts',
# 'owner_name': 'kojiadmin',
# 'parent': None,
# 'start_time': '2017-11-02 18:30:28.028843',
# 'start_ts': 1509647428.02884,
# 'create_ts': 1509647408.93375,
# 'host_id': 1, 'arch': 'noarch',
# 'request': "<?xml version='1.0'?>\n<methodCall>\n<methodName>restartHosts</methodName>\n<params>\n</params>\n</methodCall>\n",
# 'channel_id': 1,
# 'owner_type': 0}]
#
# has other restart tasks are running case
session.listTasks.return_value = [{'id': 1}, {'id': 2}, {'id': 3}]
self.assertEqual(1, handle_restart_hosts(options, session, []))
activate_session_mock.assert_called_once()
query_opt = {
'method': 'restartHosts',
'state': [koji.TASK_STATES[s] for s in ('FREE', 'OPEN', 'ASSIGNED')]
}
session.listTasks.assert_called_with(query_opt)
session.restartHosts.assert_not_called()
session.logout.assert_not_called()
expect = "Found other restartHosts tasks running.\n"
expect += "Task ids: %r\n" % \
[t['id'] for t in session.listTasks.return_value]
expect += "Use --force to run anyway\n"
self.assert_console_message(stdout, expect)