与模块python django mock同名的补丁函数

2024年2月5日 212次阅读 
|-- my_module
|   |-- __init__.py
|   |-- function.py
`-- test.py

在function.py中：

import other_function

def function():
    doStuff()
    other_function()
    return

在__init__.py中

from .function import function

在我的test.py中

from django.test import TestCase
from mock import patch
from my_module import function

class Test(TestCase):

    @patch('my_module.function.other_function')
    def function_test(self, mock_other_function):
         function()

当我跑这个我得到了
AttributeError的：

<@task: my_module.function.function of project:0x7fed6b4fc198> does not have
the attribute ‘other_function’

这意味着我正在尝试修补函数“function”而不是模块“function”.我不知道如何理解我想修补模块.

我还想避免重命名我的模块或功能.

有任何想法吗？

[编辑]
你可以在https://github.com/vthorey/example_mock找到一个例子
跑

python manage.py test

最佳答案 您可以在__init__.py中以不同的名称使模块可用：

from . import function as function_module
from .function import function

然后你可以在test.py中执行以下操作：

from django.test import TestCase
from mock import patch
from my_module import function

class Test(TestCase):

    @patch('my_module.function_module.other_function')
    def function_test(self, mock_other_function):
         function()

我不认为这是一个特别优雅的解决方案 – 对于一个随意的读者来说代码并不是很清楚.

点赞