mirror of
https://github.com/NanmiCoder/MediaCrawler.git
synced 2025-11-25 03:15:17 +08:00
feat: cache impl done
This commit is contained in:
39
test/test_expiring_local_cache.py
Normal file
39
test/test_expiring_local_cache.py
Normal file
@@ -0,0 +1,39 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# @Author : relakkes@gmail.com
|
||||
# @Name : 程序员阿江-Relakkes
|
||||
# @Time : 2024/6/2 10:35
|
||||
# @Desc :
|
||||
|
||||
import time
|
||||
import unittest
|
||||
|
||||
from cache.local_cache import ExpiringLocalCache
|
||||
|
||||
|
||||
class TestExpiringLocalCache(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.cache = ExpiringLocalCache(cron_interval=10)
|
||||
|
||||
def test_set_and_get(self):
|
||||
self.cache.set('key', 'value', 10)
|
||||
self.assertEqual(self.cache.get('key'), 'value')
|
||||
|
||||
def test_expired_key(self):
|
||||
self.cache.set('key', 'value', 1)
|
||||
time.sleep(2) # wait for the key to expire
|
||||
self.assertIsNone(self.cache.get('key'))
|
||||
|
||||
def test_clear(self):
|
||||
# 设置两个键值对,过期时间为11秒
|
||||
self.cache.set('key', 'value', 11)
|
||||
# 睡眠12秒,让cache类的定时任务执行一次
|
||||
time.sleep(12)
|
||||
self.assertIsNone(self.cache.get('key'))
|
||||
|
||||
def tearDown(self):
|
||||
del self.cache
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
40
test/test_redis_cache.py
Normal file
40
test/test_redis_cache.py
Normal file
@@ -0,0 +1,40 @@
|
||||
# -*- 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()
|
||||
Reference in New Issue
Block a user