模拟数据库连接python

Question

我有一个名为redis_db.py的文件，它有连接到redis 的代码

import os
import redis
import sys
class Database:
def __init__(self, zset_name):
redis_host = os.environ.get('REDIS_HOST', '127.0.0.1')
redis_port = os.environ.get('REDIS_PORT', 6379)
self.db = redis.StrictRedis(host=redis_host, port=redis_port)
self.zset_name = zset_name
def add(self, key):
try:
self.db.zadd(self.zset_name, {key: 0})
except redis.exceptions.ConnectionError:
print("Unable to connect to redis host.")
sys.exit(0)

我有另一个名为app.py的文件，它像这个

from flask import Flask
from redis_db import Database
app = Flask(__name__)
db = Database('zset')
@app.route('/add_word/word=<word>')
def add_word(word):
db.add(word)
return ("{} added".format(word))
if __name__ == '__main__':
app.run(host='0.0.0.0', port='8080')

现在我正在为类似的add_word函数编写单元测试

import unittest
import sys
import os
from unittest import mock
sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../api/")
from api import app  # noqa: E402

class Testing(unittest.TestCase):
def test_add_word(self):
with mock.patch('app.Database') as mockdb:
mockdb.return_value.add.return_value = ""
result = app.add_word('shivam')
self.assertEqual(result, 'shivam word added.')

我面临的问题是，即使我在模拟db方法调用，它仍然在调用类中的实际方法，而不是返回模拟值，并且在测试过程中，我收到了消息Unable to connect to redis host.的错误。

有人能帮我弄清楚如何模拟redis数据库调用吗。

我正在使用单元测试模块

Answer 1

问题是db是在模块导入时定义的，因此mock.patch不会影响db变量。要么移动的实例化add_word(word)函数中的db，或者修补db而不是Database，例如

def test_add_word():
with mock.patch('api.app.db') as mockdb:
mockdb.add = mock.MagicMock(return_value="your desired return value")
result = app.add_word('shivam')
print(result)

注意，add_word的调用必须在with块中，否则将使用未锁定的版本。