Skip to content

Commit c963b6f

Browse files
committed
test: update test_asgi.test_websocket
1 parent da2d4f7 commit c963b6f

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/integrations/asgi/test_asgi.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ async def test_trace_from_headers_if_performance_disabled(
349349

350350
@pytest.mark.asyncio
351351
async def test_websocket(sentry_init, asgi3_ws_app, capture_events, request):
352-
sentry_init(send_default_pii=True)
352+
sentry_init(send_default_pii=True, traces_sample_rate=1.0)
353353

354354
events = capture_events()
355355

@@ -362,16 +362,19 @@ async def test_websocket(sentry_init, asgi3_ws_app, capture_events, request):
362362
async with client.websocket_connect(request_url) as ws:
363363
await ws.receive_text()
364364

365-
msg_event, error_event = events
365+
msg_event, error_event, transaction_event = events
366366

367367
assert msg_event["transaction"] == request_url
368-
assert msg_event["transaction_info"]["source"] == "url"
368+
assert msg_event["transaction_info"] == {"source": "url"}
369369
assert msg_event["message"] == "Some message to the world!"
370370

371371
(exc,) = error_event["exception"]["values"]
372372
assert exc["type"] == "ValueError"
373373
assert exc["value"] == "Oh no"
374374

375+
assert transaction_event["transaction"] == request_url
376+
assert transaction_event["transaction_info"] == {"source": "url"}
377+
375378

376379
@pytest.mark.asyncio
377380
async def test_auto_session_tracking_with_aggregates(

0 commit comments

Comments
 (0)