From 5fb0138a3220161703e6ab1087319a669d14e7f4 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Sat, 4 Jul 2020 12:21:36 -0400 Subject: Implement rudimentary asyncio support w/ asyncpg Using the approach introduced at https://gist.github.com/zzzeek/6287e28054d3baddc07fa21a7227904e We can now create asyncio endpoints that are then handled in "implicit IO" form within the majority of the Core internals. Then coroutines are re-exposed at the point at which we call into asyncpg methods. Patch includes: * asyncpg dialect * asyncio package * engine, result, ORM session classes * new test fixtures, tests * some work with pep-484 and a short plugin for the pyannotate package, which seems to have so-so results Change-Id: Idbcc0eff72c4cad572914acdd6f40ddb1aef1a7d Fixes: #3414 --- lib/sqlalchemy/testing/asyncio.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 lib/sqlalchemy/testing/asyncio.py (limited to 'lib/sqlalchemy/testing/asyncio.py') diff --git a/lib/sqlalchemy/testing/asyncio.py b/lib/sqlalchemy/testing/asyncio.py new file mode 100644 index 000000000..2e274de16 --- /dev/null +++ b/lib/sqlalchemy/testing/asyncio.py @@ -0,0 +1,14 @@ +from .assertions import assert_raises as _assert_raises +from .assertions import assert_raises_message as _assert_raises_message +from ..util import await_fallback as await_ +from ..util import greenlet_spawn + + +async def assert_raises_async(except_cls, msg, coroutine): + await greenlet_spawn(_assert_raises, except_cls, await_, coroutine) + + +async def assert_raises_message_async(except_cls, msg, coroutine): + await greenlet_spawn( + _assert_raises_message, except_cls, msg, await_, coroutine + ) -- cgit v1.2.1