在Python的单元测试中，将整数与MagicMock进行比较不起作用

Question

我有一个类，使用一个类变量来选择要执行的逻辑。

#in file1:

class SomeHelper():
    def __init__(self):
        self.my_var = 0

#in file2: 
import file1
class MyClass():
    ...
    ...
    def calculate():
        inst = file1.SomeHelper()
        if x > inst.my_var:
           etc etc

我正在编写单元测试并模拟另一个文件中的SomeHelper()函数:

from file 2 import MyClass
# tried both
@patch('file2.file1') OR @patch('file2.file1.SomeHelper')
def test_calculate(self, mock_helper):
    mock_helper.my_var = 0
    to_test = MyClass.calculate()

我得到了以下错误:

TypeError: '>'不支持'MagicMock'和'int'实例之间的比较。

我认为在模块打补丁后定义了my_var。

- RebeccaK375

---

mock_helper 是 file1 的整体，你需要使用 mock_helper.SomeHelper().my_var 进行操作。但请注意，模拟这个过程的复杂性表明您需要应用其他模式，例如依赖反转，以使代码更易于测试。 - jonrsharpe

唉，我尝试了一下，但它没有起作用。 - RebeccaK375

开玩笑 - 括号起了作用。没有括号，它嘲笑的是类而不是类的实例。谢谢！如果您将此作为答案发布，我会接受它。 - RebeccaK375

Answer 1

这是针对 Python 3.7.5 的单元测试解决方案：

file1.py：

class SomeHelper():
    def __init__(self):
        self.my_var = 0

file2.py:

import file1


class MyClass():
    @classmethod
    def calculate(cls):
        x = 1
        inst = file1.SomeHelper()
        if x > inst.my_var:
            return True
        return False

test_file2.py:

import unittest
from unittest.mock import patch
from file2 import MyClass


class TestMyClass(unittest.TestCase):
    @patch('file2.file1')
    def test_calculate(self, mock_file1):
        inst = mock_file1.SomeHelper.return_value
        inst.my_var = 0.5
        to_test = MyClass.calculate()
        self.assertTrue(to_test)
        mock_file1.SomeHelper.assert_called_once()

    @patch('file2.file1')
    def test_calculate_2(self, mock_file1):
        inst = mock_file1.SomeHelper.return_value
        inst.my_var = 2
        to_test = MyClass.calculate()
        self.assertFalse(to_test)
        mock_file1.SomeHelper.assert_called_once()


if __name__ == '__main__':
    unittest.main()

单元测试结果与覆盖报告：

..
----------------------------------------------------------------------
Ran 2 tests in 0.002s

OK
Name                                       Stmts   Miss  Cover   Missing
------------------------------------------------------------------------
src/stackoverflow/50242955/file1.py            3      1    67%   3
src/stackoverflow/50242955/file2.py            8      0   100%
src/stackoverflow/50242955/test_file2.py      16      0   100%
------------------------------------------------------------------------
TOTAL                                         27      1    96%

源代码：https://github.com/mrdulin/python-codelab/tree/master/src/stackoverflow/50242955

该链接为源代码的地址。

- Lin Du