Add a Nomulus command to set domain create restricted status

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=150665065
This commit is contained in:
jianglai 2017-03-20 13:07:45 -07:00 committed by Ben McIlwain
parent 620d698479
commit a958d0a6c3
3 changed files with 43 additions and 0 deletions

View file

@ -204,6 +204,14 @@ abstract class CreateOrUpdateTldCommand extends MutatingCommand {
description = "Override restrictions on reserved list naming") description = "Override restrictions on reserved list naming")
boolean overrideReservedListRules; boolean overrideReservedListRules;
@Nullable
@Parameter(
names = {"--domain_create_restricted"},
description = "If only domains with nameserver restricted reservation can be created",
arity = 1
)
Boolean domainCreateRestricted;
@Nullable @Nullable
@Parameter( @Parameter(
names = "--claims_period_end", names = "--claims_period_end",
@ -365,6 +373,10 @@ abstract class CreateOrUpdateTldCommand extends MutatingCommand {
builder.setClaimsPeriodEnd(claimsPeriodEnd); builder.setClaimsPeriodEnd(claimsPeriodEnd);
} }
if (domainCreateRestricted != null) {
builder.setDomainCreateRestricted(domainCreateRestricted);
}
if (premiumListName != null) { if (premiumListName != null) {
if (premiumListName.isPresent()) { if (premiumListName.isPresent()) {
Optional<PremiumList> premiumList = PremiumList.get(premiumListName.get()); Optional<PremiumList> premiumList = PremiumList.get(premiumListName.get());

View file

@ -67,6 +67,7 @@ public class CreateTldCommandTest extends CommandTestCase<CreateTldCommand> {
.isEqualTo(Registry.DEFAULT_REDEMPTION_GRACE_PERIOD); .isEqualTo(Registry.DEFAULT_REDEMPTION_GRACE_PERIOD);
assertThat(registry.getPendingDeleteLength()).isEqualTo(Registry.DEFAULT_PENDING_DELETE_LENGTH); assertThat(registry.getPendingDeleteLength()).isEqualTo(Registry.DEFAULT_PENDING_DELETE_LENGTH);
assertThat(registry.getCreationTime()).isIn(Range.closed(before, after)); assertThat(registry.getCreationTime()).isIn(Range.closed(before, after));
assertThat(registry.getDomainCreateRestricted()).isFalse();
} }
@Test @Test
@ -324,6 +325,12 @@ public class CreateTldCommandTest extends CommandTestCase<CreateTldCommand> {
.containsExactly("ns1.example.com", "ns2.example.com"); .containsExactly("ns1.example.com", "ns2.example.com");
} }
@Test
public void testSuccess_setDomainCreateRestricted() throws Exception {
runCommandForced("--domain_create_restricted=true", "--roid_suffix=Q9JYB4C", "xn--q9jyb4c");
assertThat(Registry.get("xn--q9jyb4c").getDomainCreateRestricted()).isTrue();
}
@Test @Test
public void testSuccess_setCommonReservedListOnTld() throws Exception { public void testSuccess_setCommonReservedListOnTld() throws Exception {
runSuccessfulReservedListsTest("common_abuse"); runSuccessfulReservedListsTest("common_abuse");

View file

@ -407,6 +407,30 @@ public class UpdateTldCommandTest extends CommandTestCase<UpdateTldCommand> {
.containsExactly("ns2.example.com"); .containsExactly("ns2.example.com");
} }
@Test
public void testSuccess_setTldToDomainCreateRestricted() throws Exception {
persistResource(
Registry.get("xn--q9jyb4c").asBuilder().setDomainCreateRestricted(false).build());
runCommandForced("--domain_create_restricted=true", "xn--q9jyb4c");
assertThat(Registry.get("xn--q9jyb4c").getDomainCreateRestricted()).isTrue();
}
@Test
public void testSuccess_unsetTldToDomainCreateRestricted() throws Exception {
persistResource(
Registry.get("xn--q9jyb4c").asBuilder().setDomainCreateRestricted(true).build());
runCommandForced("--domain_create_restricted=false", "xn--q9jyb4c");
assertThat(Registry.get("xn--q9jyb4c").getDomainCreateRestricted()).isFalse();
}
@Test
public void testSuccess_leaveDomainCreateRestrictedStatusUnchanged() throws Exception {
persistResource(
Registry.get("xn--q9jyb4c").asBuilder().setDomainCreateRestricted(true).build());
runCommandForced("xn--q9jyb4c");
assertThat(Registry.get("xn--q9jyb4c").getDomainCreateRestricted()).isTrue();
}
@Test @Test
public void testSuccess_removeLrpPeriod() throws Exception { public void testSuccess_removeLrpPeriod() throws Exception {
persistResource( persistResource(