From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Tue, 24 May 2022 08:50:17 +0000 (+0200)
Subject: feat(crdb): add psycopg.crdb.connect() function
X-Git-Tag: 3.1~49^2~52
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=7f0a2244bf304c50ce61190513942496c7ed3168;p=thirdparty%2Fpsycopg.git

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

diff --git a/psycopg/psycopg/crdb.py b/psycopg/psycopg/crdb.py
index 051e74075..5fa3492be 100644
--- 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 9c169e3a3..1f1f64d28 100644
--- 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)