// 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.rde;

import static com.google.common.truth.Truth.assertThat;
import static google.registry.testing.DatabaseHelper.createTld;
import static google.registry.xjc.XjcXmlTransformer.marshalStrict;
import static java.nio.charset.StandardCharsets.UTF_8;
import static org.junit.jupiter.api.Assertions.assertThrows;

import com.google.common.collect.ImmutableSet;
import com.google.common.net.InetAddresses;
import google.registry.model.domain.Domain;
import google.registry.model.eppcommon.StatusValue;
import google.registry.model.host.Host;
import google.registry.persistence.transaction.JpaTestExtensions;
import google.registry.persistence.transaction.JpaTestExtensions.JpaIntegrationTestExtension;
import google.registry.testing.DatabaseHelper;
import google.registry.xjc.host.XjcHostStatusType;
import google.registry.xjc.host.XjcHostStatusValueType;
import google.registry.xjc.rdehost.XjcRdeHost;
import google.registry.xjc.rdehost.XjcRdeHostElement;
import java.io.ByteArrayOutputStream;
import org.joda.time.DateTime;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;

/**
 * Unit tests for {@link HostToXjcConverter}.
 *
 * <p>This tests the mapping between {@link Host} and {@link XjcRdeHost} as well as some exceptional
 * conditions.
 */
public class HostToXjcConverterTest {

  @RegisterExtension
  final JpaIntegrationTestExtension jpa =
      new JpaTestExtensions.Builder().buildIntegrationTestExtension();

  @BeforeEach
  void beforeEach() {
    createTld("foobar");
  }

  @Test
  void testConvertSubordinateHost() {
    Domain domain =
        DatabaseHelper.newDomain("love.foobar")
            .asBuilder()
            .setPersistedCurrentSponsorRegistrarId("LeisureDog")
            .setLastTransferTime(DateTime.parse("2010-01-01T00:00:00Z"))
            .addStatusValue(StatusValue.PENDING_TRANSFER)
            .build();
    XjcRdeHost bean =
        HostToXjcConverter.convertSubordinateHost(
            new Host.Builder()
                .setCreationRegistrarId("LawyerCat")
                .setCreationTimeForTest(DateTime.parse("1900-01-01T00:00:00Z"))
                .setPersistedCurrentSponsorRegistrarId("BusinessCat")
                .setHostName("ns1.love.foobar")
                .setInetAddresses(ImmutableSet.of(InetAddresses.forString("127.0.0.1")))
                .setLastTransferTime(DateTime.parse("1910-01-01T00:00:00Z"))
                .setLastEppUpdateRegistrarId("CeilingCat")
                .setLastEppUpdateTime(DateTime.parse("1920-01-01T00:00:00Z"))
                .setRepoId("2-roid")
                .setStatusValues(ImmutableSet.of(StatusValue.OK))
                .setSuperordinateDomain(domain.createVKey())
                .build(),
            domain);

    assertThat(bean.getAddrs()).hasSize(1);
    assertThat(bean.getAddrs().get(0).getIp().value()).isEqualTo("v4");
    assertThat(bean.getAddrs().get(0).getValue()).isEqualTo("127.0.0.1");

    assertThat(bean.getCrDate()).isEqualTo(DateTime.parse("1900-01-01T00:00:00Z"));

    // o  A <crRr> element that contains the identifier of the registrar
    //    that created the domain name object.  An OPTIONAL client attribute
    //    is used to specify the client that performed the operation.
    //    This will always be null for us since we track each registrar as a separate client.
    assertThat(bean.getCrRr().getValue()).isEqualTo("LawyerCat");
    assertThat(bean.getCrRr().getClient()).isNull();

    assertThat(bean.getName()).isEqualTo("ns1.love.foobar");

    assertThat(bean.getRoid()).isEqualTo("2-roid");

    assertThat(bean.getStatuses()).hasSize(2);
    XjcHostStatusType status0 = bean.getStatuses().get(0);
    assertThat(status0.getS()).isEqualTo(XjcHostStatusValueType.OK);
    assertThat(status0.getValue()).isNull();
    assertThat(status0.getLang()).isEqualTo("en");

    assertThat(bean.getUpDate()).isEqualTo(DateTime.parse("1920-01-01T00:00:00Z"));

    assertThat(bean.getUpRr().getValue()).isEqualTo("CeilingCat");
    assertThat(bean.getUpRr().getClient()).isNull();

    // Values that should have been copied from the superordinate domain.
    assertThat(bean.getClID()).isEqualTo("LeisureDog");
    assertThat(bean.getTrDate()).isEqualTo(DateTime.parse("2010-01-01T00:00:00Z"));
    XjcHostStatusType status1 = bean.getStatuses().get(1);
    assertThat(status1.getS()).isEqualTo(XjcHostStatusValueType.PENDING_TRANSFER);
    assertThat(status1.getValue()).isNull();
    assertThat(status1.getLang()).isEqualTo("en");
  }

