Skip to content

Commit 9ece50f

Browse files
fix tests failing due to default disabled
1 parent ea71308 commit 9ece50f

File tree

3 files changed

+46
-38
lines changed

3 files changed

+46
-38
lines changed

tests/integrations/aiohttp/test_aiohttp.py

Lines changed: 17 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -637,15 +637,24 @@ async def handler(request):
637637

638638

639639
@pytest.mark.asyncio
640+
@pytest.mark.parametrize("enable_http_request_source", [None, False])
640641
async def test_request_source_disabled(
641-
sentry_init, aiohttp_raw_server, aiohttp_client, capture_events
642+
sentry_init,
643+
aiohttp_raw_server,
644+
aiohttp_client,
645+
capture_events,
646+
enable_http_request_source,
642647
):
643-
sentry_init(
644-
integrations=[AioHttpIntegration()],
645-
traces_sample_rate=1.0,
646-
enable_http_request_source=False,
647-
http_request_source_threshold_ms=0,
648-
)
648+
sentry_options = {
649+
"integrations": [AioHttpIntegration()],
650+
"traces_sample_rate": 1.0,
651+
"http_request_source_threshold_ms": 0,
652+
}
653+
654+
if enable_http_request_source is not None:
655+
sentry_options["enable_http_request_source"] = enable_http_request_source
656+
657+
sentry_init(**sentry_options)
649658

650659
# server for making span request
651660
async def handler(request):
@@ -680,21 +689,18 @@ async def hello(request):
680689

681690

682691
@pytest.mark.asyncio
683-
@pytest.mark.parametrize("enable_http_request_source", [None, True])
684692
async def test_request_source_enabled(
685693
sentry_init,
686694
aiohttp_raw_server,
687695
aiohttp_client,
688696
capture_events,
689-
enable_http_request_source,
690697
):
691698
sentry_options = {
692699
"integrations": [AioHttpIntegration()],
693700
"traces_sample_rate": 1.0,
701+
"enable_http_request_source": True,
694702
"http_request_source_threshold_ms": 0,
695703
}
696-
if enable_http_request_source is not None:
697-
sentry_options["enable_http_request_source"] = enable_http_request_source
698704

699705
sentry_init(**sentry_options)
700706

tests/integrations/httpx/test_httpx.py

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -396,18 +396,24 @@ def test_omit_url_data_if_parsing_fails(sentry_init, capture_events, httpx_mock)
396396
assert SPANDATA.HTTP_QUERY not in event["breadcrumbs"]["values"][0]["data"]
397397

398398

399+
@pytest.mark.parametrize("enable_http_request_source", [None, False])
399400
@pytest.mark.parametrize(
400401
"httpx_client",
401402
(httpx.Client(), httpx.AsyncClient()),
402403
)
403-
def test_request_source_disabled(sentry_init, capture_events, httpx_client, httpx_mock):
404+
def test_request_source_disabled(
405+
sentry_init, capture_events, enable_http_request_source, httpx_client, httpx_mock
406+
):
404407
httpx_mock.add_response()
405-
sentry_init(
406-
integrations=[HttpxIntegration()],
407-
traces_sample_rate=1.0,
408-
enable_http_request_source=False,
409-
http_request_source_threshold_ms=0,
410-
)
408+
sentry_options = {
409+
"integrations": [HttpxIntegration()],
410+
"traces_sample_rate": 1.0,
411+
"http_request_source_threshold_ms": 0,
412+
}
413+
if enable_http_request_source is not None:
414+
sentry_options["enable_http_request_source"] = enable_http_request_source
415+
416+
sentry_init(**sentry_options)
411417

412418
events = capture_events()
413419

@@ -432,23 +438,18 @@ def test_request_source_disabled(sentry_init, capture_events, httpx_client, http
432438
assert SPANDATA.CODE_FUNCTION not in data
433439

434440

435-
@pytest.mark.parametrize("enable_http_request_source", [None, True])
436441
@pytest.mark.parametrize(
437442
"httpx_client",
438443
(httpx.Client(), httpx.AsyncClient()),
439444
)
440-
def test_request_source_enabled(
441-
sentry_init, capture_events, enable_http_request_source, httpx_client, httpx_mock
442-
):
445+
def test_request_source_enabled(sentry_init, capture_events, httpx_client, httpx_mock):
443446
httpx_mock.add_response()
444447
sentry_options = {
445448
"integrations": [HttpxIntegration()],
446449
"traces_sample_rate": 1.0,
450+
"enable_http_request_source": True,
447451
"http_request_source_threshold_ms": 0,
448452
}
449-
if enable_http_request_source is not None:
450-
sentry_options["enable_http_request_source"] = enable_http_request_source
451-
452453
sentry_init(**sentry_options)
453454

454455
events = capture_events()

tests/integrations/stdlib/test_httplib.py

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -376,12 +376,18 @@ def test_option_trace_propagation_targets(
376376
assert "baggage" not in request_headers
377377

378378

379-
def test_request_source_disabled(sentry_init, capture_events):
380-
sentry_init(
381-
traces_sample_rate=1.0,
382-
enable_http_request_source=False,
383-
http_request_source_threshold_ms=0,
384-
)
379+
@pytest.mark.parametrize("enable_http_request_source", [None, False])
380+
def test_request_source_disabled(
381+
sentry_init, capture_events, enable_http_request_source
382+
):
383+
sentry_options = {
384+
"traces_sample_rate": 1.0,
385+
"http_request_source_threshold_ms": 0,
386+
}
387+
if enable_http_request_source is not None:
388+
sentry_options["enable_http_request_source"] = enable_http_request_source
389+
390+
sentry_init(**sentry_options)
385391

386392
events = capture_events()
387393

@@ -403,17 +409,12 @@ def test_request_source_disabled(sentry_init, capture_events):
403409
assert SPANDATA.CODE_FUNCTION not in data
404410

405411

406-
@pytest.mark.parametrize("enable_http_request_source", [None, True])
407-
def test_request_source_enabled(
408-
sentry_init, capture_events, enable_http_request_source
409-
):
412+
def test_request_source_enabled(sentry_init, capture_events):
410413
sentry_options = {
411414
"traces_sample_rate": 1.0,
415+
"enable_http_request_source": True,
412416
"http_request_source_threshold_ms": 0,
413417
}
414-
if enable_http_request_source is not None:
415-
sentry_options["enable_http_request_source"] = enable_http_request_source
416-
417418
sentry_init(**sentry_options)
418419

419420
events = capture_events()

0 commit comments

Comments
 (0)