diff --git a/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py b/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py index 5e661ca5..fa704c95 100644 --- a/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py +++ b/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py @@ -48,7 +48,8 @@ @listens_for(Pool, "reset") def reset_connection(dbapi_conn, connection_record): """An event of returning a connection back to a pool.""" - dbapi_conn.connection.staleness = None + if getattr(dbapi_conn.connection, "staleness", None) is not None: + dbapi_conn.connection.staleness = None # register a method to get a single value of a JSON object diff --git a/test/test_suite.py b/test/test_suite.py index 75fc9a3c..735c0be8 100644 --- a/test/test_suite.py +++ b/test/test_suite.py @@ -1616,6 +1616,9 @@ def test_staleness(self): with self._engine.connect() as connection: assert connection.connection.staleness is None + with self._engine.connect() as connection: + del connection.staleness + class LimitOffsetTest(fixtures.TestBase): """