mirror of
https://github.com/google/nomulus.git
synced 2025-04-30 20:17:51 +02:00
5d4a88e5ce
This was an oversight I noticed ages ago, so resurrecting some old local changes I had to correct it. ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=146812322
381 lines
16 KiB
Java
381 lines
16 KiB
Java
// 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.backup;
|
|
|
|
import static com.google.common.truth.Truth.assertThat;
|
|
import static com.google.common.truth.Truth.assertWithMessage;
|
|
import static google.registry.backup.BackupUtils.GcsMetadataKeys.LOWER_BOUND_CHECKPOINT;
|
|
import static google.registry.backup.BackupUtils.GcsMetadataKeys.NUM_TRANSACTIONS;
|
|
import static google.registry.backup.BackupUtils.GcsMetadataKeys.UPPER_BOUND_CHECKPOINT;
|
|
import static google.registry.backup.BackupUtils.deserializeEntities;
|
|
import static google.registry.testing.DatastoreHelper.persistResource;
|
|
import static google.registry.util.DateTimeUtils.START_OF_TIME;
|
|
import static org.joda.time.DateTimeZone.UTC;
|
|
|
|
import com.google.appengine.tools.cloudstorage.GcsFilename;
|
|
import com.google.appengine.tools.cloudstorage.GcsService;
|
|
import com.google.appengine.tools.cloudstorage.GcsServiceFactory;
|
|
import com.google.common.collect.ImmutableMap;
|
|
import com.googlecode.objectify.Key;
|
|
import google.registry.model.ImmutableObject;
|
|
import google.registry.model.ofy.CommitLogBucket;
|
|
import google.registry.model.ofy.CommitLogCheckpoint;
|
|
import google.registry.model.ofy.CommitLogManifest;
|
|
import google.registry.model.ofy.CommitLogMutation;
|
|
import google.registry.testing.AppEngineRule;
|
|
import google.registry.testing.GcsTestingUtils;
|
|
import google.registry.testing.TestObject;
|
|
import java.util.List;
|
|
import org.joda.time.DateTime;
|
|
import org.junit.Before;
|
|
import org.junit.Rule;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
import org.junit.runners.JUnit4;
|
|
|
|
/** Unit tests for {@link ExportCommitLogDiffAction}. */
|
|
@RunWith(JUnit4.class)
|
|
public class ExportCommitLogDiffActionTest {
|
|
|
|
@Rule
|
|
public final AppEngineRule appEngine = AppEngineRule.builder()
|
|
.withDatastore()
|
|
.build();
|
|
|
|
/** Local GCS service available for testing. */
|
|
private final GcsService gcsService = GcsServiceFactory.createGcsService();
|
|
|
|
private final DateTime now = DateTime.now(UTC);
|
|
private final DateTime oneMinuteAgo = now.minusMinutes(1);
|
|
|
|
private final ExportCommitLogDiffAction task = new ExportCommitLogDiffAction();
|
|
|
|
@Before
|
|
public void before() {
|
|
task.gcsService = gcsService;
|
|
task.gcsBucket = "gcs bucket";
|
|
task.batchSize = 5;
|
|
}
|
|
|
|
@Test
|
|
public void testRun_noCommitHistory_onlyUpperCheckpointExported() throws Exception {
|
|
task.lowerCheckpointTime = oneMinuteAgo;
|
|
task.upperCheckpointTime = now;
|
|
|
|
persistResource(CommitLogCheckpoint.create(
|
|
oneMinuteAgo,
|
|
ImmutableMap.of(1, oneMinuteAgo, 2, oneMinuteAgo, 3, oneMinuteAgo)));
|
|
CommitLogCheckpoint upperCheckpoint = persistResource(CommitLogCheckpoint.create(
|
|
now,
|
|
ImmutableMap.of(1, now, 2, now, 3, now)));
|
|
|
|
// Don't persist any manifests or mutations.
|
|
|
|
task.run();
|
|
|
|
GcsFilename expectedFilename = new GcsFilename("gcs bucket", "commit_diff_until_" + now);
|
|
assertWithMessage("GCS file not found: " + expectedFilename)
|
|
.that(gcsService.getMetadata(expectedFilename)).isNotNull();
|
|
assertThat(gcsService.getMetadata(expectedFilename).getOptions().getUserMetadata())
|
|
.containsExactly(
|
|
LOWER_BOUND_CHECKPOINT,
|
|
oneMinuteAgo.toString(),
|
|
UPPER_BOUND_CHECKPOINT,
|
|
now.toString(),
|
|
NUM_TRANSACTIONS,
|
|
"0");
|
|
List<ImmutableObject> exported =
|
|
deserializeEntities(GcsTestingUtils.readGcsFile(gcsService, expectedFilename));
|
|
assertThat(exported).containsExactly(upperCheckpoint);
|
|
}
|
|
|
|
@Test
|
|
public void testRun_regularCommitHistory_exportsCorrectCheckpointDiff() throws Exception {
|
|
task.lowerCheckpointTime = oneMinuteAgo;
|
|
task.upperCheckpointTime = now;
|
|
|
|
// Persist the lower and upper checkpoints, with 3 buckets each and staggered times. We respect
|
|
// the real invariant that the time for bucket n in the lower checkpoint is <= the time for
|
|
// that bucket in the upper.
|
|
persistResource(CommitLogCheckpoint.create(
|
|
oneMinuteAgo,
|
|
ImmutableMap.of(
|
|
1, oneMinuteAgo,
|
|
2, oneMinuteAgo.minusDays(1),
|
|
3, oneMinuteAgo.minusDays(2))));
|
|
CommitLogCheckpoint upperCheckpoint = persistResource(CommitLogCheckpoint.create(
|
|
now,
|
|
ImmutableMap.of(
|
|
1, now,
|
|
2, now.minusDays(1),
|
|
3, oneMinuteAgo.minusDays(2)))); // Note that this matches the lower bound.
|
|
|
|
// Persist some fake commit log manifests.
|
|
// These shouldn't be in the diff because the lower bound is exclusive.
|
|
persistManifestAndMutation(1, oneMinuteAgo);
|
|
persistManifestAndMutation(2, oneMinuteAgo.minusDays(1));
|
|
persistManifestAndMutation(3, oneMinuteAgo.minusDays(2)); // Even though it's == upper bound.
|
|
// These shouldn't be in the diff because they are above the upper bound.
|
|
persistManifestAndMutation(1, now.plusMillis(1));
|
|
persistManifestAndMutation(2, now.minusDays(1).plusMillis(1));
|
|
persistManifestAndMutation(3, oneMinuteAgo.minusDays(2).plusMillis(1));
|
|
// These should be in the diff because they are between the bounds. (Not possible for bucket 3.)
|
|
persistManifestAndMutation(1, now.minusMillis(1));
|
|
persistManifestAndMutation(2, now.minusDays(1).minusMillis(1));
|
|
// These should be in the diff because they are at the upper bound. (Not possible for bucket 3.)
|
|
persistManifestAndMutation(1, now);
|
|
persistManifestAndMutation(2, now.minusDays(1));
|
|
|
|
task.run();
|
|
|
|
GcsFilename expectedFilename = new GcsFilename("gcs bucket", "commit_diff_until_" + now);
|
|
assertWithMessage("GCS file not found: " + expectedFilename)
|
|
.that(gcsService.getMetadata(expectedFilename)).isNotNull();
|
|
assertThat(gcsService.getMetadata(expectedFilename).getOptions().getUserMetadata())
|
|
.containsExactly(
|
|
LOWER_BOUND_CHECKPOINT,
|
|
oneMinuteAgo.toString(),
|
|
UPPER_BOUND_CHECKPOINT,
|
|
now.toString(),
|
|
NUM_TRANSACTIONS,
|
|
"4");
|
|
List<ImmutableObject> exported =
|
|
deserializeEntities(GcsTestingUtils.readGcsFile(gcsService, expectedFilename));
|
|
assertThat(exported.get(0)).isEqualTo(upperCheckpoint);
|
|
// We expect these manifests, in time order, with matching mutations.
|
|
CommitLogManifest manifest1 = createManifest(2, now.minusDays(1).minusMillis(1));
|
|
CommitLogManifest manifest2 = createManifest(2, now.minusDays(1));
|
|
CommitLogManifest manifest3 = createManifest(1, now.minusMillis(1));
|
|
CommitLogManifest manifest4 = createManifest(1, now);
|
|
assertThat(exported).containsExactly(
|
|
upperCheckpoint,
|
|
manifest1,
|
|
createMutation(manifest1),
|
|
manifest2,
|
|
createMutation(manifest2),
|
|
manifest3,
|
|
createMutation(manifest3),
|
|
manifest4,
|
|
createMutation(manifest4))
|
|
.inOrder();
|
|
}
|
|
|
|
@Test
|
|
public void testRun_simultaneousTransactions_bothExported() throws Exception {
|
|
task.lowerCheckpointTime = oneMinuteAgo;
|
|
task.upperCheckpointTime = now;
|
|
|
|
persistResource(CommitLogCheckpoint.create(
|
|
oneMinuteAgo,
|
|
ImmutableMap.of(1, START_OF_TIME, 2, START_OF_TIME, 3, START_OF_TIME)));
|
|
CommitLogCheckpoint upperCheckpoint = persistResource(CommitLogCheckpoint.create(
|
|
now,
|
|
ImmutableMap.of(1, now, 2, now, 3, now)));
|
|
|
|
// Persist some fake commit log manifests that are at the same time but in different buckets.
|
|
persistManifestAndMutation(1, oneMinuteAgo);
|
|
persistManifestAndMutation(2, oneMinuteAgo);
|
|
persistManifestAndMutation(1, now);
|
|
persistManifestAndMutation(2, now);
|
|
|
|
task.run();
|
|
|
|
GcsFilename expectedFilename = new GcsFilename("gcs bucket", "commit_diff_until_" + now);
|
|
assertWithMessage("GCS file not found: " + expectedFilename)
|
|
.that(gcsService.getMetadata(expectedFilename)).isNotNull();
|
|
assertThat(gcsService.getMetadata(expectedFilename).getOptions().getUserMetadata())
|
|
.containsExactly(
|
|
LOWER_BOUND_CHECKPOINT,
|
|
oneMinuteAgo.toString(),
|
|
UPPER_BOUND_CHECKPOINT,
|
|
now.toString(),
|
|
NUM_TRANSACTIONS,
|
|
"4");
|
|
List<ImmutableObject> exported =
|
|
deserializeEntities(GcsTestingUtils.readGcsFile(gcsService, expectedFilename));
|
|
assertThat(exported.get(0)).isEqualTo(upperCheckpoint);
|
|
// We expect these manifests, in the order below, with matching mutations.
|
|
CommitLogManifest manifest1 = createManifest(1, oneMinuteAgo);
|
|
CommitLogManifest manifest2 = createManifest(2, oneMinuteAgo);
|
|
CommitLogManifest manifest3 = createManifest(1, now);
|
|
CommitLogManifest manifest4 = createManifest(2, now);
|
|
assertThat(exported).containsExactly(
|
|
upperCheckpoint,
|
|
manifest1,
|
|
createMutation(manifest1),
|
|
manifest2,
|
|
createMutation(manifest2),
|
|
manifest3,
|
|
createMutation(manifest3),
|
|
manifest4,
|
|
createMutation(manifest4))
|
|
.inOrder();
|
|
}
|
|
|
|
@Test
|
|
public void testRun_exportsAcrossMultipleBatches() throws Exception {
|
|
task.batchSize = 2;
|
|
task.lowerCheckpointTime = oneMinuteAgo;
|
|
task.upperCheckpointTime = now;
|
|
|
|
persistResource(CommitLogCheckpoint.create(
|
|
oneMinuteAgo,
|
|
ImmutableMap.of(1, START_OF_TIME, 2, START_OF_TIME, 3, START_OF_TIME)));
|
|
CommitLogCheckpoint upperCheckpoint = persistResource(CommitLogCheckpoint.create(
|
|
now,
|
|
ImmutableMap.of(1, now, 2, now, 3, now)));
|
|
|
|
// Persist some fake commit log manifests.
|
|
persistManifestAndMutation(1, oneMinuteAgo);
|
|
persistManifestAndMutation(2, oneMinuteAgo);
|
|
persistManifestAndMutation(3, oneMinuteAgo);
|
|
persistManifestAndMutation(1, now);
|
|
persistManifestAndMutation(2, now);
|
|
persistManifestAndMutation(3, now);
|
|
|
|
task.run();
|
|
|
|
GcsFilename expectedFilename = new GcsFilename("gcs bucket", "commit_diff_until_" + now);
|
|
assertWithMessage("GCS file not found: " + expectedFilename)
|
|
.that(gcsService.getMetadata(expectedFilename)).isNotNull();
|
|
assertThat(gcsService.getMetadata(expectedFilename).getOptions().getUserMetadata())
|
|
.containsExactly(
|
|
LOWER_BOUND_CHECKPOINT,
|
|
oneMinuteAgo.toString(),
|
|
UPPER_BOUND_CHECKPOINT,
|
|
now.toString(),
|
|
NUM_TRANSACTIONS,
|
|
"6");
|
|
List<ImmutableObject> exported =
|
|
deserializeEntities(GcsTestingUtils.readGcsFile(gcsService, expectedFilename));
|
|
assertThat(exported.get(0)).isEqualTo(upperCheckpoint);
|
|
// We expect these manifests, in the order below, with matching mutations.
|
|
CommitLogManifest manifest1 = createManifest(1, oneMinuteAgo);
|
|
CommitLogManifest manifest2 = createManifest(2, oneMinuteAgo);
|
|
CommitLogManifest manifest3 = createManifest(3, oneMinuteAgo);
|
|
CommitLogManifest manifest4 = createManifest(1, now);
|
|
CommitLogManifest manifest5 = createManifest(2, now);
|
|
CommitLogManifest manifest6 = createManifest(3, now);
|
|
assertThat(exported).containsExactly(
|
|
upperCheckpoint,
|
|
manifest1,
|
|
createMutation(manifest1),
|
|
manifest2,
|
|
createMutation(manifest2),
|
|
manifest3,
|
|
createMutation(manifest3),
|
|
manifest4,
|
|
createMutation(manifest4),
|
|
manifest5,
|
|
createMutation(manifest5),
|
|
manifest6,
|
|
createMutation(manifest6))
|
|
.inOrder();
|
|
}
|
|
|
|
@Test
|
|
public void testRun_checkpointDiffWithNeverTouchedBuckets_exportsCorrectly() throws Exception {
|
|
task.lowerCheckpointTime = oneMinuteAgo;
|
|
task.upperCheckpointTime = now;
|
|
|
|
persistResource(CommitLogCheckpoint.create(
|
|
oneMinuteAgo,
|
|
ImmutableMap.of(1, START_OF_TIME, 2, START_OF_TIME, 3, START_OF_TIME)));
|
|
CommitLogCheckpoint upperCheckpoint = persistResource(CommitLogCheckpoint.create(
|
|
now,
|
|
ImmutableMap.of(1, START_OF_TIME, 2, START_OF_TIME, 3, START_OF_TIME)));
|
|
|
|
// Don't persist any commit log manifests; we're just checking that the task runs correctly
|
|
// even if the upper timestamp contains START_OF_TIME values.
|
|
|
|
task.run();
|
|
|
|
GcsFilename expectedFilename = new GcsFilename("gcs bucket", "commit_diff_until_" + now);
|
|
assertWithMessage("GCS file not found: " + expectedFilename)
|
|
.that(gcsService.getMetadata(expectedFilename)).isNotNull();
|
|
assertThat(gcsService.getMetadata(expectedFilename).getOptions().getUserMetadata())
|
|
.containsExactly(
|
|
LOWER_BOUND_CHECKPOINT,
|
|
oneMinuteAgo.toString(),
|
|
UPPER_BOUND_CHECKPOINT,
|
|
now.toString(),
|
|
NUM_TRANSACTIONS,
|
|
"0");
|
|
List<ImmutableObject> exported =
|
|
deserializeEntities(GcsTestingUtils.readGcsFile(gcsService, expectedFilename));
|
|
// We expect no manifests or mutations, only the upper checkpoint.
|
|
assertThat(exported).containsExactly(upperCheckpoint);
|
|
}
|
|
|
|
@Test
|
|
public void testRun_exportingFromStartOfTime_exportsAllCommits() throws Exception {
|
|
task.lowerCheckpointTime = START_OF_TIME;
|
|
task.upperCheckpointTime = now;
|
|
|
|
CommitLogCheckpoint upperCheckpoint = persistResource(CommitLogCheckpoint.create(
|
|
now,
|
|
ImmutableMap.of(1, now, 2, now, 3, now)));
|
|
|
|
// Persist some fake commit log manifests.
|
|
persistManifestAndMutation(1, START_OF_TIME.plusMillis(1)); // Oldest possible manifest time.
|
|
persistManifestAndMutation(2, oneMinuteAgo);
|
|
persistManifestAndMutation(3, now);
|
|
|
|
task.run();
|
|
|
|
GcsFilename expectedFilename = new GcsFilename("gcs bucket", "commit_diff_until_" + now);
|
|
assertWithMessage("GCS file not found: " + expectedFilename)
|
|
.that(gcsService.getMetadata(expectedFilename)).isNotNull();
|
|
assertThat(gcsService.getMetadata(expectedFilename).getOptions().getUserMetadata())
|
|
.containsExactly(
|
|
LOWER_BOUND_CHECKPOINT,
|
|
START_OF_TIME.toString(),
|
|
UPPER_BOUND_CHECKPOINT,
|
|
now.toString(),
|
|
NUM_TRANSACTIONS,
|
|
"3");
|
|
List<ImmutableObject> exported =
|
|
deserializeEntities(GcsTestingUtils.readGcsFile(gcsService, expectedFilename));
|
|
assertThat(exported.get(0)).isEqualTo(upperCheckpoint);
|
|
// We expect these manifests, in the order below, with matching mutations.
|
|
CommitLogManifest manifest1 = createManifest(1, START_OF_TIME.plusMillis(1));
|
|
CommitLogManifest manifest2 = createManifest(2, oneMinuteAgo);
|
|
CommitLogManifest manifest3 = createManifest(3, now);
|
|
assertThat(exported).containsExactly(
|
|
upperCheckpoint,
|
|
manifest1,
|
|
createMutation(manifest1),
|
|
manifest2,
|
|
createMutation(manifest2),
|
|
manifest3,
|
|
createMutation(manifest3))
|
|
.inOrder();
|
|
}
|
|
|
|
private CommitLogManifest createManifest(int bucketNum, DateTime commitTime) {
|
|
return CommitLogManifest.create(CommitLogBucket.getBucketKey(bucketNum), commitTime, null);
|
|
}
|
|
|
|
private CommitLogMutation createMutation(CommitLogManifest manifest) {
|
|
return CommitLogMutation.create(
|
|
Key.create(manifest),
|
|
TestObject.create(manifest.getCommitTime().toString()));
|
|
}
|
|
|
|
private void persistManifestAndMutation(int bucketNum, DateTime commitTime) {
|
|
persistResource(
|
|
createMutation(persistResource(createManifest(bucketNum, commitTime))));
|
|
}
|
|
}
|