gerrit-acceptance-tests/src/test/java/com/google/gerrit/acceptance/rest/project/ProjectLevelConfigIT.java

// Copyright (C) 2013 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.acceptance.rest.project;

import static com.google.gerrit.acceptance.GitUtil.checkout;
import static com.google.gerrit.acceptance.GitUtil.cloneProject;
import static com.google.gerrit.acceptance.GitUtil.createProject;
import static com.google.gerrit.acceptance.GitUtil.fetch;
import static org.junit.Assert.assertEquals;

import com.google.gerrit.acceptance.AbstractDaemonTest;
import com.google.gerrit.acceptance.PushOneCommit;
import com.google.gerrit.acceptance.SshSession;
import com.google.gerrit.reviewdb.client.Project;
import com.google.gerrit.reviewdb.client.RefNames;
import com.google.gerrit.reviewdb.server.ReviewDb;
import com.google.gerrit.server.config.AllProjectsNameProvider;
import com.google.gerrit.server.project.ProjectCache;
import com.google.gerrit.server.project.ProjectState;
import com.google.gwtorm.server.SchemaFactory;
import com.google.inject.Inject;

import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.api.errors.GitAPIException;
import org.eclipse.jgit.lib.Config;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.io.IOException;

public class ProjectLevelConfigIT extends AbstractDaemonTest {

  @Inject
  private SchemaFactory<ReviewDb> reviewDbProvider;

  @Inject
  private ProjectCache projectCache;

  @Inject
  private AllProjectsNameProvider allProjects;

  @Inject
  private PushOneCommit.Factory pushFactory;

  private ReviewDb db;
  private SshSession sshSession;
  private String project;
  private Git git;

  @Before
  public void setUp() throws Exception {
    sshSession = new SshSession(server, admin);

    project = "p";
    createProject(sshSession, project, null, true);
    git = cloneProject(sshSession.getUrl() + "/" + project);
    fetch(git, RefNames.REFS_CONFIG + ":refs/heads/config");
    checkout(git, "refs/heads/config");

    db = reviewDbProvider.open();
  }

  @After
  public void cleanup() {
    db.close();
  }

  @Test
  public void accessProjectSpecificConfig() throws GitAPIException, IOException {
    String configName = "test.config";
    Config cfg = new Config();
    cfg.setString("s1", null, "k1", "v1");
    cfg.setString("s2", "ss", "k2", "v2");
    PushOneCommit push =
        pushFactory.create(db, admin.getIdent(), "Create Project Level Config",
            configName, cfg.toText());
    push.to(git, RefNames.REFS_CONFIG);

    ProjectState state = projectCache.get(new Project.NameKey(project));
    assertEquals(cfg.toText(), state.getConfig(configName).get().toText());
  }

  @Test
  public void nonExistingConfig() {
    ProjectState state = projectCache.get(new Project.NameKey(project));
    assertEquals("", state.getConfig("test.config").get().toText());
  }

  @Test
  public void withInheritance() throws GitAPIException, IOException {
    String configName = "test.config";

    Config parentCfg = new Config();
    parentCfg.setString("s1", null, "k1", "parentValue1");
    parentCfg.setString("s1", null, "k2", "parentValue2");
    parentCfg.setString("s2", "ss", "k3", "parentValue3");
    parentCfg.setString("s2", "ss", "k4", "parentValue4");

    Git parentGit =
        cloneProject(sshSession.getUrl() + "/" + allProjects.get().get(), false);
    fetch(parentGit, RefNames.REFS_CONFIG + ":refs/heads/config");
    checkout(parentGit, "refs/heads/config");
    PushOneCommit push =
        pushFactory.create(db, admin.getIdent(), "Create Project Level Config",
            configName, parentCfg.toText());
    push.to(parentGit, RefNames.REFS_CONFIG);

    Config cfg = new Config();
    cfg.setString("s1", null, "k1", "childValue1");
    cfg.setString("s2", "ss", "k3", "childValue2");
    push = pushFactory.create(db, admin.getIdent(), "Create Project Level Config",
        configName, cfg.toText());
    push.to(git, RefNames.REFS_CONFIG);

    ProjectState state = projectCache.get(new Project.NameKey(project));

    Config expectedCfg = new Config();
    expectedCfg.setString("s1", null, "k1", "childValue1");
    expectedCfg.setString("s1", null, "k2", "parentValue2");
    expectedCfg.setString("s2", "ss", "k3", "childValue2");
    expectedCfg.setString("s2", "ss", "k4", "parentValue4");

    assertEquals(expectedCfg.toText(), state.getConfig(configName)
        .getWithInheritance().toText());

    assertEquals(cfg.toText(), state.getConfig(configName).get().toText());
  }
}