def tearDown(self):
global_cache.flushdb()
redis.flushdb()
cache = utils_persistence.AerospikeCache(
client=utils_aerospike.get_aerospike_client(),
default_namespace=settings.AEROSPIKE_DEFAULT_NAMESPACE,
default_set_name=settings.AEROSPIKE_DEFAULT_SET_NAME)
cache.client.truncate(settings.AEROSPIKE_DEFAULT_NAMESPACE, settings.AEROSPIKE_DEFAULT_SET_NAME, 0)
def test_key_value_aerospike(self):
cache = utils_persistence.AerospikeCache(
client=utils_aerospike.get_aerospike_client(),
default_namespace=settings.AEROSPIKE_DEFAULT_NAMESPACE,
default_set_name=settings.AEROSPIKE_DEFAULT_SET_NAME)
key = 'key'
expected = 'f' * 24
cache.put_value(key, expected)
result = cache.get_value(key)
self.assertEqual(result, expected)
def test_map_increment_aerospike(self):
cache = utils_persistence.AerospikeCache(
client=utils_aerospike.get_aerospike_client(),
default_namespace=settings.AEROSPIKE_DEFAULT_NAMESPACE,
default_set_name=settings.AEROSPIKE_DEFAULT_SET_NAME)
key = 'key'
map_key = 'map_key'
expected = 10
cache.map_increment(key, map_key, 5)
cache.map_increment(key, map_key, 5)
result = cache.map_get_value(key, map_key)
self.assertEqual(result, expected)
def test_aerospike_cache(self):
cache = utils_persistence.AerospikeCache(
client=utils_aerospike.get_aerospike_client(),
default_namespace=settings.AEROSPIKE_DEFAULT_NAMESPACE,
default_set_name=settings.AEROSPIKE_DEFAULT_SET_NAME)
cache.save_udf('../lua_scripts/set.lua')
key = 'set_key'
expected_set = {17, 18, 19, 20, 21, 11, 7, 8, 9}
s1 = {17, 18, 19, 20, 21}
s2 = {11, 7, 8, 9}
cache.set_add_many(key, [x for x in s1])
cache.set_add_many(key, [x for x in s2])
result = cache.set_scan(key)
self.assertEqual(set(['{0}'.format(e) for e in expected_set]), set(result))
cache.set_remove_many(key, [x for x in s1])
cache.set_remove_many(key, [x for x in s2])
result = cache.set_scan(key)
self.assertEqual(len(result), 0)
def test_aerospike(self):
client = utils_persistence.AerospikeCache(
client=utils_aerospike.get_aerospike_client(),
default_namespace=settings.AEROSPIKE_DEFAULT_NAMESPACE,
default_set_name=settings.AEROSPIKE_DEFAULT_SET_NAME).client
key = ('test', 'set', 'set_key')
udf_bin = 'a'
client.udf_put('../lua_scripts/set.lua')
expected_set = {17, 18, 19, 20, 21, 11, 7, 8, 9}
s1 = {17, 18, 19, 20, 21}
s2 = {11, 7, 8, 9}
client.apply(key, 'set', 'unique_set_write_many', [udf_bin, [x for x in s1]])
client.apply(key, 'set', 'unique_set_write_many', [udf_bin, [x for x in s2]])
result = client.apply(key, 'set', 'unique_set_scan', ['a'])
self.assertEqual(set(['{0}'.format(e) for e in expected_set]), set(result))
client.apply(key, 'set', 'unique_set_remove_many', [udf_bin, [x for x in s1]])
client.apply(key, 'set', 'unique_set_remove_many', [udf_bin, [x for x in s2]])
result = client.apply(key, 'set', 'unique_set_scan', ['a'])
self.assertEqual(len(result), 0)