From 84587182748de9ea515e9ced61efb770096e7573 Mon Sep 17 00:00:00 2001
From: Ajay Kannan <ajaykannan@google.com>
Date: Thu, 22 Oct 2015 14:21:00 -0700
Subject: [PATCH] Add unit tests for public methods in LocalGcdHelper

---
 .../datastore/testing/LocalGcdHelper.java     |  9 ++-
 .../gcloud/datastore/LocalGcdHelperTest.java  | 71 +++++++++++++++++++
 2 files changed, 77 insertions(+), 3 deletions(-)
 create mode 100644 gcloud-java-datastore/src/test/java/com/google/gcloud/datastore/LocalGcdHelperTest.java

diff --git a/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/testing/LocalGcdHelper.java b/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/testing/LocalGcdHelper.java
index c3afe025756b..24c0fbf9faab 100644
--- a/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/testing/LocalGcdHelper.java
+++ b/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/testing/LocalGcdHelper.java
@@ -524,7 +524,8 @@ private static void extractFile(ZipInputStream zipIn, File filePath) throws IOEx
     }
   }
 
-  public static void sendQuitRequest(int port) {
+  public static String sendQuitRequest(int port) {
+    StringBuilder result = new StringBuilder();
     try {
       URL url = new URL("http", "localhost", port, "/_ah/admin/quit");
       HttpURLConnection con = (HttpURLConnection) url.openConnection();
@@ -535,12 +536,14 @@ public static void sendQuitRequest(int port) {
       out.write("".getBytes());
       out.flush();
       InputStream in = con.getInputStream();
-      while (in.read() != -1) {
-        // consume input
+      int currByte = 0;
+      while ((currByte = in.read()) != -1) {
+        result.append(((char) currByte));
       }
     } catch (IOException ignore) {
       // ignore
     }
+    return result.toString();
   }
 
   public void stop() throws IOException, InterruptedException {
diff --git a/gcloud-java-datastore/src/test/java/com/google/gcloud/datastore/LocalGcdHelperTest.java b/gcloud-java-datastore/src/test/java/com/google/gcloud/datastore/LocalGcdHelperTest.java
new file mode 100644
index 000000000000..3d6dca5607dc
--- /dev/null
+++ b/gcloud-java-datastore/src/test/java/com/google/gcloud/datastore/LocalGcdHelperTest.java
@@ -0,0 +1,71 @@
+/*
+ * Copyright 2015 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.gcloud.datastore;
+
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+import com.google.gcloud.datastore.testing.LocalGcdHelper;
+
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+import java.io.IOException;
+import java.net.ServerSocket;
+
+@RunWith(JUnit4.class)
+public class LocalGcdHelperTest {
+
+  private static final String PROJECT_ID = LocalGcdHelper.DEFAULT_PROJECT_ID;
+  private static final int PORT = LocalGcdHelper.findAvailablePort(LocalGcdHelper.DEFAULT_PORT);
+
+  @Rule
+  public ExpectedException thrown = ExpectedException.none();
+
+  @Test
+  public void testFindAvailablePort() {
+    int chosenPort = LocalGcdHelper.findAvailablePort(LocalGcdHelper.DEFAULT_PORT);
+    try (ServerSocket tempSocket = new ServerSocket(chosenPort)) {
+      // success
+    } catch (IOException e) {
+      if (chosenPort != LocalGcdHelper.DEFAULT_PORT) {
+        fail("Chosen port not free, even though LocalGcdHelper claimed it was.");
+      }
+    }
+  }
+
+  @Test
+  public void testSendQuitRequest() throws IOException, InterruptedException {
+    LocalGcdHelper gcdHelper = LocalGcdHelper.start(PROJECT_ID, PORT);
+    assertTrue(LocalGcdHelper.sendQuitRequest(PORT).startsWith("Shutting down local server"));
+    gcdHelper.stop();
+    assertTrue(LocalGcdHelper.sendQuitRequest(PORT).isEmpty()); // shouldn't error
+  }
+
+  @Test
+  public void testStartStop() throws IOException, InterruptedException {
+    LocalGcdHelper gcdHelper = LocalGcdHelper.start(PROJECT_ID, PORT);
+    assertFalse(LocalGcdHelper.isActive("wrong-project-id", PORT));
+    assertTrue(LocalGcdHelper.isActive(PROJECT_ID, PORT));
+    gcdHelper.stop();
+    assertFalse(LocalGcdHelper.isActive(PROJECT_ID, PORT));
+  }
+}