fix: pass through kwargs in dbapi connect (#1368)

olavloite · gcf-owl-bot[bot] · web-flow · commit aae8d6161580 · 2025-05-06T01:28:15.000-07:00
* fix: pass through kwargs in dbapi connect * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
diff --git a/google/cloud/spanner_dbapi/connection.py b/google/cloud/spanner_dbapi/connection.py
@@ -798,7 +798,7 @@ def connect(
     database = None
     if database_id:
         database = instance.database(database_id, pool=pool)
-    conn = Connection(instance, database)
+    conn = Connection(instance, database, **kwargs)
     if pool is not None:
         conn._own_pool = False
 
diff --git a/tests/unit/spanner_dbapi/test_connect.py b/tests/unit/spanner_dbapi/test_connect.py
@@ -131,3 +131,17 @@ def test_w_credential_file_path(self, mock_client):
         client_info = factory.call_args_list[0][1]["client_info"]
         self.assertEqual(client_info.user_agent, USER_AGENT)
         self.assertEqual(client_info.python_version, PY_VERSION)
+
+    def test_with_kwargs(self, mock_client):
+        from google.cloud.spanner_dbapi import connect
+        from google.cloud.spanner_dbapi import Connection
+
+        client = mock_client.return_value
+        instance = client.instance.return_value
+        database = instance.database.return_value
+        self.assertIsNotNone(database)
+
+        connection = connect(INSTANCE, DATABASE, ignore_transaction_warnings=True)
+
+        self.assertIsInstance(connection, Connection)
+        self.assertTrue(connection._ignore_transaction_warnings)
