def test_rc_cache_custom(self):
"""Test that non-standard, custom resource classes hit the database and
return appropriate results, caching the results after a single
query.
"""
cache = attribute_cache.ResourceClassCache(self.context)
# Haven't added anything to the DB yet, so should raise
# ResourceClassNotFound
self.assertRaises(exception.ResourceClassNotFound,
cache.string_from_id, 1001)
self.assertRaises(exception.ResourceClassNotFound,
cache.id_from_string, "IRON_NFV")
# Now add to the database and verify appropriate results...
with self.placement_db.get_engine().connect() as conn:
ins_stmt = attribute_cache._RC_TBL.insert().values(
id=1001,
name='IRON_NFV'
)
conn.execute(ins_stmt)
self.assertEqual('IRON_NFV', cache.string_from_id(1001))
self.assertEqual(1001, cache.id_from_string('IRON_NFV'))
# Try same again and verify we don't hit the DB.
with mock.patch('sqlalchemy.select') as sel_mock:
self.assertEqual('IRON_NFV', cache.string_from_id(1001))
self.assertEqual(1001, cache.id_from_string('IRON_NFV'))
self.assertFalse(sel_mock.called)
# Verify all fields available from all_from_string
iron_nfv_class = cache.all_from_string('IRON_NFV')
self.assertEqual(1001, iron_nfv_class['id'])
self.assertEqual('IRON_NFV', iron_nfv_class['name'])
# updated_at not set on insert
self.assertIsNone(iron_nfv_class['updated_at'])
self.assertIsInstance(iron_nfv_class['created_at'], datetime.datetime)
# Update IRON_NFV (this is a no-op but will set updated_at)
with self.placement_db.get_engine().connect() as conn:
# NOTE(cdent): When using explict SQL that names columns,
# the automatic timestamp handling provided by the oslo_db
# TimestampMixin is not provided. created_at is a default
# but updated_at is an onupdate.
upd_stmt = attribute_cache._RC_TBL.update().where(
attribute_cache._RC_TBL.c.id == 1001).values(
name='IRON_NFV', updated_at=timeutils.utcnow())
conn.execute(upd_stmt)
# reset cache
cache = attribute_cache.ResourceClassCache(self.context)
iron_nfv_class = cache.all_from_string('IRON_NFV')
# updated_at set on update
self.assertIsInstance(iron_nfv_class['updated_at'], datetime.datetime)
def test_rc_cache_miss(self):
"""Test that we raise ResourceClassNotFound if an unknown resource
class ID or string is searched for.
"""
cache = attribute_cache.ResourceClassCache(self.context)
self.assertRaises(exception.ResourceClassNotFound,
cache.string_from_id, 99999999)
self.assertRaises(exception.ResourceClassNotFound,
cache.id_from_string, 'UNKNOWN')
def test_standard_has_time_fields(self):
cache = attribute_cache.ResourceClassCache(self.context)
vcpu_class = dict(cache.all_from_string('VCPU'))
expected = {'id': 0, 'name': 'VCPU', 'updated_at': None,
'created_at': None}
expected_fields = sorted(expected.keys())
self.assertEqual(expected_fields, sorted(vcpu_class.keys()))
self.assertEqual(0, vcpu_class['id'])
self.assertEqual('VCPU', vcpu_class['name'])
def test_rc_cache_std_db(self):
"""Test that looking up either an ID or a string in the resource class
cache for a standardized resource class doesn't result in a DB call
once the cache is initialized
"""
cache = attribute_cache.ResourceClassCache(self.context)
cache._refresh_from_db(self.context)
with mock.patch('sqlalchemy.select') as sel_mock:
self.assertEqual('VCPU', cache.string_from_id(0))
self.assertEqual('MEMORY_MB', cache.string_from_id(1))
self.assertEqual(0, cache.id_from_string('VCPU'))
self.assertEqual(1, cache.id_from_string('MEMORY_MB'))
self.assertFalse(sel_mock.called)
def __init__(self, *args, **kwargs):
self.config = kwargs.pop('config', None)
self.rc_cache = attribute_cache.ResourceClassCache(self)
self.trait_cache = attribute_cache.TraitCache(self)
super(RequestContext, self).__init__(*args, **kwargs)