---

ajaaym · sduskis · commit 9beecc026a70 · 2019-02-08T14:48:16.000-05:00
yaml --- r: 19035 b: refs/heads/autosynth-bigquerydatatransfer c: d6b2c21 h: refs/heads/master i: 19033: 8a5e257 19031: 7e6789c
diff --git a/[refs] b/[refs]
@@ -123,7 +123,7 @@ refs/tags/v0.68.0: 9cc799fcf68c82ab431d425fefa58ef615ce8e5b
 refs/tags/v0.69.0: 78f67a29e8b9c46ba01de566a2eae0fd1c03edea
 refs/heads/autosynth-asset: cd8251de8c40e239ad24dcf9ed93ea2708a3eed5
 refs/heads/autosynth-automl: 2a8b018cf05811fd472e5d1053e67a12ceec0b64
-refs/heads/autosynth-bigquerydatatransfer: 64f3b2be23eab15df420e5890f14409fa52c8411
+refs/heads/autosynth-bigquerydatatransfer: d6b2c21e731a4307128d79d65daa64893e421a5a
 refs/heads/autosynth-bigquerystorage: 14ab055598b943ae3f33f484e9fb1653355d08e7
 refs/heads/autosynth-bigtable: 2fbcb15847e0e89e79d6dc07420e28d7dfcea894
 refs/heads/autosynth-bigtable-admin: 6379a2bc712f2736c83de0e009b4d26da4fa82ca
diff --git a/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-spanner/src/main/java/com/google/cloud/spanner/Key.java b/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-spanner/src/main/java/com/google/cloud/spanner/Key.java
@@ -281,7 +281,9 @@ ListValue toProto() {
       } else if (part instanceof ByteArray) {
         builder.addValuesBuilder().setStringValue(((ByteArray) part).toBase64());
       } else if (part instanceof Timestamp) {
-        builder.addValuesBuilder().setStringValue(((Timestamp) part).toString());
+        builder.addValuesBuilder().setStringValue(part.toString());
+      } else if (part instanceof Date) {
+        builder.addValuesBuilder().setStringValue(part.toString());
       } else {
         throw new AssertionError("Illegal key part: " + part.getClass());
       }
diff --git a/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-spanner/src/test/java/com/google/cloud/spanner/KeyTest.java b/branches/autosynth-bigquerydatatransfer/google-cloud-clients/google-cloud-spanner/src/test/java/com/google/cloud/spanner/KeyTest.java
@@ -23,6 +23,8 @@
 import com.google.cloud.Date;
 import com.google.cloud.Timestamp;
 import com.google.common.testing.EqualsTester;
+import com.google.protobuf.ListValue;
+import com.google.protobuf.NullValue;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
@@ -189,4 +191,35 @@ public void serialization() throws Exception {
     reserializeAndAssert(Key.of(Date.parseDate("2015-09-15")));
     reserializeAndAssert(Key.of(1, 2, 3));
   }
+
+  @Test
+  public void toProto() {
+    String timestamp = "2015-09-15T00:00:00Z";
+    String date = "2015-09-15";
+    Key k =
+        Key.newBuilder()
+            .append((Boolean) null)
+            .append(true)
+            .append(32)
+            .append(64L)
+            .append(2.0f)
+            .append(4.0d)
+            .append("x")
+            .append(ByteArray.copyFrom("y"))
+            .append(Timestamp.parseTimestamp(timestamp))
+            .append(Date.parseDate(date))
+            .build();
+    ListValue.Builder builder = ListValue.newBuilder();
+    builder.addValuesBuilder().setNullValue(NullValue.NULL_VALUE);
+    builder.addValuesBuilder().setBoolValue(true);
+    builder.addValuesBuilder().setStringValue("32");
+    builder.addValuesBuilder().setStringValue("64");
+    builder.addValuesBuilder().setNumberValue(2.0f);
+    builder.addValuesBuilder().setNumberValue(4.0d);
+    builder.addValuesBuilder().setStringValue("x");
+    builder.addValuesBuilder().setStringValue("eQ==");
+    builder.addValuesBuilder().setStringValue(timestamp);
+    builder.addValuesBuilder().setStringValue(date);
+    assertThat(k.toProto()).isEqualTo(builder.build());
+  }
 }
