feat(crdb): add psycopg.crdb.connect() function

diff --git a/psycopg/psycopg/crdb.py b/psycopg/psycopg/crdb.py
index 051e74075e272c535c36f886c9e06bed8ad1a86e..5fa3492be60348581da3b1ff638962ab6718ffa3 100644 (file)
--- a/psycopg/psycopg/crdb.py
+++ b/psycopg/psycopg/crdb.py
@@ -66,6 +66,9 @@ class AsyncCrdbConnection(_CrdbConnectionMixin, AsyncConnection[Row]):
     pass
 
 
+connect = CrdbConnection.connect
+
+
 class CrdbConnectionInfo(ConnectionInfo):
     @property
     def vendor(self) -> str:

diff --git a/tests/crdb/test_connection.py b/tests/crdb/test_connection.py
index 9c169e3a3ccdc48ccfe074e86f93d0df6d737782..1f1f64d287a08a84a4b0d4c6268e7b333152d2cf 100644 (file)
--- a/tests/crdb/test_connection.py
+++ b/tests/crdb/test_connection.py
@@ -1,3 +1,4 @@
+import psycopg.crdb
 from psycopg.crdb import CrdbConnection
 
 import pytest
@@ -8,3 +9,8 @@ pytestmark = pytest.mark.crdb
 def test_is_crdb(conn):
     assert CrdbConnection.is_crdb(conn)
     assert CrdbConnection.is_crdb(conn.pgconn)
+
+
+def test_connect(dsn):
+    with psycopg.crdb.connect(dsn) as conn:
+        assert isinstance(conn, CrdbConnection)