Тест DRF Swagger Fake View в get_queryset
Я могу протестировать get_queryset() в ReadOnlyModelViewSet следующим образом.
class CatalogViewTests(APITestCase):
def setUp(self) -> None:
self.cur_user = UserFactory()
self.cur_dataset = DataSetFactory(created_by=self.cur_user)
@patch.object(CatalogView, "permission_classes", [])
def test_get_queryset(self):
url = reverse('data_tab:catalog-list')
response = self.client.get(url, format='json', **{DATASET_ID: self.cur_dataset.id})
self.assertEqual(response.status_code, status.HTTP_200_OK)
views.py
class CatalogView(ReadOnlyModelViewSet):
"""
Returns the Data tab catalog page with pagination
"""
serializer_class = CatalogSerializer
permission_classes = [UserHasDatasetChangeAccess]
pagination_class = StandardResultsSetPagination
queryset = TableMeta.objects.all()
renderer_classes = [JSONRenderer]
ordering_fields = ["created_on", "modified_on"]
ordering = ["-modified_on"]
def get_queryset(self):
if getattr(self, "swagger_fake_view", False):
# queryset just for schema generation metadata
return TableMeta.objects.none()
return TableMeta.objects.filter(
dataset=get_object_or_404(DataSet, id=self.request.META.get(DATASET_ID, ""))
).prefetch_related(
"table_columns", "table_metrics", "table_relationship_source"
)
Тестовый пример работает правильно, и я получаю вывод, как ожидалось.
Но в test_report эта строка не проверяется.
Как мы видим, красная строка все еще отображается непроверенной, потому что она ожидает, что переменная swagger_faker_view будет равна true.
знает ли кто-нибудь, как написать тестовый пример для этого сценария?
Вы можете добавить другой декоратор к вашему тесту, чтобы применить это следующим образом:
@patch.object(CatalogView, "swagger_fake_view", True)