gerrit-server/src/test/java/com/google/gerrit/rules/GerritCommonTest.java

// Copyright (C) 2011 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.rules;

import com.google.gerrit.common.data.ApprovalType;
import com.google.gerrit.common.data.ApprovalTypes;
import com.google.gerrit.reviewdb.client.ApprovalCategory;
import com.google.gerrit.reviewdb.client.ApprovalCategoryValue;
import com.google.inject.AbstractModule;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class GerritCommonTest extends PrologTestCase {
  @Override
  public void setUp() throws Exception {
    super.setUp();

    final ApprovalTypes types = new ApprovalTypes(Arrays.asList(
        codeReviewCategory(),
        verifiedCategory()
    ));

    load("gerrit", "gerrit_common_test.pl", new AbstractModule() {
      @Override
      protected void configure() {
        bind(ApprovalTypes.class).toInstance(types);
      }
    });
  }

  private static ApprovalType codeReviewCategory() {
    ApprovalCategory cat = category(0, "CRVW", "Code Review");
    List<ApprovalCategoryValue> vals = newList();
    vals.add(value(cat, 2, "Looks good to me, approved"));
    vals.add(value(cat, 1, "Looks good to me, but someone else must approve"));
    vals.add(value(cat, 0, "No score"));
    vals.add(value(cat, -1, "I would prefer that you didn't submit this"));
    vals.add(value(cat, -2, "Do not submit"));
    return new ApprovalType(cat, vals);
  }

  private static ApprovalType verifiedCategory() {
    ApprovalCategory cat = category(1, "VRIF", "Verified");
    List<ApprovalCategoryValue> vals = newList();
    vals.add(value(cat, 1, "Verified"));
    vals.add(value(cat, 0, "No score"));
    vals.add(value(cat, -1, "Fails"));
    return new ApprovalType(cat, vals);
  }

  private static ApprovalCategory category(int pos, String id, String name) {
    ApprovalCategory cat;
    cat = new ApprovalCategory(new ApprovalCategory.Id(id), name);
    cat.setPosition((short) pos);
    return cat;
  }

  private static ArrayList<ApprovalCategoryValue> newList() {
    return new ArrayList<ApprovalCategoryValue>();
  }

  private static ApprovalCategoryValue value(ApprovalCategory c, int v, String n) {
    return new ApprovalCategoryValue(
        new ApprovalCategoryValue.Id(c.getId(), (short) v),
        n);
  }
}