Import code from internal repository to git

javatests/com/google/domain/registry/export/ExportReservedTermsTaskTest.java

@@ -0,0 +1,147 @@
+// 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.domain.registry.export;
+
+import static com.google.domain.registry.export.ExportReservedTermsTask.EXPORT_MIME_TYPE;
+import static com.google.domain.registry.export.ExportReservedTermsTask.RESERVED_TERMS_FILENAME;
+import static com.google.domain.registry.testing.DatastoreHelper.createTld;
+import static com.google.domain.registry.testing.DatastoreHelper.persistReservedList;
+import static com.google.domain.registry.testing.DatastoreHelper.persistResource;
+import static java.nio.charset.StandardCharsets.UTF_8;
+import static javax.servlet.http.HttpServletResponse.SC_INTERNAL_SERVER_ERROR;
+import static javax.servlet.http.HttpServletResponse.SC_OK;
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.anyString;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import com.google.common.collect.ImmutableSet;
+import com.google.common.net.MediaType;
+import com.google.domain.registry.model.registry.Registry;
+import com.google.domain.registry.model.registry.Registry.RegistryNotFoundException;
+import com.google.domain.registry.model.registry.label.ReservedList;
+import com.google.domain.registry.request.Response;
+import com.google.domain.registry.storage.drive.DriveConnection;
+import com.google.domain.registry.testing.AppEngineRule;
+import com.google.domain.registry.testing.ExceptionRule;
+import com.google.domain.registry.testing.InjectRule;
+
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.runners.MockitoJUnitRunner;
+
+import java.io.IOException;
+
+/** Unit tests for {@link ExportReservedTermsTask}. */
+@RunWith(MockitoJUnitRunner.class)
+public class ExportReservedTermsTaskTest {
+
+  @Rule
+  public final AppEngineRule appEngine = AppEngineRule.builder()
+      .withDatastore()
+      .build();
+
+  @Rule
+  public final ExceptionRule thrown = new ExceptionRule();
+
+  @Rule
+  public final InjectRule inject = new InjectRule();
+
+  @Mock
+  private DriveConnection driveConnection;
+
+  @Mock
+  private Response response;
+
+  private void runTask(String tld) {
+    ExportReservedTermsTask task = new ExportReservedTermsTask();
+    task.response = response;
+    task.driveConnection = driveConnection;
+    task.tld = tld;
+    task.run();
+  }
+
+  @Before
+  public void init() throws Exception {
+    ReservedList rl = persistReservedList(
+        "tld-reserved",
+        "lol,FULLY_BLOCKED",
+        "cat,FULLY_BLOCKED",
+        "jimmy,UNRESERVED");
+    createTld("tld");
+    persistResource(Registry.get("tld").asBuilder()
+        .setReservedLists(rl)
+        .setDriveFolderId("brouhaha").build());
+    when(driveConnection.createOrUpdateFile(
+        anyString(),
+        any(MediaType.class),
+        anyString(),
+        any(byte[].class))).thenReturn("1001");
+  }
+
+  @Test
+  public void test_uploadFileToDrive_succeeds() throws Exception {
+    runTask("tld");
+    byte[] expected =
+        ("This is a disclaimer.\ncat\nlol\n")
+        .getBytes(UTF_8);
+    verify(driveConnection)
+        .createOrUpdateFile(RESERVED_TERMS_FILENAME, EXPORT_MIME_TYPE, "brouhaha", expected);
+    verify(response).setStatus(SC_OK);
+    verify(response).setPayload("1001");
+  }
+
+  @Test
+  public void test_uploadFileToDrive_doesNothingIfReservedListsNotConfigured() throws Exception {
+    persistResource(Registry.get("tld").asBuilder()
+        .setReservedLists(ImmutableSet.<ReservedList>of())
+        .setDriveFolderId(null)
+        .build());
+    runTask("tld");
+    verify(response).setStatus(SC_OK);
+    verify(response).setPayload("No reserved lists configured");
+  }
+
+  @Test
+  public void test_uploadFileToDrive_failsWhenDriveFolderIdIsNull() throws Exception {
+    thrown.expectRootCause(NullPointerException.class, "No drive folder associated with this TLD");
+    persistResource(Registry.get("tld").asBuilder().setDriveFolderId(null).build());
+    runTask("tld");
+    verify(response).setStatus(SC_INTERNAL_SERVER_ERROR);
+  }
+
+  @Test
+  public void test_uploadFileToDrive_failsWhenDriveCannotBeReached() throws Exception {
+    thrown.expectRootCause(IOException.class, "errorMessage");
+    when(driveConnection.createOrUpdateFile(
+        anyString(),
+        any(MediaType.class),
+        anyString(),
+        any(byte[].class))).thenThrow(new IOException("errorMessage"));
+    runTask("tld");
+    verify(response).setStatus(SC_INTERNAL_SERVER_ERROR);
+  }
+
+  @Test
+  public void test_uploadFileToDrive_failsWhenTldDoesntExist() throws Exception {
+    thrown.expectRootCause(
+        RegistryNotFoundException.class, "No registry object found for fakeTld");
+    runTask("fakeTld");
+    verify(response).setStatus(SC_INTERNAL_SERVER_ERROR);
+  }
+}