def test_remove(self):
""" Test normal removal. """
s = MemoryKeyValueStore()
s._table = {
0: 1,
'a': 'b',
}
s.remove(0)
self.assertDictEqual(s._table, {'a': 'b'})
def test_remove(self):
""" Test normal removal. """
s = MemoryKeyValueStore()
s._table = {
0: 1,
'a': 'b',
}
s.remove(0)
self.assertDictEqual(s._table, {'a': 'b'})
def test_remove_with_cache(self):
"""
Test that removal correctly updates the cache element.
"""
existing_data = {
0: 1,
'a': 'b',
}
c = DataMemoryElement(pickle.dumps(existing_data))
s = MemoryKeyValueStore(c)
self.assertDictEqual(s._table, existing_data)
s.remove('a')
self.assertDictEqual(s._table, {0: 1})
self.assertDictEqual(pickle.loads(c.get_bytes()), {0: 1})
def test_remove_with_cache(self):
"""
Test that removal correctly updates the cache element.
"""
existing_data = {
0: 1,
'a': 'b',
}
c = DataMemoryElement(pickle.dumps(existing_data))
s = MemoryKeyValueStore(c)
self.assertDictEqual(s._table, existing_data)
s.remove('a')
self.assertDictEqual(s._table, {0: 1})
self.assertDictEqual(pickle.loads(c.get_bytes()),
{0: 1})