TestGitRepo.java

/* Copyright 2014 Google Inc. All rights reserved.
 *
 *  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 hudson.plugins.git;

import org.eclipse.jgit.lib.PersonIdent;
import org.jenkinsci.plugins.gitclient.Git;
import org.jenkinsci.plugins.gitclient.GitClient;
import org.jvnet.hudson.test.HudsonTestCase;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import hudson.EnvVars;
import hudson.FilePath;
import hudson.model.TaskListener;
import hudson.model.User;
import hudson.model.UserProperty;
import hudson.tasks.Mailer;

/**
 * Since test classes aren't included in maven dependency resolution, these test classes are
 * copied from https://github.com/jenkinsci/git-plugin/
 */
public class TestGitRepo {
  protected String name; // The name of this repository.
  protected TaskListener listener;

  /**
   * This is where the commit commands create a Git repository.
   */
  public File gitDir; // was "workDir"
  public FilePath gitDirPath; // was "workspace"
  public GitClient git;

  private EnvVars envVars;

  public final PersonIdent johnDoe = new PersonIdent("John Doe", "john@doe.com");
  public final PersonIdent janeDoe = new PersonIdent("Jane Doe", "jane@doe.com");

  public TestGitRepo(String name, HudsonTestCase forTest, TaskListener listener)
      throws IOException, InterruptedException {
    this(name, forTest.createTmpDir(), listener);
  }

  public TestGitRepo(String name, File tmpDir, TaskListener listener) throws IOException, InterruptedException {
    this.name = name;
    this.listener = listener;

    envVars = new EnvVars();

    gitDir = tmpDir;
    User john = User.get(johnDoe.getName(), true);
    UserProperty johnsMailerProperty = new Mailer.UserProperty(johnDoe.getEmailAddress());
    john.addProperty(johnsMailerProperty);

    User jane = User.get(janeDoe.getName(), true);
    UserProperty janesMailerProperty = new Mailer.UserProperty(janeDoe.getEmailAddress());
    jane.addProperty(janesMailerProperty);

    // initialize the git interface.
    gitDirPath = new FilePath(gitDir);
    git = Git.with(listener, envVars).in(gitDir).getClient();

    // finally: initialize the repo
    git.init();
  }

  public void commit(final String fileName, final PersonIdent committer, final String message)
      throws GitException, InterruptedException {
    commit(fileName, fileName, committer, committer, message);
  }

  public void commit(final String fileName, final PersonIdent author, final PersonIdent committer, final String message)

      throws GitException, InterruptedException {
    commit(fileName, fileName, author, committer, message);
  }

  public void commit(final String fileName, final String fileContent, final PersonIdent committer, final String message)

      throws GitException, InterruptedException {
    commit(fileName, fileContent, committer, committer, message);
  }

  public void commit(final String fileName, final String fileContent, final PersonIdent author, final PersonIdent committer,
      final String message) throws GitException, InterruptedException {
    FilePath file = gitDirPath.child(fileName);
    try {
      file.write(fileContent, null);
    } catch (Exception e) {
      throw new GitException("unable to write file", e);
    }
    git.add(fileName);
    git.commit(message, author, committer);
  }

  public List<UserRemoteConfig> remoteConfigs() throws IOException {
    List<UserRemoteConfig> list = new ArrayList<UserRemoteConfig>();
    list.add(new UserRemoteConfig(gitDir.getAbsolutePath(), "origin", "", null));
    return list;
  }
}