fix: improve handling of response

Fixes error, when response is not from ninja or drf

Author: TheSuperiorStanislav

openapi_tester/clients.py

@@ -44,6 +44,8 @@ def request(self, *args, **kwargs) -> Response:  # type: ignore[override]
         response_handler = ResponseHandlerFactory.create(
             *args, response=response, **kwargs
         )
+        if not response_handler:
+            return response
         if self._is_successful_response(response):
             self.schema_tester.validate_request(response_handler=response_handler)
         self.schema_tester.validate_response(response_handler=response_handler)
@@ -145,6 +147,8 @@ def request(self, *args, **kwargs) -> Response:
         response_handler = ResponseHandlerFactory.create(
             *args, response=response, path_prefix=self._ninja_path_prefix, **kwargs
         )
+        if not response_handler:
+            return response
         if self._is_successful_response(response):
             self.schema_tester.validate_request(response_handler=response_handler)
         self.schema_tester.validate_response(response_handler)

openapi_tester/response_handler_factory.py

@@ -3,10 +3,9 @@
 Module that contains the factory to create response handlers.
 """
 
+import contextlib
 from typing import TYPE_CHECKING, Union
 
-from rest_framework.response import Response
-
 from openapi_tester.response_handler import (
     DjangoNinjaResponseHandler,
     DRFResponseHandler,
@@ -26,8 +25,20 @@ class ResponseHandlerFactory:
 
     @staticmethod
     def create(
-        *request_args, response: Union[Response, "HttpResponse"], **kwargs
-    ) -> "ResponseHandler":
-        if isinstance(response, Response):
-            return DRFResponseHandler(response=response)
-        return DjangoNinjaResponseHandler(*request_args, response=response, **kwargs)
+        *request_args,
+        response: "HttpResponse",
+        **kwargs,
+    ) -> Union["ResponseHandler", None]:
+        with contextlib.suppress(ImportError):
+            from rest_framework.response import Response as DRFResponse
+
+            if isinstance(response, DRFResponse):
+                return DRFResponseHandler(response=response)
+        with contextlib.suppress(ImportError):
+            from ninja.testing.client import NinjaResponse
+
+            if isinstance(response, NinjaResponse):
+                return DjangoNinjaResponseHandler(
+                    *request_args, response=response, **kwargs
+                )
+        return None