Skip to content

Commit b96b515

Browse files
add more tests for cache_utils.py
1 parent db6385b commit b96b515

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

utils/tests/test_cache_utils.py

+13-7
Original file line numberDiff line numberDiff line change
@@ -77,10 +77,16 @@ def test_8(self):
7777
expected = {b'id': cache_utils.compress_data(self.session_id)}
7878
self.assertDictEqual(expected, self.cache.hgetall(self.session_id))
7979

80-
def test_(self):
81-
82-
cache_utils.store_density(self.session_id, density_cachekey, density, self.cache)
83-
cache_utils.store_density()
84-
cache_utils.retrieve_density()
85-
cache_utils.remove_density()
86-
cache_utils.remove_all_density()
80+
def test_9(self):
81+
cachekey_1 = 'fname_1_{}_2'.format(cache_utils.CacheKeys.METADATA_TAG.value).encode()
82+
density_1 = [1, 2, 3, 3, 4, 5]
83+
cachekey_2 = 'fname_2_{}_2'.format(cache_utils.CacheKeys.METADATA_TAG.value).encode()
84+
density_2 = [5, 6, 7, 8, 9, 0]
85+
86+
cache_utils.store_density(self.session_id, cachekey_1, density_1, self.cache)
87+
cache_utils.store_density(self.session_id, cachekey_2, density_2, self.cache)
88+
output = cache_utils.retrieve_density(self.session_id, cachekey_2, self.cache)
89+
self.assertListEqual(output, density_2)
90+
expected_cache = {b'id': cache_utils.compress_data(self.session_id)}
91+
cache_utils.remove_all_density(self.session_id, self.cache)
92+
self.assertDictEqual(expected_cache, self.cache.hgetall(self.session_id))

0 commit comments

Comments
 (0)