🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

Author: gcf-owl-bot[bot]

google/cloud/spanner_v1/testing/mock_spanner.py

@@ -41,10 +41,10 @@ def __init__(self):
     def add_result(self, sql: str, result: result_set.ResultSet):
         self.results[sql.lower().strip()] = result
 
-    def add_execute_streaming_sql_results(self, sql: str,
-        partial_result_sets: list[result_set.PartialResultSet]):
-        self.execute_streaming_sql_results[
-            sql.lower().strip()] = partial_result_sets
+    def add_execute_streaming_sql_results(
+        self, sql: str, partial_result_sets: list[result_set.PartialResultSet]
+    ):
+        self.execute_streaming_sql_results[sql.lower().strip()] = partial_result_sets
 
     def get_result(self, sql: str) -> result_set.ResultSet:
         result = self.results.get(sql.lower().strip())
@@ -61,9 +61,9 @@ def pop_error(self, context):
         if error:
             context.abort_with_status(error)
 
-    def get_execute_streaming_sql_results(self, sql: str,
-        started_transaction: transaction.Transaction) -> list[
-        result_set.PartialResultSet]:
+    def get_execute_streaming_sql_results(
+        self, sql: str, started_transaction: transaction.Transaction
+    ) -> list[result_set.PartialResultSet]:
         if self.execute_streaming_sql_results[sql.lower().strip()]:
             partials = self.execute_streaming_sql_results[sql.lower().strip()]
         else:

tests/mockserver_tests/mock_server_test_base.py

@@ -39,8 +39,7 @@
 from google.cloud.spanner_v1 import TypeCode
 from google.cloud.spanner_v1.database import Database
 from google.cloud.spanner_v1.instance import Instance
-from google.cloud.spanner_v1.testing.mock_database_admin import \
-    DatabaseAdminServicer
+from google.cloud.spanner_v1.testing.mock_database_admin import DatabaseAdminServicer
 from google.cloud.spanner_v1.testing.mock_spanner import SpannerServicer
 from google.cloud.spanner_v1.testing.mock_spanner import start_mock_server
 
@@ -66,8 +65,9 @@ def aborted_status() -> _Status:
     return status
 
 
-def _make_partial_result_sets(fields: list[tuple[str, TypeCode]],
-    results: list[dict]) -> list[result_set.PartialResultSet]:
+def _make_partial_result_sets(
+    fields: list[tuple[str, TypeCode]], results: list[dict]
+) -> list[result_set.PartialResultSet]:
     partial_result_sets = []
     for result in results:
         partial_result_set = PartialResultSet()
@@ -76,14 +76,16 @@ def _make_partial_result_sets(fields: list[tuple[str, TypeCode]],
             metadata = ResultSetMetadata(row_type=StructType(fields=[]))
             for field in fields:
                 metadata.row_type.fields.append(
-                    StructType.Field(name=field[0], type_=Type(code=field[1])))
+                    StructType.Field(name=field[0], type_=Type(code=field[1]))
+                )
             partial_result_set.metadata = metadata
         for value in result["values"]:
             partial_result_set.values.append(_make_value_pb(value))
-        partial_result_set.last = result.get('last') or False
+        partial_result_set.last = result.get("last") or False
         partial_result_sets.append(partial_result_set)
     return partial_result_sets
 
+
 # Creates an UNAVAILABLE status with the smallest possible retry delay.
 def unavailable_status() -> _Status:
     error = status_pb2.Status(
@@ -128,10 +130,13 @@ def add_select1_result():
     add_single_result("select 1", "c", TypeCode.INT64, [("1",)])
 
 
-def add_execute_streaming_sql_results(sql: str,
-    partial_result_sets: list[result_set.PartialResultSet]):
+def add_execute_streaming_sql_results(
+    sql: str, partial_result_sets: list[result_set.PartialResultSet]
+):
     MockServerTestBase.spanner_service.mock_spanner.add_execute_streaming_sql_results(
-        sql, partial_result_sets)
+        sql, partial_result_sets
+    )
+
 
 def add_single_result(
     sql: str, column_name: str, type_code: spanner_type.TypeCode, row

tests/mockserver_tests/test_basics.py

@@ -24,19 +24,18 @@
 from google.cloud.spanner_v1.testing.mock_spanner import SpannerServicer
 from google.cloud.spanner_v1.transaction import Transaction
 from tests.mockserver_tests.mock_server_test_base import MockServerTestBase
-from tests.mockserver_tests.mock_server_test_base import \
-    _make_partial_result_sets
+from tests.mockserver_tests.mock_server_test_base import _make_partial_result_sets
 from tests.mockserver_tests.mock_server_test_base import add_error
-from tests.mockserver_tests.mock_server_test_base import \
-    add_execute_streaming_sql_results
+from tests.mockserver_tests.mock_server_test_base import (
+    add_execute_streaming_sql_results,
+)
 from tests.mockserver_tests.mock_server_test_base import add_select1_result
 from tests.mockserver_tests.mock_server_test_base import add_single_result
 from tests.mockserver_tests.mock_server_test_base import add_update_count
 from tests.mockserver_tests.mock_server_test_base import unavailable_status
 
 
 class TestBasics(MockServerTestBase):
-
     def setUp(self):
         super().setUp()
         super().setup_class()
@@ -183,8 +182,11 @@ def test_last_statement_query(self):
     def test_execute_streaming_sql_last_field(self):
         partial_result_sets = _make_partial_result_sets(
             [("ID", TypeCode.INT64), ("NAME", TypeCode.STRING)],
-            [{"values": ["1", "ABC", "2", "DEF"]},
-             {"values": ["3", "GHI"], "last": True}])
+            [
+                {"values": ["1", "ABC", "2", "DEF"]},
+                {"values": ["3", "GHI"], "last": True},
+            ],
+        )
 
         sql = "select * from my_table"
         add_execute_streaming_sql_results(sql, partial_result_sets)
@@ -202,6 +204,7 @@ def test_execute_streaming_sql_last_field(self):
         self.assertTrue(isinstance(requests[0], BatchCreateSessionsRequest))
         self.assertTrue(isinstance(requests[1], ExecuteSqlRequest))
 
+
 def _execute_query(transaction: Transaction, sql: str):
     rows = transaction.execute_sql(sql, last_statement=True)
     for _ in rows: