fix tests

Signed-off-by: emdneto <[email protected]>

Author: emdneto

instrumentation/opentelemetry-instrumentation-starlette/tests/test_starlette_instrumentation.py

@@ -18,7 +18,7 @@
 
 from starlette import applications
 from starlette.responses import PlainTextResponse
-from starlette.routing import Mount, Route, WebSocketRoute
+from starlette.routing import Mount, Route
 from starlette.testclient import TestClient
 from starlette.websockets import WebSocket
 
@@ -621,7 +621,10 @@ def tearDown(self) -> None:
 
     @staticmethod
     def create_starlette_app():
-        def foobar(request):
+        app = applications.Starlette()
+
+        @app.route("/foobar")
+        def _(request):
             return PlainTextResponse(
                 content="hi",
                 headers={
@@ -633,7 +636,8 @@ def foobar(request):
                 },
             )
 
-        async def foobar_web(websocket: WebSocket) -> None:
+        @app.websocket_route("/foobar_web")
+        async def _(websocket: WebSocket) -> None:
             message = await websocket.receive()
             if message.get("type") == "websocket.connect":
                 await websocket.send(
@@ -659,17 +663,11 @@ async def foobar_web(websocket: WebSocket) -> None:
             if message.get("type") == "websocket.disconnect":
                 pass
 
-        return applications.Starlette(
-            routes=[
-                Route("/foobar", foobar),
-                WebSocketRoute("/foobar_web", foobar_web),
-            ]
-        )
+        return app
 
 
 class TestHTTPAppWithCustomHeaders(TestBaseWithCustomHeaders):
     def setUp(self):
-        super().setUp()
         self.test_env_patch = patch.dict(
             "os.environ",
             {
@@ -679,6 +677,7 @@ def setUp(self):
             },
         )
         self.test_env_patch.start()
+        super().setUp()
 
     def tearDown(self):
         self.test_env_patch.stop()
@@ -812,8 +811,8 @@ def setUp(self):
         super().setUp()
 
     def tearDown(self):
-        super().tearDown()
         self.test_env_patch.stop()
+        super().tearDown()
 
     def test_custom_request_headers_in_span_attributes(self):
         expected = {