def setUp(self): core.initialize() core.ModelBase.metadata.create_all(core.get_engine()) self.context = context.Context() site_dict = { 'site_id': FAKE_SITE_ID, 'site_name': FAKE_SITE_NAME, 'az_id': FAKE_AZ } type_dict = { 'id': 1, 'service_type': FAKE_TYPE } config_dict = { 'service_id': FAKE_SERVICE_ID, 'site_id': FAKE_SITE_ID, 'service_name': FAKE_SERVICE_NAME, 'service_type': FAKE_TYPE, 'service_url': FAKE_URL } models.create_site(self.context, site_dict) models.create_service_type(self.context, type_dict) models.create_site_service_configuration(self.context, config_dict) global FAKE_RESOURCES FAKE_RESOURCES = [{'name': 'res1'}, {'name': 'res2'}] cfg.CONF.set_override(name='top_site_name', override=FAKE_SITE_NAME, group='client') self.client = client.Client() self.client.resource_service_map[FAKE_RESOURCE] = FAKE_TYPE self.client.operation_resources_map['list'].add(FAKE_RESOURCE) self.client.operation_resources_map['create'].add(FAKE_RESOURCE) self.client.operation_resources_map['delete'].add(FAKE_RESOURCE) self.client.service_handle_map[FAKE_TYPE] = FakeResHandle(None)
def test_create(self): site = {'site_id': 'test_site_uuid', 'site_name': 'test_site', 'az_id': 'test_az_uuid'} site_ret = models.create_site(self.context, site) self.assertEqual(site_ret, site) service_type = {'id': 1, 'service_type': 'nova'} type_ret = models.create_service_type(self.context, service_type) self.assertEqual(type_ret, service_type) configuration = { 'service_id': 'test_config_uuid', 'site_id': 'test_site_uuid', 'service_name': 'nova_service', 'service_type': 'nova', 'service_url': 'http://test_url' } config_ret = models.create_site_service_configuration(self.context, configuration) self.assertEqual(config_ret, configuration)