view test/org/tmatesoft/hg/test/TestClone.java @ 599:55b7987c1796

Do not instantiate intermediate arrays
author Artem Tikhomirov <tikhomirov.artem@gmail.com>
date Fri, 03 May 2013 15:29:26 +0200
parents 883f1efbcf27
children 4a0bab2c6da1
line wrap: on
line source
/*
 * Copyright (c) 2011 TMate Software Ltd
 *  
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * For information on how to redistribute this software under
 * the terms of a license other than GNU General Public License
 * contact TMate Software at support@hg4j.com
 */
package org.tmatesoft.hg.test;

import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

import org.hamcrest.CoreMatchers;
import org.junit.Rule;
import org.junit.Test;
import org.tmatesoft.hg.core.HgCloneCommand;
import org.tmatesoft.hg.repo.HgRemoteRepository;

/**
 *
 * @author Artem Tikhomirov
 * @author TMate Software Ltd.
 */
public class TestClone {

	@Rule
	public ErrorCollectorExt errorCollector = new ErrorCollectorExt();

	public static void main(String[] args) throws Throwable {
		TestClone t = new TestClone();
		t.testSimpleClone();
		t.errorCollector.verify();
	}

	public TestClone() {
	}
	
	@Test
	public void testSimpleClone() throws Exception {
		int x = 0;
		final File tempDir = Configuration.get().getTempDir();
		for (HgRemoteRepository hgRemote : Configuration.get().allRemote()) {
			HgCloneCommand cmd = new HgCloneCommand();
			cmd.source(hgRemote);
			File dest = new File(tempDir, "test-clone-" + x++);
			if (dest.exists()) {
				rmdir(dest);
			}
			cmd.destination(dest);
			cmd.execute();
			verify(hgRemote, dest);
		}
	}

	private void verify(HgRemoteRepository hgRemote, File dest) throws Exception {
		ExecHelper eh = new ExecHelper(new OutputParser.Stub(), dest);
		eh.run("hg", "verify");
		errorCollector.checkThat("Verify", eh.getExitValue(), CoreMatchers.equalTo(0));
		eh.run("hg", "out", hgRemote.getLocation());
		errorCollector.checkThat("Outgoing", eh.getExitValue(), CoreMatchers.equalTo(1));
		eh.run("hg", "in", hgRemote.getLocation());
		errorCollector.checkThat("Incoming", eh.getExitValue(), CoreMatchers.equalTo(1));
	}

	static void rmdir(File dest) throws IOException {
		LinkedList<File> queue = new LinkedList<File>();
		queue.addAll(Arrays.asList(dest.listFiles()));
		while (!queue.isEmpty()) {
			File next = queue.removeFirst();
			if (next.isDirectory()) {
				List<File> files = Arrays.asList(next.listFiles());
				if (!files.isEmpty()) {
					queue.addAll(files);
					queue.add(next);
				}
				// fall through
			} 
			next.delete();
		}
		dest.delete();
	}
}