Question

我有一个要尝试使用.endswith函数的函数，但是每次我尝试使用补丁模拟它时，都会出现错误。

with patch("killme.endswith",MagicMock()) as mock_endswith

我尝试将killme.endswith替换为以下内容：

killme.UserString.endswith
killme.__builtin__.endswith
killme.__builtin__.str.endswith
killme.str.endswith

killme.py

def foo(in_str):
 if in_str.endswith("bob"):
     return True
 return False`

killme_test.py

import killme
import unittest
from mock import MagicMock, patch


class tests(unittest.TestCase):
    def test_foo(self):
        with patch("killme.endswith", MagicMock()) as mock_endswith:
            mock_endswith.return_value = True
            result = killme.foo("xxx")
            self.assertTrue(result)

错误：

Traceback (most recent call last):
  File "C:\Python27\lib\unittest\case.py", line 329, in run
    testMethod()
  File "C:\Users\bisaacs\Desktop\gen2\tools\python\killme_test.py", line 8, in test_foo
    with patch("killme.endswith", MagicMock()) as mock_endswith:
  File "C:\Python27\lib\site-packages\mock\mock.py", line 1369, in __enter__
    original, local = self.get_original()
  File "C:\Python27\lib\site-packages\mock\mock.py", line 1343, in get_original
    "%s does not have the attribute %r" % (target, name)
AttributeError: <module 'killme' from 'C:\Users\bisaacs\Desktop\gen2\tools\python\killme.py'> does not have the attribute 'endswith'

Answer 1

endswith是内置的str方法，因此您不能简单地用killme.endswith覆盖它。代替此，您可以将模拟对象传递到foo函数中。该对象将具有与str相同的接口，但模拟了withwith方法

mocked_str = Mock()
mocked_str.endswith.return_value = True # or something else you want
mocked_str.endswith('something') # True or something else

killme.foo(mocked_str)

如何模拟/修补.endswith（）？

1 个答案: