def test_exists_many(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) asm.put(("a", "b", 1), {"a": 1}) asm.put(("a", "b", 2), {"a": 2}) asm.put(("a", "b", 3), {"a": 3}) asm.put(("a", "b", 4), {"a": 4}) keys = [ ("a", "b", 1), ("a", "b", 2), ("a", "b", 3), ("a", "b", 4), ("a", "b", 5), ] self.assertEqual([(('a', 'b', 1), { 'gen': 1, 'ttl': self.get_time(default_ttl) }), (('a', 'b', 2), { 'gen': 1, 'ttl': self.get_time(default_ttl) }), (('a', 'b', 3), { 'gen': 1, 'ttl': self.get_time(default_ttl) }), (('a', 'b', 4), { 'gen': 1, 'ttl': self.get_time(default_ttl) }), (('a', 'b', 5), None)], asm.exists_many(keys))
def test_exists_many(self): asm = AerospikeClientMock() asm.put(("a", "b", 1), {"a": 1}) asm.put(("a", "b", 2), {"a": 2}) asm.put(("a", "b", 3), {"a": 3}) asm.put(("a", "b", 4), {"a": 4}) keys = [ ("a", "b", 1), ("a", "b", 2), ("a", "b", 3), ("a", "b", 4), ("a", "b", 5), ] self.assertEqual([ (("a", "b", 1), { 'gen': 1, 'ttl': 0 }), (("a", "b", 2), { 'gen': 1, 'ttl': 0 }), (("a", "b", 3), { 'gen': 1, 'ttl': 0 }), (("a", "b", 4), { 'gen': 1, 'ttl': 0 }), (("a", "b", 5), None), ], asm.exists_many(keys))
def test_put(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) key = ("a", "b", "c") asm.put(key, {"a": 1}) self.assertEqual({('a', 'b', 'c'): {'a': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 1 }), asm.get(key)) time.sleep(1) asm.put(key, {"a": 1}) self.assertEqual((('a', 'b', 'c'), { 'gen': 2, 'ttl': self.get_time(default_ttl) }, { 'a': 1 }), asm.get(key)) asm.put(key, {"a": 1}, meta={"ttl": 1}) self.assertEqual((('a', 'b', 'c'), { 'gen': 3, 'ttl': self.get_time(1) }, { 'a': 1 }), asm.get(key))
def test_touch(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) key = ("a", "b", "c") asm.put(key, {"a": 1}) self.assertEquals((('a', 'b', 'c'), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 1 }), asm.get(key)) time.sleep(1) asm.touch(key) self.assertEquals((('a', 'b', 'c'), { 'gen': 2, 'ttl': self.get_time(default_ttl) }, { 'a': 1 }), asm.get(key)) asm.touch(key, 4) self.assertEquals((('a', 'b', 'c'), { 'gen': 3, 'ttl': self.get_time(4) }, { 'a': 1 }), asm.get(key))
def test_exists(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.put(key, {"a": 1}) self.assertEquals((True, {'gen': 1, 'ttl': 0}), asm.exists(key)) # test if not changing gen self.assertEquals((True, {'gen': 1, 'ttl': 0}), asm.exists(key))
def test_remove(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.put(key, {"a": 1}) self.assertEquals({('a', 'b', 'c'): {'a': 1}}, asm.dump()) asm.remove(key) self.assertEquals({}, asm.dump()) self.assertEquals((False, None), asm.exists(key))
def test_undefined_incr(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) key = ("a", "b", "c") asm.increment(key, "a", 1) self.assertEqual({('a', 'b', 'c'): {'a': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 1 }), asm.get(key))
def test_incr_value(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.put(key, {"a": 1}) asm.increment(key, "a", 2) self.assertEqual({('a', 'b', 'c'): {'a': 3}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 2, 'ttl': 0 }, { 'a': 3 }), asm.get(key))
def test_append(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.put(key, {"word": "ello"}) asm.prepend(key, "word", "h") self.assertEqual({('a', 'b', 'c'): {'word': 'hello'}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 2, 'ttl': 0 }, { 'word': 'hello' }), asm.get(key))
def test_prepend(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) key = ("a", "b", "c") asm.put(key, {"word": "hell"}) asm.append(key, "word", "o") self.assertEqual({('a', 'b', 'c'): {'word': 'hello'}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 2, 'ttl': self.get_time(default_ttl) }, { 'word': 'hello' }), asm.get(key))
def test_exists(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) key = ("a", "b", "c") asm.put(key, {"a": 1}) self.assertEquals((True, { 'gen': 1, 'ttl': self.get_time(default_ttl) }), asm.exists(key)) # test if not changing gen self.assertEquals((True, { 'gen': 1, 'ttl': self.get_time(default_ttl) }), asm.exists(key))
def test_get_many(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) asm.put(("a", "b", 1), {"a": 1}) asm.put(("a", "b", 2), {"a": 2}) asm.put(("a", "b", 3), {"a": 3}) asm.put(("a", "b", 4), {"a": 4}) keys = [ ("a", "b", 1), ("a", "b", 2), ("a", "b", 3), ("a", "b", 4), ("a", "b", 5), ] self.assertEqual([ (('a', 'b', 1, bytearray(b'u\x98t\x11La\x84\x9d\x94\xe3\xcdcSbn\xd7')), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 1 }), (('a', 'b', 2, bytearray( b'\xe7HY\x1f\x1f\xb8z\x8f\xf3\x0c\xf3\x04\xcc9\x14\xdc')), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 2 }), (('a', 'b', 3, bytearray(b'\xeb\x1a\x99(V\xd49\x01\xeeQ[\x92\x06-O\x08')), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 3 }), (('a', 'b', 4, bytearray( b'\xf3G\x1b\xba\xe2\xec\x11S\xc3\xc2\xab\x15\xb4\x1b\x96q')), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 4 }), (('a', 'b', 5, bytearray(b'd\t}\xc6`\xee\xe2\xf0)\x1f7\x9c\xfa\x8d\xa6\xd6')), None, None), ], asm.get_many(keys))
def test_get(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.put(key, {"a": 1}) self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': 0 }, { 'a': 1 }), asm.get(key)) # test if not changing gen self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': 0 }, { 'a': 1 }), asm.get(key))
def test_get(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) key = ("a", "b", "c") asm.put(key, {"a": 1}) self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 1 }), asm.get(key)) # test whether not changing gen self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 1 }), asm.get(key))
def test_select(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.put(key, {"a": 1, "b": 2, "c": 3}) self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': 0 }, { 'a': 1, 'c': 3 }), asm.select(key, ["a", "c"])) # test if not changing gen self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': 0 }, { 'a': 1, 'd': None }), asm.select(key, ["a", "d"]))
def test_undefined_incr(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.increment(key, "a", 1) self.assertEqual({('a', 'b', 'c'): {'a': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': 0 }, { 'a': 1 }), asm.get(key)) asm.increment(key, "b", 1) self.assertEqual({('a', 'b', 'c'): {'a': 1, 'b': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 2, 'ttl': 0 }, { 'a': 1, 'b': 1 }), asm.get(key))
def test_incr(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) key = ("a", "b", "c") asm.put(key, {"a": 1}) asm.increment(key, "a", 2) self.assertEqual({('a', 'b', 'c'): {'a': 3}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 2, 'ttl': self.get_time(default_ttl) }, { 'a': 3 }), asm.get(key)) asm.increment(key, "a", 1, meta={"ttl": 1}) self.assertEqual((('a', 'b', 'c'), { 'gen': 3, 'ttl': self.get_time(1) }, { 'a': 4 }), asm.get(key))
def test_put(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.put(key, {"a": 1}) self.assertEqual({('a', 'b', 'c'): {'a': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': 0 }, { 'a': 1 }), asm.get(key)) asm.put(key, {"b": 1}) self.assertEqual({('a', 'b', 'c'): {'a': 1, 'b': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 2, 'ttl': 0 }, { 'a': 1, 'b': 1 }), asm.get(key))
def test_get_many(self): asm = AerospikeClientMock() asm.put(("a", "b", 1), {"a": 1}) asm.put(("a", "b", 2), {"a": 2}) asm.put(("a", "b", 3), {"a": 3}) asm.put(("a", "b", 4), {"a": 4}) keys = [ ("a", "b", 1), ("a", "b", 2), ("a", "b", 3), ("a", "b", 4), ("a", "b", 5), ] self.assertEqual([ (('a', 'b', 1), { 'gen': 1, 'ttl': 0 }, { 'a': 1 }), (('a', 'b', 2), { 'gen': 1, 'ttl': 0 }, { 'a': 2 }), (('a', 'b', 3), { 'gen': 1, 'ttl': 0 }, { 'a': 3 }), (('a', 'b', 4), { 'gen': 1, 'ttl': 0 }, { 'a': 4 }), (('a', 'b', 5), None, None), ], asm.get_many(keys))
def test_remove_bin(self): default_ttl = 2 asm = AerospikeClientMock(default_ttl=2) key = ("a", "b", "c") asm.put(key, {"a": 1, "b": 1, "c": 1, "d": 1}) self.assertEquals({('a', 'b', 'c'): { 'a': 1, 'c': 1, 'b': 1, 'd': 1 }}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': self.get_time(default_ttl) }, { 'a': 1, 'c': 1, 'b': 1, 'd': 1 }), asm.get(key)) asm.remove_bin(key, ["b", "d"], meta={"ttl": 4}) self.assertEquals({('a', 'b', 'c'): {'a': 1, 'c': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 2, 'ttl': self.get_time(4) }, { 'a': 1, 'c': 1 }), asm.get(key)) asm.remove_bin(key, ["c"]) self.assertEquals({('a', 'b', 'c'): {'a': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 3, 'ttl': self.get_time(default_ttl) }, { 'a': 1 }), asm.get(key))
def test_remove_bin(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.put(key, {"a": 1, "b": 1, "c": 1, "d": 1}) self.assertEquals({('a', 'b', 'c'): { 'a': 1, 'c': 1, 'b': 1, 'd': 1 }}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 1, 'ttl': 0 }, { 'a': 1, 'c': 1, 'b': 1, 'd': 1 }), asm.get(key)) asm.remove_bin(key, ["b", "d"]) self.assertEquals({('a', 'b', 'c'): {'a': 1, 'c': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 2, 'ttl': 0 }, { 'a': 1, 'c': 1 }), asm.get(key)) asm.remove_bin(key, ["c"]) self.assertEquals({('a', 'b', 'c'): {'a': 1}}, asm.dump()) self.assertEqual((('a', 'b', 'c'), { 'gen': 3, 'ttl': 0 }, { 'a': 1 }), asm.get(key))
def test_blank_init(self): asm = AerospikeClientMock() self.assertEqual({}, asm.dump())
def test_expire_exist(self): asm = AerospikeClientMock(default_ttl=1) key = ("a", "b", "c") asm.put(key, {"a": 1}) time.sleep(2) self.assertEquals((False, None), asm.exists(key))
def test_dump(self): asm = AerospikeClientMock() key = ("a", "b", "c") asm.put(key, {"a": 1}) self.assertEqual({('a', 'b', 'c'): {'a': 1}}, asm.dump())
def test_select_many(self): asm = AerospikeClientMock() asm.put(("a", "b", 1), {"a": 1, "b": 1}) asm.put(("a", "b", 2), {"a": 2, "b": 2}) asm.put(("a", "b", 3), {"a": 3, "b": 3}) asm.put(("a", "b", 4), {"a": 4, "b": 4}) keys = [ ("a", "b", 1), ("a", "b", 2), ("a", "b", 3), ("a", "b", 4), ("a", "b", 5), ] self.assertEqual( { ('a', 'b', 3): { 'a': 3, 'b': 3 }, ('a', 'b', 2): { 'a': 2, 'b': 2 }, ('a', 'b', 4): { 'a': 4, 'b': 4 }, ('a', 'b', 1): { 'a': 1, 'b': 1 } }, asm.dump()) self.assertEqual([ (('a', 'b', 1), { 'gen': 1, 'ttl': 0 }, { 'a': 1, 'b': 1 }), (('a', 'b', 2), { 'gen': 1, 'ttl': 0 }, { 'a': 2, 'b': 2 }), (('a', 'b', 3), { 'gen': 1, 'ttl': 0 }, { 'a': 3, 'b': 3 }), (('a', 'b', 4), { 'gen': 1, 'ttl': 0 }, { 'a': 4, 'b': 4 }), None, ], asm.select_many(keys, ["a", "b"])) self.assertEqual([ (('a', 'b', 1), { 'gen': 1, 'ttl': 0 }, { 'b': 1 }), (('a', 'b', 2), { 'gen': 1, 'ttl': 0 }, { 'b': 2 }), (('a', 'b', 3), { 'gen': 1, 'ttl': 0 }, { 'b': 3 }), (('a', 'b', 4), { 'gen': 1, 'ttl': 0 }, { 'b': 4 }), None, ], asm.select_many(keys, ["b"]))
def test_not_exists(self): asm = AerospikeClientMock() key = ("a", "b", "c") self.assertEquals((False, None), asm.exists(key))
def setUp(self): self.asm = AerospikeClientMock() self.asm.put(("a", "b", 1), {"a": 1, "b": 1}) self.asm.put(("a", "b", 2), {"a": 2, "b": 2}) self.asm.put(("a", "b", 3), {"a": 3, "b": 3}) self.asm.put(("a", "c", 4), {"a": 4, "b": 4})
def test_connected(self): asm = AerospikeClientMock() self.assertTrue(asm.is_connected())