From 5e50bebdd1d43994b3d83568641d1dff1c419cc8 Mon Sep 17 00:00:00 2001 From: Astha Mohta <35952883+asthamohta@users.noreply.github.com> Date: Wed, 18 Jan 2023 00:44:01 +0530 Subject: [PATCH] fix: fix for database name in batch create request (#883) * tests * changes --- google/cloud/spanner_v1/pool.py | 4 ++-- tests/system/test_database_api.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/google/cloud/spanner_v1/pool.py b/google/cloud/spanner_v1/pool.py index 886e28d7f7..7455d0cd20 100644 --- a/google/cloud/spanner_v1/pool.py +++ b/google/cloud/spanner_v1/pool.py @@ -193,7 +193,7 @@ def bind(self, database): metadata = _metadata_with_prefix(database.name) self._database_role = self._database_role or self._database.database_role request = BatchCreateSessionsRequest( - database=database.database_id, + database=database.name, session_count=self.size - self._sessions.qsize(), session_template=Session(creator_role=self.database_role), ) @@ -406,7 +406,7 @@ def bind(self, database): self._database_role = self._database_role or self._database.database_role request = BatchCreateSessionsRequest( - database=database.database_id, + database=database.name, session_count=self.size - created_session_count, session_template=Session(creator_role=self.database_role), ) diff --git a/tests/system/test_database_api.py b/tests/system/test_database_api.py index 699b3f4a69..364c159da5 100644 --- a/tests/system/test_database_api.py +++ b/tests/system/test_database_api.py @@ -97,7 +97,7 @@ def test_database_binding_of_fixed_size_pool( default_timeout=500, database_role="parent", ) - database = shared_instance.database(temp_db.name, pool=pool) + database = shared_instance.database(temp_db_id, pool=pool) assert database._pool.database_role == "parent" @@ -125,7 +125,7 @@ def test_database_binding_of_pinging_pool( ping_interval=100, database_role="parent", ) - database = shared_instance.database(temp_db.name, pool=pool) + database = shared_instance.database(temp_db_id, pool=pool) assert database._pool.database_role == "parent"