From 2f6f9d7800653870b1ca5872a7f24a4678696005 Mon Sep 17 00:00:00 2001 From: Owl Bot Date: Mon, 29 Jan 2024 10:31:24 +0000 Subject: [PATCH] =?UTF-8?q?=F0=9F=A6=89=20Updates=20from=20OwlBot=20post-p?= =?UTF-8?q?rocessor?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --- google/cloud/spanner_v1/snapshot.py | 8 ++++++-- tests/system/test_database_api.py | 16 +++++++++++++--- tests/system/test_session_api.py | 1 - tests/unit/test_database.py | 1 - 4 files changed, 19 insertions(+), 7 deletions(-) diff --git a/google/cloud/spanner_v1/snapshot.py b/google/cloud/spanner_v1/snapshot.py index 8ba9011a82..f9295ce3c2 100644 --- a/google/cloud/spanner_v1/snapshot.py +++ b/google/cloud/spanner_v1/snapshot.py @@ -483,9 +483,13 @@ def execute_sql( if self._transaction_id is None: # lock is added to handle the inline begin for first rpc with self._lock: - return self._get_streamed_result_set(restart, request, trace_attributes, column_info) + return self._get_streamed_result_set( + restart, request, trace_attributes, column_info + ) else: - return self._get_streamed_result_set(restart, request, trace_attributes, column_info) + return self._get_streamed_result_set( + restart, request, trace_attributes, column_info + ) def _get_streamed_result_set(self, restart, request, trace_attributes, column_info): iterator = _restart_on_unavailable( diff --git a/tests/system/test_database_api.py b/tests/system/test_database_api.py index 8a145ff447..db1d48c8a2 100644 --- a/tests/system/test_database_api.py +++ b/tests/system/test_database_api.py @@ -444,7 +444,11 @@ def test_update_ddl_w_default_leader_success( def test_create_role_grant_access_success( - not_emulator, shared_instance, databases_to_delete, database_dialect, proto_descriptor_file, + not_emulator, + shared_instance, + databases_to_delete, + database_dialect, + proto_descriptor_file, ): creator_role_parent = _helpers.unique_id("role_parent", separator="_") creator_role_orphan = _helpers.unique_id("role_orphan", separator="_") @@ -469,7 +473,9 @@ def test_create_role_grant_access_success( f"GRANT SELECT ON TABLE contacts TO {creator_role_parent}", ] - operation = temp_db.update_ddl(ddl_statements, proto_descriptors=proto_descriptor_file) + operation = temp_db.update_ddl( + ddl_statements, proto_descriptors=proto_descriptor_file + ) operation.result(DBAPI_OPERATION_TIMEOUT) # raises on failure / timeout. # Perform select with orphan role on table contacts. @@ -504,7 +510,11 @@ def test_create_role_grant_access_success( def test_list_database_role_success( - not_emulator, shared_instance, databases_to_delete, database_dialect, proto_descriptor_file + not_emulator, + shared_instance, + databases_to_delete, + database_dialect, + proto_descriptor_file, ): creator_role_parent = _helpers.unique_id("role_parent", separator="_") creator_role_orphan = _helpers.unique_id("role_orphan", separator="_") diff --git a/tests/system/test_session_api.py b/tests/system/test_session_api.py index 0e6085bbf8..c8c7234c38 100644 --- a/tests/system/test_session_api.py +++ b/tests/system/test_session_api.py @@ -1338,7 +1338,6 @@ def _unit_of_work(transaction): def _set_up_proto_table(database): - sd = _sample_data def _unit_of_work(transaction): diff --git a/tests/unit/test_database.py b/tests/unit/test_database.py index f8961897c1..11754cbe86 100644 --- a/tests/unit/test_database.py +++ b/tests/unit/test_database.py @@ -390,7 +390,6 @@ def test_default_leader(self): self.assertEqual(database.default_leader, default_leader) def test_proto_descriptors(self): - instance = _Instance(self.INSTANCE_NAME) pool = _Pool() database = self._make_one(