GCF: Add HTTP unit test sample (GoogleCloudPlatform#2629)

Ace Nassri · web-flow · commit 7b872c3b28ec · 2020-04-10T13:30:22.000-07:00
* Add HTTP unit test sample

* Add newline
diff --git a/functions/snippets/helloworld/src/test/java/com/example/functions/helloworld/HelloHttpTest.java b/functions/snippets/helloworld/src/test/java/com/example/functions/helloworld/HelloHttpTest.java
@@ -0,0 +1,99 @@
+/*
+ * Copyright 2020 Google LLC
+ *
+ * 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.example.functions.helloworld;
+
+// [START functions_http_unit_test]
+import static com.google.common.truth.Truth.assertThat;
+import static org.powermock.api.mockito.PowerMockito.mock;
+import static org.powermock.api.mockito.PowerMockito.when;
+
+import com.google.cloud.functions.HttpRequest;
+import com.google.cloud.functions.HttpResponse;
+import com.google.common.testing.TestLogHandler;
+import java.io.BufferedReader;
+import java.io.BufferedWriter;
+import java.io.IOException;
+import java.io.StringReader;
+import java.io.StringWriter;
+import java.util.Optional;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+
+@RunWith(JUnit4.class)
+public class HelloHttpTest {
+  @Mock private HttpRequest request;
+  @Mock private HttpResponse response;
+
+  private BufferedWriter writerOut;
+  private StringWriter responseOut;
+
+  private static final TestLogHandler logHandler = new TestLogHandler();
+
+  @Before
+  public void beforeTest() throws IOException {
+    Mockito.mockitoSession().initMocks(this);
+
+    request = mock(HttpRequest.class);
+    response = mock(HttpResponse.class);
+
+    BufferedReader reader = new BufferedReader(new StringReader("{}"));
+    when(request.getReader()).thenReturn(reader);
+
+    responseOut = new StringWriter();
+    writerOut = new BufferedWriter(responseOut);
+    when(response.getWriter()).thenReturn(writerOut);
+
+    logHandler.clear();
+  }
+
+  @Test
+  public void helloHttp_noParamsGet() throws IOException {
+    new HelloHttp().service(request, response);
+
+    writerOut.flush();
+    assertThat(responseOut.toString()).isEqualTo("Hello world!");
+  }
+  // [END functions_http_unit_test]
+
+  @Test
+  public void helloHttp_urlParamsGet() throws IOException {
+    when(request.getFirstQueryParameter("name")).thenReturn(Optional.of("Tom"));
+
+    new HelloHttp().service(request, response);
+
+    writerOut.flush();
+    assertThat(responseOut.toString()).isEqualTo("Hello Tom!");
+  }
+
+  @Test
+  public void helloHttp_bodyParamsPost() throws IOException {
+    BufferedReader jsonReader = new BufferedReader(new StringReader("{'name': 'Jane'}"));
+
+    when(request.getReader()).thenReturn(jsonReader);
+
+    new HelloHttp().service(request, response);
+    writerOut.flush();
+
+    assertThat(responseOut.toString()).isEqualTo("Hello Jane!");
+  }
+  // [START functions_http_unit_test]
+}
+// [END functions_http_unit_test]
