123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477 |
- /*
- * Copyright (C) 2009-2010, Google Inc.
- * Copyright (C) 2008, Robin Rosenberg <robin.rosenberg@dewire.com>
- * Copyright (C) 2007, Shawn O. Pearce <spearce@spearce.org>
- * and other copyright owners as documented in the project's IP log.
- *
- * This program and the accompanying materials are made available
- * under the terms of the Eclipse Distribution License v1.0 which
- * accompanies this distribution, is reproduced below, and is
- * available at http://www.eclipse.org/org/documents/edl-v10.php
- *
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or
- * without modification, are permitted provided that the following
- * conditions are met:
- *
- * - Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * - Redistributions in binary form must reproduce the above
- * copyright notice, this list of conditions and the following
- * disclaimer in the documentation and/or other materials provided
- * with the distribution.
- *
- * - Neither the name of the Eclipse Foundation, Inc. nor the
- * names of its contributors may be used to endorse or promote
- * products derived from this software without specific prior
- * written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
- * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
- * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
- * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
- * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
- * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
- * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
- * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
- * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
-
- package org.eclipse.jgit.junit;
-
- import static org.junit.Assert.assertFalse;
- import static org.junit.Assert.fail;
-
- import java.io.File;
- import java.io.IOException;
- import java.util.ArrayList;
- import java.util.Collections;
- import java.util.HashMap;
- import java.util.List;
- import java.util.Map;
- import java.util.concurrent.TimeUnit;
-
- import org.eclipse.jgit.internal.storage.file.FileRepository;
- import org.eclipse.jgit.lib.Constants;
- import org.eclipse.jgit.lib.PersonIdent;
- import org.eclipse.jgit.lib.Repository;
- import org.eclipse.jgit.lib.RepositoryCache;
- import org.eclipse.jgit.storage.file.FileBasedConfig;
- import org.eclipse.jgit.storage.file.WindowCacheConfig;
- import org.eclipse.jgit.util.FS;
- import org.eclipse.jgit.util.FileUtils;
- import org.eclipse.jgit.util.SystemReader;
- import org.junit.After;
- import org.junit.Before;
-
- /**
- * JUnit TestCase with specialized support for temporary local repository.
- * <p>
- * A temporary directory is created for each test, allowing each test to use a
- * fresh environment. The temporary directory is cleaned up after the test ends.
- * <p>
- * Callers should not use {@link RepositoryCache} from within these tests as it
- * may wedge file descriptors open past the end of the test.
- * <p>
- * A system property {@code jgit.junit.usemmap} defines whether memory mapping
- * is used. Memory mapping has an effect on the file system, in that memory
- * mapped files in Java cannot be deleted as long as the mapped arrays have not
- * been reclaimed by the garbage collector. The programmer cannot control this
- * with precision, so temporary files may hang around longer than desired during
- * a test, or tests may fail altogether if there is insufficient file
- * descriptors or address space for the test process.
- */
- public abstract class LocalDiskRepositoryTestCase {
- private static final boolean useMMAP = "true".equals(System
- .getProperty("jgit.junit.usemmap"));
-
- /** A fake (but stable) identity for author fields in the test. */
- protected PersonIdent author;
-
- /** A fake (but stable) identity for committer fields in the test. */
- protected PersonIdent committer;
-
- private final List<Repository> toClose = new ArrayList<Repository>();
- private File tmp;
-
- private MockSystemReader mockSystemReader;
-
- @Before
- public void setUp() throws Exception {
- tmp = File.createTempFile("jgit_test_", "_tmp");
- CleanupThread.deleteOnShutdown(tmp);
- if (!tmp.delete() || !tmp.mkdir())
- throw new IOException("Cannot create " + tmp);
-
- mockSystemReader = new MockSystemReader();
- mockSystemReader.userGitConfig = new FileBasedConfig(new File(tmp,
- "usergitconfig"), FS.DETECTED);
- ceilTestDirectories(getCeilings());
- SystemReader.setInstance(mockSystemReader);
-
- final long now = mockSystemReader.getCurrentTime();
- final int tz = mockSystemReader.getTimezone(now);
- author = new PersonIdent("J. Author", "jauthor@example.com");
- author = new PersonIdent(author, now, tz);
-
- committer = new PersonIdent("J. Committer", "jcommitter@example.com");
- committer = new PersonIdent(committer, now, tz);
-
- final WindowCacheConfig c = new WindowCacheConfig();
- c.setPackedGitLimit(128 * WindowCacheConfig.KB);
- c.setPackedGitWindowSize(8 * WindowCacheConfig.KB);
- c.setPackedGitMMAP(useMMAP);
- c.setDeltaBaseCacheLimit(8 * WindowCacheConfig.KB);
- c.install();
- }
-
- protected File getTemporaryDirectory() {
- return tmp.getAbsoluteFile();
- }
-
- protected List<File> getCeilings() {
- return Collections.singletonList(getTemporaryDirectory());
- }
-
- private void ceilTestDirectories(List<File> ceilings) {
- mockSystemReader.setProperty(Constants.GIT_CEILING_DIRECTORIES_KEY, makePath(ceilings));
- }
-
- private static String makePath(List<?> objects) {
- final StringBuilder stringBuilder = new StringBuilder();
- for (Object object : objects) {
- if (stringBuilder.length() > 0)
- stringBuilder.append(File.pathSeparatorChar);
- stringBuilder.append(object.toString());
- }
- return stringBuilder.toString();
- }
-
- @After
- public void tearDown() throws Exception {
- RepositoryCache.clear();
- for (Repository r : toClose)
- r.close();
- toClose.clear();
-
- // Since memory mapping is controlled by the GC we need to
- // tell it this is a good time to clean up and unlock
- // memory mapped files.
- //
- if (useMMAP)
- System.gc();
- if (tmp != null)
- recursiveDelete(tmp, false, true);
- if (tmp != null && !tmp.exists())
- CleanupThread.removed(tmp);
-
- SystemReader.setInstance(null);
- }
-
- /** Increment the {@link #author} and {@link #committer} times. */
- protected void tick() {
- final long delta = TimeUnit.MILLISECONDS.convert(5 * 60,
- TimeUnit.SECONDS);
- final long now = author.getWhen().getTime() + delta;
- final int tz = mockSystemReader.getTimezone(now);
-
- author = new PersonIdent(author, now, tz);
- committer = new PersonIdent(committer, now, tz);
- }
-
- /**
- * Recursively delete a directory, failing the test if the delete fails.
- *
- * @param dir
- * the recursively directory to delete, if present.
- */
- protected void recursiveDelete(final File dir) {
- recursiveDelete(dir, false, true);
- }
-
- private static boolean recursiveDelete(final File dir,
- boolean silent, boolean failOnError) {
- assert !(silent && failOnError);
- if (!dir.exists())
- return silent;
- final File[] ls = dir.listFiles();
- if (ls != null)
- for (int k = 0; k < ls.length; k++) {
- final File e = ls[k];
- if (e.isDirectory())
- silent = recursiveDelete(e, silent, failOnError);
- else if (!e.delete()) {
- if (!silent)
- reportDeleteFailure(failOnError, e);
- silent = !failOnError;
- }
- }
- if (!dir.delete()) {
- if (!silent)
- reportDeleteFailure(failOnError, dir);
- silent = !failOnError;
- }
- return silent;
- }
-
- private static void reportDeleteFailure(boolean failOnError, File e) {
- String severity = failOnError ? "ERROR" : "WARNING";
- String msg = severity + ": Failed to delete " + e;
- if (failOnError)
- fail(msg);
- else
- System.err.println(msg);
- }
-
- /**
- * Creates a new empty bare repository.
- *
- * @return the newly created repository, opened for access
- * @throws IOException
- * the repository could not be created in the temporary area
- */
- protected FileRepository createBareRepository() throws IOException {
- return createRepository(true /* bare */);
- }
-
- /**
- * Creates a new empty repository within a new empty working directory.
- *
- * @return the newly created repository, opened for access
- * @throws IOException
- * the repository could not be created in the temporary area
- */
- protected FileRepository createWorkRepository() throws IOException {
- return createRepository(false /* not bare */);
- }
-
- /**
- * Creates a new empty repository.
- *
- * @param bare
- * true to create a bare repository; false to make a repository
- * within its working directory
- * @return the newly created repository, opened for access
- * @throws IOException
- * the repository could not be created in the temporary area
- */
- private FileRepository createRepository(boolean bare) throws IOException {
- File gitdir = createUniqueTestGitDir(bare);
- FileRepository db = new FileRepository(gitdir);
- assertFalse(gitdir.exists());
- db.create();
- toClose.add(db);
- return db;
- }
-
- /**
- * Adds a repository to the list of repositories which is closed at the end
- * of the tests
- *
- * @param r
- * the repository to be closed
- */
- public void addRepoToClose(Repository r) {
- toClose.add(r);
- }
-
- /**
- * Creates a unique directory for a test
- *
- * @param name
- * a subdirectory
- * @return a unique directory for a test
- * @throws IOException
- */
- protected File createTempDirectory(String name) throws IOException {
- File directory = new File(createTempFile(), name);
- FileUtils.mkdirs(directory);
- return directory.getCanonicalFile();
- }
-
- /**
- * Creates a new unique directory for a test repository
- *
- * @param bare
- * true for a bare repository; false for a repository with a
- * working directory
- * @return a unique directory for a test repository
- * @throws IOException
- */
- protected File createUniqueTestGitDir(boolean bare) throws IOException {
- String gitdirName = createTempFile().getPath();
- if (!bare)
- gitdirName += "/";
- return new File(gitdirName + Constants.DOT_GIT);
- }
-
- /**
- * Allocates a new unique file path that does not exist.
- * <p>
- * Unlike the standard {@code File.createTempFile} the returned path does
- * not exist, but may be created by another thread in a race with the
- * caller. Good luck.
- * <p>
- * This method is inherently unsafe due to a race condition between creating
- * the name and the first use that reserves it.
- *
- * @return a unique path that does not exist.
- * @throws IOException
- */
- protected File createTempFile() throws IOException {
- File p = File.createTempFile("tmp_", "", tmp);
- if (!p.delete()) {
- throw new IOException("Cannot obtain unique path " + tmp);
- }
- return p;
- }
-
- /**
- * Run a hook script in the repository, returning the exit status.
- *
- * @param db
- * repository the script should see in GIT_DIR environment
- * @param hook
- * path of the hook script to execute, must be executable file
- * type on this platform
- * @param args
- * arguments to pass to the hook script
- * @return exit status code of the invoked hook
- * @throws IOException
- * the hook could not be executed
- * @throws InterruptedException
- * the caller was interrupted before the hook completed
- */
- protected int runHook(final Repository db, final File hook,
- final String... args) throws IOException, InterruptedException {
- final String[] argv = new String[1 + args.length];
- argv[0] = hook.getAbsolutePath();
- System.arraycopy(args, 0, argv, 1, args.length);
-
- final Map<String, String> env = cloneEnv();
- env.put("GIT_DIR", db.getDirectory().getAbsolutePath());
- putPersonIdent(env, "AUTHOR", author);
- putPersonIdent(env, "COMMITTER", committer);
-
- final File cwd = db.getWorkTree();
- final Process p = Runtime.getRuntime().exec(argv, toEnvArray(env), cwd);
- p.getOutputStream().close();
- p.getErrorStream().close();
- p.getInputStream().close();
- return p.waitFor();
- }
-
- private static void putPersonIdent(final Map<String, String> env,
- final String type, final PersonIdent who) {
- final String ident = who.toExternalString();
- final String date = ident.substring(ident.indexOf("> ") + 2);
- env.put("GIT_" + type + "_NAME", who.getName());
- env.put("GIT_" + type + "_EMAIL", who.getEmailAddress());
- env.put("GIT_" + type + "_DATE", date);
- }
-
- /**
- * Create a string to a UTF-8 temporary file and return the path.
- *
- * @param body
- * complete content to write to the file. If the file should end
- * with a trailing LF, the string should end with an LF.
- * @return path of the temporary file created within the trash area.
- * @throws IOException
- * the file could not be written.
- */
- protected File write(final String body) throws IOException {
- final File f = File.createTempFile("temp", "txt", tmp);
- try {
- write(f, body);
- return f;
- } catch (Error e) {
- f.delete();
- throw e;
- } catch (RuntimeException e) {
- f.delete();
- throw e;
- } catch (IOException e) {
- f.delete();
- throw e;
- }
- }
-
- /**
- * Write a string as a UTF-8 file.
- *
- * @param f
- * file to write the string to. Caller is responsible for making
- * sure it is in the trash directory or will otherwise be cleaned
- * up at the end of the test. If the parent directory does not
- * exist, the missing parent directories are automatically
- * created.
- * @param body
- * content to write to the file.
- * @throws IOException
- * the file could not be written.
- */
- protected void write(final File f, final String body) throws IOException {
- JGitTestUtil.write(f, body);
- }
-
- protected String read(final File f) throws IOException {
- return JGitTestUtil.read(f);
- }
-
- private static String[] toEnvArray(final Map<String, String> env) {
- final String[] envp = new String[env.size()];
- int i = 0;
- for (Map.Entry<String, String> e : env.entrySet())
- envp[i++] = e.getKey() + "=" + e.getValue();
- return envp;
- }
-
- private static HashMap<String, String> cloneEnv() {
- return new HashMap<String, String>(System.getenv());
- }
-
- private static final class CleanupThread extends Thread {
- private static final CleanupThread me;
- static {
- me = new CleanupThread();
- Runtime.getRuntime().addShutdownHook(me);
- }
-
- static void deleteOnShutdown(File tmp) {
- synchronized (me) {
- me.toDelete.add(tmp);
- }
- }
-
- static void removed(File tmp) {
- synchronized (me) {
- me.toDelete.remove(tmp);
- }
- }
-
- private final List<File> toDelete = new ArrayList<File>();
-
- @Override
- public void run() {
- // On windows accidentally open files or memory
- // mapped regions may prevent files from being deleted.
- // Suggesting a GC increases the likelihood that our
- // test repositories actually get removed after the
- // tests, even in the case of failure.
- System.gc();
- synchronized (this) {
- boolean silent = false;
- boolean failOnError = false;
- for (File tmp : toDelete)
- recursiveDelete(tmp, silent, failOnError);
- }
- }
- }
- }
|