diff test/org/tmatesoft/hg/test/TestRevert.java @ 526:2f9ed6bcefa2

Initial support for Revert command with accompanying minor refactoring
author Artem Tikhomirov <tikhomirov.artem@gmail.com>
date Tue, 15 Jan 2013 17:07:19 +0100
parents
children 47b7bedf0569
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/test/org/tmatesoft/hg/test/TestRevert.java	Tue Jan 15 17:07:19 2013 +0100
@@ -0,0 +1,87 @@
+/*
+ * Copyright (c) 2013 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 static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+import java.io.File;
+import java.io.FileOutputStream;
+
+import org.junit.Rule;
+import org.junit.Test;
+import org.tmatesoft.hg.core.HgRevertCommand;
+import org.tmatesoft.hg.repo.HgLookup;
+import org.tmatesoft.hg.repo.HgRepository;
+import org.tmatesoft.hg.util.Path;
+
+
+/**
+ * 
+ * @author Artem Tikhomirov
+ * @author TMate Software Ltd.
+ */
+public class TestRevert {
+
+	@Rule
+	public ErrorCollectorExt errorCollector = new ErrorCollectorExt();
+	private HgRepository repo;
+	private ExecHelper eh;
+
+	
+	public TestRevert() {
+	}
+	
+	@Test
+	public void testCommand() throws Exception {
+		File tmpDir = Configuration.get().getTempDir();
+		tmpDir.mkdirs();
+		eh = new ExecHelper(new OutputParser.Stub(), tmpDir);
+		File testRepoLoc = TestIncoming.createEmptyDir("test-revert");
+		// get a copy of a repository
+		eh.run("hg", "clone", Configuration.get().find("log-1").getWorkingDir().toString(), testRepoLoc.getName());
+		assertEquals("[sanity]", 0, eh.getExitValue());
+		
+		repo = new HgLookup().detect(testRepoLoc);
+		Path targetFile = Path.create("b");
+		modifyFileAppend(new File(testRepoLoc, targetFile.toString()));
+		
+		StatusOutputParser statusParser = new StatusOutputParser();
+		eh = new ExecHelper(statusParser, testRepoLoc);
+		eh.run("hg", "status", "-A");
+		assertEquals("[sanity]", 1, statusParser.getModified().size());
+		assertEquals("[sanity]", 2, statusParser.getClean().size());
+		assertEquals("[sanity]", targetFile, statusParser.getModified().get(0));
+		
+		HgRevertCommand cmd = new HgRevertCommand(repo);
+		cmd.file(targetFile).execute();
+		statusParser.reset();
+		eh.run("hg", "status", "-A");
+
+		assertEquals(3, statusParser.getClean().size());
+		assertTrue(statusParser.getClean().contains(targetFile));
+		assertEquals(1, statusParser.getUnknown().size());
+		assertEquals(targetFile.toString() + ".orig", statusParser.getUnknown().get(0).toString());
+	}
+
+	private static void modifyFileAppend(File f) throws Exception {
+		assertTrue(f.isFile());
+		FileOutputStream fos = new FileOutputStream(f, true);
+		fos.write("XXX".getBytes());
+		fos.close();
+	}
+}