Refactor update snapshot view servlet into a Daggerized Action

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

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

@@ -0,0 +1,128 @@
+// 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.SNAPSHOT_DATASET_ID_PARAM;
+import static com.google.domain.registry.export.UpdateSnapshotViewAction.SNAPSHOT_KIND_PARAM;
+import static com.google.domain.registry.export.UpdateSnapshotViewAction.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.config.TestRegistryConfig;
+import com.google.domain.registry.testing.AppEngineRule;
+import com.google.domain.registry.testing.RegistryConfigRule;
+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;
+
+/** Unit tests for {@link UpdateSnapshotViewAction}. */
+@RunWith(MockitoJUnitRunner.class)
+public class UpdateSnapshotViewActionTest {
+
+  @Rule
+  public final AppEngineRule appEngine = AppEngineRule.builder()
+      .withTaskQueue()
+      .build();
+
+  @Rule
+  public final RegistryConfigRule configRule = new RegistryConfigRule(new TestRegistryConfig() {
+    @Override public String getProjectId() {
+      return "Project-Id";
+    }
+
+    @Override public String getLatestSnapshotDataset() {
+      return "testdataset";
+    }
+  });
+
+  @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"), any(String.class), any(String.class), any(Table.class)))
+        .thenReturn(bigqueryTablesUpdate);
+
+    action = new UpdateSnapshotViewAction();
+    action.bigqueryFactory = bigqueryFactory;
+    action.datasetId = "some_dataset";
+    action.tableId = "12345_fookind";
+    action.kindName = "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(SNAPSHOT_DATASET_ID_PARAM, "some_dataset")
+            .param(SNAPSHOT_TABLE_ID_PARAM, "12345_fookind")
+            .param(SNAPSHOT_KIND_PARAM, "fookind"));
+  }
+
+  @Test
+  public void testSuccess_doPost() throws Exception {
+    action.run();
+
+    // Check that we updated the view.
+    ArgumentCaptor<Table> tableArg = ArgumentCaptor.forClass(Table.class);
+    verify(bigqueryTables).update(
+        eq("Project-Id"), eq("testdataset"), eq("fookind"), tableArg.capture());
+    assertThat(tableArg.getValue().getView().getQuery())
+        .isEqualTo("SELECT * FROM [some_dataset.12345_fookind]");
+  }
+}