def test_size(self):
with open_cache(self.db_file, 1000) as cache:
self.assertEqual(cache.size(), 0)
for i in range(100):
cache.set('engine', str(i), str(i))
self.assertEqual(cache.size(), i + 1)
def test_get(self):
engine = 'engine'
tag = 'tag'
filename = 'test_file'
with open_cache(self.db_file, 1000) as cache:
self.assertEqual(cache.get(engine, tag), None)
with temp_file(filename):
cache.set(engine, tag, filename)
self.assertEqual(cache.get(engine, tag), filename)
def test_set_overwrite(self):
engine = 'engine'
tag = 'tag'
filename = 'temp_file'
filename2 = 'temp_file2'
with open_cache(self.db_file, 1000) as cache:
with temp_file(filename):
cache.set(engine, tag, filename)
self.assertEqual(cache.get(engine, tag), filename)
with temp_file(filename2):
cache.set(engine, tag, filename2)
self.assertEqual(cache.get(engine, tag), filename2)
def test_trim(self):
SIZE = 100
with open_cache(self.db_file, 1000) as cache:
for i in range(SIZE):
cache.set('engine', str(i), str(i))
self.assertEqual(cache.size(), SIZE)
NEW_SIZE=40
cache.trim(NEW_SIZE)
self.assertEqual(cache.size(), NEW_SIZE)
cache.trim(cache.size() + 1)
self.assertEqual(cache.size(), NEW_SIZE)
