Add testcases of hostmonitor

This patch added missing testcases for hostmonitor directory.
And also improved existing testcases for them.

Change-Id: Iedc86123a2bc17299119ddb52824d0c449afa993
This commit is contained in:
Kengo Takahara 2017-04-19 18:15:38 +09:00
parent dbd53c13a7
commit 097d37d505
4 changed files with 775 additions and 30 deletions

View File

@ -13,17 +13,41 @@
# limitations under the License.
import mock
import socket
import testtools
from xml.etree import ElementTree
import eventlet
from oslo_utils import timeutils
import masakarimonitors.conf
from masakarimonitors.ha import masakari
from masakarimonitors.hostmonitor.host_handler import handle_host
from masakarimonitors.hostmonitor.host_handler import hold_host_status
from masakarimonitors.hostmonitor.host_handler import parse_cib_xml
from masakarimonitors.objects import event_constants as ec
from masakarimonitors import utils
eventlet.monkey_patch(os=False)
EXECUTE_RETURN = 'Status of crmd@masakari-node: S_NOT_DC (ok)'
CONF = masakarimonitors.conf.CONF
STATUS_TAG_XML = ' <status>' \
' <node_state crmd="online" uname="node1">' \
' <test foo="foo"/>' \
' </node_state>' \
' <node_state crmd="online" uname="node2">' \
' <test foo="foo"/>' \
' </node_state>' \
' <node_state crmd="online" uname="node3">' \
' <test foo="foo"/>' \
' </node_state>' \
' <node_state crmd="offline" uname="node4">' \
' <test foo="foo"/>' \
' </node_state>' \
' <node_state crmd="other" uname="node5">' \
' <test foo="foo"/>' \
' </node_state>' \
' </status>'
class TestHandleHost(testtools.TestCase):
@ -31,39 +55,641 @@ class TestHandleHost(testtools.TestCase):
def setUp(self):
super(TestHandleHost, self).setUp()
@mock.patch.object(utils, 'execute')
def test_check_pacemaker_services(self, mock_execute):
mock_execute.return_value = ('test_stdout', '')
obj = handle_host.HandleHost()
ret = obj._check_pacemaker_services('corosync')
self.assertTrue(ret)
mock_execute.assert_called_once_with(
'systemctl', 'status', 'corosync', run_as_root=True)
@mock.patch.object(utils, 'execute')
def test_check_pacemaker_services_stderr(self, mock_execute):
mock_execute.return_value = ('test_stdout', 'test_stderr')
obj = handle_host.HandleHost()
ret = obj._check_pacemaker_services('corosync')
self.assertFalse(ret)
mock_execute.assert_called_once_with(
'systemctl', 'status', 'corosync', run_as_root=True)
@mock.patch.object(utils, 'execute')
@mock.patch.object(handle_host.HandleHost, '_check_pacemaker_services')
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)
@mock.patch.object(handle_host.HandleHost, '_check_pacemaker_services')
def test_check_hb_line_pacemaker_not_running(
self, mock_check_pacemaker_services):
mock_check_pacemaker_services.side_effect = [False, False, False]
obj = handle_host.HandleHost()
ret = obj._check_hb_line()
self.assertEqual(2, 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')
@mock.patch.object(handle_host.HandleHost, '_check_pacemaker_services')
def test_check_hb_line_work_on_pacemaker_remote(
self, mock_check_pacemaker_services):
mock_check_pacemaker_services.side_effect = [False, False, True]
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')
@mock.patch.object(handle_host.HandleHost, '_check_pacemaker_services')
def test_check_hb_line_conf_interfaces_is_none(
self, mock_check_pacemaker_services):
mock_check_pacemaker_services.side_effect = [True, True, False]
interfaces = None
ports = "5405"
CONF.host.corosync_multicast_interfaces = interfaces
CONF.host.corosync_multicast_ports = ports
obj = handle_host.HandleHost()
ret = obj._check_hb_line()
self.assertEqual(2, 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')
@mock.patch.object(handle_host.HandleHost, '_check_pacemaker_services')
def test_check_hb_line_conf_ports_is_none(
self, mock_check_pacemaker_services):
mock_check_pacemaker_services.side_effect = [True, True, False]
interfaces = "enp0s8"
ports = None
CONF.host.corosync_multicast_interfaces = interfaces
CONF.host.corosync_multicast_ports = ports
obj = handle_host.HandleHost()
ret = obj._check_hb_line()
self.assertEqual(2, 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')
@mock.patch.object(handle_host.HandleHost, '_check_pacemaker_services')
def test_check_hb_line_incorrect_interfaces(
self, mock_check_pacemaker_services):
mock_check_pacemaker_services.side_effect = [True, True, False]
interfaces = "enp0s3,enp0s8"
ports = "5405"
CONF.host.corosync_multicast_interfaces = interfaces
CONF.host.corosync_multicast_ports = ports
obj = handle_host.HandleHost()
ret = obj._check_hb_line()
self.assertEqual(2, 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')
@mock.patch.object(handle_host.HandleHost, '_check_pacemaker_services')
def test_check_hb_line_incorrect_ports(
self, mock_check_pacemaker_services):
mock_check_pacemaker_services.side_effect = [True, True, False]
interfaces = "enp0s8"
ports = "5405,5406"
CONF.host.corosync_multicast_interfaces = interfaces
CONF.host.corosync_multicast_ports = ports
obj = handle_host.HandleHost()
ret = obj._check_hb_line()
self.assertEqual(2, 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')
@mock.patch.object(utils, 'execute')
@mock.patch.object(handle_host.HandleHost, '_check_pacemaker_services')
def test_check_hb_line_tcpdump_fail(
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.side_effect = Exception("Test exception.")
obj = handle_host.HandleHost()
ret = obj._check_hb_line()
self.assertEqual(1, 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)
@mock.patch.object(utils, 'execute')
def test_check_host_status_by_crmadmin_idle(self, mock_execute):
my_hostname = socket.gethostname()
crmadmin_stdout = "Status of crmd@%s: S_IDLE (ok)" % my_hostname
mock_execute.return_value = (crmadmin_stdout, '')
obj = handle_host.HandleHost()
ret = obj._check_host_status_by_crmadmin()
self.assertEqual(0, ret)
mock_execute.assert_called_once_with(
'crmadmin', '-S', my_hostname, run_as_root=True)
@mock.patch.object(utils, 'execute')
def test_check_host_status_by_crmadmin_not_dc(self, mock_execute):
my_hostname = socket.gethostname()
crmadmin_stdout = "Status of crmd@%s: S_NOT_DC (ok)" % my_hostname
mock_execute.return_value = (crmadmin_stdout, '')
obj = handle_host.HandleHost()
ret = obj._check_host_status_by_crmadmin()
self.assertEqual(0, ret)
mock_execute.assert_called_once_with(
'crmadmin', '-S', my_hostname, run_as_root=True)
@mock.patch.object(utils, 'execute')
def test_check_host_status_by_crmadmin_output_stderr(self, mock_execute):
my_hostname = socket.gethostname()
mock_execute.return_value = ('', 'crmadmin: command not found')
obj = handle_host.HandleHost()
ret = obj._check_host_status_by_crmadmin()
self.assertEqual(1, ret)
mock_execute.assert_called_once_with(
'crmadmin', '-S', my_hostname, run_as_root=True)
@mock.patch.object(utils, 'execute')
def test_check_host_status_by_crmadmin_output_unexpected_message(
self, mock_execute):
my_hostname = socket.gethostname()
mock_execute.return_value = ('Unexpected message.', '')
obj = handle_host.HandleHost()
ret = obj._check_host_status_by_crmadmin()
self.assertEqual(1, ret)
mock_execute.assert_called_once_with(
'crmadmin', '-S', my_hostname, run_as_root=True)
@mock.patch.object(utils, 'execute')
def test_get_cib_xml(self, mock_execute):
mock_execute.return_value = ('test_stdout', '')
obj = handle_host.HandleHost()
ret = obj._get_cib_xml()
self.assertEqual('test_stdout', ret)
mock_execute.assert_called_once_with(
'cibadmin', '--query', run_as_root=True)
@mock.patch.object(utils, 'execute')
def test_get_cib_xml_output_stderr(self, mock_execute):
mock_execute.return_value = ('test_stdout', 'test_stderr')
obj = handle_host.HandleHost()
ret = obj._get_cib_xml()
self.assertIsNone(ret)
mock_execute.assert_called_once_with(
'cibadmin', '--query', run_as_root=True)
@mock.patch.object(utils, 'execute')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'get_stonith_ipmi_params')
def test_is_poweroff(self, mock_get_stonith_ipmi_params, mock_execute):
ipmi_values = {
'userid': 'admin',
'passwd': 'password',
'interface': 'lanplus',
'ipaddr': '0.0.0.0'
}
mock_get_stonith_ipmi_params.return_value = ipmi_values
mock_execute.return_value = ('Chassis Power is off', '')
obj = handle_host.HandleHost()
ret = obj._is_poweroff('test_hostname')
self.assertTrue(ret)
cmd_str = ("timeout %s ipmitool -U %s -P %s -I %s -H %s "
"power status") \
% (str(CONF.host.ipmi_timeout), ipmi_values['userid'],
ipmi_values['passwd'], ipmi_values['interface'],
ipmi_values['ipaddr'])
command = cmd_str.split(' ')
mock_execute.assert_called_once_with(*command, run_as_root=False)
@mock.patch.object(utils, 'execute')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'get_stonith_ipmi_params')
def test_is_poweroff_ipmi_values_is_none(
self, mock_get_stonith_ipmi_params, mock_execute):
mock_get_stonith_ipmi_params.return_value = None
mock_execute.return_value = ('Chassis Power is off', '')
obj = handle_host.HandleHost()
ret = obj._is_poweroff('test_hostname')
self.assertFalse(ret)
mock_execute.assert_not_called()
@mock.patch.object(eventlet.greenthread, 'sleep')
@mock.patch.object(utils, 'execute')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'get_stonith_ipmi_params')
def test_is_poweroff_output_stderr(
self, mock_get_stonith_ipmi_params, mock_execute, mock_sleep):
ipmi_values = {
'userid': 'admin',
'passwd': 'password',
'interface': 'lanplus',
'ipaddr': '0.0.0.0'
}
mock_get_stonith_ipmi_params.return_value = ipmi_values
mock_execute.return_value = ('Chassis Power is off', 'test_stderr')
mock_sleep.return_value = None
obj = handle_host.HandleHost()
ret = obj._is_poweroff('test_hostname')
self.assertFalse(ret)
cmd_str = ("timeout %s ipmitool -U %s -P %s -I %s -H %s "
"power status") \
% (str(CONF.host.ipmi_timeout), ipmi_values['userid'],
ipmi_values['passwd'], ipmi_values['interface'],
ipmi_values['ipaddr'])
command = cmd_str.split(' ')
calls = [mock.call(*command, run_as_root=False),
mock.call(*command, run_as_root=False),
mock.call(*command, run_as_root=False),
mock.call(*command, run_as_root=False)]
mock_execute.assert_has_calls(calls)
@mock.patch.object(eventlet.greenthread, 'sleep')
@mock.patch.object(utils, 'execute')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'get_stonith_ipmi_params')
def test_is_poweroff_output_unexpected_message(
self, mock_get_stonith_ipmi_params, mock_execute, mock_sleep):
ipmi_values = {
'userid': 'admin',
'passwd': 'password',
'interface': 'lanplus',
'ipaddr': '0.0.0.0'
}
mock_get_stonith_ipmi_params.return_value = ipmi_values
mock_execute.return_value = ('Unexpected message.', '')
mock_sleep.return_value = None
obj = handle_host.HandleHost()
ret = obj._is_poweroff('test_hostname')
self.assertFalse(ret)
cmd_str = ("timeout %s ipmitool -U %s -P %s -I %s -H %s "
"power status") \
% (str(CONF.host.ipmi_timeout), ipmi_values['userid'],
ipmi_values['passwd'], ipmi_values['interface'],
ipmi_values['ipaddr'])
command = cmd_str.split(' ')
calls = [mock.call(*command, run_as_root=False),
mock.call(*command, run_as_root=False),
mock.call(*command, run_as_root=False),
mock.call(*command, run_as_root=False)]
mock_execute.assert_has_calls(calls)
@mock.patch.object(timeutils, 'utcnow')
def test_make_event_online(self, mock_utcnow):
current_time = timeutils.utcnow()
mock_utcnow.return_value = current_time
obj = handle_host.HandleHost()
hostname = 'test_hostname'
current_status = 'online'
ret = obj._make_event(hostname, current_status)
event_type = ec.EventConstants.EVENT_STARTED
cluster_status = current_status.upper()
host_status = ec.EventConstants.HOST_STATUS_NORMAL
event = {
'notification': {
'type': ec.EventConstants.TYPE_COMPUTE_HOST,
'hostname': hostname,
'generated_time': current_time,
'payload': {
'event': event_type,
'cluster_status': cluster_status,
'host_status': host_status
}
}
}
self.assertEqual(event, ret)
@mock.patch.object(timeutils, 'utcnow')
@mock.patch.object(handle_host.HandleHost, '_is_poweroff')
def test_make_event_offline_disable_ipmi_check_false_poweroff(
self, mock_is_poweroff, mock_utcnow):
mock_is_poweroff.return_value = True
current_time = timeutils.utcnow()
mock_utcnow.return_value = current_time
CONF.host.disable_ipmi_check = False
obj = handle_host.HandleHost()
hostname = 'test_hostname'
current_status = 'offline'
ret = obj._make_event(hostname, current_status)
event_type = ec.EventConstants.EVENT_STOPPED
cluster_status = current_status.upper()
host_status = ec.EventConstants.HOST_STATUS_NORMAL
event = {
'notification': {
'type': ec.EventConstants.TYPE_COMPUTE_HOST,
'hostname': hostname,
'generated_time': current_time,
'payload': {
'event': event_type,
'cluster_status': cluster_status,
'host_status': host_status
}
}
}
self.assertEqual(event, ret)
mock_is_poweroff.assert_called_once_with(hostname)
@mock.patch.object(timeutils, 'utcnow')
@mock.patch.object(handle_host.HandleHost, '_is_poweroff')
def test_make_event_offline_disable_ipmi_check_false_poweron(
self, mock_is_poweroff, mock_utcnow):
mock_is_poweroff.return_value = False
current_time = timeutils.utcnow()
mock_utcnow.return_value = current_time
CONF.host.disable_ipmi_check = False
obj = handle_host.HandleHost()
hostname = 'test_hostname'
current_status = 'offline'
ret = obj._make_event(hostname, current_status)
event_type = ec.EventConstants.EVENT_STOPPED
cluster_status = current_status.upper()
host_status = ec.EventConstants.HOST_STATUS_UNKNOWN
event = {
'notification': {
'type': ec.EventConstants.TYPE_COMPUTE_HOST,
'hostname': hostname,
'generated_time': current_time,
'payload': {
'event': event_type,
'cluster_status': cluster_status,
'host_status': host_status
}
}
}
self.assertEqual(event, ret)
mock_is_poweroff.assert_called_once_with(hostname)
@mock.patch.object(timeutils, 'utcnow')
def test_make_event_offline_disable_ipmi_check_true(self, mock_utcnow):
current_time = timeutils.utcnow()
mock_utcnow.return_value = current_time
CONF.host.disable_ipmi_check = True
obj = handle_host.HandleHost()
hostname = 'test_hostname'
current_status = 'offline'
ret = obj._make_event(hostname, current_status)
event_type = ec.EventConstants.EVENT_STOPPED
cluster_status = current_status.upper()
host_status = ec.EventConstants.HOST_STATUS_NORMAL
event = {
'notification': {
'type': ec.EventConstants.TYPE_COMPUTE_HOST,
'hostname': hostname,
'generated_time': current_time,
'payload': {
'event': event_type,
'cluster_status': cluster_status,
'host_status': host_status
}
}
}
self.assertEqual(event, ret)
@mock.patch.object(masakari.SendNotification, 'send_notification')
@mock.patch.object(handle_host.HandleHost, '_make_event')
@mock.patch.object(hold_host_status.HostHoldStatus, 'set_host_status')
@mock.patch.object(hold_host_status.HostHoldStatus, 'get_host_status')
@mock.patch.object(socket, 'gethostname')
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)
@mock.patch.object(handle_host.HandleHost, '_check_if_status_changed')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'get_node_state_tag_list')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'have_quorum')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'set_cib_xml')
@mock.patch.object(utils, 'execute')
@mock.patch.object(handle_host.HandleHost, '_get_cib_xml')
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 = status_tag.getchildren()
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)
@mock.patch.object(handle_host.HandleHost, '_check_if_status_changed')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'get_node_state_tag_list')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'have_quorum')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'set_cib_xml')
@mock.patch.object(handle_host.HandleHost, '_get_cib_xml')
def test_check_host_status_by_cibadmin_no_quorum(
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 = 0
status_tag = ElementTree.fromstring(STATUS_TAG_XML)
node_state_tag_list = status_tag.getchildren()
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)
@mock.patch.object(handle_host.HandleHost, '_get_cib_xml')
def test_check_host_status_by_cibadmin_cib_xml_is_None(
self, mock_get_cib_xml):
mock_get_cib_xml.return_value = None
obj = handle_host.HandleHost()
ret = obj._check_host_status_by_cibadmin()
self.assertEqual(1, ret)
mock_get_cib_xml.assert_called_once_with()
@mock.patch.object(parse_cib_xml.ParseCibXml, 'get_node_state_tag_list')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'have_quorum')
@mock.patch.object(parse_cib_xml.ParseCibXml, 'set_cib_xml')
@mock.patch.object(handle_host.HandleHost, '_get_cib_xml')
def test_check_host_status_by_cibadmin_not_have_node_state_tag(
self, mock_get_cib_xml, mock_set_cib_xml, mock_have_quorum,
mock_get_node_state_tag_list):
mock_get_cib_xml.return_value = STATUS_TAG_XML
mock_set_cib_xml.return_value = None
mock_have_quorum.return_value = 1
mock_get_node_state_tag_list.return_value = []
obj = handle_host.HandleHost()
self.assertRaisesRegexp(
Exception, "Failed to get node_state tag from cib xml.",
obj._check_host_status_by_cibadmin)
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()
def test_stop(self):
obj = handle_host.HandleHost()
obj.stop()
self.assertFalse(obj.running)
@mock.patch.object(eventlet.greenthread, 'sleep')
@mock.patch.object(handle_host.HandleHost,
'_check_host_status_by_cibadmin')
@mock.patch.object(handle_host.HandleHost,
'_check_host_status_by_crmadmin')
@mock.patch.object(handle_host.HandleHost, '_check_pacemaker_services')
@mock.patch.object(handle_host.HandleHost, '_check_hb_line')
def test_monitor_hosts(self,
mock_check_hb_line,
mock_check_pacemaker_services,
mock_execute,
mock_set_cib_xml,
mock_have_quorum,
mock_get_node_state_tag_list):
mock_check_host_status_by_cibadmin,
mock_check_host_status_by_crmadmin,
mock_sleep):
mock_check_hb_line.side_effect = \
[0, 1, 2, 0, Exception("Test exception.")]
mock_check_pacemaker_services.side_effect = [True, False, False]
mock_check_host_status_by_cibadmin.side_effect = [0, 1]
mock_check_host_status_by_crmadmin.side_effect = [0, 1]
mock_sleep.return_value = None
obj = handle_host.HandleHost()
obj.monitor_hosts()
mock_check_hb_line.return_value = 0
mock_check_pacemaker_services.return_value = False
mock_execute.return_value = (EXECUTE_RETURN, '')
mock_set_cib_xml.return_value = None
mock_have_quorum.return_value = 0
mock_get_node_state_tag_list.return_value = []
ret = obj.monitor_hosts()
self.assertEqual(None, ret)
@mock.patch.object(utils, 'execute')
def test_check_hb_line(self,
mock_execute):
obj = handle_host.HandleHost()
mock_execute.return_value = ('', '')
ret = obj._check_hb_line()
self.assertEqual(2, ret)
self.assertEqual(5, mock_check_hb_line.call_count)
self.assertEqual(3, mock_check_pacemaker_services.call_count)
mock_check_pacemaker_services.assert_called_with('pacemaker_remote')
self.assertEqual(2, mock_check_host_status_by_cibadmin.call_count)
self.assertEqual(2, mock_check_host_status_by_crmadmin.call_count)

