chore: update sample docstring (#146)

GoogleCloudPlatform · Oct 30, 2023 · 1799674 · 1799674
1 parent 8121df6
commit 1799674
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/system/test_sqlalchemy_asyncpg_direct_connection.py b/tests/system/test_sqlalchemy_asyncpg_direct_connection.py
@@ -48,9 +48,10 @@ def create_sqlalchemy_engine(
                 db,
         )
 
-        with engine.connect() as conn:
-            time = conn.execute(sqlalchemy.text("SELECT NOW()")).fetchone()
+        async with engine.connect() as conn:
+            result = await conn.execute(sqlalchemy.text("SELECT NOW()"))
             conn.commit()
+            time = result.fetchone()
 
     Args:
         ip_address (str):