src/test/java/com/googlesource/gerrit/plugins/simplesubmitrules/config/ConfigTranslatorTest.java

// Copyright (C) 2018 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.simplesubmitrules.config;

import static com.google.common.truth.Truth.assertThat;
import static com.google.common.truth.Truth.assertWithMessage;
import static com.google.gerrit.testing.GerritJUnit.assertThrows;

import com.google.common.collect.ImmutableSet;
import com.google.gerrit.common.data.LabelType;
import com.google.gerrit.extensions.restapi.BadRequestException;
import com.googlesource.gerrit.plugins.simplesubmitrules.api.LabelDefinition;
import java.util.function.BiConsumer;
import java.util.function.Predicate;
import org.junit.Test;

public class ConfigTranslatorTest {
  @Test
  public void checkLabelTranslation() throws Exception {
    checkTranslation("copyMinScore", LabelType::isCopyMinScore, LabelType.Builder::setCopyMinScore);
    checkTranslation("copyMaxScore", LabelType::isCopyMaxScore, LabelType.Builder::setCopyMaxScore);

    checkTranslation(
        "copyAllScoresIfNoChange",
        LabelType::isCopyAllScoresIfNoChange,
        LabelType.Builder::setCopyAllScoresIfNoChange);

    checkTranslation(
        "copyAllScoresIfNoCodeChange",
        LabelType::isCopyAllScoresIfNoCodeChange,
        LabelType.Builder::setCopyAllScoresIfNoCodeChange);

    checkTranslation(
        "copyAllScoresOnMergeFirstParentUpdate",
        LabelType::isCopyAllScoresOnMergeFirstParentUpdate,
        LabelType.Builder::setCopyAllScoresOnMergeFirstParentUpdate);

    checkTranslation(
        "copyAllScoresOnTrivialRebase",
        LabelType::isCopyAllScoresOnTrivialRebase,
        LabelType.Builder::setCopyAllScoresOnTrivialRebase);
  }

  @Test
  public void checkDisallowedCopyScoreThrowsBadRequest() throws Exception {
    BadRequestException thrown =
        assertThrows(
            BadRequestException.class,
            () ->
                ConfigTranslator.applyCopyScoreRulesTo(
                    ImmutableSet.of(
                        "copyAllScoresIfNoChange", "copyAllScoresOnMergeFirstParentUpdate"),
                    ImmutableSet.of("copyAllScoresIfNoChange"),
                    LabelType.withDefaultValues("Verified").toBuilder()));
    assertThat(thrown)
        .hasMessageThat()
        .contains("copy score rules [copyAllScoresIfNoChange] are forbidden");
  }

  /** Helper method to check that conversion from/to Gerrit works for both true and false values */
  private static void checkTranslation(
      String copyScoreName,
      Predicate<LabelType> functionToCheck,
      BiConsumer<LabelType.Builder, Boolean> functionToSet)
      throws Exception {
    checkLabelToGerritPresent(copyScoreName, functionToCheck);
    checkLabelToGerritAbsent(copyScoreName, functionToCheck);

    checkLabelFromGerritPresent(copyScoreName, functionToSet);
    checkLabelFromGerritAbsent(copyScoreName, functionToSet);
  }

  private static void checkLabelFromGerritPresent(
      String copyScoreName, BiConsumer<LabelType.Builder, Boolean> functionToSet) {

    LabelType.Builder label = LabelType.withDefaultValues("Verified").toBuilder();
    LabelDefinition labelDefinition = new LabelDefinition();

    functionToSet.accept(label, false);
    ConfigTranslator.extractLabelCopyScoreRules(label.build(), labelDefinition);

    assertWithMessage("[case %s:false]", copyScoreName)
        .that(labelDefinition.copyScoreRules)
        .doesNotContain(copyScoreName);
  }

  private static void checkLabelFromGerritAbsent(
      String copyScoreName, BiConsumer<LabelType.Builder, Boolean> functionToSet) {

    LabelType.Builder label = LabelType.withDefaultValues("Verified").toBuilder();
    LabelDefinition labelDefinition = new LabelDefinition();

    functionToSet.accept(label, true);
    ConfigTranslator.extractLabelCopyScoreRules(label.build(), labelDefinition);

    assertWithMessage("[case %s:true]", copyScoreName)
        .that(labelDefinition.copyScoreRules)
        .contains(copyScoreName);
  }

  private static void checkLabelToGerritPresent(
      String copyScoreName, Predicate<LabelType> functionToCheck) throws Exception {
    LabelType.Builder label = LabelType.withDefaultValues("Verified").toBuilder();
    ConfigTranslator.applyCopyScoreRulesTo(
        ImmutableSet.of(copyScoreName), ImmutableSet.of(), label);
    assertWithMessage("[case %s:true]", copyScoreName)
        .that(functionToCheck.test(label.build()))
        .isTrue();
  }

  private static void checkLabelToGerritAbsent(
      String copyScoreName, Predicate<LabelType> functionToCheck) throws Exception {
    LabelType.Builder label = LabelType.withDefaultValues("Verified").toBuilder();

    ConfigTranslator.applyCopyScoreRulesTo(ImmutableSet.of(), ImmutableSet.of(), label);
    assertWithMessage("[case %s:false]", copyScoreName)
        .that(functionToCheck.test(label.build()))
        .isFalse();
  }
}