fix more tests

Author: Zylphrex

tests/integrations/django/asgi/test_asgi.py

@@ -104,7 +104,9 @@ async def test_async_views(sentry_init, capture_events, application):
 @pytest.mark.skipif(
     django.VERSION < (3, 1), reason="async views have been introduced in Django 3.1"
 )
-async def test_active_thread_id(sentry_init, capture_envelopes, endpoint, application):
+async def test_active_thread_id(
+    sentry_init, capture_envelopes, teardown_profiling, endpoint, application
+):
     with mock.patch(
         "sentry_sdk.profiler.transaction_profiler.PROFILE_MINIMUM_SAMPLES", 0
     ):
@@ -121,17 +123,18 @@ async def test_active_thread_id(sentry_init, capture_envelopes, endpoint, applic
         await comm.wait()
 
         assert response["status"] == 200, response["body"]
-        assert len(envelopes) == 1
 
-        profiles = [item for item in envelopes[0].items if item.type == "profile"]
-        assert len(profiles) == 1
+    assert len(envelopes) == 1
+
+    profiles = [item for item in envelopes[0].items if item.type == "profile"]
+    assert len(profiles) == 1
 
-        data = json.loads(response["body"])
+    data = json.loads(response["body"])
 
-        for item in profiles:
-            transactions = item.payload.json["transactions"]
-            assert len(transactions) == 1
-            assert str(data["active"]) == transactions[0]["active_thread_id"]
+    for item in profiles:
+        transactions = item.payload.json["transactions"]
+        assert len(transactions) == 1
+        assert str(data["active"]) == transactions[0]["active_thread_id"]
 
     transactions = [item for item in envelopes[0].items if item.type == "transaction"]
     assert len(transactions) == 1

tests/integrations/quart/test_quart.py

@@ -539,41 +539,47 @@ async def dispatch_request(self):
 
 
 @pytest.mark.parametrize("endpoint", ["/sync/thread_ids", "/async/thread_ids"])
-@mock.patch("sentry_sdk.profiler.transaction_profiler.PROFILE_MINIMUM_SAMPLES", 0)
 @pytest.mark.asyncio
-async def test_active_thread_id(sentry_init, capture_envelopes, endpoint):
-    sentry_init(
-        traces_sample_rate=1.0,
-        profiles_sample_rate=1.0,
-    )
-    app = quart_app_factory()
+async def test_active_thread_id(
+    sentry_init, capture_envelopes, teardown_profiling, endpoint
+):
+    with mock.patch(
+        "sentry_sdk.profiler.transaction_profiler.PROFILE_MINIMUM_SAMPLES", 0
+    ):
+        sentry_init(
+            traces_sample_rate=1.0,
+            profiles_sample_rate=1.0,
+        )
+        app = quart_app_factory()
 
-    envelopes = capture_envelopes()
+        envelopes = capture_envelopes()
 
-    async with app.test_client() as client:
-        response = await client.get(endpoint)
-        assert response.status_code == 200
+        async with app.test_client() as client:
+            response = await client.get(endpoint)
+            assert response.status_code == 200
 
-    data = json.loads(await response.get_data(as_text=True))
+        data = json.loads(await response.get_data(as_text=True))
 
-    envelopes = [envelope for envelope in envelopes]
-    assert len(envelopes) == 1
+        envelopes = [envelope for envelope in envelopes]
+        assert len(envelopes) == 1
 
-    profiles = [item for item in envelopes[0].items if item.type == "profile"]
-    assert len(profiles) == 1, envelopes[0].items
+        profiles = [item for item in envelopes[0].items if item.type == "profile"]
+        assert len(profiles) == 1, envelopes[0].items
 
-    for item in profiles:
-        transactions = item.payload.json["transactions"]
-        assert len(transactions) == 1
-        assert str(data["active"]) == transactions[0]["active_thread_id"]
+        for item in profiles:
+            transactions = item.payload.json["transactions"]
+            assert len(transactions) == 1
+            assert str(data["active"]) == transactions[0]["active_thread_id"]
 
-    transactions = [item for item in envelopes[0].items if item.type == "transaction"]
-    assert len(transactions) == 1
+        transactions = [
+            item for item in envelopes[0].items if item.type == "transaction"
+        ]
+        assert len(transactions) == 1
 
-    for item in transactions:
-        transaction = item.payload.json
-        trace_context = transaction["contexts"]["trace"]
-        assert str(data["active"]) == trace_context["data"]["thread.id"]
+        for item in transactions:
+            transaction = item.payload.json
+            trace_context = transaction["contexts"]["trace"]
+            assert str(data["active"]) == trace_context["data"]["thread.id"]
 
 
 @pytest.mark.asyncio