feat: support configuring logger in dbapi kwargs

Allow the kwargs for dbapi connections to inlcude a logger, and use this as the
logger for the database that is used.

Also set a default logger that only logs at WARN level for the mock server tests
to stop them from spamming the test log with a bunch of "Created multiplexed session."
messages that are logged at INFO level.

Author: olavloite

google/cloud/spanner_dbapi/connection.py

@@ -819,8 +819,9 @@ def connect(
     instance = client.instance(instance_id)
     database = None
     if database_id:
+        logger = kwargs.get("logger")
         database = instance.database(
-            database_id, pool=pool, database_role=database_role
+            database_id, pool=pool, database_role=database_role, logger=logger
         )
     conn = Connection(instance, database, **kwargs)
     if pool is not None:

tests/mockserver_tests/mock_server_test_base.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
+import logging
 import unittest
 
 import grpc
@@ -170,12 +170,15 @@ class MockServerTestBase(unittest.TestCase):
     spanner_service: SpannerServicer = None
     database_admin_service: DatabaseAdminServicer = None
     port: int = None
+    logger: logging.Logger = None
 
     def __init__(self, *args, **kwargs):
         super(MockServerTestBase, self).__init__(*args, **kwargs)
         self._client = None
         self._instance = None
         self._database = None
+        self.logger = logging.getLogger("MockServerTestBase")
+        self.logger.setLevel(logging.WARN)
 
     @classmethod
     def setup_class(cls):
@@ -227,6 +230,7 @@ def database(self) -> Database:
                 "test-database",
                 pool=FixedSizePool(size=10),
                 enable_interceptors_in_tests=True,
+                logger=self.logger,
             )
         return self._database
 

tests/mockserver_tests/test_request_id_header.py

@@ -227,10 +227,6 @@ def test_database_execute_partitioned_dml_request_id(self):
                 (1, REQ_RAND_PROCESS_ID, NTH_CLIENT, CHANNEL_ID, exec_sql_seq, 1),
             )
         ]
-        print(f"Filtered unary segments: {filtered_unary_segments}")
-        print(f"Want unary segments: {want_unary_segments}")
-        print(f"Got stream segments: {got_stream_segments}")
-        print(f"Want stream segments: {want_stream_segments}")
         assert all(seg in filtered_unary_segments for seg in want_unary_segments)
         assert got_stream_segments == want_stream_segments
 
@@ -269,8 +265,6 @@ def test_unary_retryable_error(self):
                 (1, REQ_RAND_PROCESS_ID, NTH_CLIENT, CHANNEL_ID, exec_sql_seq, 1),
             )
         ]
-        print(f"Got stream segments: {got_stream_segments}")
-        print(f"Want stream segments: {want_stream_segments}")
         assert got_stream_segments == want_stream_segments
 
     def test_streaming_retryable_error(self):