Example #1
0
    def test_queryset(self, mock_super_get_queryset, mock_get_form_queryset, mock_get_object_or_404):
        request = mock.Mock()
        viewset = FilledPDFFormViewSet(request=request, kwargs={"form_pk": 123})

        actual = viewset.get_queryset()

        assert actual == mock_super_get_queryset.return_value.filter.return_value
        assert viewset.form == mock_get_object_or_404.return_value
        mock_super_get_queryset.return_value.filter.assert_called_once_with(form__owner_id=request.user.pk, form_id=123)
        mock_get_form_queryset.assert_called_once_with()
        mock_get_object_or_404.assert_called_once_with(mock_get_form_queryset.return_value, pk=123)
Example #2
0
    def test_queryset(self, mock_super_get_queryset, mock_get_form_queryset,
                      mock_get_object_or_404):
        request = mock.Mock()
        viewset = FilledPDFFormViewSet(request=request,
                                       kwargs={'form_pk': 123})

        actual = viewset.get_queryset()

        assert actual == mock_super_get_queryset.return_value.filter.return_value
        assert viewset.form == mock_get_object_or_404.return_value
        mock_super_get_queryset.return_value.filter.assert_called_once_with(
            form__owner_id=request.user.pk, form_id=123)
        mock_get_form_queryset.assert_called_once_with()
        mock_get_object_or_404.assert_called_once_with(
            mock_get_form_queryset.return_value, pk=123)