summaryrefslogtreecommitdiff
path: root/test/dialect/postgresql/test_async_pg_py3k.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/dialect/postgresql/test_async_pg_py3k.py')
-rw-r--r--test/dialect/postgresql/test_async_pg_py3k.py17
1 files changed, 17 insertions, 0 deletions
diff --git a/test/dialect/postgresql/test_async_pg_py3k.py b/test/dialect/postgresql/test_async_pg_py3k.py
index d9116a7ce..49014fcaf 100644
--- a/test/dialect/postgresql/test_async_pg_py3k.py
+++ b/test/dialect/postgresql/test_async_pg_py3k.py
@@ -1,4 +1,5 @@
import random
+import uuid
from sqlalchemy import Column
from sqlalchemy import exc
@@ -272,3 +273,19 @@ class AsyncPgTest(fixtures.TestBase):
await conn.close()
eq_(codec_meth.mock_calls, [mock.call(adapted_conn)])
+
+ @async_test
+ async def test_name_connection_func(self, metadata, async_testing_engine):
+ cache = []
+
+ def name_f():
+ name = str(uuid.uuid4())
+ cache.append(name)
+ return name
+
+ engine = async_testing_engine(
+ options={"connect_args": {"prepared_statement_name_func": name_f}},
+ )
+ async with engine.begin() as conn:
+ await conn.execute(select(1))
+ assert len(cache) > 0