Add serialization test for PubSub classes

mziccard · mziccard · commit 551e9fdb6638 · 2016-06-23T02:06:09.000+02:00
diff --git a/gcloud-java-pubsub/pom.xml b/gcloud-java-pubsub/pom.xml
@@ -30,13 +30,20 @@
     <dependency>
       <groupId>io.grpc</groupId>
       <artifactId>grpc-all</artifactId>
-      <version>0.12.0</version>
+      <version>0.14.0</version>
     </dependency>
     <dependency>
       <groupId>com.google.auto.value</groupId>
       <artifactId>auto-value</artifactId>
       <version>1.1</version>
     </dependency>
+    <dependency>
+      <groupId>${project.groupId}</groupId>
+      <artifactId>gcloud-java-core</artifactId>
+      <version>${project.version}</version>
+      <type>test-jar</type>
+      <scope>test</scope>
+    </dependency>
     <dependency>
       <groupId>junit</groupId>
       <artifactId>junit</artifactId>
diff --git a/gcloud-java-pubsub/src/test/java/com/google/cloud/pubsub/SerializationTest.java b/gcloud-java-pubsub/src/test/java/com/google/cloud/pubsub/SerializationTest.java
@@ -0,0 +1,101 @@
+/*
+ * Copyright 2016 Google Inc. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.cloud.pubsub;
+
+import com.google.cloud.AuthCredentials;
+import com.google.cloud.BaseSerializationTest;
+import com.google.cloud.GrpcServiceOptions.ExecutorFactory;
+import com.google.cloud.Restorable;
+import com.google.cloud.pubsub.PubSub.ListOption;
+import com.google.cloud.pubsub.PubSub.PullOption;
+
+import java.io.Serializable;
+import java.util.concurrent.ScheduledExecutorService;
+
+public class SerializationTest extends BaseSerializationTest {
+
+  private static final PubSub PUB_SUB = PubSubOptions.builder()
+      .projectId("p")
+      .authCredentials(AuthCredentials.noAuth())
+      .build().service();
+  private static final Message MESSAGE = Message.of("payload");
+  private static final com.google.pubsub.v1.ReceivedMessage RECEIVED_MESSAGE_PB =
+      com.google.pubsub.v1.ReceivedMessage.newBuilder()
+          .setMessage(MESSAGE.toPb())
+          .setAckId("ackId")
+          .build();
+  private static final ReceivedMessage RECEIVED_MESSAGE =
+      ReceivedMessage.fromPb(PUB_SUB, "subscription", RECEIVED_MESSAGE_PB);
+  private static final SubscriptionInfo SUBSCRIPTION_INFO = SubscriptionInfo.of("topic", "sub");
+  private static final Subscription SUBSCRIPTION =
+      new Subscription(PUB_SUB, new SubscriptionInfo.BuilderImpl(SUBSCRIPTION_INFO));
+  private static final SubscriptionId SUBSCRIPTION_ID = new SubscriptionId("project", "sub");
+  private static final TopicInfo TOPIC_INFO = TopicInfo.of("topic");
+  private static final Topic TOPIC =
+      new Topic(PUB_SUB, new TopicInfo.BuilderImpl(TOPIC_INFO));
+  private static final ListOption PAGE_TOKEN_OPTION = ListOption.pageToken("cursor");
+  private static final ListOption PAGE_SIZE_OPTION = ListOption.pageSize(42);
+  private static final PullOption MAX_QUEUED_CALLBACKS_OPTION = PullOption.maxQueuedCallbacks(42);
+  private static final PullOption EXECUTOR_FACTORY_OPTION =
+      PullOption.executorFactory(new TestExecutorFactory());
+
+  public static class TestExecutorFactory
+      implements ExecutorFactory<ScheduledExecutorService>, Serializable {
+
+    private static final long serialVersionUID = -2154875338174302704L;
+
+    @Override
+    public ScheduledExecutorService get() {
+      return null;
+    }
+
+    @Override
+    public void release(ScheduledExecutorService executor) {
+      // do nothing
+    }
+
+    @Override
+    public boolean equals(Object obj) {
+      return obj instanceof TestExecutorFactory;
+    }
+
+    @Override
+    public int hashCode() {
+      return 1;
+    }
+  }
+
+  @Override
+  protected Serializable[] serializableObjects() {
+    PubSubOptions options = PubSubOptions.builder()
+        .projectId("p1")
+        .initialTimeout(1234)
+        .build();
+    PubSubOptions otherOptions = options.toBuilder()
+        .projectId("p2")
+        .executorFactory(new TestExecutorFactory())
+        .build();
+    return new Serializable[]{options, otherOptions, MESSAGE, RECEIVED_MESSAGE, SUBSCRIPTION_INFO,
+        SUBSCRIPTION, SUBSCRIPTION_ID, TOPIC_INFO, TOPIC, PAGE_TOKEN_OPTION, PAGE_SIZE_OPTION,
+        MAX_QUEUED_CALLBACKS_OPTION, EXECUTOR_FACTORY_OPTION};
+  }
+
+  @Override
+  protected Restorable<?>[] restorableObjects() {
+    return null;
+  }
+}
