3

摘要：我正在做一个实验，试图创建一个简单的mock来替换redis。我正在尝试做的事情应该从代码中显而易见。简短的版本是，模拟不起作用 - 它仍然会 redis 并创建密钥。

测试.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)

实用程序.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 不符合我的预期？

4

3 回答 3

12

链接

http://www.voidspace.org.uk/python/mock/patch.html#where-to-patch

帮助我更好地理解如何在 python 中修补函数调用。

于 2014-09-29T10:15:12.000 回答
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 的函数，并在测试中调用该函数。

于 2013-03-12T08:26:42.513 回答
3

而不是导入get_redis，尝试import util and call util.get_redis()。

于 2019-12-09T16:24:04.347 回答