尝试修补成员函数会产生错误属性：＆＃x27;模块＆＃x27;对象没有属性＆＃x27;对象＆＃x27;

发布于 2025-02-09 05:33:32 字数 944 浏览 0 评论 0原文

我想从ut中断言testrunner.test_run，其中一些深嵌套的函数stript.run_cmd用字符串参数“ unique cmd” 。我的设置与以下内容相似：

# Module application/engine/prompt.py
class Prompt:
    def run_cmd(self, input):
        pass
        
# Module: application/scheduler/runner.py   
class Runner:
    def __init__(self):
        self.prompt = application.engine.prompt.Prompt()

    def run(self):
        self.prompt.run_cmd("unique cmd")
        
        
# Module tests/application/scheduler/runner_test.py
class TestRunner(unittest.TestCase):
    ...
    def test_run(self):
        # calls Runner.run
        # Objective assert that Prompt.run is called with the argument "unique cmd"
        # Current attempt below:
        with mock.patch(application.engine.prompt, "run_cmd") as mock_run_cmd:
           pass

不幸的是，我尝试模拟stript.run_cmd失败的尝试使用错误消息

attributeError：'module'对象没有属性'对象'

原文

I would like to assert from a UT, TestRunner.test_run that some deeply nested function Prompt.run_cmd is called with the string argument "unique cmd". My setup besically resembles the following:

# Module application/engine/prompt.py
class Prompt:
    def run_cmd(self, input):
        pass
        
# Module: application/scheduler/runner.py   
class Runner:
    def __init__(self):
        self.prompt = application.engine.prompt.Prompt()

    def run(self):
        self.prompt.run_cmd("unique cmd")
        
        
# Module tests/application/scheduler/runner_test.py
class TestRunner(unittest.TestCase):
    ...
    def test_run(self):
        # calls Runner.run
        # Objective assert that Prompt.run is called with the argument "unique cmd"
        # Current attempt below:
        with mock.patch(application.engine.prompt, "run_cmd") as mock_run_cmd:
           pass

Unfortunately my attempts to mock the Prompt.run_cmd fail with the error message

AttributeError: 'module' object has no attribute 'object'

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

雨的味道风的声音 2025-02-16 05:33:32

如果您想修补一个具体实例，则可以使用mock.patch.Object.Object和wraps（例如，请参见此问题

。您确实必须使用mock.patch。 wraps在这里（至少我不知道一种方法）

。然后，您可以通过您自己的实现来修补。

class MyPrompt(Prompt):
    calls = []

    def run_cmd(self, input):
        super().run_cmd(input)
        # we just add a string in the call list - this could be more sophisticated
        self.__class__.calls.append(input)

class TestRunner(unittest.TestCase):

    def test_run(self):
        with mock.patch("application.engine.prompt.Prompt", MyPrompt) as mock_prompt:
            runner = Runner()
            runner.run()
            self.assertEqual(["unique cmd"], mock_prompt.calls)

If you wanted to patch a concrete instance, you could easily do this using mock.patch.object and wraps (see for example this question.

If you want to patch your function for all instances instead, you indeed have to use mock.patch. In this case you could only mock the class itself, as mocking the method would not work (because it is used on instances, not classes), so you cannot use wraps here (at least I don't know a way to do this).

What you could do instead is derive your own class from Prompt and overwrite the method to collect the calls yourself. You could then patch Prompt by your own implementation. Here is a simple example:

class MyPrompt(Prompt):
    calls = []

    def run_cmd(self, input):
        super().run_cmd(input)
        # we just add a string in the call list - this could be more sophisticated
        self.__class__.calls.append(input)

class TestRunner(unittest.TestCase):

    def test_run(self):
        with mock.patch("application.engine.prompt.Prompt", MyPrompt) as mock_prompt:
            runner = Runner()
            runner.run()
            self.assertEqual(["unique cmd"], mock_prompt.calls)

回复收藏 0 原文

~没有更多了~