def test_add_by_key(self):
params = ("k1", 1), ("k2", ("k3", 2))
obj = "object_example"
h = CacheHandler._hash(params)
CacheHandler.add_by_key(h, obj)
filename = CacheHandler._build_filename(h, CacheObjectType.OTHER)
with open(filename, "rb") as file:
obj2 = pickle.load(file)
self.assertEqual(obj, obj2)
os.remove(filename)
def test_get_by_key(self):
params = (("k1", 1), ("k2", 2))
obj = "object_example"
h = CacheHandler._hash(params)
filename = CacheHandler._build_filename(h, CacheObjectType.OTHER)
with open(filename, "wb") as file:
pickle.dump(obj, file)
obj2 = CacheHandler.get_by_key(h, CacheObjectType.OTHER)
self.assertEqual(obj, obj2)
os.remove(filename)