在类中使用的 python 模拟全局函数

Question

我似乎无法理解 Python 中的模拟。我有一个全局函数:

a.py:

def has_permission(args):
    ret_val = ...get-true-or-false...
    return ret_val

b.py:

class MySerializer(HyperlinkedModelSerializer):

     def get_fields():
         fields = super().get_fields()
         for f in :
             if has_permission(...):
                 ret_val[f.name] = fields[f]
         return ret_val

c.py:

class CountrySerializer(MySerializer):
    class Meta:
        model = Country

问题:现在我想测试c.py，但我想模拟a.py中定义的has_permission函数，但在get_fields中调用- 在 b.py 中定义的类 MySerializer 的方法...我该怎么做？

我试过这样的事情:

@patch('b.MySerializer.has_permission')

和

@patch('b.MySerializer.get_fields.has_permission')

和

@patch('a.has_permission')

但是我尝试的所有方法要么不起作用并且 has_permission 仍在执行，要么 python 提示它找不到属性“has_permission”

补丁完成时间:

测试.py

class TestSerializerFields(TestCase):
    @patch(... the above examples....)
    def test_my_country_serializer():
        s = CountrySerializer()
        self..assertTrue(issubclass(my_serializer_fields.MyCharField, type(s.get_fields()['field1'])))

Answer 1

您需要在 b 模块中修补全局:

@patch('b.has_permission')

因为这是您的代码查找它的地方。

另见 Where to patch section mock 文档。