Pytest PropertyMock как return_value для MagicMock не вызывается - PullRequest
Новая
       28

Pytest PropertyMock как return_value для MagicMock не вызывается

0 голосов
/ 27 августа 2018

Я пытаюсь смоделировать return_value для экземпляра MagicMock как 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

1 Ответ

0 голосов
/ 27 августа 2018

Мне не нужно было устанавливать 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
# ...
...