我可以像這樣在 ReadOnlyModelViewSet 中測驗 get_queryset() 。
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)
視圖.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 變數為真。
有誰知道如何為這種情況撰寫測驗用例？
uj5u.com熱心網友回復：
您可以在測驗中添加另一個裝飾器來應用它，如下所示：
@patch.object(CatalogView, "swagger_fake_view", True, create=True)
轉載請註明出處，本文鏈接：https://www.uj5u.com/yidong/416385.html
標籤：