class SvcMonitorInitTest(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
@patch.object(ServiceMonitorDB, '_cassandra_init')
def test_vm_instance(self, cassandra_init_mock):
logging.debug("init")
self._api_client = mock.Mock()
arguments = Arguments()
with patch.object(logging.handlers, 'RotatingFileHandler'):
self._svc_monitor = SvcMonitor(arguments)
self._svc_monitor.post_init(self._api_client, arguments)
self._svc_monitor.db._svc_si_cf = {}
self._svc_monitor.db._svc_vm_cf = mock.Mock()
self._svc_monitor._novaclient_get = mock.Mock()
identities = {
'service-template': 'default-domain:test:template1',
'service-instance': 'default-domain:test:service1'
}
tmpl_attr = ServiceTemplateType()
tmpl_attr.service_mode = 'in-network-nat'
tmpl_attr.service_type = 'firewall'
tmpl_attr.image_name = 'test-template'
tmpl_attr.service_virtualization_type = 'virtual-machine'
template = ServiceTemplate(service_template_properties=tmpl_attr)
template.uuid = 'aaa'
svc_attr = ServiceInstanceType()
svc_attr.left_virtual_network = 'default-project:demo:test'
svc_attr.right_virtual_network = 'default-project:admin:public'
svc_attr.scale_out = ServiceScaleOutType()
service = ServiceInstance('test-instance',
service_instance_properties=svc_attr)
service.uuid = 'bbb'
self._api_client.service_template_read.return_value = template
self._api_client.service_instance_read.return_value = service
with contextlib.nested(
patch.object(self._svc_monitor.vm_manager,
'_create_svc_vm'),
patch.object(self._svc_monitor.db,
'service_instance_insert')) as (create_svc_vm, svc_insert):
class Vm(object):
@property
def id(self):
return 'ccc'
create_svc_vm.return_value = Vm()
self._svc_monitor.\
_addmsg_service_instance_service_template(identities)
svc_insert.assert_called_with(
'default-domain:default-project:test-instance',
{'vm0-state': 'pending', 'vm0-preference': '0',
'vm0-uuid': 'ccc', 'vm0-vrouter': 'None',
'vm0-name': 'default-domain__default-project__bbb__1'})
def test_vm_instance(self, cassandra_init_mock):
logging.debug("init")
self._api_client = mock.Mock()
arguments = Arguments()
with patch.object(logging.handlers, 'RotatingFileHandler'):
self._svc_monitor = SvcMonitor(arguments)
self._svc_monitor.post_init(self._api_client, arguments)
self._svc_monitor.db._svc_si_cf = {}
self._svc_monitor.db._svc_vm_cf = mock.Mock()
self._svc_monitor._novaclient_get = mock.Mock()
identities = {
'service-template': 'default-domain:test:template1',
'service-instance': 'default-domain:test:service1'
}
tmpl_attr = ServiceTemplateType()
tmpl_attr.service_mode = 'in-network-nat'
tmpl_attr.service_type = 'firewall'
tmpl_attr.image_name = 'test-template'
tmpl_attr.service_virtualization_type = 'virtual-machine'
template = ServiceTemplate(service_template_properties=tmpl_attr)
template.uuid = 'aaa'
svc_attr = ServiceInstanceType()
svc_attr.left_virtual_network = 'default-project:demo:test'
svc_attr.right_virtual_network = 'default-project:admin:public'
svc_attr.scale_out = ServiceScaleOutType()
service = ServiceInstance('test-instance',
service_instance_properties=svc_attr)
service.uuid = 'bbb'
self._api_client.service_template_read.return_value = template
self._api_client.service_instance_read.return_value = service
with contextlib.nested(
patch.object(self._svc_monitor.vm_manager,
'_create_svc_vm'),
patch.object(self._svc_monitor.db,
'service_instance_insert')) as (create_svc_vm,
svc_insert):
class Vm(object):
@property
def id(self):
return 'ccc'
create_svc_vm.return_value = Vm()
self._svc_monitor.\
_addmsg_service_instance_service_template(identities)
svc_insert.assert_called_with(
'default-domain:default-project:test-instance', {
'vm0-state': 'pending',
'vm0-preference': '0',
'vm0-uuid': 'ccc',
'vm0-vrouter': 'None',
'vm0-name': 'default-domain__default-project__bbb__1'
})
class SvcMonitorTest(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
def test_svc_monitor_init(self):
self.args = Arguments()
VncCassandraClient.__init__ = mock.MagicMock()
VncCassandraClient._cf_dict = {'service_instance_table':None, 'pool_table':None}
VncKombuClient.__init__ = mock.MagicMock(return_value=None)
ServiceMonitorLogger.__init__ = mock.MagicMock(return_value=None)
ServiceMonitorLogger.log = mock.MagicMock()
self._svc_monitor = SvcMonitor(self.args)
self.vnc_mock = mock.MagicMock()
def db_list(obj_type):
return (False, None)
DBBaseSM._cassandra = mock.MagicMock()
DBBaseSM._cassandra.list = db_list
self._svc_monitor.post_init(self.vnc_mock, self.args)
def test_svc_monitor_init(self):
self.args = Arguments()
VncCassandraClient.__init__ = mock.MagicMock()
VncCassandraClient._cf_dict = {"service_instance_table": None, "pool_table": None}
VncKombuClient.__init__ = mock.MagicMock(return_value=None)
ServiceMonitorLogger.__init__ = mock.MagicMock(return_value=None)
ServiceMonitorLogger.log = mock.MagicMock()
self._svc_monitor = SvcMonitor(self.args)
self.vnc_mock = mock.MagicMock()
def db_list(obj_type):
return (False, None)
DBBaseSM._cassandra = mock.MagicMock()
DBBaseSM._cassandra.list = mock.MagicMock(side_effect=db_list)
self._svc_monitor.post_init(self.vnc_mock, self.args)
def test_vm_instance(self, uve_mock, cassandra_init_mock):
logging.debug("init")
self._api_client = mock.Mock()
arguments = Arguments()
with patch.object(logging.handlers, 'RotatingFileHandler'):
self._svc_monitor = SvcMonitor(self._api_client, arguments)
self._svc_monitor._svc_si_cf = mock.Mock()
self._svc_monitor._svc_vm_cf = mock.Mock()
self._svc_monitor._novaclient_get = mock.Mock()
identities = {
'service-template': 'default-domain:test:template1',
'service-instance': 'default-domain:test:service1'
}
tmpl_attr = ServiceTemplateType()
tmpl_attr.service_mode = 'in-network-nat'
tmpl_attr.service_type = 'firewall'
tmpl_attr.image_name = 'test-template'
tmpl_attr.service_virtualization_type = 'virtual-machine'
template = ServiceTemplate(service_template_properties=tmpl_attr)
svc_attr = ServiceInstanceType()
svc_attr.left_virtual_network = 'default-project:demo:test'
svc_attr.right_virtual_network = 'default-project:admin:public'
svc_attr.scale_out = ServiceScaleOutType()
service = ServiceInstance('test-instance',
service_instance_properties=svc_attr)
self._api_client.service_template_read.return_value = template
self._api_client.service_instance_read.return_value = service
self._svc_monitor.\
_addmsg_service_instance_service_template(identities)
expected = {
'instance_name': 'test-instance_1',
'si_fq_str': 'default-domain:default-project:test-instance',
'instance_type': 'virtual-machine'
}
self._svc_monitor._svc_vm_cf.insert.assert_called_with(
mock.ANY, expected)
