Esempio n. 1
0
    def test_get_object(self, mock_get_queryset):
        field = MultiplePKsHyperlinkedIdentityField(
            'foo',
            lookup_fields=['one', 'two'],
            lookup_url_kwargs=['foo', 'bar'],
        )

        actual = field.get_object(None, None, {'foo': 'value', 'bar': 'other'})

        assert actual == mock_get_queryset.return_value.get.return_value
        mock_get_queryset.assert_called_once_with()
        mock_get_queryset.return_value.get.assert_called_once_with(
            one='value', two='other'
        )