google-nomulus/javatests/google/registry/tools/UpdateServerLocksCommandTest.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.tools;

import static google.registry.testing.JUnitBackports.assertThrows;

import com.beust.jcommander.ParameterException;
import org.junit.Test;

/** Unit tests for {@link UpdateServerLocksCommand}. */
public class UpdateServerLocksCommandTest extends EppToolCommandTestCase<UpdateServerLocksCommand> {

  @Test
  public void testSuccess_applyOne() throws Exception {
    runCommandForced("--client=NewRegistrar", "--registrar_request=true", "--reason=Test",
        "--domain_name=example.tld", "--apply=serverRenewProhibited");
    eppVerifier.verifySent("update_server_locks_apply_one.xml");
  }

  @Test
  public void testSuccess_multipleWordReason() throws Exception {
    runCommandForced("--client=NewRegistrar", "--registrar_request=false",
        "--reason=\"Test this\"", "--domain_name=example.tld", "--apply=serverRenewProhibited");
    eppVerifier.verifySent("update_server_locks_multiple_word_reason.xml");
  }

  @Test
  public void testSuccess_removeOne() throws Exception {
    runCommandForced("--client=NewRegistrar", "--registrar_request=true", "--reason=Test",
        "--domain_name=example.tld", "--remove=serverRenewProhibited");
    eppVerifier.verifySent("update_server_locks_remove_one.xml");
  }

  @Test
  public void testSuccess_applyAll() throws Exception {
    runCommandForced("--client=NewRegistrar", "--registrar_request=true", "--reason=Test",
        "--domain_name=example.tld", "--apply=all");
    eppVerifier.verifySent("update_server_locks_apply_all.xml");
  }

  @Test
  public void testSuccess_removeAll() throws Exception {
    runCommandForced("--client=NewRegistrar", "--registrar_request=true", "--reason=Test",
        "--domain_name=example.tld", "--remove=all");
    eppVerifier.verifySent("update_server_locks_remove_all.xml");
  }

  @Test
  public void testFailure_applyAllRemoveOne_failsDueToOverlap() throws Exception {
    assertThrows(
        IllegalArgumentException.class,
        () ->
            runCommandForced(
                "--client=NewRegistrar",
                "--registrar_request=true",
                "--reason=Test",
                "--domain_name=example.tld",
                "--apply=all",
                "--remove=serverRenewProhibited"));
  }

  @Test
  public void testFailure_illegalStatus() throws Exception {
    // The EPP status is a valid one by RFC, but invalid for this command.
    assertThrows(
        IllegalArgumentException.class,
        () ->
            runCommandForced(
                "--client=NewRegistrar",
                "--registrar_request=true",
                "--reason=Test",
                "--domain_name=example.tld",
                "--apply=clientRenewProhibited"));
  }

  @Test
  public void testFailure_unrecognizedStatus() throws Exception {
    // Handles a status passed to the command that doesn't correspond to any
    // EPP-valid status.
    assertThrows(
        IllegalArgumentException.class,
        () ->
            runCommandForced(
                "--client=NewRegistrar",
                "--registrar_request=true",
                "--reason=Test",
                "--domain_name=example.tld",
                "--apply=foo"));
  }

  @Test
  public void testFailure_mainParameter() throws Exception {
    assertThrows(
        ParameterException.class,
        () ->
            runCommandForced(
                "--client=NewRegistrar",
                "--registrar_request=true",
                "--reason=Test",
                "--domain_name=example.tld",
                "example2.tld",
                "--apply=serverRenewProhibited"));
  }

  @Test
  public void testFailure_noOp() throws Exception {
    assertThrows(
        IllegalArgumentException.class,
        () ->
            runCommandForced(
                "--client=NewRegistrar",
                "--domain_name=example.tld",
                "--apply=all",
                "--remove=serverRenewProhibited,serverTransferProhibited,"
                    + "serverDeleteProhibited,serverUpdateProhibited,serverHold",
                "--registrar_request=true",
                "--reason=Test"));
  }

  @Test
  public void testFailure_missingClientId() throws Exception {
    assertThrows(
        ParameterException.class,
        () ->
            runCommandForced(
                "--domain_name=example.tld",
                "--registrar_request=true",
                "--apply=serverRenewProhibited",
                "--reason=Test"));
  }

  @Test
  public void testFailure_unknownFlag() throws Exception {
    assertThrows(
        ParameterException.class,
        () ->
            runCommandForced(
                "--client=NewRegistrar",
                "--registrar_request=true",
                "--reason=Test",
                "--domain_name=example.tld",
                "--apply=serverRenewProhibited",
                "--foo=bar"));
  }

  @Test
  public void testFailure_noReasonWhenNotRegistrarRequested() throws Exception {
    assertThrows(
        IllegalArgumentException.class,
        () ->
            runCommandForced(
                "--client=NewRegistrar",
                "--registrar_request=false",
                "--domain_name=example.tld",
                "--apply=serverRenewProhibited"));
  }

  @Test
  public void testFailure_missingRegistrarRequest() throws Exception {
    assertThrows(
        ParameterException.class,
        () ->
            runCommandForced(
                "--client=NewRegistrar",
                "--reason=Test",
                "--domain_name=example.tld",
                "--apply=serverRenewProhibited"));
  }
}