def test_removes_key_and_value(self):
db = Database(self.engine, self.config)
db.put(r"key1", r"value1")
self.assertTrue(db.exists(r"key1"))
self.assertEqual(db.get(r"key1"), r"value1")
self.assertTrue(db.remove(r"key1"))
self.assertFalse(db.remove(r"key1"))
self.assertFalse(db.exists(r"key1"))
self.assertEqual(db.get(r"key1"), None)
db.stop()
def test_removes_key_and_value(self):
db = Database(self.engine, self.config)
db.put(r"key1", r"value1")
self.assertTrue(db.exists(r"key1"))
self.assertEqual(db.get_string(r"key1"), r"value1")
self.assertTrue(db.remove(r"key1"))
self.assertFalse(db.remove(r"key1"))
self.assertFalse(db.exists(r"key1"))
with self.assertRaises(KeyError):
db.get_string(r"key1")
db.stop()
def test_get_copy_to_class_member(self):
class Callback:
def __init__(self):
self.result = None
def __call__(self, key):
self.result = bytes(key)
callback = Callback()
db = Database(self.engine, self.config)
key = "dict_test"
val = "123"
db[key] = val
db.get(key, callback)
db.remove(key)
db[key] = "Some other string"
self.assertEqual(callback.result.decode(), val)
db.stop()
def test_blackhole_engine(self):
db = Database(r"blackhole", self.config)
self.assertEqual(db.count_all(), 0)
self.assertFalse(db.exists(r"key1"))
self.assertEqual(db.get(r"key1"), None)
db.put(r"key1", r"value123")
self.assertEqual(db.count_all(), 0)
self.assertFalse(db.exists(r"key1"))
self.assertEqual(db.get(r"key1"), None)
self.assertTrue(db.remove(r"key1"))
self.assertFalse(db.exists(r"key1"))
self.assertEqual(db.get(r"key1"), None)
db.stop()
def test_blackhole_engine(self):
db = Database(r"blackhole", self.config)
self.assertEqual(db.count_all(), 0)
self.assertFalse(db.exists(r"key1"))
with self.assertRaises(KeyError):
db.get_string(r"key1")
db.put(r"key1", r"value123")
self.assertEqual(db.count_all(), 0)
self.assertFalse(db.exists(r"key1"))
with self.assertRaises(KeyError):
db.get_string(r"key1")
self.assertTrue(db.remove(r"key1"))
self.assertFalse(db.exists(r"key1"))
with self.assertRaises(KeyError):
db.get_string(r"key1")
db.stop()