View File

@ -22,7 +22,7 @@ from masakarimonitors.hostmonitor.host_handler import hold_host_status
eventlet.monkey_patch(os=False)
NODE_STATE_XML = '<node_state uname="masakari-node" crmd="online">' \
' <test hoge="hoge"/>' \
' <test foo="foo"/>' \
'</node_state>'
NODE_STATE_TAG = ElementTree.fromstring(NODE_STATE_XML)

View File

@ -87,16 +87,16 @@ CIB_XML = '<cib have-quorum="1">' \
' </resources>' \
' <constraints>' \
' <rsc_location id="loc_grpStonith1" rsc="grpStonith1">' \
' <rule test="hoge"/>' \
' <rule test="foo"/>' \
' </rsc_location>' \
' </constraints>' \
' </configuration>' \
' <status>' \
' <node_state uname="masakari-node" crmd="online">' \
' <test hoge="hoge"/>' \
' <test foo="foo"/>' \
' </node_state>' \
' <node_state crmd="online" uname="compute-node">' \
' <test hoge="hoge"/>' \
' <test foo="foo"/>' \
' </node_state>' \
' </status>' \
'</cib>'

View File

@ -0,0 +1,119 @@
# Copyright(c) 2017 Nippon Telegraph and Telephone Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import mock
import os
import testtools
from stevedore import driver
from masakarimonitors.hostmonitor import host
class TestHostmonitorManager(testtools.TestCase):
def setUp(self):
super(TestHostmonitorManager, self).setUp()
@mock.patch.object(driver, 'DriverManager')
def test_init_host(self, mock_DriverManager):
mock_driver = mock.Mock()
mock_DriverManager.return_value = mock_driver
host_manager = host.HostmonitorManager()
host_manager.init_host()
mock_DriverManager.assert_called_once_with(
namespace='hostmonitor.driver',
name='default',
invoke_on_load=True,
invoke_args=(),
)
@mock.patch.object(os, '_exit')
@mock.patch.object(driver, 'DriverManager')
def test_init_host_exception(self, mock_DriverManager, mock_exit):
mock_DriverManager.side_effect = Exception("Test exception.")
mock_exit.return_value = None
host_manager = host.HostmonitorManager()
host_manager.init_host()
mock_DriverManager.assert_called_once_with(
namespace='hostmonitor.driver',
name='default',
invoke_on_load=True,
invoke_args=(),
)
mock_exit.assert_called_once_with(1)
@mock.patch.object(driver, 'DriverManager')
def test_stop(self, mock_DriverManager):
mock_driver = mock.Mock()
mock_DriverManager.return_value = mock_driver
host_manager = host.HostmonitorManager()
host_manager.init_host()
host_manager.stop()
mock_DriverManager.assert_called_once_with(
namespace='hostmonitor.driver',
name='default',
invoke_on_load=True,
invoke_args=(),
)
mock_driver.driver.stop.assert_called_once_with()
@mock.patch.object(driver, 'DriverManager')
def test_main(self, mock_DriverManager):
mock_driver = mock.Mock()
mock_DriverManager.return_value = mock_driver
host_manager = host.HostmonitorManager()
host_manager.init_host()
ret = host_manager.main()
mock_DriverManager.assert_called_once_with(
namespace='hostmonitor.driver',
name='default',
invoke_on_load=True,
invoke_args=(),
)
mock_driver.driver.monitor_hosts.assert_called_once_with()
self.assertIsNone(ret)
@mock.patch.object(driver, 'DriverManager')
def test_main_exception(self, mock_DriverManager):
mock_driver = mock.Mock()
mock_DriverManager.return_value = mock_driver
mock_driver.driver.monitor_hosts.side_effect = \
Exception("Test exception.")
host_manager = host.HostmonitorManager()
host_manager.init_host()
ret = host_manager.main()
mock_DriverManager.assert_called_once_with(
namespace='hostmonitor.driver',
name='default',
invoke_on_load=True,
invoke_args=(),
)
mock_driver.driver.monitor_hosts.assert_called_once_with()
self.assertIsNone(ret)