chore: update test cases

Author: olavloite

tests/unit/spanner_dbapi/test_connect.py

@@ -59,7 +59,7 @@ def test_w_implicit(self, mock_client):
 
         self.assertIs(connection.database, database)
         instance.database.assert_called_once_with(
-            DATABASE, pool=None, database_role=None
+            DATABASE, pool=None, database_role=None, logger=None
         )
         # Database constructs its own pool
         self.assertIsNotNone(connection.database._pool)
@@ -107,7 +107,7 @@ def test_w_explicit(self, mock_client):
 
         self.assertIs(connection.database, database)
         instance.database.assert_called_once_with(
-            DATABASE, pool=pool, database_role=role
+            DATABASE, pool=pool, database_role=role, logger=None
         )
 
     def test_w_credential_file_path(self, mock_client):

tests/unit/spanner_dbapi/test_connection.py

@@ -888,8 +888,9 @@ def database(
         pool=None,
         database_dialect=DatabaseDialect.GOOGLE_STANDARD_SQL,
         database_role=None,
+        logger=None,
     ):
-        return _Database(database_id, pool, database_dialect, database_role)
+        return _Database(database_id, pool, database_dialect, database_role, logger)
 
 
 class _Database(object):
@@ -899,8 +900,10 @@ def __init__(
         pool=None,
         database_dialect=DatabaseDialect.GOOGLE_STANDARD_SQL,
         database_role=None,
+        logger=None,
     ):
         self.name = database_id
         self.pool = pool
         self.database_dialect = database_dialect
         self.database_role = database_role
+        self.logger = logger