@@ -77,10 +77,16 @@ def test_8(self):
77
77
expected = {b'id' : cache_utils .compress_data (self .session_id )}
78
78
self .assertDictEqual (expected , self .cache .hgetall (self .session_id ))
79
79
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