mv com/google/domain/registry google/registry

This change renames directories in preparation for the great package
rename. The repository is now in a broken state because the code
itself hasn't been updated. However this should ensure that git
correctly preserves history for each file.

javatests/google/registry/export/UpdateSnapshotViewActionTest.java

@@ -0,0 +1,135 @@
+// Copyright 2016 The Domain Registry 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 com.google.domain.registry.export;
+
+import static com.google.common.truth.Truth.assertThat;
+import static com.google.domain.registry.export.UpdateSnapshotViewAction.UPDATE_SNAPSHOT_DATASET_ID_PARAM;
+import static com.google.domain.registry.export.UpdateSnapshotViewAction.UPDATE_SNAPSHOT_KIND_PARAM;
+import static com.google.domain.registry.export.UpdateSnapshotViewAction.UPDATE_SNAPSHOT_TABLE_ID_PARAM;
+import static com.google.domain.registry.testing.TaskQueueHelper.assertTasksEnqueued;
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.anyString;
+import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import com.google.api.services.bigquery.Bigquery;
+import com.google.api.services.bigquery.model.Dataset;
+import com.google.api.services.bigquery.model.Table;
+import com.google.appengine.api.taskqueue.QueueFactory;
+import com.google.domain.registry.bigquery.BigqueryFactory;
+import com.google.domain.registry.request.HttpException.InternalServerErrorException;
+import com.google.domain.registry.testing.AppEngineRule;
+import com.google.domain.registry.testing.ExceptionRule;
+import com.google.domain.registry.testing.TaskQueueHelper.TaskMatcher;
+
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.ArgumentCaptor;
+import org.mockito.Mock;
+import org.mockito.runners.MockitoJUnitRunner;
+
+import java.io.IOException;
+
+/** Unit tests for {@link UpdateSnapshotViewAction}. */
+@RunWith(MockitoJUnitRunner.class)
+public class UpdateSnapshotViewActionTest {
+
+  @Rule
+  public final AppEngineRule appEngine = AppEngineRule.builder()
+      .withTaskQueue()
+      .build();
+
+  @Rule
+  public final ExceptionRule thrown = new ExceptionRule();
+
+  @Mock
+  private BigqueryFactory bigqueryFactory;
+
+  @Mock
+  private Bigquery bigquery;
+
+  @Mock
+  private Bigquery.Datasets bigqueryDatasets;
+
+  @Mock
+  private Bigquery.Datasets.Insert bigqueryDatasetsInsert;
+
+  @Mock
+  private Bigquery.Tables bigqueryTables;
+
+  @Mock
+  private Bigquery.Tables.Update bigqueryTablesUpdate;
+
+  private UpdateSnapshotViewAction action;
+
+  @Before
+  public void before() throws Exception {
+    when(bigqueryFactory.create(anyString(), anyString())).thenReturn(bigquery);
+    when(bigquery.datasets()).thenReturn(bigqueryDatasets);
+    when(bigqueryDatasets.insert(eq("Project-Id"), any(Dataset.class)))
+        .thenReturn(bigqueryDatasetsInsert);
+    when(bigquery.tables()).thenReturn(bigqueryTables);
+    when(bigqueryTables.update(
+            eq("Project-Id"), anyString(), anyString(), any(Table.class)))
+        .thenReturn(bigqueryTablesUpdate);
+
+    action = new UpdateSnapshotViewAction();
+    action.bigqueryFactory = bigqueryFactory;
+    action.datasetId = "some_dataset";
+    action.kindName = "fookind";
+    action.projectId = "Project-Id";
+    action.tableId = "12345_fookind";
+  }
+
+  @Test
+  public void testSuccess_createViewUpdateTask() throws Exception {
+    QueueFactory.getDefaultQueue().add(
+        UpdateSnapshotViewAction.createViewUpdateTask("some_dataset", "12345_fookind", "fookind"));
+    assertTasksEnqueued("default",
+        new TaskMatcher()
+            .url(UpdateSnapshotViewAction.PATH)
+            .method("POST")
+            .param(UPDATE_SNAPSHOT_DATASET_ID_PARAM, "some_dataset")
+            .param(UPDATE_SNAPSHOT_TABLE_ID_PARAM, "12345_fookind")
+            .param(UPDATE_SNAPSHOT_KIND_PARAM, "fookind"));
+  }
+
+  @Test
+  public void testSuccess_doPost() throws Exception {
+    action.run();
+
+    // Check that the BigQuery factory was called in such a way that the dataset would be created
+    // if it didn't already exist.
+    verify(bigqueryFactory).create("Project-Id", "latest_snapshot");
+
+    // Check that we updated the view.
+    ArgumentCaptor<Table> tableArg = ArgumentCaptor.forClass(Table.class);
+    verify(bigqueryTables).update(
+        eq("Project-Id"), eq("latest_snapshot"), eq("fookind"), tableArg.capture());
+    assertThat(tableArg.getValue().getView().getQuery())
+        .isEqualTo("SELECT * FROM [some_dataset.12345_fookind]");
+  }
+
+  @Test
+  public void testFailure_bigqueryConnectionThrowsError() throws Exception {
+    when(bigqueryTables.update(anyString(), anyString(), anyString(), any(Table.class)))
+        .thenThrow(new IOException("I'm sorry Dave, I can't let you do that"));
+    thrown.expect(InternalServerErrorException.class, "Error in update snapshot view action");
+    action.run();
+  }
+}