tikhomirov@648: /*
tikhomirov@648:  * Copyright (c) 2013 TMate Software Ltd
tikhomirov@648:  *  
tikhomirov@648:  * This program is free software; you can redistribute it and/or modify
tikhomirov@648:  * it under the terms of the GNU General Public License as published by
tikhomirov@648:  * the Free Software Foundation; version 2 of the License.
tikhomirov@648:  *
tikhomirov@648:  * This program is distributed in the hope that it will be useful,
tikhomirov@648:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@648:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@648:  * GNU General Public License for more details.
tikhomirov@648:  *
tikhomirov@648:  * For information on how to redistribute this software under
tikhomirov@648:  * the terms of a license other than GNU General Public License
tikhomirov@648:  * contact TMate Software at support@hg4j.com
tikhomirov@648:  */
tikhomirov@648: package org.tmatesoft.hg.test;
tikhomirov@648: 
tikhomirov@648: import java.util.Arrays;
tikhomirov@648: 
tikhomirov@648: import org.junit.Rule;
tikhomirov@648: import org.junit.Test;
tikhomirov@648: import org.tmatesoft.hg.core.Nodeid;
tikhomirov@648: import org.tmatesoft.hg.internal.RevisionSet;
tikhomirov@651: import org.tmatesoft.hg.repo.HgChangelog;
tikhomirov@651: import org.tmatesoft.hg.repo.HgParentChildMap;
tikhomirov@651: import org.tmatesoft.hg.repo.HgRepository;
tikhomirov@648: 
tikhomirov@648: /**
tikhomirov@648:  * 
tikhomirov@648:  * @author Artem Tikhomirov
tikhomirov@648:  * @author TMate Software Ltd.
tikhomirov@648:  */
tikhomirov@648: public class TestRevisionSet {
tikhomirov@648: 	
tikhomirov@648: 	@Rule
tikhomirov@648: 	public ErrorCollectorExt errorCollector = new ErrorCollectorExt();
tikhomirov@648: 	
tikhomirov@648: 	@Test
tikhomirov@648: 	public void testRegularSetOperations() {
tikhomirov@648: 		Nodeid n1 = Nodeid.fromAscii("c75297c1786734589175c673db40e8ecaa032b09");
tikhomirov@648: 		Nodeid n2 = Nodeid.fromAscii("3b7d51ed4c65082f9235e3459e282d7ff723aa97");
tikhomirov@648: 		Nodeid n3 = Nodeid.fromAscii("14dac192aa262feb8ff6645a102648498483a188");
tikhomirov@648: 		Nodeid n4 = Nodeid.fromAscii("1deea2f332183c947937f6df988c2c6417efc217");
tikhomirov@651: 		RevisionSet a = rs(n1, n2, n3);
tikhomirov@651: 		RevisionSet b = rs(n3, n4);
tikhomirov@651: 		RevisionSet union_ab = rs(n1, n2, n3, n4);
tikhomirov@651: 		RevisionSet intersect_ab = rs(n3);
tikhomirov@651: 		RevisionSet subtract_ab = rs(n1, n2);
tikhomirov@651: 		RevisionSet subtract_ba = rs(n4);
tikhomirov@651: 		RevisionSet symDiff_ab = rs(n1, n2, n4);
tikhomirov@648: 		
tikhomirov@648: 		errorCollector.assertEquals(union_ab, a.union(b));
tikhomirov@648: 		errorCollector.assertEquals(union_ab, b.union(a));
tikhomirov@648: 		errorCollector.assertEquals(intersect_ab, a.intersect(b));
tikhomirov@648: 		errorCollector.assertEquals(intersect_ab, b.intersect(a));
tikhomirov@648: 		errorCollector.assertEquals(subtract_ab, a.subtract(b));
tikhomirov@648: 		errorCollector.assertEquals(subtract_ba, b.subtract(a));
tikhomirov@648: 		errorCollector.assertEquals(symDiff_ab, a.symmetricDifference(b));
tikhomirov@648: 		errorCollector.assertEquals(symDiff_ab, b.symmetricDifference(a));
tikhomirov@651: 		errorCollector.assertTrue(rs(n1, n2, n4).equals(rs(n4, n1, n2)));
tikhomirov@651: 		errorCollector.assertTrue(rs().equals(rs()));
tikhomirov@651: 		errorCollector.assertFalse(rs(n1).equals(rs(n2)));
tikhomirov@651: 	}
tikhomirov@651: 	
tikhomirov@651: 	@Test
tikhomirov@651: 	public void testRootsAndHeads() throws Exception {
tikhomirov@651: 		final HgRepository repo = Configuration.get().find("test-annotate");
tikhomirov@651: 		Nodeid[] allRevs = allRevisions(repo);
tikhomirov@651: 		HgParentChildMap<HgChangelog> parentHelper = new HgParentChildMap<HgChangelog>(repo.getChangelog());
tikhomirov@651: 		parentHelper.init();
tikhomirov@651: 		final RevisionSet complete = rs(allRevs);
tikhomirov@651: 		// roots
tikhomirov@651: 		errorCollector.assertEquals(rs(allRevs[0]), complete.roots(parentHelper));
tikhomirov@651: 		RevisionSet fromR2 = complete.subtract(rs(allRevs[0], allRevs[1]));
tikhomirov@651: 		RevisionSet fromR3 = complete.subtract(rs(allRevs[0], allRevs[1], allRevs[2]));
tikhomirov@651: 		errorCollector.assertEquals(rs(allRevs[2], allRevs[3]), fromR2.roots(parentHelper));
tikhomirov@651: 		errorCollector.assertEquals(rs(allRevs[3], allRevs[4], allRevs[5]), fromR3.roots(parentHelper));
tikhomirov@651: 		// heads
tikhomirov@651: 		errorCollector.assertEquals(rs(allRevs[9], allRevs[7]), complete.heads(parentHelper));
tikhomirov@651: 		RevisionSet toR7 = complete.subtract(rs(allRevs[9], allRevs[8]));
tikhomirov@651: 		errorCollector.assertEquals(rs(allRevs[7], allRevs[6], allRevs[4]), toR7.heads(parentHelper));
tikhomirov@651: 		RevisionSet withoutNoMergeBranch = toR7.subtract(rs(allRevs[5], allRevs[7]));
tikhomirov@651: 		errorCollector.assertEquals(rs(allRevs[6], allRevs[4]), withoutNoMergeBranch.heads(parentHelper));
tikhomirov@651: 		errorCollector.assertEquals(complete.heads(parentHelper), complete.heads(parentHelper).heads(parentHelper));
tikhomirov@651: 	}
tikhomirov@651: 	
tikhomirov@651: 	@Test
tikhomirov@651: 	public void testAncestorsAndChildren() throws Exception {
tikhomirov@651: 		final HgRepository repo = Configuration.get().find("test-annotate");
tikhomirov@651: 		Nodeid[] allRevs = allRevisions(repo);
tikhomirov@651: 		HgParentChildMap<HgChangelog> parentHelper = new HgParentChildMap<HgChangelog>(repo.getChangelog());
tikhomirov@651: 		parentHelper.init();
tikhomirov@651: 		final RevisionSet complete = rs(allRevs);
tikhomirov@651: 		// children
tikhomirov@651: 		errorCollector.assertTrue(rs().children(parentHelper).isEmpty());
tikhomirov@651: 		errorCollector.assertEquals(rs(allRevs[8], allRevs[9]), rs(allRevs[4]).children(parentHelper));
tikhomirov@651: 		// default branch and no-merge branch both from r2 
tikhomirov@651: 		RevisionSet s1 = rs(allRevs[8], allRevs[9], allRevs[4], allRevs[5], allRevs[7]);
tikhomirov@651: 		errorCollector.assertEquals(s1, rs(allRevs[2]).children(parentHelper));
tikhomirov@651: 		// ancestors
tikhomirov@651: 		RevisionSet fromR2 = complete.subtract(rs(allRevs[0], allRevs[1]));
tikhomirov@651: 		// no-merge branch and r9 are not in ancestors of r8 (as well as r8 itself)
tikhomirov@651: 		RevisionSet s3 = fromR2.subtract(rs(allRevs[9], allRevs[5], allRevs[7], allRevs[8]));
tikhomirov@651: 		errorCollector.assertEquals(s3, fromR2.ancestors(rs(allRevs[8]), parentHelper));
tikhomirov@651: 		// ancestors of no-merge branch
tikhomirov@651: 		RevisionSet branchNoMerge = rs(allRevs[5], allRevs[7]);
tikhomirov@651: 		errorCollector.assertEquals(rs(allRevs[0], allRevs[1], allRevs[2]), complete.ancestors(branchNoMerge, parentHelper));
tikhomirov@651: 		errorCollector.assertEquals(rs(allRevs[2]), fromR2.ancestors(branchNoMerge, parentHelper));
tikhomirov@651: 	}
tikhomirov@651: 	
tikhomirov@651: 	private static Nodeid[] allRevisions(HgRepository repo) {
tikhomirov@651: 		Nodeid[] allRevs = new Nodeid[repo.getChangelog().getRevisionCount()];
tikhomirov@651: 		for (int i = 0; i < allRevs.length; i++) {
tikhomirov@651: 			allRevs[i] = repo.getChangelog().getRevision(i);
tikhomirov@651: 		}
tikhomirov@651: 		return allRevs;
tikhomirov@648: 	}
tikhomirov@648: 
tikhomirov@648: 	
tikhomirov@651: 	private static RevisionSet rs(Nodeid... nodes) {
tikhomirov@648: 		return new RevisionSet(Arrays.asList(nodes));
tikhomirov@648: 	}
tikhomirov@648: }