WIP

Author: huynguyengl99

tests/ext/fast_channels/test_views.py

@@ -12,7 +12,7 @@
     asyncapi_spec_yaml,
     generate_asyncapi_schema,
 )
-from starlette.applications import Starlette
+from fastapi import FastAPI
 from starlette.requests import Request
 from starlette.responses import HTMLResponse, JSONResponse, Response
 from starlette.routing import WebSocketRoute
@@ -31,7 +31,7 @@ def setup_method(self) -> None:
         async def websocket_endpoint(websocket: WebSocket):
             pass
 
-        self.app = Starlette(
+        self.app = FastAPI(
             routes=[
                 WebSocketRoute("/ws", websocket_endpoint),
             ]
@@ -93,7 +93,7 @@ class TestAsyncAPISpecJSON:
     def setup_method(self) -> None:
         """Set up test fixtures."""
         self.mock_request = Mock(spec=Request)
-        self.app = Starlette()
+        self.app = FastAPI()
         # Add FastAPI-like attributes
         self.app.title = "Test API"
         self.app.version = "1.0.0"
@@ -163,7 +163,7 @@ class TestAsyncAPISpecYAML:
     def setup_method(self) -> None:
         """Set up test fixtures."""
         self.mock_request = Mock(spec=Request)
-        self.app = Starlette()
+        self.app = FastAPI()
         # Add FastAPI-like attributes
         self.app.title = "Test API"
         self.app.version = "1.0.0"
@@ -256,7 +256,7 @@ class TestAsyncAPIDocs:
     def setup_method(self) -> None:
         """Set up test fixtures."""
         self.mock_request = Mock(spec=Request)
-        self.app = Starlette()
+        self.app = FastAPI()
         # Add FastAPI-like attributes
         self.app.title = "Test API"
         self.app.version = "1.0.0"