From 1f1f2b953a095451dfb6160cec16fa3f8a1bcc2f Mon Sep 17 00:00:00 2001
From: datadog-bits <263423550+datadog-bits@users.noreply.github.com>
Date: Wed, 20 May 2026 20:50:00 +0000
Subject: [PATCH] Stabilize asyncpg patch cycle test setup

---
 tests/contrib/asyncpg/test_asyncpg.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/tests/contrib/asyncpg/test_asyncpg.py b/tests/contrib/asyncpg/test_asyncpg.py
index bed8d81affd..7a06ce21b91 100644
--- a/tests/contrib/asyncpg/test_asyncpg.py
+++ b/tests/contrib/asyncpg/test_asyncpg.py
@@ -414,11 +414,22 @@ def test_patch_unpatch_patch_cycle():
 
 
 @pytest.mark.asyncio
-async def test_execute_after_patch_unpatch_patch(patched_conn):
+async def test_execute_after_patch_unpatch_patch():
     # Regression: executing a query must succeed after patch/unpatch/patch.
     unpatch()
     patch()
-    await patched_conn.execute("SELECT 1")
+
+    conn = await asyncpg.connect(
+        host=POSTGRES_CONFIG["host"],
+        port=POSTGRES_CONFIG["port"],
+        user=POSTGRES_CONFIG["user"],
+        database=POSTGRES_CONFIG["dbname"],
+        password=POSTGRES_CONFIG["password"],
+    )
+    try:
+        await conn.execute("SELECT 1")
+    finally:
+        await conn.close()
 
 
 class AsyncPgTestCase(AsyncioTestCase):