def setUp(self):
"""
set up DAS core module
"""
debug = 0
config = das_readconfig()
logger = DASLogger(verbose=debug, stdout=debug)
config['logger'] = logger
config['verbose'] = debug
self.memcache = DASMemcache(config)
def setUp(self):
"""
set up DAS core module
"""
debug = 0
config = das_readconfig()
logger = DASLogger(verbose=debug, stdout=debug)
config["logger"] = logger
config["verbose"] = debug
self.memcache = DASMemcache(config)
class testDASMemcache(unittest.TestCase):
"""
A test class for the DAS cache module
"""
def setUp(self):
"""
set up DAS core module
"""
debug = 0
config = das_readconfig()
logger = DASLogger(verbose=debug, stdout=debug)
config['logger'] = logger
config['verbose'] = debug
self.memcache = DASMemcache(config)
def test_key(self):
"""test DAS cache key generator"""
query = "find site where site=T2_UK"
result = genkey(query)
try:
hash = hashlib.md5()
except:
# prior python 2.5
hash = md5.new()
hash.update(query)
expect = hash.hexdigest()
self.assertEqual(expect, result)
def test_result(self):
"""test DAS memcache result method"""
self.memcache.delete_cache()
query = "find site where site=T2_UK"
expire = 60
expect = [0,1,2,3,4,5,6,7,8,9]
expect = self.memcache.update_cache(query, expect, expire)
expect = [i for i in expect]
result = [i for i in self.memcache.get_from_cache(query)]
result.sort()
self.assertEqual(expect, result)
def test_pagintation(self):
"""test DAS memcache result method with pagination"""
self.memcache.delete_cache()
query = "find site where site=T2_UK"
expire = 60
expect = [0,1,2,3,4,5,6,7,8,9]
expect = self.memcache.update_cache(query, expect, expire)
expect = [i for i in expect]
idx = 1
limit = 3
result = [i for i in self.memcache.get_from_cache(query, idx, limit)]
result.sort()
self.assertEqual(expect[idx:limit+1], result)
def test_sorting(self):
"""test DAS memcache result method with sorting"""
self.memcache.delete_cache()
query = "find site where site=T2_UK"
expire = 60
data = [
{'id':0, 'data':'a', 'run':1},
{'id':1, 'data':'b', 'run':3},
{'id':2, 'data':'c', 'run':2},
]
gen = self.memcache.update_cache(query, data, expire)
res = [i for i in gen]
skey = 'run'
order = 'desc'
result = [i for i in \
self.memcache.get_from_cache(query, skey=skey, order=order)]
expect = [
{'id':1, 'data':'b', 'run':3},
{'id':2, 'data':'c', 'run':2},
{'id':0, 'data':'a', 'run':1},
]
self.assertEqual(expect, result)
skey = 'run'
order = 'asc'
result = [i for i in \
self.memcache.get_from_cache(query, skey=skey, order=order)]
expect = [
{'id':0, 'data':'a', 'run':1},
{'id':2, 'data':'c', 'run':2},
{'id':1, 'data':'b', 'run':3},
]
self.assertEqual(expect, result)
def test_incache(self):
"""test DAS memcache incache method"""
self.memcache.delete_cache()
query = "find site where site=T2_UK"
expire = 1
expect = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
expect = self.memcache.update_cache(query, expect, expire)
expect = [i for i in expect]
result = self.memcache.incache(query)
self.assertEqual(1, result)
time.sleep(2)
result = self.memcache.incache(query)
self.assertEqual(0, result)
class testDASMemcache(unittest.TestCase):
"""
A test class for the DAS cache module
"""
def setUp(self):
"""
set up DAS core module
"""
debug = 0
config = das_readconfig()
logger = DASLogger(verbose=debug, stdout=debug)
config["logger"] = logger
config["verbose"] = debug
self.memcache = DASMemcache(config)
def test_key(self):
"""test DAS cache key generator"""
query = "find site where site=T2_UK"
result = genkey(query)
try:
hash = hashlib.md5()
except:
# prior python 2.5
hash = md5.new()
hash.update(query)
expect = hash.hexdigest()
self.assertEqual(expect, result)
def test_result(self):
"""test DAS memcache result method"""
self.memcache.delete_cache()
query = "find site where site=T2_UK"
expire = 60
expect = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
expect = self.memcache.update_cache(query, expect, expire)
expect = [i for i in expect]
result = [i for i in self.memcache.get_from_cache(query)]
result.sort()
self.assertEqual(expect, result)
def test_pagintation(self):
"""test DAS memcache result method with pagination"""
self.memcache.delete_cache()
query = "find site where site=T2_UK"
expire = 60
expect = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
expect = self.memcache.update_cache(query, expect, expire)
expect = [i for i in expect]
idx = 1
limit = 3
result = [i for i in self.memcache.get_from_cache(query, idx, limit)]
result.sort()
self.assertEqual(expect[idx : limit + 1], result)
def test_sorting(self):
"""test DAS memcache result method with sorting"""
self.memcache.delete_cache()
query = "find site where site=T2_UK"
expire = 60
data = [{"id": 0, "data": "a", "run": 1}, {"id": 1, "data": "b", "run": 3}, {"id": 2, "data": "c", "run": 2}]
gen = self.memcache.update_cache(query, data, expire)
res = [i for i in gen]
skey = "run"
order = "desc"
result = [i for i in self.memcache.get_from_cache(query, skey=skey, order=order)]
expect = [{"id": 1, "data": "b", "run": 3}, {"id": 2, "data": "c", "run": 2}, {"id": 0, "data": "a", "run": 1}]
self.assertEqual(expect, result)
skey = "run"
order = "asc"
result = [i for i in self.memcache.get_from_cache(query, skey=skey, order=order)]
expect = [{"id": 0, "data": "a", "run": 1}, {"id": 2, "data": "c", "run": 2}, {"id": 1, "data": "b", "run": 3}]
self.assertEqual(expect, result)
def test_incache(self):
"""test DAS memcache incache method"""
self.memcache.delete_cache()
query = "find site where site=T2_UK"
expire = 1
expect = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
expect = self.memcache.update_cache(query, expect, expire)
expect = [i for i in expect]
result = self.memcache.incache(query)
self.assertEqual(1, result)
time.sleep(2)
result = self.memcache.incache(query)
self.assertEqual(0, result)