def test_handle_enable_host_no_argument(self, activate_session_mock,
stdout):
"""Test %s function without arguments""" % handle_enable_host.__name__
options = mock.MagicMock()
session = mock.MagicMock()
session.getHost.return_value = None
session.multiCall.return_value = [[None]]
session.enableHost.return_value = True
session.editHost.return_value = True
handle_enable_host(options, session, [])
activate_session_mock.assert_called_once()
session.getHost.assert_not_called()
session.multiCall.assert_called()
session.enableHost.assert_not_called()
session.editHost.assert_not_called()
self.assert_console_message(stdout, '')
def test_handle_enable_host(self, activate_session_mock, stdout):
"""Test %s function""" % handle_enable_host.__name__
arguments = []
options = mock.MagicMock()
session = mock.MagicMock()
session.getHost.return_value = None
session.enableHost.return_value = True
session.editHost.return_value = True
#
# session.multiCall returns:
#
# [[{'comment': None,
# 'capacity': 2.0,
# 'name': 'kbuilder01',
# 'enabled': True,
# 'arches': 'x86_64',
# 'task_load': 0.0,
# 'ready': False,
# 'user_id': 4,
# 'id': 2, 'description': None}],
# [{'comment': None,
# 'capacity': 2.0,
# 'name': 'kbuilder02' ...}]
#
session.multiCall.return_value = [[None], [None]]
arguments = ['host1', 'host2']
self.assertEqual(1, handle_enable_host(options, session, arguments))
activate_session_mock.assert_called_once()
session.getHost.assert_has_calls([call('host1'), call('host2')])
session.multiCall.assert_called_once()
session.enableHost.assert_not_called()
session.editHost.assert_not_called()
expect = ''
for host in arguments:
expect += "Host %s does not exist\n" % host
expect += "No changes made. Please correct the command line.\n"
self.assert_console_message(stdout, expect)
# reset session mocks
activate_session_mock.reset_mock()
session.multiCall.reset_mock()
session.disableHost.reset_mock()
session.editHost.reset_mock()
session.multiCall.return_value = [[{
'id': 1,
'name': 'host1'
}], [{
'id': 2,
'name': 'host2'
}]]
arguments = ['host1', 'host2', '--comment', 'disable host test']
handle_enable_host(options, session, arguments)
activate_session_mock.assert_called_once()
session.getHost.assert_has_calls([call('host1'), call('host2')])
self.assertEqual(2, session.multiCall.call_count)
session.enableHost.assert_has_calls([call('host1'), call('host2')])
session.editHost.assert_has_calls([
call('host1', comment='disable host test'),
call('host2', comment='disable host test')
])
self.assert_console_message(stdout, '')