11

我一直在使用 Python 的unittest.mock库，但现在我正在努力解决一个可能无法正确处理的用例。

考虑一个mymodule/code.py包含以下代码段的文件：

def sum():
  pass

def mul():
  pass

def div():
  pass

def get_functions():
  return [sum, mul, div]

def foo():
  functions = get_functions()
  for func in functions:
    func()

我想测试foo功能，修补sum功能，然后保持mul原样div。这是我最初尝试的：

class TestFoo(unittest.TestCase):
  @mock.patch('mymodule.code.foo.sum')
  def test_foo(foo_sum_mock):
    foo()
    foo_sum_mock.assert_called_once()

但是，上面说明的修补方法不起作用。我相信该sum函数在加载时已正确修补mymodule.code.py，但由于def sum()块而重新定义。

通过阅读官方文档，我也尝试使用该库的start和stop函数unittest.mock如下：

def test_foo():
  patcher = mock.patch('module.code.sum')
  mocked_sum_fun = patcher.start()

  foo()

  mocked_sum_fun.assert_called_once()
  mock_sum_fun.stop()

这种方法也没有奏效。我希望它会在文件加载sum后避免函数覆盖。modules/code.py

是否可以修补本地功能，例如sum？或者将sum函数移动到另一个文件是修补的唯一选项？

提前谢谢了！

4

2 回答 2

3

mock.patch您可以使用此模块来模拟同一模块的功能并将其引用为__main__

代码.py

from unittest.mock import patch

def sum():
    print("called method sum")
    pass

def call_sum():
    sum()

def return_mock():
    print("I'm a mocked method")
    return True

with patch('__main__.sum', return_value=return_mock()) as mock_test:
    call_sum()
    mock_test.assert_called_once() # assure that mocked method was called, not original.

您也可以使用 lib( my_project.code.sum) 的路径而不是__main__.sum.

于 2020-07-15T19:22:06.947 回答
-3

一般来说，您希望将测试代码与生产代码分开：

代码.py

def sum():
    pass

def mul():
    pass

def div():
    pass

def get_functions():
    return [sum, mul, div]

def foo():
    functions = get_functions()
    for func in functions:
        func()

code_test.py

import unittest
import mock_test as mock
import code

class TestFoo(unittest.TestCase):
    @mock.patch('code.sum')
    def test_foo(self, sum_mock):
        def new_sum_mock(*args, **kwargs):
            # mock code here
            pass
        sum_mock.side_effect = new_sum_mock
        code.foo()
        sum_mock.assert_called_once()

但是，是的，您可以将它们全部放入一个文件中：

代码测试.py：

import unittest
import mock_test as mock
import code

def sum():
    pass


def mul():
    pass


def div():
    pass


def get_functions():
    return [sum, mul, div]


def foo():
    functions = get_functions()
    for func in functions:
        func()


class TestFoo(unittest.TestCase):

    @mock.patch('code_test.sum')
    def test_foo(self, sum_mock):
        def new_sum_mock(*args, **kwargs):
            # mock code here
            pass

        sum_mock.side_effect = new_sum_mock
        code.foo()
        sum_mock.assert_called_once()
于 2019-01-07T23:04:04.870 回答