fix: connection reset fails when an additional dialect is used (#188)

Author: Ilya Gurov

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

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):
     """