用于编写单元测试以供应用的模拟红色

Question

我有一个烧瓶应用程序，在其中我使用“烧瓶缓存”与redis。我将一些值保存在缓存中，并检索它以进行处理。我必须为这件事写一封信。如何在烧瓶缓存配置中模拟redis实例并传递该服务器的主机。

我确实尝试过"fakeredis"，但这并没有给我红色主机，我可以在烧瓶缓存和测试我的应用程序。

我如何编写这方面的单元测试用例？

def func_1():
    # some steps
    x = func2() # this is where the code for flask caching (redis) data retrieval is
    # process x

我必须测试func_1()

Answer 1

模拟对象库可用于为特定方法调用的Redis创建模拟实例，例如：

from unittest.mock import Mock

def test_redis():
    mock_redis = Mock()
    mock_redis.get.return_value = "2020-12-31T00:00:00".encode()

    function_to_test(mock_redis)

如果需要对函数进行模拟，以便不实际调用该函数，并且在正在测试的函数中使用，则可以使用贴片装饰器，如下所示：

from unittest.mock import patch

@patch("package.module.function_to_mock")
def test_redis(mock_function_to_mock):
    mock_function_to_mock.get.return_value = "2020-12-31T00:00:00".encode()

    function_to_test()

这可以用来模拟您正在测试的函数中与烧瓶缓存相关的代码。

编辑

就你的情况而言，我相信你可能想做以下工作：

from unittest.mock import patch

@patch("package.module.func2")
def test_func_1(mock_func2):
    mock_func2.return_value = "Test value"

    func_1()

    mock_func2.assert_called_once()

在这里，func2()被模拟，它的返回值被设置为"Test value"，所以当在func_1()变量中调用它时，x等于"Test value"

注意:传递给patch()的路径必须是导入func2()的位置，而不是定义func2()的模块。