Coverage for test / storage_m_redis_test.py: 97%

1# SPDX-FileCopyrightText: 2023 Arianna Moretti <arianna.moretti4@unibo.it> 

2# SPDX-FileCopyrightText: 2026 Arcangelo Massari <arcangelo.massari@unibo.it> 

3# 

4# SPDX-License-Identifier: ISC 

5 

6import unittest 

7from oc_ds_converter.oc_idmanager.oc_data_storage.redis_manager import RedisStorageManager 

8 

9 

10class TestRedisStorageManager(unittest.TestCase): 

11 

12 def test_storage_management_testing(self): 

13 rsm = RedisStorageManager(testing=True) 

14 rsm.set_value("pmid:9", True) 

15 rsm.set_value("pmid:0", False) 

16 

17 # test set_value 

18 # test get_all_keys 

19 self.assertCountEqual(rsm.get_all_keys(), {"pmid:9", "pmid:0"}) 

20 

21 #test delete_storage 

22 rsm.delete_storage() 

23 self.assertCountEqual(rsm.get_all_keys(), {}) 

24 

25 #test set_multi_value 

26 rsm.set_multi_value([("pmid:1020", True), ("pmid:2020", False)]) 

27 self.assertCountEqual(rsm.get_all_keys(), {"pmid:1020", "pmid:2020"}) 

28 

29 #test get_value 

30 get_val = rsm.get_value("pmid:1020") 

31 exp = True 

32 self.assertEqual(get_val, exp) 

33 

34 get_val = rsm.get_value("pmid:2020") 

35 exp = False 

36 self.assertEqual(get_val, exp) 

37 

38 get_val = rsm.get_value("pmid:3020") 

39 exp = None 

40 self.assertEqual(get_val, exp) 

41 

42 #test set_full_value 

43 

44 rsm.set_full_value("pmid:1212", {"valid":True}) 

45 get_val = rsm.get_value("pmid:1212") 

46 exp = True 

47 self.assertEqual(get_val, exp) 

48 

49 rsm.delete_storage() 

50 emp_set = set() 

51 self.assertEqual(emp_set, rsm.get_all_keys()) 

52 

53 

54if __name__ == '__main__': 

55 unittest.main()