Skip to content

Commit f20a4cf

Browse files
test: Remove unnecessary filter clauses
1 parent b71f4ff commit f20a4cf

20 files changed

Lines changed: 174 additions & 182 deletions

tests/integrations/anthropic/test_anthropic.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5074,7 +5074,7 @@ def test_system_prompt_with_complex_structure(
50745074
assert response == EXAMPLE_MESSAGE
50755075

50765076
sentry_sdk.flush()
5077-
spans = [item.payload for item in items if item.type == "span"]
5077+
spans = [item.payload for item in items]
50785078
assert len(spans) == 1
50795079
(span,) = spans
50805080

@@ -5426,7 +5426,7 @@ def test_message_with_url_image(
54265426
client.messages.create(max_tokens=1024, messages=messages, model="model")
54275427

54285428
sentry_sdk.flush()
5429-
spans = [item.payload for item in items if item.type == "span"]
5429+
spans = [item.payload for item in items]
54305430
(span,) = spans
54315431

54325432
stored_messages = json.loads(
@@ -5498,7 +5498,7 @@ def test_message_with_file_image(
54985498
client.messages.create(max_tokens=1024, messages=messages, model="model")
54995499

55005500
sentry_sdk.flush()
5501-
spans = [item.payload for item in items if item.type == "span"]
5501+
spans = [item.payload for item in items]
55025502
(span,) = spans
55035503

55045504
stored_messages = json.loads(
@@ -5614,7 +5614,7 @@ def test_message_with_url_pdf(
56145614
client.messages.create(max_tokens=1024, messages=messages, model="model")
56155615

56165616
sentry_sdk.flush()
5617-
spans = [item.payload for item in items if item.type == "span"]
5617+
spans = [item.payload for item in items]
56185618
(span,) = spans
56195619

56205620
stored_messages = json.loads(
@@ -5686,7 +5686,7 @@ def test_message_with_file_document(
56865686
client.messages.create(max_tokens=1024, messages=messages, model="model")
56875687

56885688
sentry_sdk.flush()
5689-
spans = [item.payload for item in items if item.type == "span"]
5689+
spans = [item.payload for item in items]
56905690
(span,) = spans
56915691

56925692
stored_messages = json.loads(
@@ -5916,7 +5916,7 @@ def test_binary_content_not_stored_when_pii_disabled(
59165916
client.messages.create(max_tokens=1024, messages=messages, model="model")
59175917

59185918
sentry_sdk.flush()
5919-
spans = [item.payload for item in items if item.type == "span"]
5919+
spans = [item.payload for item in items]
59205920
(span,) = spans
59215921

59225922
# Messages should not be stored
@@ -5980,7 +5980,7 @@ def test_binary_content_not_stored_when_prompts_disabled(
59805980
client.messages.create(max_tokens=1024, messages=messages, model="model")
59815981

59825982
sentry_sdk.flush()
5983-
spans = [item.payload for item in items if item.type == "span"]
5983+
spans = [item.payload for item in items]
59845984
(span,) = spans
59855985

59865986
# Messages should not be stored
@@ -6213,7 +6213,7 @@ def test_input_tokens_include_cache_read_nonstreaming(
62136213
)
62146214

62156215
sentry_sdk.flush()
6216-
(span,) = [item.payload for item in items if item.type == "span"]
6216+
(span,) = [item.payload for item in items]
62176217

62186218
# input_tokens should be total: 19 (non-cached) + 2846 (cache_read) = 2865
62196219
assert span["attributes"][SPANDATA.GEN_AI_USAGE_INPUT_TOKENS] == 2865

tests/integrations/boto3/test_s3.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def test_basic(
4747
span.end()
4848

4949
sentry_sdk.flush()
50-
spans = [item.payload for item in items if item.type == "span"]
50+
spans = [item.payload for item in items]
5151
assert len(spans) == 2
5252
span = spans[0]
5353
assert span["attributes"]["sentry.op"] == "http.client"
@@ -102,7 +102,7 @@ def test_streaming(
102102
span.end()
103103

104104
sentry_sdk.flush()
105-
spans = [item.payload for item in items if item.type == "span"]
105+
spans = [item.payload for item in items]
106106
assert len(spans) == 3
107107

108108
span1 = spans[0]
@@ -196,7 +196,7 @@ def test_streaming_close(
196196
span.end()
197197

198198
sentry_sdk.flush()
199-
spans = [item.payload for item in items if item.type == "span"]
199+
spans = [item.payload for item in items]
200200
assert len(spans) == 3
201201
span1 = spans[0]
202202
assert span1["attributes"]["sentry.op"] == "http.client"
@@ -258,7 +258,7 @@ def test_omit_url_data_if_parsing_fails(
258258
span.end()
259259

260260
sentry_sdk.flush()
261-
spans = [item.payload for item in items if item.type == "span"]
261+
spans = [item.payload for item in items]
262262
assert spans[0]["attributes"] == ApproxDict(
263263
{
264264
"http.request.method": "GET",
@@ -334,7 +334,7 @@ def test_span_origin(
334334
_ = [obj for obj in bucket.objects.all()]
335335

336336
sentry_sdk.flush()
337-
spans = [item.payload for item in items if item.type == "span"]
337+
spans = [item.payload for item in items]
338338

339339
assert spans[1]["attributes"]["sentry.origin"] == "manual"
340340
assert spans[0]["attributes"]["sentry.origin"] == "auto.http.boto3"

tests/integrations/django/asgi/test_asgi.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,7 @@ async def test_active_thread_id(
235235
data = json.loads(response["body"])
236236

237237
sentry_sdk.flush()
238-
spans = [item.payload for item in items if item.type == "span"]
238+
spans = [item.payload for item in items]
239239

240240
for span in spans:
241241
if span["is_segment"] is False:
@@ -393,7 +393,7 @@ async def test_async_middleware_spans(
393393
assert response["status"] == 200
394394

395395
sentry_sdk.flush()
396-
spans = [item.payload for item in items if item.type == "span"]
396+
spans = [item.payload for item in items]
397397

398398
# Filter out signal-receiver spans — their ordering depends on Django
399399
# module import order and is not what this middleware test verifies.
@@ -911,7 +911,7 @@ async def test_async_view(
911911
await comm.wait()
912912

913913
sentry_sdk.flush()
914-
spans = [item.payload for item in items if item.type == "span"]
914+
spans = [item.payload for item in items]
915915
assert spans[5]["name"] == "/simple_async_view"
916916
else:
917917
events = capture_events()
@@ -961,7 +961,7 @@ async def test_transaction_http_method_default(
961961
await comm.wait()
962962

963963
sentry_sdk.flush()
964-
spans = [item.payload for item in items if item.type == "span"]
964+
spans = [item.payload for item in items]
965965
assert spans[5]["attributes"]["http.request.method"] == "GET"
966966
else:
967967
events = capture_events()
@@ -1025,7 +1025,7 @@ async def test_transaction_http_method_custom(
10251025
await comm.wait()
10261026

10271027
sentry_sdk.flush()
1028-
spans = [item.payload for item in items if item.type == "span"]
1028+
spans = [item.payload for item in items]
10291029

10301030
assert spans[10]["attributes"][SPANDATA.HTTP_REQUEST_METHOD] == "OPTIONS"
10311031
assert spans[16]["attributes"][SPANDATA.HTTP_REQUEST_METHOD] == "HEAD"

tests/integrations/django/test_basic.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -807,7 +807,7 @@ def test_response_trace(
807807
assert status == "200 OK"
808808

809809
sentry_sdk.flush()
810-
spans = [item.payload for item in items if item.type == "span"]
810+
spans = [item.payload for item in items]
811811

812812
assert (
813813
'- sentry.op="view.response.render": name="serialize response"'
@@ -1022,7 +1022,7 @@ def test_django_connect_trace(
10221022
assert status == "200 OK"
10231023

10241024
sentry_sdk.flush()
1025-
spans = [item.payload for item in items if item.type == "span"]
1025+
spans = [item.payload for item in items]
10261026

10271027
for span in spans:
10281028
if span["attributes"].get("sentry.op") == "db":
@@ -1138,7 +1138,7 @@ def test_db_connection_span_data(
11381138
assert status == "200 OK"
11391139

11401140
sentry_sdk.flush()
1141-
spans = [item.payload for item in items if item.type == "span"]
1141+
spans = [item.payload for item in items]
11421142

11431143
for span in spans:
11441144
if span["attributes"].get("sentry.op") == "db":
@@ -1448,7 +1448,7 @@ def test_template_tracing_meta(
14481448
content, _, _ = unpack_werkzeug_response(client.get(reverse("template_test3")))
14491449
rendered_meta = content.decode("utf-8")
14501450

1451-
events = [item.payload for item in items if item.type == "event"]
1451+
events = [item.payload for item in items]
14521452
else:
14531453
events = capture_events()
14541454

@@ -1677,7 +1677,7 @@ def test_render_spans(
16771677
items = capture_items("span")
16781678
client.get(url)
16791679
sentry_sdk.flush()
1680-
spans = [item.payload for item in items if item.type == "span"]
1680+
spans = [item.payload for item in items]
16811681
assert expected_line in render_span_tree(spans)
16821682
else:
16831683
views_tests = [
@@ -2000,7 +2000,7 @@ def test_signals_spans_filtering(
20002000
client.get(reverse("send_myapp_custom_signal"))
20012001

20022002
sentry_sdk.flush()
2003-
spans = [item.payload for item in items if item.type == "span"]
2003+
spans = [item.payload for item in items]
20042004
assert (
20052005
render_span_tree(spans)
20062006
== """\
@@ -2242,7 +2242,7 @@ def test_span_origin(
22422242
client.get(reverse("view_with_signal"))
22432243

22442244
sentry_sdk.flush()
2245-
spans = [item.payload for item in items if item.type == "span"]
2245+
spans = [item.payload for item in items]
22462246

22472247
assert spans[-1]["attributes"]["sentry.origin"] == "auto.http.django"
22482248

@@ -2291,7 +2291,7 @@ def test_transaction_http_method_default(
22912291
client.head("/nomessage")
22922292

22932293
sentry_sdk.flush()
2294-
spans = [item.payload for item in items if item.type == "span"]
2294+
spans = [item.payload for item in items]
22952295

22962296
assert spans[2]["attributes"][SPANDATA.HTTP_REQUEST_METHOD] == "GET"
22972297
else:
@@ -2335,7 +2335,7 @@ def test_transaction_http_method_custom(
23352335
client.head("/nomessage")
23362336

23372337
sentry_sdk.flush()
2338-
spans = [item.payload for item in items if item.type == "span"]
2338+
spans = [item.payload for item in items]
23392339

23402340
assert spans[4]["attributes"][SPANDATA.HTTP_REQUEST_METHOD] == "OPTIONS"
23412341
assert spans[7]["attributes"][SPANDATA.HTTP_REQUEST_METHOD] == "HEAD"

tests/integrations/django/test_cache_module.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def test_cache_spans_disabled_middleware(
119119
client.get(reverse("not_cached_view"))
120120

121121
sentry_sdk.flush()
122-
spans = [item.payload for item in items if item.type == "span"]
122+
spans = [item.payload for item in items]
123123
assert len(spans) == 2
124124
else:
125125
events = capture_events()
@@ -162,7 +162,7 @@ def test_cache_spans_disabled_decorator(
162162
client.get(reverse("cached_view"))
163163

164164
sentry_sdk.flush()
165-
spans = [item.payload for item in items if item.type == "span"]
165+
spans = [item.payload for item in items]
166166
assert len(spans) == 2
167167
else:
168168
events = capture_events()
@@ -205,7 +205,7 @@ def test_cache_spans_disabled_templatetag(
205205
client.get(reverse("view_with_cached_template_fragment"))
206206

207207
sentry_sdk.flush()
208-
spans = [item.payload for item in items if item.type == "span"]
208+
spans = [item.payload for item in items]
209209
assert len(spans) == 2
210210
else:
211211
events = capture_events()
@@ -250,7 +250,7 @@ def test_cache_spans_middleware(
250250
client.get(reverse("not_cached_view"))
251251

252252
sentry_sdk.flush()
253-
spans = [item.payload for item in items if item.type == "span"]
253+
spans = [item.payload for item in items]
254254
# first_event - cache.get
255255
assert spans[0]["attributes"]["sentry.op"] == "cache.get"
256256
assert spans[0]["name"].startswith("views.decorators.cache.cache_header.")
@@ -370,7 +370,7 @@ def test_cache_spans_decorator(
370370
client.get(reverse("cached_view"))
371371

372372
sentry_sdk.flush()
373-
spans = [item.payload for item in items if item.type == "span"]
373+
spans = [item.payload for item in items]
374374
# first_event - cache.get
375375
assert spans[0]["attributes"]["sentry.op"] == "cache.get"
376376
assert spans[0]["name"].startswith("views.decorators.cache.cache_header.")
@@ -470,7 +470,7 @@ def test_cache_spans_templatetag(
470470
client.get(reverse("view_with_cached_template_fragment"))
471471

472472
sentry_sdk.flush()
473-
spans = [item.payload for item in items if item.type == "span"]
473+
spans = [item.payload for item in items]
474474
assert len(spans) == 5
475475
# first_event - cache.get
476476
assert spans[0]["attributes"]["sentry.op"] == "cache.get"
@@ -607,7 +607,7 @@ def test_cache_spans_location_with_port(
607607
client.get(reverse("cached_view"))
608608

609609
sentry_sdk.flush()
610-
spans = [item.payload for item in items if item.type == "span"]
610+
spans = [item.payload for item in items]
611611

612612
for span in spans:
613613
if span["is_segment"] is True:
@@ -660,7 +660,7 @@ def test_cache_spans_location_without_port(
660660
client.get(reverse("cached_view"))
661661

662662
sentry_sdk.flush()
663-
spans = [item.payload for item in items if item.type == "span"]
663+
spans = [item.payload for item in items]
664664

665665
for span in spans:
666666
if span["is_segment"] is True:
@@ -709,7 +709,7 @@ def test_cache_spans_location_with_cluster(
709709
client.get(reverse("cached_view"))
710710

711711
sentry_sdk.flush()
712-
spans = [item.payload for item in items if item.type == "span"]
712+
spans = [item.payload for item in items]
713713

714714
for span in spans:
715715
# because it is a cluster we do not know what host is actually accessed, so we omit the data
@@ -757,7 +757,7 @@ def test_cache_spans_item_size(
757757
client.get(reverse("cached_view"))
758758

759759
sentry_sdk.flush()
760-
spans = [item.payload for item in items if item.type == "span"]
760+
spans = [item.payload for item in items]
761761
assert len(spans) == 7
762762
assert spans[0]["attributes"]["sentry.op"] == "cache.get"
763763
assert not spans[0]["attributes"]["cache.hit"]
@@ -851,7 +851,7 @@ def test_cache_spans_get_custom_default(
851851
cache.get(f"S{id + 2}", default="null")
852852

853853
sentry_sdk.flush()
854-
spans = [item.payload for item in items if item.type == "span"]
854+
spans = [item.payload for item in items]
855855
assert len(spans) == 9
856856

857857
assert spans[0]["attributes"]["sentry.op"] == "cache.put"
@@ -956,7 +956,7 @@ def test_cache_spans_get_many(
956956
cache.get_many([f"S{id}", f"S{id + 1}"])
957957

958958
sentry_sdk.flush()
959-
spans = [item.payload for item in items if item.type == "span"]
959+
spans = [item.payload for item in items]
960960
assert len(spans) == 8
961961

962962
assert spans[2]["attributes"]["sentry.op"] == "cache.get"
@@ -1058,7 +1058,7 @@ def test_cache_spans_set_many(
10581058
cache.get(f"S{id}")
10591059

10601060
sentry_sdk.flush()
1061-
spans = [item.payload for item in items if item.type == "span"]
1061+
spans = [item.payload for item in items]
10621062
assert len(spans) == 5
10631063

10641064
assert spans[2]["attributes"]["sentry.op"] == "cache.put"
@@ -1125,7 +1125,7 @@ def test_span_origin_cache(
11251125
client.get(reverse("cached_view"))
11261126

11271127
sentry_sdk.flush()
1128-
spans = [item.payload for item in items if item.type == "span"]
1128+
spans = [item.payload for item in items]
11291129

11301130
assert spans[1]["attributes"]["sentry.origin"] == "auto.http.django"
11311131

0 commit comments

Comments
 (0)