mirror of
https://github.com/google/nomulus.git
synced 2025-05-20 11:19:35 +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:
parent
a41677aea1
commit
5012893c1d
2396 changed files with 0 additions and 0 deletions
192
javatests/google/registry/tools/AllocateDomainCommandTest.java
Normal file
192
javatests/google/registry/tools/AllocateDomainCommandTest.java
Normal file
|
@ -0,0 +1,192 @@
|
|||
// 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.io.BaseEncoding.base16;
|
||||
import static com.google.common.io.Resources.getResource;
|
||||
import static com.google.common.io.Resources.toByteArray;
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
import static com.google.domain.registry.flows.EppServletUtils.APPLICATION_EPP_XML_UTF8;
|
||||
import static com.google.domain.registry.flows.FlowRegistry.getFlowClass;
|
||||
import static com.google.domain.registry.model.domain.DesignatedContact.Type.ADMIN;
|
||||
import static com.google.domain.registry.model.domain.DesignatedContact.Type.BILLING;
|
||||
import static com.google.domain.registry.model.domain.DesignatedContact.Type.TECH;
|
||||
import static com.google.domain.registry.model.domain.launch.ApplicationStatus.VALIDATED;
|
||||
import static com.google.domain.registry.model.registry.Registry.TldState.QUIET_PERIOD;
|
||||
import static com.google.domain.registry.testing.DatastoreHelper.createTld;
|
||||
import static com.google.domain.registry.testing.DatastoreHelper.newDomainApplication;
|
||||
import static com.google.domain.registry.testing.DatastoreHelper.persistActiveContact;
|
||||
import static com.google.domain.registry.testing.DatastoreHelper.persistActiveHost;
|
||||
import static com.google.domain.registry.testing.DatastoreHelper.persistResource;
|
||||
import static com.google.domain.registry.util.DateTimeUtils.START_OF_TIME;
|
||||
import static com.google.domain.registry.util.ResourceUtils.readResourceUtf8;
|
||||
import static com.google.domain.registry.xml.XmlTestUtils.assertXmlEquals;
|
||||
import static java.nio.charset.StandardCharsets.UTF_8;
|
||||
import static org.mockito.Matchers.eq;
|
||||
import static org.mockito.Mockito.times;
|
||||
import static org.mockito.Mockito.verify;
|
||||
|
||||
import com.google.common.collect.ImmutableMap;
|
||||
import com.google.common.collect.ImmutableSet;
|
||||
import com.google.domain.registry.flows.EppXmlTransformer;
|
||||
import com.google.domain.registry.flows.domain.DomainAllocateFlow;
|
||||
import com.google.domain.registry.model.domain.DesignatedContact;
|
||||
import com.google.domain.registry.model.domain.DomainApplication;
|
||||
import com.google.domain.registry.model.domain.ReferenceUnion;
|
||||
import com.google.domain.registry.model.domain.launch.LaunchNotice;
|
||||
import com.google.domain.registry.model.domain.secdns.DelegationSignerData;
|
||||
import com.google.domain.registry.model.eppcommon.Trid;
|
||||
import com.google.domain.registry.model.eppinput.EppInput;
|
||||
import com.google.domain.registry.model.reporting.HistoryEntry;
|
||||
import com.google.domain.registry.tools.ServerSideCommand.Connection;
|
||||
|
||||
import com.beust.jcommander.ParameterException;
|
||||
|
||||
import org.joda.time.DateTime;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.mockito.ArgumentCaptor;
|
||||
import org.mockito.Captor;
|
||||
import org.mockito.Mock;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.List;
|
||||
|
||||
/** Unit tests for {@link AllocateDomainCommand}. */
|
||||
public class AllocateDomainCommandTest extends CommandTestCase<AllocateDomainCommand> {
|
||||
|
||||
private static final String EXPECTED_XML_ONE =
|
||||
readResourceUtf8(AllocateDomainCommandTest.class, "testdata/allocate_domain.xml");
|
||||
private static final String EXPECTED_XML_TWO =
|
||||
readResourceUtf8(AllocateDomainCommandTest.class, "testdata/allocate_domain2.xml");
|
||||
|
||||
@Mock
|
||||
Connection connection;
|
||||
|
||||
@Captor
|
||||
ArgumentCaptor<byte[]> xml;
|
||||
|
||||
@Before
|
||||
public void init() throws IOException {
|
||||
command.setConnection(connection);
|
||||
createTld("tld", QUIET_PERIOD);
|
||||
createApplication("example-one.tld", "testdata/domain_create_sunrush.xml", "1-TLD");
|
||||
createApplication("example-two.tld", "testdata/domain_create_sunrush2.xml", "2-TLD");
|
||||
}
|
||||
|
||||
private void createApplication(String name, String xmlFile, String repoId) throws IOException {
|
||||
DomainApplication application =
|
||||
persistResource(newDomainApplication(name)
|
||||
.asBuilder()
|
||||
.setRepoId(repoId)
|
||||
.setCreationTimeForTest(START_OF_TIME)
|
||||
.setRegistrant(ReferenceUnion.create(persistActiveContact("registrant")))
|
||||
.setContacts(ImmutableSet.of(
|
||||
DesignatedContact.create(
|
||||
ADMIN,
|
||||
ReferenceUnion.create(persistActiveContact("adminContact"))),
|
||||
DesignatedContact.create(
|
||||
BILLING,
|
||||
ReferenceUnion.create(persistActiveContact("billingContact"))),
|
||||
DesignatedContact.create(
|
||||
TECH,
|
||||
ReferenceUnion.create(persistActiveContact("techContact")))))
|
||||
.setNameservers(ImmutableSet.of(
|
||||
ReferenceUnion.create(persistActiveHost("ns1.example.com")),
|
||||
ReferenceUnion.create(persistActiveHost("ns2.example.com"))))
|
||||
.setApplicationStatus(VALIDATED)
|
||||
.setDsData(ImmutableSet.of(
|
||||
DelegationSignerData.create(
|
||||
12345, 3, 1, base16().decode("49FD46E6C4B45C55D4AC")),
|
||||
DelegationSignerData.create(
|
||||
56789, 2, 4, base16().decode("69FD46E6C4A45C55D4AC"))))
|
||||
.setLaunchNotice(LaunchNotice.create(
|
||||
"370d0b7c9223372036854775807",
|
||||
"tmch",
|
||||
DateTime.parse("2010-08-16T09:00:00.0Z"),
|
||||
DateTime.parse("2009-08-16T09:00:00.0Z")))
|
||||
.build());
|
||||
|
||||
persistResource(
|
||||
new HistoryEntry.Builder()
|
||||
.setParent(application)
|
||||
.setClientId("NewRegistrar")
|
||||
.setModificationTime(application.getCreationTime())
|
||||
.setTrid(Trid.create("ABC-123"))
|
||||
.setXmlBytes(toByteArray(getResource(AllocateDomainCommandTest.class, xmlFile)))
|
||||
.build());
|
||||
}
|
||||
|
||||
private void verifySent(boolean dryRun, String clientId, String... expectedXml) throws Exception {
|
||||
ImmutableMap<String, ?> params = ImmutableMap.of(
|
||||
"dryRun", dryRun,
|
||||
"clientIdentifier", clientId,
|
||||
"superuser", true);
|
||||
verify(connection, times(expectedXml.length))
|
||||
.send(eq("/_dr/epptool"), eq(params), eq(APPLICATION_EPP_XML_UTF8), xml.capture());
|
||||
|
||||
List<byte[]> allCapturedXml = xml.getAllValues();
|
||||
assertThat(allCapturedXml).hasSize(expectedXml.length);
|
||||
int capturedXmlIndex = 0;
|
||||
for (String expected : expectedXml) {
|
||||
assertXmlEquals(expected, new String(allCapturedXml.get(capturedXmlIndex++), UTF_8));
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSuccess() throws Exception {
|
||||
runCommand("--ids=1-TLD", "--force", "--superuser");
|
||||
// NB: These commands are all sent on behalf of the sponsoring registrar, in this case
|
||||
// "TheRegistrar".
|
||||
verifySent(false, "TheRegistrar", EXPECTED_XML_ONE);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSuccess_multiple() throws Exception {
|
||||
runCommand("--ids=1-TLD,2-TLD", "--force", "--superuser");
|
||||
verifySent(false, "TheRegistrar", EXPECTED_XML_ONE, EXPECTED_XML_TWO);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSuccess_dryRun() throws Exception {
|
||||
runCommand("--ids=1-TLD", "--dry_run", "--superuser");
|
||||
verifySent(true, "TheRegistrar", EXPECTED_XML_ONE);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testFailure_notAsSuperuser() throws Exception {
|
||||
thrown.expect(IllegalArgumentException.class);
|
||||
runCommand("--ids=1-TLD", "--force");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testFailure_forceAndDryRunIncompatible() throws Exception {
|
||||
thrown.expect(IllegalArgumentException.class);
|
||||
runCommand("--ids=1-TLD", "--force", "--dry_run", "--superuser");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testFailure_unknownFlag() throws Exception {
|
||||
thrown.expect(ParameterException.class);
|
||||
runCommand("--ids=1-TLD", "--force", "--unrecognized=foo", "--superuser");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testXmlInstantiatesFlow() throws Exception {
|
||||
assertThat(
|
||||
getFlowClass(EppXmlTransformer.<EppInput>unmarshal(EXPECTED_XML_ONE.getBytes(UTF_8))))
|
||||
.isEqualTo(DomainAllocateFlow.class);
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue