google-nomulus/javatests/google/registry/tldconfig/idn/IdnTableTest.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.tldconfig.idn;

import static com.google.common.truth.Truth.assertThat;
import static google.registry.testing.JUnitBackports.assertThrows;

import com.google.common.collect.ImmutableList;
import java.net.URI;
import java.util.Optional;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

/** Unit tests for {@link IdnTable}. */
@RunWith(JUnit4.class)
public class IdnTableTest {
  @Test
  public void testDigits() {
    ImmutableList<String> of = ImmutableList.of(
        "# URL: https://love.example/lolcatattack.txt",
        "# Policy: https://love.example/policy.html",
        "U+0030",
        "U+0031",
        "U+0032",
        "U+0033",
        "U+0034",
        "U+0035",
        "U+0036",
        "U+0037",
        "U+0038",
        "U+0039");
    IdnTable idnTable = IdnTable.createFrom("lolcatattack", of, Optional.empty());
    assertThat(idnTable.isValidLabel("0123456789")).isTrue();
    assertThat(idnTable.isValidLabel("54321a")).isFalse();
    assertThat(idnTable.isValidLabel("AAA000")).isFalse();
  }

  @Test
  public void testIgnoreCommentAndEmptyLines() {
    IdnTable idnTable = IdnTable.createFrom("lolcatattack", ImmutableList.of(
        "# URL: https://love.example/lolcatattack.txt",
        "# Policy: https://love.example/policy.html",
        "U+0030",
        "#U+0031",
        "",
        "U+0032",
        "             ",
        "U+0033   # U+0031",
        "U+0034",
        "U+0035",
        "U+0036",
        "U+0037",
        "U+0038",
        "U+0039"), Optional.empty());
    assertThat(idnTable.isValidLabel("0123456789")).isFalse();
    assertThat(idnTable.isValidLabel("023456789")).isTrue();  // Works when you remove 1
  }

  @Test
  public void testSurrogates() {
    IdnTable idnTable =
        IdnTable.createFrom(
            "lolcatattack",
            ImmutableList.of(
                "# URL: https://love.example/lolcatattack.txt",
                "# Policy: https://love.example/policy.html",
                "U+0035",
                "U+0036",
                "U+0037",
                "U+2070E",
                "U+20731"),
            Optional.empty());
    assertThat(idnTable.getName()).isEqualTo("lolcatattack");
    assertThat(idnTable.isValidLabel("𠜎")).isTrue();
    assertThat(idnTable.isValidLabel("𠜱")).isTrue();
    assertThat(idnTable.isValidLabel("𠝹 ")).isFalse();
    assertThat(idnTable.isValidLabel("𠝹 0")).isFalse();
    assertThat(idnTable.isValidLabel("𠜎567𠜱")).isTrue();
  }

  @Test
  public void testSpecialComments_getParsed() {
    ImmutableList<String> of =
        ImmutableList.of(
            "# URL: https://love.example/lolcatattack.txt",
            "# Policy: https://love.example/policy.html");
    IdnTable idnTable =
        IdnTable.createFrom("lolcatattack", of, Optional.empty());
    assertThat(idnTable.getUrl()).isEqualTo(URI.create("https://love.example/lolcatattack.txt"));
    assertThat(idnTable.getPolicy()).isEqualTo(URI.create("https://love.example/policy.html"));
  }

  @Test
  public void testMissingUrl_throwsNpe() {
    ImmutableList<String> of = ImmutableList.of("# Policy: https://love.example/policy.html");
    NullPointerException thrown =
        assertThrows(
            NullPointerException.class, () -> IdnTable.createFrom("sloth", of, Optional.empty()));
    assertThat(thrown).hasMessageThat().contains("sloth missing '# URL:");
  }

  @Test
  public void testMissingPolicy_throwsNpe() {
    ImmutableList<String> of = ImmutableList.of("# URL: https://love.example/sloth.txt");
    NullPointerException thrown =
        assertThrows(
            NullPointerException.class, () -> IdnTable.createFrom("sloth", of, Optional.empty()));
    assertThat(thrown).hasMessageThat().contains("sloth missing '# Policy:");
  }
}