From d3fcb86bbea5f8cd1a92435cbd3d5b31b9664cb3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Knut=20Olav=20L=C3=B8ite?= <koloite@gmail.com>
Date: Thu, 23 Mar 2023 17:52:00 +0100
Subject: [PATCH] test: fix mock server ARRAY<BYTES> parameter cast (#2344)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* test: fix mock server ARRAY<BYTES> parameter cast

The recent change that made decoding BYTES and ARRAY<BYTES> lazy did not
update all code paths in the mock Spanner server that is used for testing.
This meant that tests that tried to use ARRAY<BYTES> query parameters would
run into a ClassCastException. This problem only occurs in (internal) test
code and does not affect user code.

* 🦉 Updates from OwlBot post-processor

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

---------

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
---
 .../cloud/spanner/DatabaseClientImplTest.java | 19 +++++++++++++++++++
 .../cloud/spanner/MockSpannerServiceImpl.java |  9 ++++++---
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/DatabaseClientImplTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/DatabaseClientImplTest.java
index ac0f0a98d29..9885e4f83d5 100644
--- a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/DatabaseClientImplTest.java
+++ b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/DatabaseClientImplTest.java
@@ -21,6 +21,7 @@
 import static com.google.cloud.spanner.MockSpannerTestUtil.READ_ONE_KEY_VALUE_STATEMENT;
 import static com.google.cloud.spanner.MockSpannerTestUtil.READ_TABLE_NAME;
 import static com.google.cloud.spanner.MockSpannerTestUtil.SELECT1;
+import static com.google.cloud.spanner.MockSpannerTestUtil.SELECT1_RESULTSET;
 import static com.google.cloud.spanner.SpannerApiFutures.get;
 import static com.google.common.truth.Truth.assertThat;
 import static org.junit.Assert.assertEquals;
@@ -2943,6 +2944,24 @@ public void testMetadataUnknownTypes() {
     }
   }
 
+  @Test
+  public void testStatementWithBytesArrayParameter() {
+    Statement statement =
+        Statement.newBuilder("select id from test where b=@p1")
+            .bind("p1")
+            .toBytesArray(
+                ImmutableList.of(ByteArray.copyFrom("test1"), ByteArray.copyFrom("test2")))
+            .build();
+    mockSpanner.putStatementResult(StatementResult.query(statement, SELECT1_RESULTSET));
+    DatabaseClient client =
+        spanner.getDatabaseClient(DatabaseId.of(TEST_PROJECT, TEST_INSTANCE, TEST_DATABASE));
+    try (ResultSet resultSet = client.singleUse().executeQuery(statement)) {
+      assertTrue(resultSet.next());
+      assertEquals(1L, resultSet.getLong(0));
+      assertFalse(resultSet.next());
+    }
+  }
+
   static void assertAsString(String expected, ResultSet resultSet, int col) {
     assertEquals(expected, resultSet.getValue(col).getAsString());
     assertEquals(ImmutableList.of(expected), resultSet.getValue(col).getAsStringList());
diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/MockSpannerServiceImpl.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/MockSpannerServiceImpl.java
index 29ab79450b5..567279c511c 100644
--- a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/MockSpannerServiceImpl.java
+++ b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/MockSpannerServiceImpl.java
@@ -20,6 +20,7 @@
 import com.google.cloud.ByteArray;
 import com.google.cloud.Date;
 import com.google.cloud.spanner.AbstractResultSet.GrpcStruct;
+import com.google.cloud.spanner.AbstractResultSet.LazyByteArray;
 import com.google.cloud.spanner.SessionPool.SessionPoolTransactionContext;
 import com.google.cloud.spanner.TransactionRunnerImpl.TransactionContextImpl;
 import com.google.common.base.Optional;
@@ -1362,9 +1363,11 @@ private Statement buildStatement(
                 builder
                     .bind(fieldName)
                     .toBytesArray(
-                        (Iterable<ByteArray>)
-                            GrpcStruct.decodeArrayValue(
-                                com.google.cloud.spanner.Type.bytes(), value.getListValue()));
+                        Iterables.transform(
+                            (Iterable<LazyByteArray>)
+                                GrpcStruct.decodeArrayValue(
+                                    com.google.cloud.spanner.Type.bytes(), value.getListValue()),
+                            LazyByteArray::getByteArray));
                 break;
               case DATE:
                 builder