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)