不会调用Pytest PropertyMock作为MagicMock的返回值

2024-09-27 07:33:37 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图将MagicMock实例的return_value模拟为PropertyMock,但它似乎没有调用PropertyMock对象,即没有返回实际值。在

class MySerializer(serializers.ModelSerializer):

    class Meta:
        model = MyModel
        # ...
        fields = (
            "my_field"
        )

    my_field = serializers.SerializerMethodField()

    def get_my_field(self, instance):
        other_models = instance.other_models.filter(stuff="stuff")
        if other_models:
            # returns <class 'unittest.mock.PropertyMock'> instead of an actual value
            print(type(other_models.get().my_property))
            # throws error since it isn't receiving a string
            return ujson.loads(other_models.get().my_property)
        return {}


# test case
def test_hello(mocker):
    expected_value = ujson.dumps({'hello': 2})
    instance_mock = mocker.MagicMock()
    other_model_mock = mocker.MagicMock()
    other_model_mock.get().my_property = mocker.PropertyMock(return_value=expected_value)
    instance_mock.other_models.filter.return_value = other_model_mock

    serializer = MySerializer()
    observed_value = serializer.get_my_field(instance_mock)
    assert expected_value == observed_value

Tags: instancefieldgetmodelreturnvaluemodelsmy
1条回答
网友
1楼 · 发布于 2024-09-27 07:33:37

我不需要将other_model_mock设置为MagicMock,只需将其设置为PropertyMock工作

expected_value = {'hello', 2}
instance_mock = mocker.MagicMock()
other_model_mock = mocker.PropertyMock()
other_model_mock.get().my_property.return_value = ujson.dumps(expected_value)
instance_mock.other_models.filter.return_value = other_model_mock
# ... 

相关问题 更多 >

    热门问题