google-nomulus/javatests/google/registry/flows/FlowRunnerTest.java

// 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 google.registry.flows;

import static com.google.common.io.BaseEncoding.base64;
import static com.google.common.truth.Truth.assertThat;
import static google.registry.testing.TestDataHelper.loadFileWithSubstitutions;
import static java.nio.charset.StandardCharsets.UTF_8;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import com.google.appengine.api.users.User;
import com.google.common.base.Joiner;
import com.google.common.base.Optional;
import com.google.common.base.Predicate;
import com.google.common.base.Splitter;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Iterables;
import com.google.common.testing.TestLogHandler;
import google.registry.model.eppcommon.Trid;
import google.registry.model.eppinput.EppInput;
import google.registry.model.eppoutput.EppOutput;
import google.registry.model.eppoutput.EppResponse;
import google.registry.monitoring.whitebox.EppMetric;
import google.registry.testing.AppEngineRule;
import google.registry.testing.FakeClock;
import google.registry.testing.FakeHttpSession;
import google.registry.testing.Providers;
import google.registry.testing.ShardableTestCase;
import java.util.List;
import java.util.Map;
import java.util.logging.LogRecord;
import java.util.logging.Logger;
import org.joda.time.DateTime;
import org.json.simple.JSONValue;
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 FlowRunner}. */
@RunWith(JUnit4.class)
public class FlowRunnerTest extends ShardableTestCase {

  @Rule
  public final AppEngineRule appEngineRule = new AppEngineRule.Builder().build();

  private final FlowRunner flowRunner = new FlowRunner();

  private final TestLogHandler handler = new TestLogHandler();

  @Before
  public void before() {
    Logger.getLogger(FlowRunner.class.getCanonicalName()).addHandler(handler);

    final EppOutput eppOutput = mock(EppOutput.class);
    EppResponse eppResponse = mock(EppResponse.class);
    when(eppResponse.getCreatedRepoId()).thenReturn("foo");
    when(eppResponse.getExecutionTime()).thenReturn(new DateTime(1337));
    when(eppOutput.getResponse()).thenReturn(eppResponse);

    flowRunner.clientId = "TheRegistrar";
    flowRunner.clock = new FakeClock();
    flowRunner.credentials = new PasswordOnlyTransportCredentials();
    flowRunner.eppInput = new EppInput();
    flowRunner.eppRequestSource = EppRequestSource.UNIT_TEST;
    flowRunner.flowProvider =
        Providers.<Flow>of(
            new Flow() {
              @Override
              protected EppOutput run() {
                return eppOutput;
              }
            });
    flowRunner.inputXmlBytes = "<xml/>".getBytes(UTF_8);
    flowRunner.isDryRun = false;
    flowRunner.isSuperuser = false;
    flowRunner.isTransactional = false;
    flowRunner.metric = new EppMetric.Builder();
    flowRunner.sessionMetadata =
        new StatelessRequestSessionMetadata("TheRegistrar", ImmutableSet.<String>of());
    flowRunner.trid = Trid.create("client-123", "server-456");
  }

  @Test
  public void testRun_reportingLogStatement_basic() throws Exception {
    flowRunner.run();
    assertThat(parseJsonMap(findLogMessageByPrefix(handler, "EPP-REPORTING-LOG-SIGNATURE: ")))
        .containsExactly(
              "trid", "server-456",
              "clientId", "TheRegistrar",
              "xml", "<?xml version=\"1.0\" encoding=\"UTF-8\"?><xml/>\n",
              // Base64-encoding of "<xml/>":
              "xmlBytes", "PHhtbC8+");
  }

  @Test
  public void testRun_notIsTransactional_incrementsMetricAttempts() throws Exception {
    flowRunner.run();
    assertThat(flowRunner.metric.build().getAttempts()).isEqualTo(1);
  }

  @Test
  public void testRun_isTransactional_incrementsMetricAttempts() throws Exception {
    flowRunner.isTransactional = true;
    flowRunner.run();
    assertThat(flowRunner.metric.build().getAttempts()).isEqualTo(1);
  }

  @Test
  public void testRun_reportingLogStatement_noClientId() throws Exception {
    flowRunner.clientId = null;
    flowRunner.run();
    assertThat(parseJsonMap(findLogMessageByPrefix(handler, "EPP-REPORTING-LOG-SIGNATURE: ")))
        .containsExactly(
              "trid", "server-456",
              "clientId", "",
              "xml", "<?xml version=\"1.0\" encoding=\"UTF-8\"?><xml/>\n",
              // Base64-encoding of "<xml/>":
              "xmlBytes", "PHhtbC8+");
  }

