tikhomirov@311: /* tikhomirov@552: * Copyright (c) 2011-2013 TMate Software Ltd tikhomirov@311: * tikhomirov@311: * This program is free software; you can redistribute it and/or modify tikhomirov@311: * it under the terms of the GNU General Public License as published by tikhomirov@311: * the Free Software Foundation; version 2 of the License. tikhomirov@311: * tikhomirov@311: * This program is distributed in the hope that it will be useful, tikhomirov@311: * but WITHOUT ANY WARRANTY; without even the implied warranty of tikhomirov@311: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the tikhomirov@311: * GNU General Public License for more details. tikhomirov@311: * tikhomirov@311: * For information on how to redistribute this software under tikhomirov@311: * the terms of a license other than GNU General Public License tikhomirov@311: * contact TMate Software at support@hg4j.com tikhomirov@311: */ tikhomirov@311: package org.tmatesoft.hg.test; tikhomirov@311: tikhomirov@472: import static java.lang.Integer.toBinaryString; tikhomirov@472: import static org.junit.Assert.*; tikhomirov@313: import static org.tmatesoft.hg.repo.HgRepository.TIP; tikhomirov@443: import static org.tmatesoft.hg.util.Path.CompareResult.*; tikhomirov@313: tikhomirov@322: import java.io.IOException; tikhomirov@322: import java.nio.ByteBuffer; tikhomirov@657: import java.util.Arrays; tikhomirov@322: tikhomirov@311: import org.junit.Assert; tikhomirov@443: import org.junit.Rule; tikhomirov@311: import org.junit.Test; tikhomirov@322: import org.tmatesoft.hg.core.HgCatCommand; tikhomirov@312: import org.tmatesoft.hg.core.Nodeid; tikhomirov@311: import org.tmatesoft.hg.internal.ArrayHelper; tikhomirov@574: import org.tmatesoft.hg.internal.ByteVector; tikhomirov@674: import org.tmatesoft.hg.internal.IntSliceSeq; tikhomirov@674: import org.tmatesoft.hg.internal.IntTuple; tikhomirov@552: import org.tmatesoft.hg.internal.IntVector; tikhomirov@443: import org.tmatesoft.hg.internal.PathScope; tikhomirov@472: import org.tmatesoft.hg.internal.RevisionDescendants; tikhomirov@703: import org.tmatesoft.hg.internal.diff.RangePairSeq; tikhomirov@312: import org.tmatesoft.hg.repo.HgChangelog; tikhomirov@312: import org.tmatesoft.hg.repo.HgChangelog.RawChangeset; tikhomirov@327: import org.tmatesoft.hg.repo.HgDataFile; tikhomirov@313: import org.tmatesoft.hg.repo.HgManifest; tikhomirov@313: import org.tmatesoft.hg.repo.HgManifest.Flags; tikhomirov@312: import org.tmatesoft.hg.repo.HgRepository; tikhomirov@628: import org.tmatesoft.hg.repo.HgRuntimeException; tikhomirov@312: import org.tmatesoft.hg.util.Adaptable; tikhomirov@322: import org.tmatesoft.hg.util.ByteChannel; tikhomirov@312: import org.tmatesoft.hg.util.CancelSupport; tikhomirov@312: import org.tmatesoft.hg.util.CancelledException; tikhomirov@313: import org.tmatesoft.hg.util.Path; tikhomirov@478: import org.tmatesoft.hg.util.ProgressSupport; tikhomirov@311: tikhomirov@311: /** tikhomirov@311: * tikhomirov@311: * @author Artem Tikhomirov tikhomirov@311: * @author TMate Software Ltd. tikhomirov@311: */ tikhomirov@311: public class TestAuxUtilities { tikhomirov@311: tikhomirov@443: @Rule tikhomirov@443: public ErrorCollectorExt errorCollector = new ErrorCollectorExt(); tikhomirov@443: tikhomirov@311: @Test tikhomirov@311: public void testArrayHelper() { tikhomirov@311: String[] initial = {"d", "w", "k", "b", "c", "i", "a", "r", "e", "h" }; tikhomirov@657: ArrayHelper ah = new ArrayHelper(initial); tikhomirov@311: String[] result = initial.clone(); tikhomirov@657: ah.sort(result, false, false); tikhomirov@657: String[] restored = restore(result, ah.getReverseIndexes()); tikhomirov@472: assertArrayEquals(initial, restored); tikhomirov@311: // tikhomirov@311: // few elements are on the right place from the very start and do not shift during sort. tikhomirov@311: // make sure for them we've got correct reversed indexes as well tikhomirov@311: initial = new String[] {"d", "h", "c", "b", "k", "i", "a", "r", "e", "w" }; tikhomirov@657: ah = new ArrayHelper(initial); tikhomirov@657: ah.sort(result = new String[initial.length], true, true); tikhomirov@657: restored = restore(result, ah.getReverseIndexes()); tikhomirov@472: assertArrayEquals(initial, restored); tikhomirov@657: for (int i = 0; i < initial.length; i++) { tikhomirov@657: String s = initial[i]; tikhomirov@657: errorCollector.assertEquals(i, ah.binarySearch(s, -1)); tikhomirov@657: errorCollector.assertEquals(Arrays.binarySearch(result, s), ah.binarySearchSorted(s)); tikhomirov@657: } tikhomirov@311: } tikhomirov@311: tikhomirov@311: private static String[] restore(String[] sorted, int[] sortReverse) { tikhomirov@311: String[] rebuilt = new String[sorted.length]; tikhomirov@311: for (int i = 0; i < sorted.length; i++) { tikhomirov@311: int indexInOriginal = sortReverse[i]; tikhomirov@657: rebuilt[indexInOriginal] = sorted[i]; tikhomirov@311: } tikhomirov@311: return rebuilt; tikhomirov@311: } tikhomirov@478: tikhomirov@478: tikhomirov@478: @Test tikhomirov@478: public void checkSubProgress() { tikhomirov@478: // no repo tikhomirov@478: class PS implements ProgressSupport { tikhomirov@478: tikhomirov@478: @SuppressWarnings("unused") tikhomirov@478: public int units; tikhomirov@478: public int worked; tikhomirov@478: public boolean done = false; tikhomirov@478: tikhomirov@478: public void start(int totalUnits) { tikhomirov@478: units = totalUnits; tikhomirov@478: } tikhomirov@478: public void worked(int wu) { tikhomirov@478: worked += wu; tikhomirov@478: } tikhomirov@478: public void done() { tikhomirov@478: done = true; tikhomirov@478: } tikhomirov@478: }; tikhomirov@478: PS ps = new PS(); tikhomirov@478: ps.start(10); tikhomirov@478: ProgressSupport.Sub s1 = new ProgressSupport.Sub(ps, 3); tikhomirov@478: ProgressSupport.Sub s2 = new ProgressSupport.Sub(ps, 7); tikhomirov@478: s1.start(10); tikhomirov@478: s1.worked(1); tikhomirov@478: s1.worked(1); tikhomirov@478: s1.worked(1); tikhomirov@478: s1.worked(1); tikhomirov@478: // so far s1 consumed 40% of total 3 units tikhomirov@478: assertEquals(1, ps.worked); tikhomirov@478: s1.done(); tikhomirov@478: // now s1 consumed 100% of total 3 units tikhomirov@478: assertEquals(3, ps.worked); tikhomirov@478: assertFalse(ps.done); tikhomirov@478: // tikhomirov@478: s2.start(5); tikhomirov@478: s2.worked(3); tikhomirov@478: // s2 consumed 60% (3/5) of ps's 7 units tikhomirov@478: // 3+4 == 3 from s1 + 0.6*7 tikhomirov@478: assertEquals(3 + 4, ps.worked); tikhomirov@478: s2.worked(2); tikhomirov@478: assertEquals(3 + 7, ps.worked); tikhomirov@478: assertFalse(ps.done); tikhomirov@478: s2.done(); tikhomirov@478: //assertTrue(ps.done); tikhomirov@478: } tikhomirov@478: tikhomirov@312: tikhomirov@313: static class CancelImpl implements CancelSupport { tikhomirov@313: private boolean shallStop = false; tikhomirov@313: public void stop() { tikhomirov@313: shallStop = true; tikhomirov@313: } tikhomirov@313: public void checkCancelled() throws CancelledException { tikhomirov@313: if (shallStop) { tikhomirov@313: throw new CancelledException(); tikhomirov@312: } tikhomirov@312: } tikhomirov@313: } tikhomirov@520: tikhomirov@520: static class CancelAtValue { tikhomirov@520: public int lastSeen; tikhomirov@520: public final int stopValue; tikhomirov@520: protected final CancelImpl cancelImpl = new CancelImpl(); tikhomirov@520: tikhomirov@520: protected CancelAtValue(int value) { tikhomirov@520: stopValue = value; tikhomirov@520: } tikhomirov@520: tikhomirov@520: protected void nextValue(int value) { tikhomirov@520: lastSeen = value; tikhomirov@520: if (value == stopValue) { tikhomirov@520: cancelImpl.stop(); tikhomirov@520: } tikhomirov@520: } tikhomirov@520: } tikhomirov@313: tikhomirov@313: @Test tikhomirov@313: public void testChangelogCancelSupport() throws Exception { tikhomirov@313: HgRepository repository = Configuration.get().find("branches-1"); // any repo with more revisions tikhomirov@520: class InspectorImplementsCancel extends CancelAtValue implements HgChangelog.Inspector, CancelSupport { tikhomirov@312: tikhomirov@312: public InspectorImplementsCancel(int limit) { tikhomirov@520: super(limit); tikhomirov@312: } tikhomirov@312: tikhomirov@312: public void next(int revisionNumber, Nodeid nodeid, RawChangeset cset) { tikhomirov@520: nextValue(revisionNumber); tikhomirov@312: } tikhomirov@312: tikhomirov@312: public void checkCancelled() throws CancelledException { tikhomirov@312: cancelImpl.checkCancelled(); tikhomirov@312: } tikhomirov@312: }; tikhomirov@520: class InspectorImplementsAdaptable extends CancelAtValue implements HgChangelog.Inspector, Adaptable { tikhomirov@312: public InspectorImplementsAdaptable(int limit) { tikhomirov@520: super(limit); tikhomirov@312: } tikhomirov@312: tikhomirov@312: public void next(int revisionNumber, Nodeid nodeid, RawChangeset cset) { tikhomirov@520: nextValue(revisionNumber); tikhomirov@312: } tikhomirov@520: tikhomirov@312: public T getAdapter(Class adapterClass) { tikhomirov@312: if (CancelSupport.class == adapterClass) { tikhomirov@322: return adapterClass.cast(cancelImpl); tikhomirov@312: } tikhomirov@312: return null; tikhomirov@312: } tikhomirov@312: } tikhomirov@312: // tikhomirov@312: InspectorImplementsCancel insp1; tikhomirov@312: repository.getChangelog().all(insp1= new InspectorImplementsCancel(2)); tikhomirov@520: Assert.assertEquals(insp1.stopValue, insp1.lastSeen); tikhomirov@312: repository.getChangelog().all(insp1 = new InspectorImplementsCancel(12)); tikhomirov@520: Assert.assertEquals(insp1.stopValue, insp1.lastSeen); tikhomirov@312: // tikhomirov@312: InspectorImplementsAdaptable insp2; tikhomirov@312: repository.getChangelog().all(insp2= new InspectorImplementsAdaptable(3)); tikhomirov@520: Assert.assertEquals(insp2.stopValue, insp2.lastSeen); tikhomirov@312: repository.getChangelog().all(insp2 = new InspectorImplementsAdaptable(10)); tikhomirov@520: Assert.assertEquals(insp2.stopValue, insp2.lastSeen); tikhomirov@312: } tikhomirov@313: tikhomirov@313: @Test tikhomirov@313: public void testManifestCancelSupport() throws Exception { tikhomirov@313: HgRepository repository = Configuration.get().find("branches-1"); // any repo with as many revisions as possible tikhomirov@520: class InspectorImplementsAdaptable extends CancelAtValue implements HgManifest.Inspector, Adaptable { tikhomirov@313: public InspectorImplementsAdaptable(int limit) { tikhomirov@520: super(limit); tikhomirov@313: } tikhomirov@313: tikhomirov@313: public boolean begin(int mainfestRevision, Nodeid nid, int changelogRevision) { tikhomirov@520: nextValue(lastSeen+1); tikhomirov@313: return true; tikhomirov@313: } tikhomirov@313: tikhomirov@313: public boolean end(int manifestRevision) { tikhomirov@313: return true; tikhomirov@313: } tikhomirov@313: tikhomirov@313: public T getAdapter(Class adapterClass) { tikhomirov@313: if (CancelSupport.class == adapterClass) { tikhomirov@322: return adapterClass.cast(cancelImpl); tikhomirov@313: } tikhomirov@313: return null; tikhomirov@313: } tikhomirov@313: tikhomirov@313: public boolean next(Nodeid nid, Path fname, Flags flags) { tikhomirov@313: return true; tikhomirov@313: } tikhomirov@313: } tikhomirov@313: InspectorImplementsAdaptable insp1; tikhomirov@313: repository.getManifest().walk(0, TIP, insp1= new InspectorImplementsAdaptable(3)); tikhomirov@520: Assert.assertEquals(insp1.stopValue, insp1.lastSeen); tikhomirov@313: repository.getManifest().walk(0, TIP, insp1 = new InspectorImplementsAdaptable(10)); tikhomirov@520: Assert.assertEquals(insp1.stopValue, insp1.lastSeen); tikhomirov@313: } tikhomirov@313: tikhomirov@322: @Test tikhomirov@322: public void testCatCommandCancelSupport() throws Exception { tikhomirov@322: HgRepository repository = Configuration.get().find("branches-1"); // any repo tikhomirov@322: final HgCatCommand cmd = new HgCatCommand(repository); tikhomirov@322: cmd.file(Path.create("file1")); tikhomirov@322: cmd.set(new CancelSupport() { tikhomirov@322: int i = 0; tikhomirov@322: public void checkCancelled() throws CancelledException { tikhomirov@322: if (i++ == 2) { tikhomirov@322: throw new CancelledException(); tikhomirov@322: } tikhomirov@322: } tikhomirov@322: }); tikhomirov@322: try { tikhomirov@322: cmd.execute(new ByteChannel() { tikhomirov@322: tikhomirov@322: public int write(ByteBuffer buffer) throws IOException, CancelledException { tikhomirov@322: Assert.fail("Shall not get that far provided cancellation from command's CancelSupport is functional"); tikhomirov@322: return 0; tikhomirov@322: } tikhomirov@322: }); tikhomirov@322: Assert.fail("Command execution shall not fail silently, exception shall propagate"); tikhomirov@322: } catch (CancelledException ex) { tikhomirov@322: // good! tikhomirov@322: } tikhomirov@322: } tikhomirov@327: tikhomirov@327: @Test tikhomirov@418: public void testRevlogInspectors() throws Exception { // TODO move to better place tikhomirov@327: HgRepository repository = Configuration.get().find("branches-1"); // any repo tikhomirov@431: repository.getChangelog().indexWalk(0, TIP, new HgChangelog.RevisionInspector() { tikhomirov@327: tikhomirov@327: public void next(int localRevision, Nodeid revision, int linkedRevision) { tikhomirov@327: Assert.assertEquals(localRevision, linkedRevision); tikhomirov@327: } tikhomirov@327: }); tikhomirov@327: final HgDataFile fileNode = repository.getFileNode("file1"); tikhomirov@431: fileNode.indexWalk(0, TIP, new HgDataFile.RevisionInspector() { tikhomirov@327: int i = 0; tikhomirov@327: tikhomirov@628: public void next(int localRevision, Nodeid revision, int linkedRevision) throws HgRuntimeException { tikhomirov@472: assertEquals(i++, localRevision); tikhomirov@472: assertEquals(fileNode.getChangesetRevisionIndex(localRevision), linkedRevision); tikhomirov@472: assertEquals(fileNode.getRevision(localRevision), revision); tikhomirov@327: } tikhomirov@327: }); tikhomirov@445: class ParentInspectorCheck implements HgDataFile.ParentInspector { tikhomirov@448: private int i, c; tikhomirov@445: private Nodeid[] all; tikhomirov@448: private final int start; tikhomirov@445: tikhomirov@445: public ParentInspectorCheck(int start, int total) { tikhomirov@448: this.start = start; tikhomirov@448: i = start; // revision index being iterated tikhomirov@448: c = 0; // index/counter of visited revisions tikhomirov@445: all = new Nodeid[total]; tikhomirov@445: } tikhomirov@327: tikhomirov@327: public void next(int localRevision, Nodeid revision, int parent1, int parent2, Nodeid nidParent1, Nodeid nidParent2) { tikhomirov@472: assertEquals(i++, localRevision); tikhomirov@448: all[c++] = revision; tikhomirov@472: assertNotNull(revision); tikhomirov@472: assertFalse(localRevision == 0 && (parent1 != -1 || parent2 != -1)); tikhomirov@472: assertFalse(localRevision > 0 && parent1 == -1 && parent2 == -1); tikhomirov@327: if (parent1 != -1) { tikhomirov@327: Assert.assertNotNull(nidParent1); tikhomirov@448: if (parent1 >= start) { tikhomirov@448: // deliberately ==, not asserEquals to ensure same instance tikhomirov@448: Assert.assertTrue(nidParent1 == all[parent1-start]); tikhomirov@448: } tikhomirov@327: } tikhomirov@327: if (parent2 != -1) { tikhomirov@327: Assert.assertNotNull(nidParent2); tikhomirov@448: if (parent2 >= start) { tikhomirov@448: Assert.assertTrue(nidParent2 == all[parent2-start]); tikhomirov@448: } tikhomirov@327: } tikhomirov@327: } tikhomirov@445: }; tikhomirov@471: fileNode.indexWalk(0, TIP, new ParentInspectorCheck(0, fileNode.getRevisionCount())); tikhomirov@445: assert fileNode.getRevisionCount() > 2 : "prereq"; // need at least few revisions tikhomirov@445: // there used to be a defect in #walk impl, assumption all parents come prior to a revision tikhomirov@471: fileNode.indexWalk(1, 3, new ParentInspectorCheck(1, 3)); tikhomirov@327: } tikhomirov@331: tikhomirov@472: /* tikhomirov@472: * This test checks not only RevisionDescendants class, but also tikhomirov@472: * Revlog.indexWalk implementation defect, aka: tikhomirov@472: * Issue 31: Revlog#walk doesn't handle ParentInspector correctly with start revision other than 0, fails with AIOOBE tikhomirov@472: */ tikhomirov@472: @Test tikhomirov@472: public void testRevisionDescendants() throws Exception { tikhomirov@472: HgRepository hgRepo = Configuration.get().find("branches-1"); tikhomirov@472: int[] roots = new int[] {0, 1, 2, 3, 4, 5}; tikhomirov@472: // 0: all revisions are descendants, 17 total. tikhomirov@472: // 1: 2, 4, 7, 8, 9 tikhomirov@472: // 2: 7, 8, 9 tikhomirov@472: // 3: 5,6, 10-16 tikhomirov@472: // 4: no children tikhomirov@472: // 5: 6, 10-16 tikhomirov@472: // array values represent bit mask, '1' for revision that shall re reported as descendant tikhomirov@472: // least significant bit is revision 0, and so on, so that 1<