java/com/google/gerrit/plugins/checks/acceptance/testsuite/TestCheckerUpdate.java

// Copyright (C) 2019 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.google.gerrit.plugins.checks.acceptance.testsuite;

import static com.google.common.collect.ImmutableSortedSet.toImmutableSortedSet;
import static java.util.Comparator.naturalOrder;

import com.google.auto.value.AutoValue;
import com.google.common.collect.ImmutableSortedSet;
import com.google.gerrit.acceptance.testsuite.ThrowingConsumer;
import com.google.gerrit.plugins.checks.api.BlockingCondition;
import com.google.gerrit.plugins.checks.api.CheckerStatus;
import com.google.gerrit.reviewdb.client.Project;
import java.util.Arrays;
import java.util.Optional;
import java.util.stream.Stream;

@AutoValue
public abstract class TestCheckerUpdate {
  public abstract Optional<String> name();

  public abstract Optional<String> description();

  public abstract Optional<String> url();

  public abstract Optional<Project.NameKey> repository();

  public abstract Optional<CheckerStatus> status();

  public abstract Optional<ImmutableSortedSet<BlockingCondition>> blockingConditions();

  public abstract Optional<String> query();

  public abstract boolean forceInvalidConfig();

  public abstract boolean forceInvalidBlockingCondition();

  public abstract boolean deleteRef();

  abstract ThrowingConsumer<TestCheckerUpdate> checkerUpdater();

  public static Builder builder(ThrowingConsumer<TestCheckerUpdate> checkerUpdater) {
    return new AutoValue_TestCheckerUpdate.Builder()
        .checkerUpdater(checkerUpdater)
        .forceInvalidConfig(false)
        .forceInvalidBlockingCondition(false)
        .deleteRef(false);
  }

  @AutoValue.Builder
  public abstract static class Builder {

    public abstract Builder name(String name);

    public abstract Builder description(String description);

    public Builder clearDescription() {
      return description("");
    }

    public abstract Builder url(String url);

    public Builder clearUrl() {
      return url("");
    }

    public abstract Builder repository(Project.NameKey repository);

    public abstract Builder status(CheckerStatus status);

    public Builder enable() {
      return status(CheckerStatus.ENABLED);
    }

    public Builder disable() {
      return status(CheckerStatus.DISABLED);
    }

    public Builder forceInvalidConfig() {
      return forceInvalidConfig(true);
    }

    abstract Builder forceInvalidConfig(boolean forceInvalidConfig);

    public Builder forceInvalidBlockingCondition() {
      return forceInvalidBlockingCondition(true);
    }

    abstract Builder forceInvalidBlockingCondition(boolean forceInvalidBlockingCondition);

    public Builder deleteRef() {
      return deleteRef(true);
    }

    abstract Builder deleteRef(boolean deleteRef);

    public abstract Builder blockingConditions(
        ImmutableSortedSet<BlockingCondition> blockingConditions);

    public Builder blockingConditions(BlockingCondition first, BlockingCondition... rest) {
      return blockingConditions(
          Stream.concat(Stream.of(first), Arrays.stream(rest))
              .collect(toImmutableSortedSet(naturalOrder())));
    }

    public Builder clearBlockingConditions() {
      return blockingConditions(ImmutableSortedSet.of());
    }

    public abstract Builder query(String query);

    public Builder clearQuery() {
      return query("");
    }

    abstract Builder checkerUpdater(ThrowingConsumer<TestCheckerUpdate> checkerUpdater);

    abstract TestCheckerUpdate autoBuild();

    /** Executes the checker update as specified. */
    public void update() {
      TestCheckerUpdate checkerUpdater = autoBuild();
      checkerUpdater.checkerUpdater().acceptAndThrowSilently(checkerUpdater);
    }
  }
}