def test_create_resource_provider_already_exists(self): utils.create_test_resource_provider( context=self.context, uuid='123') with self.assertRaisesRegex(exception.ResourceProviderAlreadyExists, 'A resource provider with UUID 123.*'): utils.create_test_resource_provider( context=self.context, uuid='123')
def test_list_resource_providers_with_filters(self): provider1 = utils.create_test_resource_provider( name='provider-one', uuid=uuidutils.generate_uuid(), context=self.context) provider2 = utils.create_test_resource_provider( name='provider-two', uuid=uuidutils.generate_uuid(), context=self.context) res = dbapi.list_resource_providers( self.context, filters={'name': 'provider-one'}) self.assertEqual([provider1.id], [r.id for r in res]) res = dbapi.list_resource_providers( self.context, filters={'name': 'provider-two'}) self.assertEqual([provider2.id], [r.id for r in res]) res = dbapi.list_resource_providers( self.context, filters={'name': 'bad-provider'}) self.assertEqual([], [r.id for r in res]) res = dbapi.list_resource_providers( self.context, filters={'name': provider1.name}) self.assertEqual([provider1.id], [r.id for r in res])
def test_list_resource_providers_with_filters(self): provider1 = utils.create_test_resource_provider( name='provider-one', uuid=uuidutils.generate_uuid(), context=self.context) provider2 = utils.create_test_resource_provider( name='provider-two', uuid=uuidutils.generate_uuid(), context=self.context) res = dbapi.list_resource_providers(self.context, filters={'name': 'provider-one'}) self.assertEqual([provider1.id], [r.id for r in res]) res = dbapi.list_resource_providers(self.context, filters={'name': 'provider-two'}) self.assertEqual([provider2.id], [r.id for r in res]) res = dbapi.list_resource_providers(self.context, filters={'name': 'bad-provider'}) self.assertEqual([], [r.id for r in res]) res = dbapi.list_resource_providers(self.context, filters={'name': provider1.name}) self.assertEqual([provider1.id], [r.id for r in res])
def test_get_inventory_by_id(self): provider = utils.create_test_resource_provider( context=self.context) inventory = utils.create_test_inventory( resource_provider_id=provider.id, context=self.context) res = dbapi.get_inventory(self.context, inventory.id) self.assertEqual(inventory.id, res.id)
def test_list_inventories_with_filters(self): provider = utils.create_test_resource_provider( context=self.context) inventory1 = utils.create_test_inventory( total=10, resource_provider_id=provider.id, resource_class_id=1, context=self.context) inventory2 = utils.create_test_inventory( total=20, resource_provider_id=provider.id, resource_class_id=2, context=self.context) res = dbapi.list_inventories( self.context, filters={'total': 10}) self.assertEqual([inventory1.id], [r.id for r in res]) res = dbapi.list_inventories( self.context, filters={'total': 20}) self.assertEqual([inventory2.id], [r.id for r in res]) res = dbapi.list_inventories( self.context, filters={'total': 11111}) self.assertEqual([], [r.id for r in res]) res = dbapi.list_inventories( self.context, filters={'total': inventory1.total}) self.assertEqual([inventory1.id], [r.id for r in res])
def test_list_allocations_with_filters(self): provider = utils.create_test_resource_provider( id=1, uuid=uuidutils.generate_uuid(), context=self.context) allocation1 = utils.create_test_allocation( used=0, resource_provider_id=provider.id, context=self.context) allocation2 = utils.create_test_allocation( used=1, resource_provider_id=provider.id, context=self.context) res = dbapi.list_allocations( self.context, filters={'used': 0}) self.assertEqual([allocation1.id], [r.id for r in res]) res = dbapi.list_allocations( self.context, filters={'used': 1}) self.assertEqual([allocation2.id], [r.id for r in res]) res = dbapi.list_allocations( self.context, filters={'used': 11111}) self.assertEqual([], [r.id for r in res]) res = dbapi.list_allocations( self.context, filters={'used': allocation1.used}) self.assertEqual([allocation1.id], [r.id for r in res])
def test_get_allocation_by_id(self): provider = utils.create_test_resource_provider( context=self.context) allocation = utils.create_test_allocation( resource_provider_id=provider.id, context=self.context) res = dbapi.get_allocation(self.context, allocation.id) self.assertEqual(allocation.id, res.id)
def test_update_resource_provider(self): provider = utils.create_test_resource_provider(context=self.context) old_name = provider.name new_name = 'new-name' self.assertNotEqual(old_name, new_name) res = dbapi.update_resource_provider(self.context, provider.id, {'name': new_name}) self.assertEqual(new_name, res.name)
def test_update_resource_provider(self): provider = utils.create_test_resource_provider(context=self.context) old_name = provider.name new_name = 'new-name' self.assertNotEqual(old_name, new_name) res = dbapi.update_resource_provider( self.context, provider.id, {'name': new_name}) self.assertEqual(new_name, res.name)
def test_list_resource_providers(self): uuids = [] for i in range(1, 6): provider = utils.create_test_resource_provider( uuid=uuidutils.generate_uuid(), context=self.context, name='provider' + str(i)) uuids.append(six.text_type(provider['uuid'])) res = dbapi.list_resource_providers(self.context) res_uuids = [r.uuid for r in res] self.assertEqual(sorted(uuids), sorted(res_uuids))
def test_list_inventories(self): totals = [] for i in range(1, 6): provider = utils.create_test_resource_provider( id=i, uuid=uuidutils.generate_uuid(), context=self.context) inventory = utils.create_test_inventory( id=i, resource_provider_id=provider.id, total=i, context=self.context) totals.append(inventory['total']) res = dbapi.list_inventories(self.context) res_totals = [r.total for r in res] self.assertEqual(sorted(totals), sorted(res_totals))
def test_list_allocations(self): cids = [] for i in range(1, 6): provider = utils.create_test_resource_provider( id=i, uuid=uuidutils.generate_uuid(), context=self.context) allocation = utils.create_test_allocation( id=i, resource_provider_id=provider.id, consumer_id=uuidutils.generate_uuid(), context=self.context) cids.append(allocation['consumer_id']) res = dbapi.list_allocations(self.context) res_cids = [r.consumer_id for r in res] self.assertEqual(sorted(cids), sorted(res_cids))
def test_list_resource_providers_sorted(self): uuids = [] for i in range(5): provider = utils.create_test_resource_provider( uuid=uuidutils.generate_uuid(), context=self.context, name='provider' + str(i)) uuids.append(six.text_type(provider.uuid)) res = dbapi.list_resource_providers(self.context, sort_key='uuid') res_uuids = [r.uuid for r in res] self.assertEqual(sorted(uuids), res_uuids) self.assertRaises(exception.InvalidParameterValue, dbapi.list_resource_providers, self.context, sort_key='foo')
def test_list_allocations_sorted(self): cids = [] for i in range(5): provider = utils.create_test_resource_provider( id=i, uuid=uuidutils.generate_uuid(), context=self.context) allocation = utils.create_test_allocation( id=i, resource_provider_id=provider.id, consumer_id=uuidutils.generate_uuid(), context=self.context) cids.append(allocation['consumer_id']) res = dbapi.list_allocations(self.context, sort_key='consumer_id') res_cids = [r.consumer_id for r in res] self.assertEqual(sorted(cids), res_cids) self.assertRaises(exception.InvalidParameterValue, dbapi.list_allocations, self.context, sort_key='foo')
def test_list_inventories_sorted(self): totals = [] for i in range(5): provider = utils.create_test_resource_provider( id=i, uuid=uuidutils.generate_uuid(), context=self.context) inventory = utils.create_test_inventory( id=i, resource_provider_id=provider.id, total=10 - i, context=self.context) totals.append(inventory['total']) res = dbapi.list_inventories(self.context, sort_key='total') res_totals = [r.total for r in res] self.assertEqual(sorted(totals), res_totals) self.assertRaises(exception.InvalidParameterValue, dbapi.list_inventories, self.context, sort_key='foo')
def test_list_allocations_with_filters(self): provider = utils.create_test_resource_provider( id=1, uuid=uuidutils.generate_uuid(), context=self.context) allocation1 = utils.create_test_allocation( used=0, resource_provider_id=provider.id, context=self.context) allocation2 = utils.create_test_allocation( used=1, resource_provider_id=provider.id, context=self.context) res = dbapi.list_allocations(self.context, filters={'used': 0}) self.assertEqual([allocation1.id], [r.id for r in res]) res = dbapi.list_allocations(self.context, filters={'used': 1}) self.assertEqual([allocation2.id], [r.id for r in res]) res = dbapi.list_allocations(self.context, filters={'used': 11111}) self.assertEqual([], [r.id for r in res]) res = dbapi.list_allocations(self.context, filters={'used': allocation1.used}) self.assertEqual([allocation1.id], [r.id for r in res])
def test_destroy_resource_provider_by_uuid(self): provider = utils.create_test_resource_provider(context=self.context) dbapi.destroy_resource_provider(self.context, provider.uuid) self.assertRaises(exception.ResourceProviderNotFound, dbapi.get_resource_provider, self.context, provider.uuid)
def test_update_resource_provider_uuid(self): provider = utils.create_test_resource_provider(context=self.context) self.assertRaises(exception.InvalidParameterValue, dbapi.update_resource_provider, self.context, provider.id, {'uuid': ''})
def test_get_resource_provider_by_name(self): provider = utils.create_test_resource_provider(context=self.context) res = dbapi.get_resource_provider(self.context, provider.name) self.assertEqual(provider.id, res.id) self.assertEqual(provider.uuid, res.uuid)
def test_create_resource_provider(self): utils.create_test_resource_provider(context=self.context)
def test_get_resource_provider_by_name(self): provider = utils.create_test_resource_provider(context=self.context) res = dbapi.get_resource_provider( self.context, provider.name) self.assertEqual(provider.id, res.id) self.assertEqual(provider.uuid, res.uuid)
def test_get_allocation_by_id(self): provider = utils.create_test_resource_provider(context=self.context) allocation = utils.create_test_allocation( resource_provider_id=provider.id, context=self.context) res = dbapi.get_allocation(self.context, allocation.id) self.assertEqual(allocation.id, res.id)