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@74: import static org.tmatesoft.hg.repo.HgRepository.TIP;
tikhomirov@68: 
tikhomirov@61: import java.util.Collection;
tikhomirov@75: import java.util.HashMap;
tikhomirov@61: import java.util.LinkedList;
tikhomirov@61: import java.util.List;
tikhomirov@75: import java.util.Map;
tikhomirov@61: 
tikhomirov@68: import org.tmatesoft.hg.core.StatusCommand;
tikhomirov@74: import org.tmatesoft.hg.repo.HgRepository;
tikhomirov@74: import org.tmatesoft.hg.repo.Lookup;
tikhomirov@74: import org.tmatesoft.hg.repo.StatusCollector;
tikhomirov@74: import org.tmatesoft.hg.repo.WorkingCopyStatusCollector;
tikhomirov@68: 
tikhomirov@61: 
tikhomirov@61: /**
tikhomirov@76:  * 
tikhomirov@66:  * @author Artem Tikhomirov
tikhomirov@66:  * @author TMate Software Ltd.
tikhomirov@61:  */
tikhomirov@61: public class TestStatus {
tikhomirov@61: 
tikhomirov@68: 	private final HgRepository repo;
tikhomirov@66: 	private StatusOutputParser statusParser;
tikhomirov@66: 	private ExecHelper eh;
tikhomirov@66: 
tikhomirov@61: 	public static void main(String[] args) throws Exception {
tikhomirov@74: 		HgRepository repo = new Lookup().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@74: 		final WorkingCopyStatusCollector wcc = new WorkingCopyStatusCollector(repo);
tikhomirov@68: 		statusParser.reset();
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@68: 		final StatusCommand sc = new StatusCommand(repo).all();
tikhomirov@68: 		StatusCollector.Record r;
tikhomirov@68: 		statusParser.reset();
tikhomirov@68: 		eh.run("hg", "status", "-A");
tikhomirov@68: 		sc.execute(r = new StatusCollector.Record());
tikhomirov@68: 		report("hg status -A", r, statusParser);
tikhomirov@68: 		//
tikhomirov@68: 		statusParser.reset();
tikhomirov@68: 		int revision = 3;
tikhomirov@68: 		eh.run("hg", "status", "-A", "--rev", String.valueOf(revision));
tikhomirov@68: 		sc.base(revision).execute(r = new StatusCollector.Record());
tikhomirov@68: 		report("status -A --rev " + revision, r, statusParser);
tikhomirov@68: 		//
tikhomirov@68: 		statusParser.reset();
tikhomirov@68: 		eh.run("hg", "status", "-A", "--change", String.valueOf(revision));
tikhomirov@68: 		sc.base(TIP).revision(revision).execute(r = new StatusCollector.Record());
tikhomirov@68: 		report("status -A --change " + revision, r, statusParser);
tikhomirov@68: 		
tikhomirov@68: 		// TODO check not -A, but defaults()/custom set of modifications 
tikhomirov@66: 	}
tikhomirov@66: 	
tikhomirov@76: 	public void testRemovedAgainstNonTip() {
tikhomirov@76: 		/*
tikhomirov@76: 		 status --rev N when a file added past revision N was removed ((both physically and in dirstate), but not yet committed
tikhomirov@76: 
tikhomirov@76: 		 Reports extra REMOVED file (the one added and removed in between). Shall not
tikhomirov@76: 		 */
tikhomirov@76: 	}
tikhomirov@76: 	
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@75: 		List<String> copiedKeyDiff = difference(r.getCopied().keySet(), statusParser.getCopied().keySet());
tikhomirov@75: 		HashMap<String, String> copyDiff = new HashMap<String,String>();
tikhomirov@75: 		if (copiedKeyDiff.isEmpty()) {
tikhomirov@75: 			for (String jk : r.getCopied().keySet()) {
tikhomirov@75: 				String jv = r.getCopied().get(jk);
tikhomirov@75: 				if (statusParser.getCopied().containsKey(jk)) {
tikhomirov@75: 					String cmdv = statusParser.getCopied().get(jk);
tikhomirov@75: 					if (!jv.equals(cmdv)) {
tikhomirov@75: 						copyDiff.put(jk, jv + " instead of " + cmdv);
tikhomirov@75: 					}
tikhomirov@75: 				} else {
tikhomirov@75: 					copyDiff.put(jk, "ERRONEOUSLY REPORTED IN JAVA");
tikhomirov@75: 				}
tikhomirov@75: 			}
tikhomirov@75: 		}
tikhomirov@75: 		System.out.println("COPIED" + (copiedKeyDiff.isEmpty() && copyDiff.isEmpty() ? " are the same" : " are NOT the same:"));
tikhomirov@75: 		for (String s : copiedKeyDiff) {
tikhomirov@75: 			System.out.println("\tNon-matching key:" + s);
tikhomirov@75: 		}
tikhomirov@75: 		for (String s : copyDiff.keySet()) {
tikhomirov@75: 			System.out.println(s + " : " + copyDiff.get(s));
tikhomirov@75: 		}
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: }