fixed unit tests

Author: sagnghos

tests/unit/spanner_dbapi/test_connect.py

@@ -55,6 +55,10 @@ def test_w_implicit(self, mock_client):
             client_info=mock.ANY,
             client_options=mock.ANY,
             route_to_leader_enabled=True,
+            use_plain_text=False,
+            ca_certificate=None,
+            client_certificate=None,
+            client_key=None,
         )
 
         self.assertIs(connection.database, database)
@@ -97,6 +101,10 @@ def test_w_explicit(self, mock_client):
             client_info=mock.ANY,
             client_options=mock.ANY,
             route_to_leader_enabled=False,
+            use_plain_text=False,
+            ca_certificate=None,
+            client_certificate=None,
+            client_key=None,
         )
         client_info = mock_client.call_args_list[0][1]["client_info"]
         self.assertEqual(client_info.user_agent, USER_AGENT)

tests/unit/test_database.py

@@ -3543,6 +3543,8 @@ def __init__(
         self.credentials.expiry = None
         self.credentials.valid = True
 
+        self._experimental_host = None
+
         # Mock the spanner API to return proper session names
         self._spanner_api = mock.Mock()
 
@@ -3560,14 +3562,11 @@ def _next_nth_request(self):
 
 
 class _Instance(object):
-    def __init__(
-        self, name, client=_Client(), emulator_host=None, experimental_host=None
-    ):
+    def __init__(self, name, client=_Client(), emulator_host=None):
         self.name = name
         self.instance_id = name.rsplit("/", 1)[1]
         self._client = client
         self.emulator_host = emulator_host
-        self.experimental_host = experimental_host
 
 
 class _Backup(object):

tests/unit/test_instance.py

@@ -1023,6 +1023,7 @@ def __init__(self, project, timeout_seconds=None):
         self.route_to_leader_enabled = True
         self.directed_read_options = None
         self.default_transaction_options = DefaultTransactionOptions()
+        self._experimental_host = None
 
     def copy(self):
         from copy import deepcopy