google-cloud-spanner 3.67.0__tar.gz → 3.68.0__tar.gz

{google_cloud_spanner-3.67.0/google_cloud_spanner.egg-info → google_cloud_spanner-3.68.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-spanner
-Version: 3.67.0
+Version: 3.68.0
 Summary: Google Cloud Spanner API client library
 Home-page: https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-spanner
 Author: Google LLC
@@ -40,6 +40,13 @@ Requires-Dist: google-cloud-monitoring>=2.16.0
 Requires-Dist: mmh3>=4.1.0
 Provides-Extra: libcst
 Requires-Dist: libcst>=0.2.5; extra == "libcst"
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: mock; extra == "test"
+Requires-Dist: asyncmock; extra == "test"
+Requires-Dist: pytest-cov; extra == "test"
+Requires-Dist: pytest-asyncio; extra == "test"
+Requires-Dist: pytest-xdist; extra == "test"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{google_cloud_spanner-3.67.0/google/cloud/spanner_admin_instance → google_cloud_spanner-3.68.0/google/cloud/spanner}/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "3.67.0"  # {x-release-please-version}
+__version__ = "3.68.0"  # {x-release-please-version}

{google_cloud_spanner-3.67.0/google/cloud/spanner_admin_database_v1 → google_cloud_spanner-3.68.0/google/cloud/spanner_admin_database}/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "3.67.0"  # {x-release-please-version}
+__version__ = "3.68.0"  # {x-release-please-version}

{google_cloud_spanner-3.67.0/google/cloud/spanner_admin_database → google_cloud_spanner-3.68.0/google/cloud/spanner_admin_database_v1}/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "3.67.0"  # {x-release-please-version}
+__version__ = "3.68.0"  # {x-release-please-version}

{google_cloud_spanner-3.67.0/google/cloud/spanner → google_cloud_spanner-3.68.0/google/cloud/spanner_admin_instance}/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "3.67.0"  # {x-release-please-version}
+__version__ = "3.68.0"  # {x-release-please-version}

google_cloud_spanner-3.68.0/google/cloud/spanner_admin_instance_v1/gapic_version.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+#
+__version__ = "3.68.0"  # {x-release-please-version}

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/google/cloud/spanner_dbapi/partition_helper.py

@@ -90,23 +90,6 @@ def _deserialize_value(val: Any) -> Any:
     return val
 
 
-def _unpack_value_pb(value):
-    which = value.WhichOneof("kind")
-    if which == "null_value":
-        return None
-    elif which == "number_value":
-        return value.number_value
-    elif which == "string_value":
-        return value.string_value
-    elif which == "bool_value":
-        return value.bool_value
-    elif which == "struct_value":
-        return {k: _unpack_value_pb(v) for k, v in value.struct_value.fields.items()}
-    elif which == "list_value":
-        return [_unpack_value_pb(v) for v in value.list_value.values]
-    return None
-
-
 def decode_from_string(encoded_partition_id):
     gzip_bytes = base64.b64decode(bytes(encoded_partition_id, "utf-8"))
     partition_id_bytes = gzip.decompress(gzip_bytes)
@@ -124,9 +107,7 @@ def decode_from_string(encoded_partition_id):
     if "query" in partition_result and "params" in partition_result["query"]:
         params_pb = partition_result["query"]["params"]
         if params_pb:
-            partition_result["query"]["params"] = {
-                k: _unpack_value_pb(v) for k, v in params_pb.fields.items()
-            }
+            partition_result["query"]["params"] = MessageToDict(params_pb)
 
     return PartitionId(btid, partition_result)
 

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/google/cloud/spanner_dbapi/version.py

@@ -15,6 +15,6 @@
 import platform
 
 PY_VERSION = platform.python_version()
-__version__ = "3.67.0"
+__version__ = "3.68.0"
 VERSION = __version__
 DEFAULT_USER_AGENT = "gl-dbapi/" + VERSION

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/google/cloud/spanner_v1/_async/streamed.py

@@ -129,16 +129,39 @@ class StreamedResultSet(object):
         decoders = self._decoders
         width = len(self.fields)
         index = len(self._current_row)
-        for value in values:
-            if self._lazy_decode:
-                self._current_row.append(value)
-            else:
-                self._current_row.append(_parse_nullable(value, decoders[index]))
-            index += 1
-            if index == width:
-                self._rows.append(self._current_row)
-                self._current_row = []
-                index = 0
+        current_row = self._current_row
+        rows = self._rows
+
+        current_row_append = current_row.append
+        rows_append = rows.append
+
+        if self._lazy_decode:
+            for value in values:
+                current_row_append(value)
+                index += 1
+                if index == width:
+                    rows_append(current_row)
+                    current_row = []
+                    current_row_append = current_row.append
+                    index = 0
+        else:
+            for value in values:
+                # Note: We manually check value.HasField("null_value") here instead of
+                # wrapping every decoder in _parse_nullable to avoid the overhead of
+                # an extra Python function call layer for every cell value decoded in this loop.
+                # If the nullable check logic is updated in _parse_nullable, update this check.
+                if value.HasField("null_value"):
+                    current_row_append(None)
+                else:
+                    current_row_append(decoders[index](value))
+                index += 1
+                if index == width:
+                    rows_append(current_row)
+                    current_row = []
+                    current_row_append = current_row.append
+                    index = 0
+
+        self._current_row = current_row
 
     @CrossSync.convert
     async def _consume_next(self):

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/google/cloud/spanner_v1/_helpers.py

@@ -19,6 +19,7 @@ import datetime
 import decimal
 import logging
 import math
+import operator
 import threading
 import time
 import uuid
@@ -26,7 +27,6 @@ from contextlib import contextmanager
 
 from google.api_core import datetime_helpers
 from google.api_core.exceptions import Aborted
-from google.cloud._helpers import _date_from_iso8601_date
 from google.protobuf.internal.enum_type_wrapper import EnumTypeWrapper
 from google.protobuf.message import DecodeError, Message
 from google.protobuf.struct_pb2 import ListValue, Value
@@ -465,6 +465,12 @@ def _parse_value_pb(value_pb, field_type, field_name, column_info=None):
     return _parse_nullable(value_pb, decoder)
 
 
+_date_fromisoformat = datetime.date.fromisoformat
+_Decimal = decimal.Decimal
+_json_from_str = JsonObject.from_str
+_uuid_UUID = uuid.UUID
+
+
 def _get_type_decoder(field_type, field_name, column_info=None):
     """Returns a function that converts a Value protobuf to cell data.
 
@@ -489,28 +495,30 @@ def _get_type_decoder(field_type, field_name, column_info=None):
     """
 
     type_code = field_type.code
+    # Note: STRING and BOOL use operator.attrgetter because direct attribute extraction
+    # is faster in Python. Other types require type transformation, so they use lambdas.
     if type_code == TypeCode.STRING:
-        return _parse_string
+        return operator.attrgetter("string_value")
     elif type_code == TypeCode.BYTES:
-        return _parse_bytes
+        return lambda value_pb: value_pb.string_value.encode("utf8")
     elif type_code == TypeCode.BOOL:
-        return _parse_bool
+        return operator.attrgetter("bool_value")
     elif type_code == TypeCode.INT64:
-        return _parse_int64
+        return lambda value_pb: int(value_pb.string_value)
     elif type_code == TypeCode.FLOAT64:
         return _parse_float
     elif type_code == TypeCode.FLOAT32:
         return _parse_float
     elif type_code == TypeCode.DATE:
-        return _parse_date
+        return lambda value_pb: _date_fromisoformat(value_pb.string_value)
     elif type_code == TypeCode.TIMESTAMP:
         return _parse_timestamp
     elif type_code == TypeCode.NUMERIC:
-        return _parse_numeric
+        return lambda value_pb: _Decimal(value_pb.string_value)
     elif type_code == TypeCode.JSON:
-        return _parse_json
+        return lambda value_pb: _json_from_str(value_pb.string_value)
     elif type_code == TypeCode.UUID:
-        return _parse_uuid
+        return lambda value_pb: _uuid_UUID(value_pb.string_value)
     elif type_code == TypeCode.PROTO:
         return lambda value_pb: _parse_proto(value_pb, column_info, field_name)
     elif type_code == TypeCode.ENUM:
@@ -553,48 +561,81 @@ def _parse_list_value_pbs(rows, row_type):
     return result
 
 
-def _parse_string(value_pb) -> str:
-    return value_pb.string_value
-
-
-def _parse_bytes(value_pb):
-    return value_pb.string_value.encode("utf8")
-
-
-def _parse_bool(value_pb) -> bool:
-    return value_pb.bool_value
-
-
-def _parse_int64(value_pb) -> int:
-    return int(value_pb.string_value)
-
-
 def _parse_float(value_pb) -> float:
-    if value_pb.HasField("string_value"):
-        return float(value_pb.string_value)
-    else:
-        return value_pb.number_value
-
-
-def _parse_date(value_pb):
-    return _date_from_iso8601_date(value_pb.string_value)
+    # Note: Storing val = value_pb.string_value and doing a truthiness check is faster
+    # than calling value_pb.HasField("string_value") because it avoids the C-extension
+    # method lookup/call overhead and accesses the attribute only once.
+    val = value_pb.string_value
+    return float(val) if val else value_pb.number_value
+
+
+_POWERS_OF_10 = (
+    1,
+    10,
+    100,
+    1000,
+    10000,
+    100000,
+    1000000,
+    10000000,
+    100000000,
+    1000000000,
+)
 
 
 def _parse_timestamp(value_pb):
-    DatetimeWithNanoseconds = datetime_helpers.DatetimeWithNanoseconds
-    return DatetimeWithNanoseconds.from_rfc3339(value_pb.string_value)
-
-
-def _parse_numeric(value_pb):
-    return decimal.Decimal(value_pb.string_value)
-
-
-def _parse_json(value_pb):
-    return JsonObject.from_str(value_pb.string_value)
-
-
-def _parse_uuid(value_pb):
-    return uuid.UUID(value_pb.string_value)
+    val = value_pb.string_value
+    try:
+        if len(val) < 20 or val[10] != "T":
+            raise ValueError()
+        no_fraction = val[:19]
+        bare = datetime.datetime.fromisoformat(no_fraction)
+        if val[19] == ".":
+            if val.endswith("Z"):
+                offset = "Z"
+                fraction = val[20:-1]
+            elif val[-6] in ("+", "-"):
+                offset = val[-6:]
+                fraction = val[20:-6]
+            else:
+                raise ValueError()
+            if not fraction or len(fraction) > 9 or not fraction.isdigit():
+                raise ValueError()
+            scale = 9 - len(fraction)
+            nanos = int(fraction) * _POWERS_OF_10[scale]
+        else:
+            nanos = 0
+            if val.endswith("Z"):
+                offset = "Z"
+            elif val[-6] in ("+", "-"):
+                offset = val[-6:]
+            else:
+                raise ValueError()
+
+        if offset != "Z":
+            sign = offset[0]
+            hours = int(offset[1:3])
+            minutes = int(offset[4:6])
+            if offset[3] != ":":
+                raise ValueError()
+            delta = datetime.timedelta(hours=hours, minutes=minutes)
+            if sign == "-":
+                delta = -delta
+            tzinfo = datetime.timezone(delta)
+            bare = bare.replace(tzinfo=tzinfo).astimezone(datetime.timezone.utc)
+
+        return datetime_helpers.DatetimeWithNanoseconds(
+            bare.year,
+            bare.month,
+            bare.day,
+            bare.hour,
+            bare.minute,
+            bare.second,
+            nanosecond=nanos,
+            tzinfo=datetime.timezone.utc,
+        )
+    except (IndexError, ValueError) as e:
+        raise ValueError("Timestamp: {} does not match pattern".format(val)) from e
 
 
 def _parse_proto(value_pb, column_info, field_name):

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/google/cloud/spanner_v1/data_types.py

@@ -99,6 +99,11 @@ class JsonObject(dict):
         return json.dumps(self, sort_keys=True, separators=(",", ":"))
 
 
+_INTERVAL_PATTERN = re.compile(
+    r"^P(-?\d+Y)?(-?\d+M)?(-?\d+D)?(T(-?\d+H)?(-?\d+M)?(-?((\d+([.,]\d{1,9})?)|([.,]\d{1,9}))S)?)?$"
+)
+
+
 @dataclass
 class Interval:
     """Represents a Spanner INTERVAL type.
@@ -187,8 +192,7 @@ class Interval:
     @classmethod
     def from_str(cls, s: str) -> "Interval":
         """Parse an ISO8601 duration format string into an Interval."""
-        pattern = r"^P(-?\d+Y)?(-?\d+M)?(-?\d+D)?(T(-?\d+H)?(-?\d+M)?(-?((\d+([.,]\d{1,9})?)|([.,]\d{1,9}))S)?)?$"
-        match = re.match(pattern, s)
+        match = _INTERVAL_PATTERN.match(s)
         if not match or len(s) == 1:
             raise ValueError(f"Invalid interval format: {s}")
 

google_cloud_spanner-3.68.0/google/cloud/spanner_v1/gapic_version.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+#
+__version__ = "3.68.0"  # {x-release-please-version}

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/google/cloud/spanner_v1/streamed.py

@@ -35,8 +35,7 @@ class StreamedResultSet(object):
         instances.
 
     :type source: :class:`~google.cloud.spanner_v1.snapshot.Snapshot`
-    :param source: Deprecated. Snapshot from which the result set was fetched.
-    """
+    :param source: Deprecated. Snapshot from which the result set was fetched."""
 
     def __init__(
         self,
@@ -117,16 +116,36 @@ class StreamedResultSet(object):
         decoders = self._decoders
         width = len(self.fields)
         index = len(self._current_row)
-        for value in values:
-            if self._lazy_decode:
-                self._current_row.append(value)
-            else:
-                self._current_row.append(_parse_nullable(value, decoders[index]))
-            index += 1
-            if index == width:
-                self._rows.append(self._current_row)
-                self._current_row = []
-                index = 0
+        current_row = self._current_row
+        rows = self._rows
+        current_row_append = current_row.append
+        rows_append = rows.append
+        if self._lazy_decode:
+            for value in values:
+                current_row_append(value)
+                index += 1
+                if index == width:
+                    rows_append(current_row)
+                    current_row = []
+                    current_row_append = current_row.append
+                    index = 0
+        else:
+            for value in values:
+                # Note: We manually check value.HasField("null_value") here instead of
+                # wrapping every decoder in _parse_nullable to avoid the overhead of
+                # an extra Python function call layer for every cell value decoded in this loop.
+                # If the nullable check logic is updated in _parse_nullable, update this check.
+                if value.HasField("null_value"):
+                    current_row_append(None)
+                else:
+                    current_row_append(decoders[index](value))
+                index += 1
+                if index == width:
+                    rows_append(current_row)
+                    current_row = []
+                    current_row_append = current_row.append
+                    index = 0
+        self._current_row = current_row
 
     def _consume_next(self):
         """Consume the next partial result set from the stream.

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0/google_cloud_spanner.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-spanner
-Version: 3.67.0
+Version: 3.68.0
 Summary: Google Cloud Spanner API client library
 Home-page: https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-spanner
 Author: Google LLC
@@ -40,6 +40,13 @@ Requires-Dist: google-cloud-monitoring>=2.16.0
 Requires-Dist: mmh3>=4.1.0
 Provides-Extra: libcst
 Requires-Dist: libcst>=0.2.5; extra == "libcst"
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: mock; extra == "test"
+Requires-Dist: asyncmock; extra == "test"
+Requires-Dist: pytest-cov; extra == "test"
+Requires-Dist: pytest-asyncio; extra == "test"
+Requires-Dist: pytest-xdist; extra == "test"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/google_cloud_spanner.egg-info/requires.txt

@@ -22,3 +22,11 @@ grpcio<2.0.0,>=1.75.1
 
 [libcst]
 libcst>=0.2.5
+
+[test]
+pytest
+mock
+asyncmock
+pytest-cov
+pytest-asyncio
+pytest-xdist

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/setup.py

@@ -60,7 +60,17 @@ dependencies = [
     "google-cloud-monitoring >= 2.16.0",
     "mmh3 >= 4.1.0",
 ]
-extras = {"libcst": "libcst >= 0.2.5"}
+extras = {
+    "libcst": "libcst >= 0.2.5",
+    "test": [
+        "pytest",
+        "mock",
+        "asyncmock",
+        "pytest-cov",
+        "pytest-asyncio",
+        "pytest-xdist",
+    ],
+}
 
 url = "https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-spanner"
 

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/tests/mockserver_tests/test_dbapi_partition_query.py

@@ -1,5 +1,4 @@
-# Copyright 2024 Google LLC All rights reserved.
-#
+# Copyright 2026 Google LLC All rights reserved.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/tests/system/_async/test_database_api.py

@@ -179,7 +179,10 @@ async def test_transaction_manual_abort_retry(shared_database):
         transaction.insert_or_update(sd.TABLE, sd.COLUMNS, sd.ROW_DATA)
 
     await shared_database.run_in_transaction(_unit_of_work)
-    assert attempts == 2
+    # Expect at least 2 attempts due to our simulated manual abort on first try.
+    # We use >= 2 rather than == 2 because the live Spanner server can also
+    # trigger transient abort retries depending on real-world GCP resource contention.
+    assert attempts >= 2
 
 
 @pytest.mark.asyncio

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/tests/system/conftest.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import datetime
+import os
 import time
 
 import pytest
@@ -25,6 +26,10 @@ from google.cloud.spanner_admin_database_v1.types.backup import (
 
 from . import _helpers
 
+# Disable builtin metrics for system tests by default to avoid 401 errors
+# from the background thread exporting to Cloud Monitoring without permissions.
+os.environ["SPANNER_DISABLE_BUILTIN_METRICS"] = "true"
+
 
 @pytest.fixture(scope="function")
 def if_create_instance():

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/tests/unit/_async/test_client.py

@@ -799,7 +799,9 @@ class TestClient(IsolatedAsyncioTestCase):
             info_logger.assert_not_called()
 
         # Also test when the environment variable is not set at all
-        with mock.patch.dict(os.environ, {}, clear=True):
+        with mock.patch.dict(
+            os.environ, {"SPANNER_DISABLE_BUILTIN_METRICS": "true"}, clear=True
+        ):
             with mock.patch.object(logger, "info") as info_logger:
                 client = self._make_one(project=self.PROJECT, credentials=creds)
                 self.assertIsNotNone(client)

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/tests/unit/_async/test_client_extra.py

@@ -131,7 +131,21 @@ class TestClientExtra(unittest.IsolatedAsyncioTestCase):
                                 self.assertIsNotNone(ia_api)
                                 self.assertIsNotNone(da_api)
 
-    def test_initialize_metrics_double_check(self):
+    # Safety shield mocks: We intercept the OpenTelemetry metric classes at the client module namespace level
+    # to prevent instantiating real exporter objects. This prevents spawning live background worker threads
+    # that periodically wake up and trigger 401 credential errors inside unauthenticated unit test runs.
+    @mock.patch("google.cloud.spanner_v1._async.client.CloudMonitoringMetricsExporter")
+    @mock.patch("google.cloud.spanner_v1._async.client.PeriodicExportingMetricReader")
+    @mock.patch("google.cloud.spanner_v1._async.client.MeterProvider")
+    # Global state reset: Temporarily override the module's process-wide global boolean _metrics_monitor_initialized
+    # to False so that the client enters the initialization logic instead of returning early.
+    @mock.patch(
+        "google.cloud.spanner_v1._async.client._metrics_monitor_initialized",
+        False,
+    )
+    def test_initialize_metrics_double_check(
+        self, mock_provider, mock_reader, mock_exporter
+    ):
         # coverage for line 143->exit
         from google.cloud.spanner_v1._async import client as MUT
 
@@ -147,15 +161,17 @@ class TestClientExtra(unittest.IsolatedAsyncioTestCase):
             def __exit__(self, *args):
                 return original_lock.__exit__(*args)
 
+        # Concurrency race condition simulator: Replace the process synchronization lock with our custom SettingLock.
+        # When this lock enters, it toggles _metrics_monitor_initialized to True to simulate another thread
+        # completing metrics setup while this thread was waiting for the lock.
         with mock.patch(
-            "google.cloud.spanner_v1._async.client._metrics_monitor_initialized", False
+            "google.cloud.spanner_v1._async.client._metrics_monitor_lock",
+            SettingLock(),
         ):
-            with mock.patch(
-                "google.cloud.spanner_v1._async.client._metrics_monitor_lock",
-                SettingLock(),
-            ):
-                MUT._initialize_metrics("project", self.credentials)
-                self.assertTrue(MUT._metrics_monitor_initialized)
+            # Trigger the initialization function and verify Spanner's double-checked lock safely
+            # checks the flag again and aborts cleanly to prevent dual-registration.
+            MUT._initialize_metrics("project", self.credentials)
+            self.assertTrue(MUT._metrics_monitor_initialized)
 
     def test_default_transaction_options_validation(self):
         # coverage for line 344

{google_cloud_spanner-3.67.0 → google_cloud_spanner-3.68.0}/tests/unit/_async/test_session.py

@@ -1,4 +1,5 @@
 import datetime
+import threading
 from datetime import timezone
 
 import google.api_core.gapic_v1.method
@@ -1800,11 +1801,16 @@ class TestSession(OpenTelemetryBase):
             called_with.append((txn, args, kw))
             txn.insert(TABLE_NAME, COLUMNS, VALUES)
 
+        main_thread = threading.current_thread()
+        _results = [1, 1.5]
+
         # retry once w/ timeout_secs=1
-        def _time(_results=[1, 1.5]):
-            if len(_results) > 1:
-                return _results.pop(0)
-            return _results[0]
+        def _time():
+            if threading.current_thread() is main_thread:
+                if len(_results) > 1:
+                    return _results.pop(0)
+                return _results[0]
+            return 1.0
 
         with mock.patch("time.time", _time):
             with mock.patch(
@@ -1877,9 +1883,14 @@ class TestSession(OpenTelemetryBase):
             called_with.append((txn, args, kw))
             txn.insert(TABLE_NAME, COLUMNS, VALUES)
 
+        main_thread = threading.current_thread()
+        _results = [1] * 100
+
         # retry several times to check backoff
-        def _time(_results=[1] * 100):
-            return _results.pop(0)
+        def _time():
+            if threading.current_thread() is main_thread:
+                return _results.pop(0)
+            return 1.0
 
         with (
             mock.patch("time.time", _time),