From b96b515de91a7e12cfcac4396283c30c3eddd876 Mon Sep 17 00:00:00 2001 From: FilomenoSanchez Date: Fri, 16 Apr 2021 11:31:23 +0100 Subject: [PATCH] add more tests for cache_utils.py --- utils/tests/test_cache_utils.py | 20 +++++++++++++------- 1 file changed, 13 insertions(+), 7 deletions(-) diff --git a/utils/tests/test_cache_utils.py b/utils/tests/test_cache_utils.py index dbfc832..626a04c 100644 --- a/utils/tests/test_cache_utils.py +++ b/utils/tests/test_cache_utils.py @@ -77,10 +77,16 @@ def test_8(self): expected = {b'id': cache_utils.compress_data(self.session_id)} self.assertDictEqual(expected, self.cache.hgetall(self.session_id)) - def test_(self): - - cache_utils.store_density(self.session_id, density_cachekey, density, self.cache) - cache_utils.store_density() - cache_utils.retrieve_density() - cache_utils.remove_density() - cache_utils.remove_all_density() \ No newline at end of file + def test_9(self): + cachekey_1 = 'fname_1_{}_2'.format(cache_utils.CacheKeys.METADATA_TAG.value).encode() + density_1 = [1, 2, 3, 3, 4, 5] + cachekey_2 = 'fname_2_{}_2'.format(cache_utils.CacheKeys.METADATA_TAG.value).encode() + density_2 = [5, 6, 7, 8, 9, 0] + + cache_utils.store_density(self.session_id, cachekey_1, density_1, self.cache) + cache_utils.store_density(self.session_id, cachekey_2, density_2, self.cache) + output = cache_utils.retrieve_density(self.session_id, cachekey_2, self.cache) + self.assertListEqual(output, density_2) + expected_cache = {b'id': cache_utils.compress_data(self.session_id)} + cache_utils.remove_all_density(self.session_id, self.cache) + self.assertDictEqual(expected_cache, self.cache.hgetall(self.session_id))