Add unit tests to RequestStatusCheckerImpl

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=167742899

javatests/google/registry/util/RequestStatusCheckerImplTest.java

@@ -0,0 +1,129 @@
+// Copyright 2017 The Nomulus Authors. 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 google.registry.util;
+
+import static com.google.common.truth.Truth.assertThat;
+import static google.registry.testing.LogsSubject.assertAboutLogs;
+import static org.mockito.Matchers.argThat;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import com.google.appengine.api.log.LogQuery;
+import com.google.appengine.api.log.LogService;
+import com.google.appengine.api.log.RequestLogs;
+import com.google.apphosting.api.ApiProxy;
+import com.google.common.collect.ImmutableList;
+import com.google.common.testing.TestLogHandler;
+import google.registry.testing.AppEngineRule;
+import java.util.logging.Level;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+import org.mockito.ArgumentMatcher;
+
+/** Unit tests for {@link RequestStatusCheckerImpl}. */
+@RunWith(JUnit4.class)
+public final class RequestStatusCheckerImplTest {
+
+  private static final TestLogHandler logHandler = new TestLogHandler();
+
+  private static final RequestStatusChecker requestStatusChecker = new RequestStatusCheckerImpl();
+
+  /**
+   * Matcher for the expected LogQuery in {@link RequestStatusCheckerImpl#isRunning}.
+   *
+   * Because LogQuery doesn't have a .equals function, we have to create an actual matcher to make
+   * sure we have the right argument in our mocks.
+   */
+  private static final LogQuery expectedLogQuery(final String requestLogId) {
+    return argThat(
+        new ArgumentMatcher<LogQuery>() {
+          @Override
+          public boolean matches(Object object) {
+            assertThat(object).isInstanceOf(LogQuery.class);
+            LogQuery logQuery = (LogQuery) object;
+            assertThat(logQuery.getRequestIds()).containsExactly(requestLogId);
+            assertThat(logQuery.getIncludeAppLogs()).isFalse();
+            assertThat(logQuery.getIncludeIncomplete()).isTrue();
+            return true;
+          }
+        });
+  }
+
+  @Rule
+  public AppEngineRule appEngineRule = AppEngineRule.builder().build();
+
+  @Before public void setUp() {
+    RequestStatusCheckerImpl.logger.addHandler(logHandler);
+    RequestStatusCheckerImpl.logService = mock(LogService.class);
+  }
+
+  @After public void tearDown() {
+    RequestStatusCheckerImpl.logger.removeHandler(logHandler);
+  }
+
+  // If a logId is unrecognized, it could be that the log hasn't been uploaded yet - so we assume
+  // it's a request that has just started running recently.
+  @Test public void testIsRunning_unrecognized() {
+    when(RequestStatusCheckerImpl.logService.fetch(expectedLogQuery("12345678")))
+        .thenReturn(ImmutableList.of());
+    assertThat(requestStatusChecker.isRunning("12345678")).isTrue();
+    assertAboutLogs()
+        .that(logHandler)
+        .hasLogAtLevelWithMessage(Level.INFO, "Queried an unrecognized requestLogId");
+  }
+
+  @Test public void testIsRunning_notFinished() {
+    RequestLogs requestLogs = new RequestLogs();
+    requestLogs.setFinished(false);
+
+    when(RequestStatusCheckerImpl.logService.fetch(expectedLogQuery("12345678")))
+        .thenReturn(ImmutableList.of(requestLogs));
+
+    assertThat(requestStatusChecker.isRunning("12345678")).isTrue();
+    assertAboutLogs()
+        .that(logHandler)
+        .hasLogAtLevelWithMessage(Level.INFO, "isFinished: false");
+  }
+
+  @Test public void testIsRunning_finished() {
+    RequestLogs requestLogs = new RequestLogs();
+    requestLogs.setFinished(true);
+
+    when(RequestStatusCheckerImpl.logService.fetch(expectedLogQuery("12345678")))
+        .thenReturn(ImmutableList.of(requestLogs));
+
+    assertThat(requestStatusChecker.isRunning("12345678")).isFalse();
+    assertAboutLogs()
+        .that(logHandler)
+        .hasLogAtLevelWithMessage(Level.INFO, "isFinished: true");
+  }
+
+  @Test public void testGetLogId_returnsRequestLogId() {
+    String expectedLogId = ApiProxy.getCurrentEnvironment().getAttributes().get(
+        "com.google.appengine.runtime.request_log_id").toString();
+    assertThat(requestStatusChecker.getLogId()).isEqualTo(expectedLogId);
+  }
+
+  @Test public void testGetLogId_createsLog() {
+    requestStatusChecker.getLogId();
+    assertAboutLogs()
+        .that(logHandler)
+        .hasLogAtLevelWithMessage(Level.INFO, "Current requestLogId: ");
+  }
+}