  @Test
  void testConvertExternalHost() {
    XjcRdeHost bean =
        HostToXjcConverter.convertExternalHost(
            new Host.Builder()
                .setCreationRegistrarId("LawyerCat")
                .setCreationTimeForTest(DateTime.parse("1900-01-01T00:00:00Z"))
                .setPersistedCurrentSponsorRegistrarId("BusinessCat")
                .setHostName("ns1.love.lol")
                .setInetAddresses(ImmutableSet.of(InetAddresses.forString("127.0.0.1")))
                .setLastTransferTime(DateTime.parse("1910-01-01T00:00:00Z"))
                .setLastEppUpdateRegistrarId("CeilingCat")
                .setLastEppUpdateTime(DateTime.parse("1920-01-01T00:00:00Z"))
                .setRepoId("2-roid")
                .setStatusValues(ImmutableSet.of(StatusValue.OK))
                .build());

    assertThat(bean.getAddrs()).hasSize(1);
    assertThat(bean.getAddrs().get(0).getIp().value()).isEqualTo("v4");
    assertThat(bean.getAddrs().get(0).getValue()).isEqualTo("127.0.0.1");

    assertThat(bean.getClID()).isEqualTo("BusinessCat");

    assertThat(bean.getCrDate()).isEqualTo(DateTime.parse("1900-01-01T00:00:00Z"));

    // o  A <crRr> element that contains the identifier of the registrar
    //    that created the domain name object.  An OPTIONAL client attribute
    //    is used to specify the client that performed the operation.
    //    This will always be null for us since we track each registrar as a separate client.
    assertThat(bean.getCrRr().getValue()).isEqualTo("LawyerCat");
    assertThat(bean.getCrRr().getClient()).isNull();

    assertThat(bean.getName()).isEqualTo("ns1.love.lol");

    assertThat(bean.getRoid()).isEqualTo("2-roid");

    assertThat(bean.getStatuses()).hasSize(1);
    assertThat(bean.getStatuses().get(0).getS()).isEqualTo(XjcHostStatusValueType.OK);
    assertThat(bean.getStatuses().get(0).getValue()).isNull();
    assertThat(bean.getStatuses().get(0).getLang()).isEqualTo("en");

    assertThat(bean.getTrDate()).isEqualTo(DateTime.parse("1910-01-01T00:00:00Z"));

    assertThat(bean.getUpDate()).isEqualTo(DateTime.parse("1920-01-01T00:00:00Z"));

    assertThat(bean.getUpRr().getValue()).isEqualTo("CeilingCat");
    assertThat(bean.getUpRr().getClient()).isNull();
  }

  @Test
  void testConvertExternalHost_ipv6() {
    XjcRdeHost bean =
        HostToXjcConverter.convertExternalHost(
            new Host.Builder()
                .setCreationRegistrarId("LawyerCat")
                .setCreationTimeForTest(DateTime.parse("1900-01-01T00:00:00Z"))
                .setPersistedCurrentSponsorRegistrarId("BusinessCat")
                .setHostName("ns1.love.lol")
                .setInetAddresses(ImmutableSet.of(InetAddresses.forString("cafe::abba")))
                .setLastTransferTime(DateTime.parse("1910-01-01T00:00:00Z"))
                .setLastEppUpdateRegistrarId("CeilingCat")
                .setLastEppUpdateTime(DateTime.parse("1920-01-01T00:00:00Z"))
                .setRepoId("2-LOL")
                .setStatusValues(ImmutableSet.of(StatusValue.OK))
                .build());
    assertThat(bean.getAddrs()).hasSize(1);
    assertThat(bean.getAddrs().get(0).getIp().value()).isEqualTo("v6");
    assertThat(bean.getAddrs().get(0).getValue()).isEqualTo("cafe::abba");
  }

  @Test
  void testHostStatusValueIsInvalid() {
    assertThrows(
        IllegalArgumentException.class,
        () ->
            HostToXjcConverter.convertExternalHost(
                new Host.Builder()
                    .setCreationRegistrarId("LawyerCat")
                    .setCreationTimeForTest(DateTime.parse("1900-01-01T00:00:00Z"))
                    .setPersistedCurrentSponsorRegistrarId("BusinessCat")
                    .setHostName("ns1.love.lol")
                    .setInetAddresses(ImmutableSet.of(InetAddresses.forString("cafe::abba")))
                    .setLastTransferTime(DateTime.parse("1910-01-01T00:00:00Z"))
                    .setLastEppUpdateRegistrarId("CeilingCat")
                    .setLastEppUpdateTime(DateTime.parse("1920-01-01T00:00:00Z"))
                    .setRepoId("2-LOL")
                    .setStatusValues(ImmutableSet.of(StatusValue.SERVER_HOLD)) // <-- OOPS
                    .build()));
  }

  @Test
  void testMarshal() throws Exception {
    // Bean! Bean! Bean!
    XjcRdeHostElement bean =
        HostToXjcConverter.convertExternal(
            new Host.Builder()
                .setCreationRegistrarId("LawyerCat")
                .setCreationTimeForTest(DateTime.parse("1900-01-01T00:00:00Z"))
                .setPersistedCurrentSponsorRegistrarId("BusinessCat")
                .setHostName("ns1.love.lol")
                .setInetAddresses(ImmutableSet.of(InetAddresses.forString("cafe::abba")))
                .setLastTransferTime(DateTime.parse("1910-01-01T00:00:00Z"))
                .setLastEppUpdateRegistrarId("CeilingCat")
                .setLastEppUpdateTime(DateTime.parse("1920-01-01T00:00:00Z"))
                .setRepoId("2-LOL")
                .setStatusValues(ImmutableSet.of(StatusValue.OK))
                .build());
    marshalStrict(bean, new ByteArrayOutputStream(), UTF_8);
  }
}
