Django的mock patch不按预期工作
总结：我正在做一个实验，想要创建一个简单的模拟工具来替代redis。我的代码应该能让你明白我在做什么。简单来说，这个模拟工具没有成功——它还是在访问redis并创建键。
tests.py:
from django.test import TestCase
import mock
from redis_mock.simple_redis_mock import redisMockGetRedis, redisMockFlushDB
from account.util import get_redis
class SimpleTest(TestCase):
def setUp(self):
redisMockFlushDB()
@mock.patch("account.util.get_redis", redisMockGetRedis)
def test_redis(self):
key = "hello123"
value = "world123"
r = get_redis()
r.set(key, value)
value2 = r.get(key)
self.assertEqual(value, value2)
util.py:
import redis
REDIS_HOST = 'localhost'
REDIS_PORT = 6379
REDIS_DEFAULT_DB = 0
def get_redis():
print "account.util.get_redis"
return redis.StrictRedis(
REDIS_HOST,
REDIS_PORT,
REDIS_DEFAULT_DB
)
simple_redis_mock.py:
"""
A simple mock for Redis. Just mocks set, get and expire commands.
"""
class SimpleRedisMockDB:
db = {}
def redisMockFlushDB():
"""
Helper function to flush the RedisMock db between test runs
"""
print "redisMockFlushDB"
SimpleRedisMock.db = {}
class SimpleRedisMock:
def get(self, key):
val = None
try:
val = SimpleRedisMockDB.db[key]
except:
pass
print "SimpleRedisMock get(" + str(key) + "):" + str(val)
return val
def set(self, key, val):
print "SimpleRedisMock set(" + str(key) + "," + str(val) +")"
SimpleRedisMockDB.db[key] = val
def expire(self, key):
pass
def redisMockGetRedis():
print "redisMockGetRedis"
return SimpleRedisMock()
现在，我希望在运行我的测试时，不会设置任何redis的键。实际上发生的情况是：
twang$ redis-cli
redis 127.0.0.1:6379> del hello123
(integer) 1
redis 127.0.0.1:6379> exit
twang$ ./manage.py test account
Creating test database for alias 'default'...
redisMockFlushDB
account.util.get_redis
.
----------------------------------------------------------------------
Ran 1 test in 0.001s
OK
Destroying test database for alias 'default'...
twang$ redis-cli
redis 127.0.0.1:6379> get hello123
"world123"
简单的问题：为什么mock.patch没有按照我预期的那样工作？
3 个回答
3
与其直接引入 get_redis，不如试试先引入 util，然后用 util.get_redis() 来调用它。
8
补丁（Patch）只会在你指定的位置对对象进行模拟。下面的示例代码可以帮助你理解发生了什么。
from mock import Mock, patch
import unittest
from patch_code import anotherfunc, thefunc
print 'Function imported', thefunc
class SomeTest(unittest.TestCase):
@patch('patch_code.thefunc', Mock())
def test_method(self):
anotherfunc()
print 'Inside test method', thefunc
if __name__ == '__main__':
unittest.main()
而被测试的代码其实只有两个函数：
def thefunc():
pass
def anotherfunc():
print 'Inside code under test', thefunc
运行这个测试会得到以下输出：
Function imported <function thefunc at 0xb740e614>
Inside code under test <Mock id='3071597132'>
Inside test method <function thefunc at 0xb740e614>
你可以清楚地看到，补丁只在被测试的代码中模拟了“thefunc”。
如果你想测试函数 get_redis，而不想引起任何副作用，那么你应该模拟 account.util.redis.StrictRedis，并检查它是否用正确的参数被调用。
如果你想测试那些使用 get_redis 的函数，那么你应该模拟 get_redis，然后导入使用了 get_redis 的函数，并在你的测试中调用那个函数。
12