def test_builds_conditional_prop_exeption(self):
model = MagicMock()
model.db_model = CachingPynamoModel({})
command = CommandTest2Mock(model)
filters = [FilterModel('kind', 'my_operator', 'my_param')]
with self.assertRaises(Base422Exception) as err:
conditions = command._build_conditionals(filters)
self.assertEqual(str(err.exception),
'Model has no property named "kind"')
def test_query_cache(self):
# TODO Stop this test from flapping.
get_mock.reset_mock()
model = CachingPynamoModel({})
CachingPynamoModel.cache = TTLCache(maxsize=128, ttl=0.001)
result = model.query('1234')
self.assertIsInstance(result, QueryResult)
self.assertEqual(len(result.items), 4)
item = model.get(result.items[0].hash_id, result.items[0].range_id)
self.assertEqual(item.kind, 'abcd')
# Should have pulled from cache
get_mock.assert_not_called()
self.assertEqual(len(model.cache), 4)
sleep(.05)
self.assertEqual(len(model.cache), 0)
item = model.get(result.items[0].hash_id, result.items[0].range_id)
self.assertEqual(item, get_mock_result)
get_mock.assert_called_once()
def test_builds_conditional_operator_exeption(self):
model = MagicMock()
model.db_model = CachingPynamoModel({})
setattr(model.db_model, 'kind', {'prop': 'no operators'})
command = CommandTest2Mock(model)
filters = [FilterModel('kind', 'my_operator', 'my_param')]
with self.assertRaises(Base422Exception) as err:
conditions = command._build_conditionals(filters)
self.assertEqual(
str(err.exception),
'Attribute kind does not support the operator "my_operator"')
def test_delete_cache(self):
get_mock.reset_mock()
model = CachingPynamoModel({})
CachingPynamoModel.cache = TTLCache(maxsize=128, ttl=0.001)
key = [random_key(), random_key()]
model._get_keys = MagicMock(return_value=key)
model.save()
item = model.get(*key)
self.assertEqual(item, model)
get_mock.assert_not_called()
model.delete()
item = model.get(*key)
self.assertEqual(item, get_mock_result)
get_mock.assert_called_once()
def test_update_cache(self):
get_mock.reset_mock()
model = CachingPynamoModel({})
CachingPynamoModel.cache = TTLCache(maxsize=128, ttl=0.001)
key = [random_key(), random_key()]
model._get_keys = MagicMock(return_value=key)
model.update(actions=[])
item = model.get(*key)
self.assertEqual(item, model)
get_mock.assert_not_called()
def test_get_cache(self):
get_mock.reset_mock()
model = CachingPynamoModel({})
CachingPynamoModel.cache = TTLCache(maxsize=128, ttl=0.001)
item = model.get(get_mock_result.hash_id, get_mock_result.range_id)
self.assertEqual(item.kind, 'abcd')
get_mock.assert_called_once()
item = model.get(get_mock_result.hash_id, get_mock_result.range_id)
self.assertEqual(item, get_mock_result)
# Should have pulled from cache
get_mock.assert_called_once()
sleep(.005)
model.get(get_mock_result.hash_id, get_mock_result.range_id)
self.assertEqual(get_mock.call_count, 2)
def test_init(self):
model = CachingPynamoModel({})
self.assertIsInstance(model.cache, TTLCache)