mirror of
https://github.com/google/nomulus.git
synced 2025-05-15 00:47:11 +02:00
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.
This commit is contained in:
Justine Tunney
parent
a41677aea1
commit
5012893c1d
2396 changed files with 0 additions and 0 deletions
|
|
@ -0,0 +1,240 @@
|
|||
// 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.tools;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
import static com.google.domain.registry.testing.DatastoreHelper.createTld;
|
||||
import static com.google.domain.registry.testing.DatastoreHelper.createTlds;
|
||||
import static com.google.domain.registry.testing.DatastoreHelper.generateNewContactHostRoid;
|
||||
import static com.google.domain.registry.testing.DatastoreHelper.persistResourceWithCommitLog;
|
||||
import static com.google.domain.registry.util.ResourceUtils.readResourceUtf8;
|
||||
import static java.nio.charset.StandardCharsets.UTF_8;
|
||||
import static java.util.Arrays.asList;
|
||||
|
||||
import com.google.common.collect.ImmutableMap;
|
||||
import com.google.common.collect.ImmutableSet;
|
||||
import com.google.common.net.InetAddresses;
|
||||
import com.google.domain.registry.config.TestRegistryConfig;
|
||||
import com.google.domain.registry.model.eppcommon.StatusValue;
|
||||
import com.google.domain.registry.model.host.HostResource;
|
||||
import com.google.domain.registry.model.ofy.Ofy;
|
||||
import com.google.domain.registry.rde.RdeCounter;
|
||||
import com.google.domain.registry.rde.RdeResourceType;
|
||||
import com.google.domain.registry.rde.RdeUtil;
|
||||
import com.google.domain.registry.testing.BouncyCastleProviderRule;
|
||||
import com.google.domain.registry.testing.FakeClock;
|
||||
import com.google.domain.registry.testing.InjectRule;
|
||||
import com.google.domain.registry.util.Idn;
|
||||
import com.google.domain.registry.xjc.XjcXmlTransformer;
|
||||
import com.google.domain.registry.xjc.rde.XjcRdeContentType;
|
||||
import com.google.domain.registry.xjc.rde.XjcRdeDeposit;
|
||||
import com.google.domain.registry.xjc.rde.XjcRdeDepositTypeType;
|
||||
import com.google.domain.registry.xjc.rdeheader.XjcRdeHeader;
|
||||
import com.google.domain.registry.xjc.rdeheader.XjcRdeHeaderCount;
|
||||
import com.google.domain.registry.xjc.rdehost.XjcRdeHost;
|
||||
import com.google.domain.registry.xjc.rderegistrar.XjcRdeRegistrar;
|
||||
import com.google.domain.registry.xml.XmlException;
|
||||
import com.google.domain.registry.xml.XmlTestUtils;
|
||||
|
||||
import com.beust.jcommander.ParameterException;
|
||||
|
||||
import org.joda.time.DateTime;
|
||||
import org.junit.Before;
|
||||
import org.junit.Rule;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Paths;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
import javax.xml.bind.JAXBElement;
|
||||
|
||||
/** Unit tests for {@link GenerateEscrowDepositCommand}. */
|
||||
public class GenerateEscrowDepositCommandTest
|
||||
extends CommandTestCase<GenerateEscrowDepositCommand> {
|
||||
|
||||
@Rule
|
||||
public final InjectRule inject = new InjectRule();
|
||||
|
||||
@Rule
|
||||
public final BouncyCastleProviderRule bouncy = new BouncyCastleProviderRule();
|
||||
|
||||
private final FakeClock clock = new FakeClock(DateTime.parse("2010-10-17T04:20:00Z"));
|
||||
private final List<? super XjcRdeContentType> alreadyExtracted = new ArrayList<>();
|
||||
|
||||
@Before
|
||||
public void before() throws Exception {
|
||||
inject.setStaticField(Ofy.class, "clock", clock);
|
||||
command.encryptor = EncryptEscrowDepositCommandTest.createEncryptor();
|
||||
command.counter = new RdeCounter();
|
||||
command.eppResourceIndexBucketCount = new TestRegistryConfig().getEppResourceIndexBucketCount();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRun_randomDomain_generatesXmlAndEncryptsItToo() throws Exception {
|
||||
createTld("xn--q9jyb4c");
|
||||
runCommand(
|
||||
"--outdir=" + tmpDir.getRoot(),
|
||||
"--tld=xn--q9jyb4c",
|
||||
"--watermark=" + clock.nowUtc().withTimeAtStartOfDay());
|
||||
assertThat(tmpDir.getRoot().list()).asList().containsExactly(
|
||||
"xn--q9jyb4c_2010-10-17_full_S1_R0.xml",
|
||||
"xn--q9jyb4c_2010-10-17_full_S1_R0-report.xml",
|
||||
"xn--q9jyb4c_2010-10-17_full_S1_R0.ryde",
|
||||
"xn--q9jyb4c_2010-10-17_full_S1_R0.sig",
|
||||
"xn--q9jyb4c.pub");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRun_missingTldName_fails() throws Exception {
|
||||
thrown.expect(ParameterException.class);
|
||||
runCommand(
|
||||
"--outdir=" + tmpDir.getRoot(),
|
||||
"--watermark=" + clock.nowUtc());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRun_nonexistentTld_fails() throws Exception {
|
||||
thrown.expect(IllegalArgumentException.class);
|
||||
runCommand(
|
||||
"--outdir=" + tmpDir.getRoot(),
|
||||
"--tld=xn--q9jyb4c",
|
||||
"--watermark=" + clock.nowUtc());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRun_oneHostNotDeletedOrFuture_producesValidDepositXml() throws Exception {
|
||||
createTlds("lol", "ussr", "xn--q9jyb4c");
|
||||
clock.setTo(DateTime.parse("1980-01-01TZ"));
|
||||
persistResourceWithCommitLog(
|
||||
newHostResource("communism.ussr", "dead:beef::cafe").asBuilder()
|
||||
.setDeletionTime(DateTime.parse("1991-12-25TZ")) // nope you're deleted
|
||||
.build());
|
||||
clock.setTo(DateTime.parse("1999-12-31TZ"));
|
||||
saveHostResource("ns1.cat.lol", "feed::a:bee"); // different tld doesn't matter
|
||||
clock.setTo(DateTime.parse("2020-12-31TZ"));
|
||||
saveHostResource("ns2.cat.lol", "a:fed::acab"); // nope you're from the future
|
||||
clock.setTo(DateTime.parse("2010-10-17TZ"));
|
||||
runCommand(
|
||||
"--outdir=" + tmpDir.getRoot(),
|
||||
"--tld=xn--q9jyb4c",
|
||||
"--watermark=" + clock.nowUtc());
|
||||
XmlTestUtils.assertXmlEquals(
|
||||
readResourceUtf8(getClass(), "testdata/xn--q9jyb4c_2010-10-17_full_S1_R0.xml"),
|
||||
new String(
|
||||
Files.readAllBytes(
|
||||
Paths.get(tmpDir.getRoot().toString(), "xn--q9jyb4c_2010-10-17_full_S1_R0.xml")),
|
||||
UTF_8),
|
||||
"deposit.contents.registrar.crDate",
|
||||
"deposit.contents.registrar.upDate");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRun_generatedXml_isSchemaValidAndHasStuffInIt() throws Exception {
|
||||
clock.setTo(DateTime.parse("1984-12-17TZ"));
|
||||
createTld("xn--q9jyb4c");
|
||||
saveHostResource("ns1.cat.lol", "feed::a:bee");
|
||||
clock.setTo(DateTime.parse("1984-12-18TZ"));
|
||||
runCommand(
|
||||
"--outdir=" + tmpDir.getRoot(),
|
||||
"--tld=xn--q9jyb4c",
|
||||
"--watermark=" + clock.nowUtc());
|
||||
XjcRdeDeposit deposit = (XjcRdeDeposit)
|
||||
unmarshal(Files.readAllBytes(
|
||||
Paths.get(tmpDir.getRoot().toString(), "xn--q9jyb4c_1984-12-18_full_S1_R0.xml")));
|
||||
assertThat(deposit.getType()).isEqualTo(XjcRdeDepositTypeType.FULL);
|
||||
assertThat(deposit.getId()).isEqualTo(RdeUtil.timestampToId(DateTime.parse("1984-12-18TZ")));
|
||||
assertThat(deposit.getWatermark()).isEqualTo(DateTime.parse("1984-12-18TZ"));
|
||||
XjcRdeRegistrar registrar1 = extractAndRemoveContentWithType(XjcRdeRegistrar.class, deposit);
|
||||
XjcRdeRegistrar registrar2 = extractAndRemoveContentWithType(XjcRdeRegistrar.class, deposit);
|
||||
XjcRdeHost host = extractAndRemoveContentWithType(XjcRdeHost.class, deposit);
|
||||
XjcRdeHeader header = extractAndRemoveContentWithType(XjcRdeHeader.class, deposit);
|
||||
assertThat(host.getName()).isEqualTo("ns1.cat.lol");
|
||||
assertThat(asList(registrar1.getName(), registrar2.getName()))
|
||||
.containsExactly("New Registrar", "The Registrar");
|
||||
assertThat(mapifyCounts(header)).containsEntry(RdeResourceType.HOST.getUri(), 1L);
|
||||
assertThat(mapifyCounts(header)).containsEntry(RdeResourceType.REGISTRAR.getUri(), 2L);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRun_thinBrdaDeposit_hostsGetExcluded() throws Exception {
|
||||
clock.setTo(DateTime.parse("1984-12-17TZ"));
|
||||
createTld("xn--q9jyb4c");
|
||||
saveHostResource("ns1.cat.lol", "feed::a:bee");
|
||||
clock.setTo(DateTime.parse("1984-12-18TZ"));
|
||||
runCommand(
|
||||
"--outdir=" + tmpDir.getRoot(),
|
||||
"--tld=xn--q9jyb4c",
|
||||
"--watermark=" + clock.nowUtc(),
|
||||
"--mode=THIN");
|
||||
XjcRdeDeposit deposit = (XjcRdeDeposit)
|
||||
unmarshal(Files.readAllBytes(
|
||||
Paths.get(tmpDir.getRoot().toString(), "xn--q9jyb4c_1984-12-18_thin_S1_R0.xml")));
|
||||
assertThat(deposit.getType()).isEqualTo(XjcRdeDepositTypeType.FULL);
|
||||
assertThat(deposit.getId()).isEqualTo(RdeUtil.timestampToId(DateTime.parse("1984-12-18TZ")));
|
||||
assertThat(deposit.getWatermark()).isEqualTo(DateTime.parse("1984-12-18TZ"));
|
||||
XjcRdeHeader header = extractAndRemoveContentWithType(XjcRdeHeader.class, deposit);
|
||||
assertThat(mapifyCounts(header)).doesNotContainKey(RdeResourceType.HOST.getUri());
|
||||
assertThat(mapifyCounts(header)).containsEntry(RdeResourceType.REGISTRAR.getUri(), 2L);
|
||||
}
|
||||
|
||||
private HostResource saveHostResource(String fqdn, String ip) {
|
||||
clock.advanceOneMilli();
|
||||
return persistResourceWithCommitLog(newHostResource(fqdn, ip));
|
||||
}
|
||||
|
||||
private HostResource newHostResource(String fqdn, String ip) {
|
||||
return new HostResource.Builder()
|
||||
.setRepoId(generateNewContactHostRoid())
|
||||
.setCreationClientId("LawyerCat")
|
||||
.setCreationTimeForTest(clock.nowUtc())
|
||||
.setCurrentSponsorClientId("BusinessCat")
|
||||
.setFullyQualifiedHostName(Idn.toASCII(fqdn))
|
||||
.setInetAddresses(ImmutableSet.of(InetAddresses.forString(ip)))
|
||||
.setLastTransferTime(DateTime.parse("1910-01-01T00:00:00Z"))
|
||||
.setLastEppUpdateClientId("CeilingCat")
|
||||
.setLastEppUpdateTime(clock.nowUtc())
|
||||
.setStatusValues(ImmutableSet.of(
|
||||
StatusValue.OK,
|
||||
StatusValue.PENDING_UPDATE))
|
||||
.build();
|
||||
}
|
||||
|
||||
public static Object unmarshal(byte[] xml) throws XmlException {
|
||||
return XjcXmlTransformer.unmarshal(new ByteArrayInputStream(xml));
|
||||
}
|
||||
|
||||
private static ImmutableMap<String, Long> mapifyCounts(XjcRdeHeader header) {
|
||||
ImmutableMap.Builder<String, Long> builder = new ImmutableMap.Builder<>();
|
||||
for (XjcRdeHeaderCount count : header.getCounts()) {
|
||||
builder.put(count.getUri(), count.getValue());
|
||||
}
|
||||
return builder.build();
|
||||
}
|
||||
|
||||
private <T extends XjcRdeContentType>
|
||||
T extractAndRemoveContentWithType(Class<T> type, XjcRdeDeposit deposit) {
|
||||
for (JAXBElement<? extends XjcRdeContentType> content : deposit.getContents().getContents()) {
|
||||
XjcRdeContentType piece = content.getValue();
|
||||
if (type.isInstance(piece) && !alreadyExtracted.contains(piece)) {
|
||||
alreadyExtracted.add(piece);
|
||||
return type.cast(piece);
|
||||
}
|
||||
}
|
||||
throw new AssertionError("Expected deposit to contain another " + type.getSimpleName());
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue