def test_monitor_hosts(self,
mock_check_hb_line,
mock_check_pacemaker_services,
mock_check_host_status_by_crmadmin,
mock_check_host_status_by_cibadmin,
mock_sleep):
mock_check_hb_line.side_effect = \
[0, 1, 2, 0, Exception("Test exception."), 0, KeyboardInterrupt()]
mock_check_pacemaker_services.side_effect = [True, False, False, True]
mock_check_host_status_by_crmadmin.side_effect = [0, 1]
mock_check_host_status_by_cibadmin.side_effect = [0, 1, 0]
mock_sleep.return_value = None
obj = handle_host.HandleHost()
self.assertRaises(KeyboardInterrupt, obj.monitor_hosts)
self.assertEqual(7, mock_check_hb_line.call_count)
self.assertEqual(4, mock_check_pacemaker_services.call_count)
mock_check_pacemaker_services.assert_called_with('pacemaker_remote')
self.assertEqual(2, mock_check_host_status_by_crmadmin.call_count)
self.assertEqual(3, mock_check_host_status_by_cibadmin.call_count)
def test_check_hb_line(
self, mock_check_pacemaker_services, mock_execute):
mock_check_pacemaker_services.side_effect = [True, True, False]
interfaces = "enp0s8"
ports = "5405"
CONF.host.corosync_multicast_interfaces = interfaces
CONF.host.corosync_multicast_ports = ports
mock_execute.return_value = ('', '')
obj = handle_host.HandleHost()
ret = obj._check_hb_line()
self.assertEqual(0, ret)
self.assertEqual(3, mock_check_pacemaker_services.call_count)
mock_check_pacemaker_services.assert_any_call('corosync')
mock_check_pacemaker_services.assert_any_call('pacemaker')
mock_check_pacemaker_services.assert_any_call('pacemaker_remote')
cmd_str = ("timeout %s tcpdump -n -c 1 -p -i %s port %s") \
% (CONF.host.tcpdump_timeout, interfaces, ports)
command = cmd_str.split()
mock_execute.assert_called_once_with(*command, run_as_root=True)
def test_check_host_status_by_cibadmin(self, mock_get_cib_xml,
mock_set_cib_xml, mock_have_quorum,
mock_get_node_state_tag_list,
mock_check_if_status_changed):
mock_get_cib_xml.return_value = STATUS_TAG_XML
mock_set_cib_xml.return_value = None
mock_have_quorum.return_value = 1
status_tag = ElementTree.fromstring(STATUS_TAG_XML)
node_state_tag_list = list(status_tag)
mock_get_node_state_tag_list.return_value = node_state_tag_list
mock_check_if_status_changed.return_value = None
obj = handle_host.HandleHost()
ret = obj._check_host_status_by_cibadmin()
self.assertEqual(0, ret)
mock_get_cib_xml.assert_called_once_with()
mock_set_cib_xml.assert_called_once_with(STATUS_TAG_XML)
mock_have_quorum.assert_called_once_with()
mock_get_node_state_tag_list.assert_called_once_with()
mock_check_if_status_changed.assert_called_once_with(
node_state_tag_list)
def test_check_host_status_by_crm_mon(
self, mock_get_crmmon_xml, mock_set_crmmon_xml,
mock_has_quorum, mock_get_node_state_tag_list,
mock_check_if_status_changed):
mock_get_crmmon_xml.return_value = CRMMON_NODES_TAG_XML
mock_set_crmmon_xml.return_value = None
mock_has_quorum.return_value = True
status_tag = ElementTree.fromstring(CRMMON_NODES_TAG_XML)
node_state_tag_list = list(status_tag)
mock_get_node_state_tag_list.return_value = node_state_tag_list
mock_check_if_status_changed.return_value = None
obj = handle_host.HandleHost()
ret = obj._check_host_status_by_crm_mon()
self.assertEqual(0, ret)
mock_set_crmmon_xml.assert_called_once_with(CRMMON_NODES_TAG_XML)
mock_get_node_state_tag_list.assert_called_once_with()
mock_check_if_status_changed.assert_called_once_with(
[
{'uname': 'remote1', 'crmd': 'online'},
{'uname': 'remote2', 'crmd': 'online'},
{'uname': 'remote3', 'crmd': 'online'}])
def test_check_if_status_changed(
self, mock_gethostname, mock_get_host_status, mock_set_host_status,
mock_make_event, mock_send_notification):
mock_gethostname.return_value = 'node1'
mock_get_host_status.side_effect = \
[None, 'online', 'online', 'online']
mock_set_host_status.return_value = None
test_event = {'notification': 'test'}
mock_make_event.return_value = test_event
status_tag = ElementTree.fromstring(STATUS_TAG_XML)
node_state_tag_list = status_tag.getchildren()
obj = handle_host.HandleHost()
obj._check_if_status_changed(node_state_tag_list)
node_state_node2 = node_state_tag_list[1]
node_state_node3 = node_state_tag_list[2]
node_state_node4 = node_state_tag_list[3]
node_state_node5 = node_state_tag_list[4]
node2 = node_state_node2.get('uname')
node3 = node_state_node3.get('uname')
node4 = node_state_node4.get('uname')
node5 = node_state_node5.get('uname')
calls_get_host_status = [mock.call(node2),
mock.call(node3),
mock.call(node4),
mock.call(node5)]
calls_set_host_status = [mock.call(node_state_node2),
mock.call(node_state_node3),
mock.call(node_state_node4),
mock.call(node_state_node5)]
mock_get_host_status.assert_has_calls(calls_get_host_status)
mock_set_host_status.assert_has_calls(calls_set_host_status)
mock_make_event.assert_called_once_with(node4, 'offline')
mock_send_notification.assert_called_once_with(
CONF.host.api_retry_max, CONF.host.api_retry_interval, test_event)
def test_stop(self):
obj = handle_host.HandleHost()
obj.stop()
self.assertFalse(obj.running)