  @Test
  public void testRun_reportingLogStatement_complexEppInput() throws Exception {
    String domainCreateXml = loadFileWithSubstitutions(
        getClass(), "domain_create_prettyprinted.xml", ImmutableMap.<String, String>of());
    flowRunner.inputXmlBytes = domainCreateXml.getBytes(UTF_8);
    flowRunner.run();
    assertThat(parseJsonMap(findLogMessageByPrefix(handler, "EPP-REPORTING-LOG-SIGNATURE: ")))
        .containsExactly(
              "trid", "server-456",
              "clientId", "TheRegistrar",
              "xml", domainCreateXml,
              "xmlBytes", base64().encode(domainCreateXml.getBytes(UTF_8)));
  }

  @Test
  public void testRun_legacyLoggingStatement_basic() throws Exception {
    flowRunner.run();
    assertThat(Splitter.on("\n\t").split(findLogMessageByPrefix(handler, "EPP Command\n\t")))
        .containsExactly(
            "server-456",
            "TheRegistrar",
            "StatelessRequestSessionMetadata"
                + "{clientId=TheRegistrar, failedLoginAttempts=0, serviceExtensionUris=}",
            "<?xml version=\"1.0\" encoding=\"UTF-8\"?><xml/>",
            "",  // Extra newline at the end of the XML.
            "PasswordOnlyTransportCredentials{}",
            "UNIT_TEST",
            "LIVE")
        .inOrder();
  }

  @Test
  public void testRun_legacyLoggingStatement_httpSessionMetadata() throws Exception {
    flowRunner.sessionMetadata = new HttpSessionMetadata(new FakeHttpSession());
    flowRunner.sessionMetadata.setClientId("TheRegistrar");
    flowRunner.run();
    assertThat(Splitter.on("\n\t").split(findLogMessageByPrefix(handler, "EPP Command\n\t")))
        .contains(
            "HttpSessionMetadata"
                + "{clientId=TheRegistrar, failedLoginAttempts=0, serviceExtensionUris=}");
  }

  @Test
  public void testRun_legacyLoggingStatement_gaeUserCredentials() throws Exception {
    flowRunner.credentials = new GaeUserCredentials(new User("user@example.com", "authDomain"));
    flowRunner.run();
    assertThat(Splitter.on("\n\t").split(findLogMessageByPrefix(handler, "EPP Command\n\t")))
        .contains("GaeUserCredentials{gaeUser=user@example.com}");
  }

  @Test
  public void testRun_legacyLoggingStatement_tlsCredentials() throws Exception {
    flowRunner.credentials = new TlsCredentials("abc123def", Optional.of("127.0.0.1"), "sni");
    flowRunner.run();
    assertThat(Splitter.on("\n\t").split(findLogMessageByPrefix(handler, "EPP Command\n\t")))
        .contains(
            "TlsCredentials{clientCertificateHash=abc123def, clientAddress=/127.0.0.1, sni=sni}");
  }

  @Test
  public void testRun_legacyLoggingStatement_dryRun() throws Exception {
    flowRunner.isDryRun = true;
    flowRunner.run();
    assertThat(Splitter.on("\n\t").split(findLogMessageByPrefix(handler, "EPP Command\n\t")))
        .contains("DRY_RUN");
  }

  @Test
  public void testRun_legacyLoggingStatement_complexEppInput() throws Exception {
    String domainCreateXml = loadFileWithSubstitutions(
        getClass(), "domain_create_prettyprinted.xml", ImmutableMap.<String, String>of());
    flowRunner.inputXmlBytes = domainCreateXml.getBytes(UTF_8);
    flowRunner.run();
    String logMessage = findLogMessageByPrefix(handler, "EPP Command\n\t");
    List<String> lines = Splitter.on("\n\t").splitToList(logMessage);
    assertThat(lines.size()).named("number of lines in log message").isAtLeast(9);
    String xml = Joiner.on('\n').join(lines.subList(3, lines.size() - 3));
    assertThat(xml).isEqualTo(domainCreateXml);
  }

  @SuppressWarnings("unchecked")
  private static Map<String, Object> parseJsonMap(String json) throws Exception {
    return (Map<String, Object>) JSONValue.parseWithException(json);
  }

  /**
   * Find the first log message stored in the handler that has the provided prefix, and return
   * that message with the prefix stripped off.
   */
  private static String findLogMessageByPrefix(TestLogHandler handler, final String prefix) {
    return Iterables.find(
            handler.getStoredLogRecords(),
            new Predicate<LogRecord>() {
              @Override
              public boolean apply(LogRecord logRecord) {
                return logRecord.getMessage().startsWith(prefix);
              }
            })
        .getMessage()
        .replaceFirst("^" + prefix, "");
  }
}