---

ajaaym · sduskis · commit ee2c01d998a6 · 2019-02-08T14:48:16.000-05:00
yaml --- r: 14765 b: refs/heads/autosynth-asset c: d6b2c21 h: refs/heads/master i: 14763: 72d8ec4
diff --git a/[refs] b/[refs]
@@ -122,7 +122,7 @@ refs/heads/autosynth-vision: b8e47d76578b5f150ef530072ea7e485e2b02ca0
 refs/heads/spanner: b01127f885b4611bf1852abb0ce481eeb7fcc131
 refs/tags/v0.68.0: 9cc799fcf68c82ab431d425fefa58ef615ce8e5b
 refs/tags/v0.69.0: 78f67a29e8b9c46ba01de566a2eae0fd1c03edea
-refs/heads/autosynth-asset: 64f3b2be23eab15df420e5890f14409fa52c8411
+refs/heads/autosynth-asset: d6b2c21e731a4307128d79d65daa64893e421a5a
 refs/heads/autosynth-automl: d4315b3596bac160e3439432c54435f44b09953e
 refs/heads/autosynth-bigquerydatatransfer: 2a9f3938237f85a8919602d74011326580ff387f
 refs/heads/autosynth-bigquerystorage: 99aee05df348f39d98b6fb23c292006f1d2a6c28
diff --git a/branches/autosynth-asset/google-cloud-clients/google-cloud-spanner/src/main/java/com/google/cloud/spanner/Key.java b/branches/autosynth-asset/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-asset/google-cloud-clients/google-cloud-spanner/src/test/java/com/google/cloud/spanner/KeyTest.java b/branches/autosynth-asset/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());
+  }
 }
