@@ -229,6 +229,7 @@ def update_expected_user_agent(expected):
229229
230230 def test_websocket (self ):
231231 async def app_asgi (scope , receive , send ):
232+ assert scope ["type" ] == "websocket"
232233 payload = await receive ()
233234 if payload .get ("type" ) == "http.request" :
234235 await send (
@@ -244,7 +245,6 @@ async def app_asgi(scope, receive, send):
244245 app = otel_asgi .OpenTelemetryMiddleware (app_asgi )
245246 self .seed_app (app )
246247 self .send_default_request ()
247- outputs = self .get_all_output ()
248248 span_list = self .memory_exporter .get_finished_spans ()
249249 self .assertEqual (len (span_list ), 4 )
250250 expected = [
@@ -257,14 +257,14 @@ async def app_asgi(scope, receive, send):
257257
258258 def test_lifespan (self ):
259259 async def app_asgi (scope , receive , send ):
260+ assert scope ["type" ] == "lifespan"
260261 await receive ()
261262 await send ({})
262263
263264 self .scope ["type" ] = "lifespan"
264265 app = otel_asgi .OpenTelemetryMiddleware (app_asgi )
265266 self .seed_app (app )
266267 self .send_default_request ()
267- outputs = self .get_all_output ()
268268 span_list = self .memory_exporter .get_finished_spans ()
269269 self .assertEqual (len (span_list ), 0 )
270270
0 commit comments