// Copyright (C) 2017 The Android Open Source Project
//
// 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 com.googlesource.gerrit.plugins.automerger;

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

import java.util.Map;
import java.util.TreeMap;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

@RunWith(JUnit4.class)
public class FailedMergeExceptionTest {
  private Map<String, String> failedMergeBranches;
  private String currentRevision;
  private String hostName;
  private String projectName;
  private int changeNumber;
  private int patchsetNumber;
  private String topic;

  @Before
  public void setUp() throws Exception {
    failedMergeBranches = new TreeMap<String, String>();
    failedMergeBranches.put("branch1", "branch1 merge conflict");
    failedMergeBranches.put("branch2", "branch2 merge conflict");
    failedMergeBranches.put("branch3", "branch3 merge conflict");
    currentRevision = "d3adb33f";
    hostName = "livingbeef.example.com";
    projectName = "testproject";
    changeNumber = 15;
    patchsetNumber = 2;
    topic = "testtopic";
  }

  @Test
  public void conflictMessageTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "Merge conflict found on ${branch}",
            topic);
    String branch1String = "Merge conflict found on branch1";
    String branch2String = "Merge conflict found on branch2";
    String branch3String = "Merge conflict found on branch3";
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly(branch1String, branch2String, branch3String);
  }

  @Test
  public void customConflictMessageTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "asdf",
            topic);
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly("asdf", "asdf", "asdf");
  }

  @Test
  public void customConflictMessageWithRevisionSubstitutionTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "${branch} ${revision}",
            topic);
    String branch1String = "branch1 " + currentRevision;
    String branch2String = "branch2 " + currentRevision;
    String branch3String = "branch3 " + currentRevision;
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly(branch1String, branch2String, branch3String);
  }

  @Test
  public void customConflictMessageWithConflictSubstitutionTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "${branch}: ${conflict}",
            topic);
    String branch1String = "branch1: branch1 merge conflict";
    String branch2String = "branch2: branch2 merge conflict";
    String branch3String = "branch3: branch3 merge conflict";
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly(branch1String, branch2String, branch3String);
  }

  @Test
  public void customConflictMessageWithHostNameTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "${hostname}: ${conflict}",
            topic);
    String branch1String = hostName + ": branch1 merge conflict";
    String branch2String = hostName + ": branch2 merge conflict";
    String branch3String = hostName + ": branch3 merge conflict";
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly(branch1String, branch2String, branch3String);
  }

  @Test
  public void customConflictMessageWithProjectNameTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "${project}: ${conflict}",
            topic);
    String branch1String = projectName + ": branch1 merge conflict";
    String branch2String = projectName + ": branch2 merge conflict";
    String branch3String = projectName + ": branch3 merge conflict";
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly(branch1String, branch2String, branch3String);
  }

  @Test
  public void customConflictMessageWithChangeNumberTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "${changeNumber}: hooray",
            topic);
    String changeNumberString = "15: hooray";
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly(changeNumberString, changeNumberString, changeNumberString);
  }

  @Test
  public void customConflictMessageWithPatchsetNumberTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "${patchsetNumber}: hooray",
            topic);
    String patchsetNumberString = "2: hooray";
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly(patchsetNumberString, patchsetNumberString, patchsetNumberString);
  }

  @Test
  public void customConflictMessageWitTopicTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "i am ${topic}",
            topic);
    String conflictMessage = "i am testtopic";
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly(conflictMessage, conflictMessage, conflictMessage);
  }

  @Test
  public void customConflictMessageMultilineTest() throws Exception {
    FailedMergeException fme =
        new FailedMergeException(
            failedMergeBranches,
            currentRevision,
            hostName,
            projectName,
            changeNumber,
            patchsetNumber,
            "${branch}a\nb\nc\nd",
            topic);
    assertThat(fme.getDisplayString().split("\\n"))
        .asList()
        .containsExactly(
            "branch1a", "b", "c", "d", "branch2a", "b", "c", "d", "branch3a", "b", "c", "d");
  }
}