[Python] How can I properly test swagger_auto_schema for methods, request_body, and responses...

I’m working on testing a Django REST Framework (DRF) CartViewSet using pytest, and I need to verify the swagger_auto_schema properties like the HTTP method, request body, and responses for different actions (e.g., add, remove, clear).

I have the following code in my CartViewSet:

class CartViewSet(GenericViewSet, RetrieveModelMixin, ListModelMixin):
# Other viewset code...

@swagger_auto_schema(
method="post",
request_body=AddToCartSerializer,
responses={
201: openapi.Response(description="Item added successfully."),
400: openapi.Response(description="Invalid input data"),
},
)
@action(detail=False, methods=["post"], url_path="add")
def add(self, request):
# Logic for adding an item to the cart
pass


Now, I want to write a pytest unit test to check the following for the add method:

1. HTTP Method: Ensure the swagger_auto_schema method is POST.
2. Request Body: Ensure the correct serializer (AddToCartSerializer) is set for the request body.
3. Responses: Verify that the response status codes (201 and 400) and their descriptions are properly set.

Could someone guide me on how to properly test the swagger_auto_schema properties for method, request body, and responses in pytest?

Any help or insights would be greatly appreciated!

Continue reading...