tikhomirov@61: /*
tikhomirov@66:  * Copyright (c) 2011 TMate Software Ltd
tikhomirov@66:  *  
tikhomirov@66:  * This program is free software; you can redistribute it and/or modify
tikhomirov@66:  * it under the terms of the GNU General Public License as published by
tikhomirov@66:  * the Free Software Foundation; version 2 of the License.
tikhomirov@66:  *
tikhomirov@66:  * This program is distributed in the hope that it will be useful,
tikhomirov@66:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@66:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@66:  * GNU General Public License for more details.
tikhomirov@66:  *
tikhomirov@66:  * For information on how to redistribute this software under
tikhomirov@66:  * the terms of a license other than GNU General Public License
tikhomirov@66:  * contact TMate Software at support@svnkit.com
tikhomirov@61:  */
tikhomirov@66: package org.tmatesoft.hg.test;
tikhomirov@61: 
tikhomirov@61: import java.io.File;
tikhomirov@61: import java.util.Collection;
tikhomirov@61: import java.util.LinkedList;
tikhomirov@61: import java.util.List;
tikhomirov@61: 
tikhomirov@61: import com.tmate.hgkit.fs.FileWalker;
tikhomirov@61: import com.tmate.hgkit.fs.RepositoryLookup;
tikhomirov@61: import com.tmate.hgkit.ll.HgRepository;
tikhomirov@61: import com.tmate.hgkit.ll.StatusCollector;
tikhomirov@61: import com.tmate.hgkit.ll.WorkingCopyStatusCollector;
tikhomirov@61: 
tikhomirov@61: /**
tikhomirov@61:  *
tikhomirov@66:  * @author Artem Tikhomirov
tikhomirov@66:  * @author TMate Software Ltd.
tikhomirov@61:  */
tikhomirov@61: public class TestStatus {
tikhomirov@61: 
tikhomirov@66: 	private StatusOutputParser statusParser;
tikhomirov@66: 	private ExecHelper eh;
tikhomirov@66: 	private final HgRepository repo;
tikhomirov@66: 
tikhomirov@61: 	public static void main(String[] args) throws Exception {
tikhomirov@62: 		HgRepository repo = new RepositoryLookup().detectFromWorkingDir();
tikhomirov@66: 		TestStatus test = new TestStatus(repo);
tikhomirov@66: 		test.testLowLevel();
tikhomirov@66: 		test.testStatusCommand();
tikhomirov@66: 	}
tikhomirov@66: 	
tikhomirov@66: 	public TestStatus(HgRepository hgRepo) {
tikhomirov@66: 		repo = hgRepo;
tikhomirov@66: 		statusParser = new StatusOutputParser();
tikhomirov@66: 		eh = new ExecHelper(statusParser, null);
tikhomirov@66: 	}
tikhomirov@66: 	
tikhomirov@66: 	public void testLowLevel() throws Exception {
tikhomirov@62: 		final WorkingCopyStatusCollector wcc = new WorkingCopyStatusCollector(repo, new FileWalker(new File(System.getProperty("user.dir"))));
tikhomirov@61: 		eh.run("hg", "status", "-A");
tikhomirov@61: 		StatusCollector.Record r = wcc.status(HgRepository.TIP);
tikhomirov@62: 		report("hg status -A", r, statusParser);
tikhomirov@62: 		//
tikhomirov@62: 		statusParser.reset();
tikhomirov@62: 		int revision = 3;
tikhomirov@62: 		eh.run("hg", "status", "-A", "--rev", String.valueOf(revision));
tikhomirov@62: 		r = wcc.status(revision);
tikhomirov@62: 		report("status -A --rev " + revision, r, statusParser);
tikhomirov@62: 		//
tikhomirov@62: 		statusParser.reset();
tikhomirov@62: 		eh.run("hg", "status", "-A", "--change", String.valueOf(revision));
tikhomirov@62: 		r = new StatusCollector.Record();
tikhomirov@62: 		new StatusCollector(repo).change(revision, r);
tikhomirov@62: 		report("status -A --change " + revision, r, statusParser);
tikhomirov@62: 	}
tikhomirov@62: 	
tikhomirov@66: 	public void testStatusCommand() throws Exception {
tikhomirov@66: 		throw HgRepository.notImplemented();
tikhomirov@66: 	}
tikhomirov@66: 	
tikhomirov@62: 	private static void report(String what, StatusCollector.Record r, StatusOutputParser statusParser) {
tikhomirov@62: 		System.out.println(">>>" + what);
tikhomirov@61: 		reportNotEqual("MODIFIED", r.getModified(), statusParser.getModified());
tikhomirov@61: 		reportNotEqual("ADDED", r.getAdded(), statusParser.getAdded());
tikhomirov@61: 		reportNotEqual("REMOVED", r.getRemoved(), statusParser.getRemoved());
tikhomirov@61: 		reportNotEqual("CLEAN", r.getClean(), statusParser.getClean());
tikhomirov@61: 		reportNotEqual("IGNORED", r.getIgnored(), statusParser.getIgnored());
tikhomirov@61: 		reportNotEqual("MISSING", r.getMissing(), statusParser.getMissing());
tikhomirov@61: 		reportNotEqual("UNKNOWN", r.getUnknown(), statusParser.getUnknown());
tikhomirov@61: 		// TODO compare equals
tikhomirov@62: 		System.out.println("<<<\n");
tikhomirov@61: 	}
tikhomirov@61: 	
tikhomirov@61: 	private static <T> void reportNotEqual(String what, Collection<T> l1, Collection<T> l2) {
tikhomirov@61: 		List<T> diff = difference(l1, l2);
tikhomirov@61: 		System.out.print(what);
tikhomirov@61: 		if (!diff.isEmpty()) {
tikhomirov@61: 			System.out.print(" are NOT the same: ");
tikhomirov@61: 			for (T t : diff) {
tikhomirov@61: 				System.out.print(t);
tikhomirov@61: 				System.out.print(", ");
tikhomirov@61: 			}
tikhomirov@61: 			System.out.println();
tikhomirov@61: 		} else {
tikhomirov@61: 			System.out.println(" are the same");
tikhomirov@61: 		}
tikhomirov@61: 	}
tikhomirov@61: 
tikhomirov@61: 	private static <T> List<T> difference(Collection<T> l1, Collection<T> l2) {
tikhomirov@61: 		LinkedList<T> result = new LinkedList<T>(l2);
tikhomirov@61: 		for (T t : l1) {
tikhomirov@61: 			if (l2.contains(t)) {
tikhomirov@61: 				result.remove(t);
tikhomirov@61: 			} else {
tikhomirov@61: 				result.add(t);
tikhomirov@61: 			}
tikhomirov@61: 		}
tikhomirov@61: 		return result;
tikhomirov@61: 	}
tikhomirov@61: }