Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions google/cloud/spanner_dbapi/connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -235,6 +235,7 @@ def close(self):
):
self._transaction.rollback()

self.database._pool.clear()
self.is_closed = True

def commit(self):
Expand Down
8 changes: 8 additions & 0 deletions tests/spanner_dbapi/test_connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,3 +77,11 @@ def test_instance_property(self):

with self.assertRaises(AttributeError):
connection.instance = None

def test_clearing_pool_on_close(self):
connection = self._make_connection()
with mock.patch.object(
connection.database._pool, "clear"
) as pool_clear_mock:
connection.close()
pool_clear_mock.assert_called_once()