kitaev@213: /*
kitaev@213:  * Copyright (c) 2011 TMate Software Ltd
kitaev@213:  *  
kitaev@213:  * This program is free software; you can redistribute it and/or modify
kitaev@213:  * it under the terms of the GNU General Public License as published by
kitaev@213:  * the Free Software Foundation; version 2 of the License.
kitaev@213:  *
kitaev@213:  * This program is distributed in the hope that it will be useful,
kitaev@213:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
kitaev@213:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
kitaev@213:  * GNU General Public License for more details.
kitaev@213:  *
kitaev@213:  * For information on how to redistribute this software under
kitaev@213:  * the terms of a license other than GNU General Public License
kitaev@213:  * contact TMate Software at support@hg4j.com
kitaev@213:  */
kitaev@213: package org.tmatesoft.hg.test;
kitaev@213: 
kitaev@213: import static org.hamcrest.CoreMatchers.equalTo;
kitaev@213: import static org.hamcrest.CoreMatchers.notNullValue;
kitaev@213: import static org.junit.Assert.assertTrue;
kitaev@213: import static org.tmatesoft.hg.repo.HgRepository.TIP;
kitaev@213: 
kitaev@213: import java.util.Collections;
kitaev@213: import java.util.LinkedHashMap;
kitaev@213: import java.util.LinkedList;
kitaev@213: import java.util.Map;
kitaev@213: 
kitaev@213: import org.junit.Rule;
kitaev@213: import org.junit.Test;
kitaev@213: import org.tmatesoft.hg.core.HgLogCommand.FileRevision;
kitaev@213: import org.tmatesoft.hg.core.HgManifestCommand;
kitaev@213: import org.tmatesoft.hg.core.Nodeid;
kitaev@213: import org.tmatesoft.hg.repo.HgLookup;
kitaev@213: import org.tmatesoft.hg.repo.HgRepository;
kitaev@213: import org.tmatesoft.hg.util.Path;
kitaev@213: 
kitaev@213: 
kitaev@213: /**
kitaev@213:  *
kitaev@213:  * @author Artem Tikhomirov
kitaev@213:  * @author TMate Software Ltd.
kitaev@213:  */
kitaev@213: public class TestManifest {
kitaev@213: 
kitaev@213: 	@Rule
kitaev@213: 	public ErrorCollectorExt errorCollector = new ErrorCollectorExt();
kitaev@213: 
kitaev@213: 	private final HgRepository repo;
kitaev@213: 	private ManifestOutputParser manifestParser;
kitaev@213: 	private ExecHelper eh;
kitaev@213: 	final LinkedList<FileRevision> revisions = new LinkedList<FileRevision>();
kitaev@213: 	private HgManifestCommand.Handler handler  = new HgManifestCommand.Handler() {
kitaev@213: 		
kitaev@213: 		public void file(FileRevision fileRevision) {
kitaev@213: 			revisions.add(fileRevision);
kitaev@213: 		}
kitaev@213: 		
kitaev@213: 		public void end(Nodeid manifestRevision) {}
kitaev@213: 		public void dir(Path p) {}
kitaev@213: 		public void begin(Nodeid manifestRevision) {}
kitaev@213: 	};
kitaev@213: 
kitaev@213: 	public static void main(String[] args) throws Throwable {
kitaev@213: 		TestManifest tm = new TestManifest();
kitaev@213: 		tm.testTip();
kitaev@213: 		tm.testFirstRevision();
kitaev@213: 		tm.testRevisionInTheMiddle();
kitaev@213: 		tm.errorCollector.verify();
kitaev@213: 	}
kitaev@213: 	
kitaev@213: 	public TestManifest() throws Exception {
kitaev@213: 		this(new HgLookup().detectFromWorkingDir());
kitaev@213: 	}
kitaev@213: 
kitaev@213: 	private TestManifest(HgRepository hgRepo) {
kitaev@213: 		repo = hgRepo;
kitaev@213: 		assertTrue(!repo.isInvalid());
kitaev@213: 		eh = new ExecHelper(manifestParser = new ManifestOutputParser(), null);
kitaev@213: 	}
kitaev@213: 
kitaev@213: 	@Test
kitaev@213: 	public void testTip() throws Exception {
kitaev@213: 		testRevision(TIP);
kitaev@213: 	}
kitaev@213: 
kitaev@213: 	@Test
kitaev@213: 	public void testFirstRevision() throws Exception {
kitaev@213: 		testRevision(0);
kitaev@213: 	}
kitaev@213: 	
kitaev@213: 	@Test
kitaev@213: 	public void testRevisionInTheMiddle() throws Exception {
kitaev@213: 		int rev = repo.getManifest().getRevisionCount() / 2;
kitaev@213: 		if (rev == 0) {
kitaev@213: 			throw new IllegalStateException("Need manifest with few revisions");
kitaev@213: 		}
kitaev@213: 		testRevision(rev);
kitaev@213: 	}
kitaev@213: 
kitaev@213: 	private void testRevision(int rev) throws Exception {
kitaev@213: 		manifestParser.reset();
kitaev@213: 		eh.run("hg", "manifest", "--debug", "--rev", String.valueOf(rev == TIP ? -1 : rev));
kitaev@213: 		revisions.clear();
kitaev@213: 		new HgManifestCommand(repo).revision(rev).execute(handler);
kitaev@213: 		report("manifest " + (rev == TIP ? "TIP:" : "--rev " + rev));
kitaev@213: 	}
kitaev@213: 
kitaev@213: 	private void report(String what) throws Exception {
kitaev@213: 		final Map<Path, Nodeid> cmdLineResult = new LinkedHashMap<Path, Nodeid>(manifestParser.getResult());
kitaev@213: 		for (FileRevision fr : revisions) {
kitaev@213: 			Nodeid nid = cmdLineResult.remove(fr.getPath());
kitaev@213: 			errorCollector.checkThat("Extra " + fr.getPath() + " in Java result", nid, notNullValue());
kitaev@213: 			if (nid != null) {
kitaev@213: 				errorCollector.checkThat("Non-matching nodeid:" + nid, nid, equalTo(fr.getRevision()));
kitaev@213: 			}
kitaev@213: 		}
kitaev@213: 		errorCollector.checkThat("Non-matched entries from command line:", cmdLineResult, equalTo(Collections.<Path,Nodeid>emptyMap()));
kitaev@213: 	}
kitaev@213: }