41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# @Author : relakkes@gmail.com
|
||
|
|
# @Name : 程序员阿江-Relakkes
|
||
|
|
# @Time : 2024/6/2 19:54
|
||
|
|
# @Desc :
|
||
|
|
|
||
|
|
import time
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from cache.redis_cache import RedisCache
|
||
|
|
|
||
|
|
|
||
|
|
class TestRedisCache(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.redis_cache = RedisCache()
|
||
|
|
|
||
|
|
def test_set_and_get(self):
|
||
|
|
self.redis_cache.set('key', 'value', 10)
|
||
|
|
self.assertEqual(self.redis_cache.get('key'), 'value')
|
||
|
|
|
||
|
|
def test_expired_key(self):
|
||
|
|
self.redis_cache.set('key', 'value', 1)
|
||
|
|
time.sleep(2) # wait for the key to expire
|
||
|
|
self.assertIsNone(self.redis_cache.get('key'))
|
||
|
|
|
||
|
|
def test_keys(self):
|
||
|
|
self.redis_cache.set('key1', 'value1', 10)
|
||
|
|
self.redis_cache.set('key2', 'value2', 10)
|
||
|
|
keys = self.redis_cache.keys('*')
|
||
|
|
self.assertIn('key1', keys)
|
||
|
|
self.assertIn('key2', keys)
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
# self.redis_cache._redis_client.flushdb() # 清空redis数据库
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|