window 测试redis 通不通

转载

lazihuman 2024-09-09 20:06:52

文章标签 window 测试redis 通不通 redis 数据库缓存 Redis 文章分类 Redis 数据库

1.定义了一个名为RedisDb的类，用于简化与Redis数据库的交互操作。代码如下：

import redis
from config.setting import REDIS_HOST, REDIS_PORT, REDIS_PASSWD, EXPIRE_TIME

class RedisDb():

    def __init__(self, host, port, passwd):
        try:
        # 建立数据库连接
            self.r = redis.Redis(
                host=host,
                port=port,
                password=passwd,
                decode_responses=True # get() 得到字符串类型的数据
            )
        except redis.ConnectionError as e :
            print(f"Connection error : {e}")
            raise

    def handle_redis_token(self, key, value=None):
        if value: # 如果value非空，那么就设置key和value，EXPIRE_TIME为过期时间
            self.r.set(key, value, ex=EXPIRE_TIME)
        else: # 如果value为空，那么直接通过key从redis中取值
            redis_token = self.r.get(key)
            return redis_token


redis_db = RedisDb(REDIS_HOST, REDIS_PORT, REDIS_PASSWD)

2.那么如何测试redis是否能正常工作呢？

RedisDb类中有两个方法，一个是初始化连接，一个是设置键值

针对此进行简单的单元测试

完整代码如下：

from datetime import time

import redis
from config.setting import REDIS_HOST, REDIS_PORT, REDIS_PASSWD, EXPIRE_TIME
import unittest

import logging
#logging.basicConfig(level=logging.DEBUG)
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[
        logging.StreamHandler()  # 控制台输出
    ]
)
# 添加一个FileHandler来记录所有级别的日志
file_handler = logging.FileHandler('app.log')
file_handler.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
file_handler.setFormatter(formatter)
logging.getLogger().addHandler(file_handler)




class RedisDb():

    def __init__(self, host, port, passwd):
        try:
        # 建立数据库连接
            self.r = redis.Redis(
                host=host,
                port=port,
                password=passwd,
                decode_responses=True # get() 得到字符串类型的数据
            )
        except redis.ConnectionError as e :
            print(f"Connection error : {e}")
            raise

    def handle_redis_token(self, key, value=None):
        if value: # 如果value非空，那么就设置key和value，EXPIRE_TIME为过期时间
            self.r.set(key, value, ex=EXPIRE_TIME)
        else: # 如果value为空，那么直接通过key从redis中取值
            redis_token = self.r.get(key)
            return redis_token


redis_db = RedisDb(REDIS_HOST, REDIS_PORT, REDIS_PASSWD)



# 假定的常量
EXPIRE_TIME = 60  # 示例过期时间为60秒

class TestRedisDb(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        # 这里使用真实的或者测试环境的Redis配置
        cls.REDIS_HOST = REDIS_HOST  # Redis服务器地址
        cls.REDIS_PORT = REDIS_PORT      # Redis服务器端口
        cls.REDIS_PASSWD = REDIS_PASSWD        # Redis密码，如果没有设置则留空
        cls.redis_db = RedisDb(cls.REDIS_HOST, cls.REDIS_PORT, cls.REDIS_PASSWD)

    @classmethod
    def tearDownClass(cls):
        # 清理环境，例如删除测试中创建的键值对
        pass

    def test_redis_connection(self):
        # 测试Redis连接是否成功
        logging.info(f"Redis connection: {self.redis_db.r}")
        self.assertIsNotNone(self.redis_db.r, "Redis connection should not be None")

    def test_handle_redis_token_set(self):
        # 测试设置键值对
        test_key = "test_key"
        test_value = "test_value"
        self.redis_db.handle_redis_token(test_key, test_value)
        actual_value = self.redis_db.r.get(test_key)
        logging.info(f"Actual value for key '{test_key}': {actual_value}")
        self.assertEqual(actual_value, test_value, f"Value for key '{test_key}' should match after setting.")

    def test_handle_redis_token_get(self):
        # 测试获取键值
        test_key = "test_key_get"
        test_value = "test_value_get"
        self.redis_db.r.set(test_key, test_value, ex=EXPIRE_TIME)  # 直接使用redis.Redis对象设置值，绕过handle_redis_token
        retrieved_value = self.redis_db.handle_redis_token(test_key)
        logging.info(f"Retrieved value for key '{test_key}': {retrieved_value}")
        self.assertEqual(retrieved_value, test_value, f"Retrieved value for key '{test_key}' should match.")


if __name__ == '__main__':
    # 创建一个测试套件
    suite = unittest.TestSuite()
    # 添加测试用例到套件中
    suite.addTest(unittest.defaultTestLoader.loadTestsFromTestCase(TestRedisDb))    # 创建一个测试运行器

    runner = unittest.TextTestRunner()
    # 运行测试套件
    runner.run(suite)

window 测试redis 通不通_redis