--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/test/org/tmatesoft/hg/test/TestRevisionMaps.java	Wed Jul 03 14:38:30 2013 +0200
@@ -0,0 +1,103 @@
+/*
+ * 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 java.util.Arrays;
+import java.util.Collections;
+import java.util.HashSet;
+
+import org.junit.Rule;
+import org.junit.Test;
+import org.tmatesoft.hg.core.HgException;
+import org.tmatesoft.hg.core.Nodeid;
+import org.tmatesoft.hg.repo.HgChangelog;
+import org.tmatesoft.hg.repo.HgParentChildMap;
+import org.tmatesoft.hg.repo.HgRepository;
+
+/**
+ * 
+ * @author Artem Tikhomirov
+ * @author TMate Software Ltd.
+ */
+public class TestRevisionMaps {
+
+	@Rule
+	public ErrorCollectorExt errorCollector = new ErrorCollectorExt();
+
+	@Test
+	public void testParentChildMap() throws HgException {
+		final HgRepository repo = Configuration.get().find("test-annotate");
+		Nodeid[] allRevs = RepoUtils.allRevisions(repo);
+		HgParentChildMap<HgChangelog> parentHelper = new HgParentChildMap<HgChangelog>(repo.getChangelog());
+		parentHelper.init();
+		errorCollector.assertEquals(Arrays.asList(allRevs), parentHelper.all());
+		for (Nodeid n : allRevs) {
+			errorCollector.assertTrue(parentHelper.knownNode(n));
+			// parents
+			final Nodeid p1 = parentHelper.safeFirstParent(n);
+			final Nodeid p2 = parentHelper.safeSecondParent(n);
+			errorCollector.assertFalse(p1 == null);
+			errorCollector.assertFalse(p2 == null);
+			errorCollector.assertEquals(p1.isNull() ? null : p1, parentHelper.firstParent(n));
+			errorCollector.assertEquals(p2.isNull() ? null : p2, parentHelper.secondParent(n));
+			HashSet<Nodeid> parents = new HashSet<Nodeid>();
+			boolean modified = parentHelper.appendParentsOf(n, parents);
+			errorCollector.assertEquals(p1.isNull() && p2.isNull(), !modified);
+			HashSet<Nodeid> cp = new HashSet<Nodeid>();
+			cp.add(parentHelper.firstParent(n));
+			cp.add(parentHelper.secondParent(n));
+			cp.remove(null);
+			errorCollector.assertEquals(cp, parents);
+			modified = parentHelper.appendParentsOf(n, parents);
+			errorCollector.assertFalse(modified);
+			//
+			// isChild, hasChildren, childrenOf, directChildren
+			if (!p1.isNull()) {
+				errorCollector.assertTrue(parentHelper.isChild(p1, n));
+				errorCollector.assertTrue(parentHelper.hasChildren(p1));
+				errorCollector.assertTrue(parentHelper.childrenOf(Collections.singleton(p1)).contains(n));
+				errorCollector.assertTrue(parentHelper.directChildren(p1).contains(n));
+			}
+			if (!p2.isNull()) {
+				errorCollector.assertTrue(parentHelper.isChild(p2, n));
+				errorCollector.assertTrue(parentHelper.hasChildren(p2));
+				errorCollector.assertTrue(parentHelper.childrenOf(Collections.singleton(p2)).contains(n));
+				errorCollector.assertTrue(parentHelper.directChildren(p2).contains(n));
+			}
+			errorCollector.assertFalse(parentHelper.isChild(n, p1));
+			errorCollector.assertFalse(parentHelper.isChild(n, p2));
+			//
+			
+		}
+		// heads
+		errorCollector.assertEquals(Arrays.asList(allRevs[7], allRevs[9]), parentHelper.heads());
+		// isChild
+		errorCollector.assertTrue(parentHelper.isChild(allRevs[1], allRevs[9]));
+		errorCollector.assertTrue(parentHelper.isChild(allRevs[0], allRevs[7]));
+		errorCollector.assertFalse(parentHelper.isChild(allRevs[4], allRevs[7]));
+		errorCollector.assertFalse(parentHelper.isChild(allRevs[2], allRevs[6]));
+		// childrenOf
+		errorCollector.assertEquals(Arrays.asList(allRevs[7]), parentHelper.childrenOf(Collections.singleton(allRevs[5])));
+		errorCollector.assertEquals(Arrays.asList(allRevs[8], allRevs[9]), parentHelper.childrenOf(Arrays.asList(allRevs[4], allRevs[6])));
+		errorCollector.assertEquals(Arrays.asList(allRevs[6], allRevs[8], allRevs[9]), parentHelper.childrenOf(Collections.singleton(allRevs[3])));
+		// directChildren
+		errorCollector.assertEquals(Arrays.asList(allRevs[2], allRevs[3]), parentHelper.directChildren(allRevs[1]));
+		errorCollector.assertEquals(Arrays.asList(allRevs[8]), parentHelper.directChildren(allRevs[6]));
+		errorCollector.assertEquals(Collections.emptyList(), parentHelper.directChildren(allRevs[7]));
+	}
+
+}