test_media_resource_field.py 文件源码

def test_get_queryset(km_user_factory, media_resource_factory):
    """
    The queryset should return all the items owned by the Know Me user
    that is provided as context.
    """
    km_user = km_user_factory()
    media_resource_factory(km_user=km_user)

    media_resource_factory()

    with mock.patch(
            'know_me.serializers.fields.MediaResourceField.context',
            new_callable=mock.PropertyMock) as mock_context:
        mock_context.return_value = {
            'km_user': km_user,
        }

        field = fields.MediaResourceField()

        result = field.get_queryset()
        expected = km_user.media_resources.all()

        assert list(result) == list(expected)