Skip to content

Commit 213eb58

Browse files
test: Remove unnecessary filter clauses (#6548)
1 parent 4b35d66 commit 213eb58

20 files changed

Lines changed: 173 additions & 181 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
@@ -840,7 +840,7 @@ def test_response_trace(
840840
assert status == "200 OK"
841841

842842
sentry_sdk.flush()
843-
spans = [item.payload for item in items if item.type == "span"]
843+
spans = [item.payload for item in items]
844844

845845
assert (
846846
'- sentry.op="view.response.render": name="serialize response"'
@@ -1055,7 +1055,7 @@ def test_django_connect_trace(
10551055
assert status == "200 OK"
10561056

10571057
sentry_sdk.flush()
1058-
spans = [item.payload for item in items if item.type == "span"]
1058+
spans = [item.payload for item in items]
10591059

10601060
for span in spans:
10611061
if span["attributes"].get("sentry.op") == "db":
@@ -1171,7 +1171,7 @@ def test_db_connection_span_data(
11711171
assert status == "200 OK"
11721172

11731173
sentry_sdk.flush()
1174-
spans = [item.payload for item in items if item.type == "span"]
1174+
spans = [item.payload for item in items]
11751175

11761176
for span in spans:
11771177
if span["attributes"].get("sentry.op") == "db":
@@ -1481,7 +1481,7 @@ def test_template_tracing_meta(
14811481
content, _, _ = unpack_werkzeug_response(client.get(reverse("template_test3")))
14821482
rendered_meta = content.decode("utf-8")
14831483

1484-
events = [item.payload for item in items if item.type == "event"]
1484+
events = [item.payload for item in items]
14851485
else:
14861486
events = capture_events()
14871487

@@ -1710,7 +1710,7 @@ def test_render_spans(
17101710
items = capture_items("span")
17111711
client.get(url)
17121712
sentry_sdk.flush()
1713-
spans = [item.payload for item in items if item.type == "span"]
1713+
spans = [item.payload for item in items]
17141714
assert expected_line in render_span_tree(spans)
17151715
else:
17161716
views_tests = [
@@ -2033,7 +2033,7 @@ def test_signals_spans_filtering(
20332033
client.get(reverse("send_myapp_custom_signal"))
20342034

20352035
sentry_sdk.flush()
2036-
spans = [item.payload for item in items if item.type == "span"]
2036+
spans = [item.payload for item in items]
20372037
assert (
20382038
render_span_tree(spans)
20392039
== """\
@@ -2275,7 +2275,7 @@ def test_span_origin(
22752275
client.get(reverse("view_with_signal"))
22762276

22772277
sentry_sdk.flush()
2278-
spans = [item.payload for item in items if item.type == "span"]
2278+
spans = [item.payload for item in items]
22792279

22802280
assert spans[-1]["attributes"]["sentry.origin"] == "auto.http.django"
22812281

@@ -2324,7 +2324,7 @@ def test_transaction_http_method_default(
23242324
client.head("/nomessage")
23252325

23262326
sentry_sdk.flush()
2327-
spans = [item.payload for item in items if item.type == "span"]
2327+
spans = [item.payload for item in items]
23282328

23292329
assert spans[2]["attributes"][SPANDATA.HTTP_REQUEST_METHOD] == "GET"
23302330
else:
@@ -2368,7 +2368,7 @@ def test_transaction_http_method_custom(
23682368
client.head("/nomessage")
23692369

23702370
sentry_sdk.flush()
2371-
spans = [item.payload for item in items if item.type == "span"]
2371+
spans = [item.payload for item in items]
23722372

23732373
assert spans[4]["attributes"][SPANDATA.HTTP_REQUEST_METHOD] == "OPTIONS"
23742374
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)