tikhomirov@311: /*
tikhomirov@311:  * Copyright (c) 2011 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@313: import static org.tmatesoft.hg.repo.HgRepository.TIP;
tikhomirov@313: 
tikhomirov@322: import java.io.IOException;
tikhomirov@322: import java.nio.ByteBuffer;
tikhomirov@322: 
tikhomirov@311: import org.junit.Assert;
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@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@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@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@311: 	@Test
tikhomirov@311: 	public void testArrayHelper() {
tikhomirov@311: 		String[] initial = {"d", "w", "k", "b", "c", "i", "a", "r", "e", "h" };
tikhomirov@311: 		ArrayHelper ah = new ArrayHelper();
tikhomirov@311: 		String[] result = initial.clone();
tikhomirov@311: 		ah.sort(result);
tikhomirov@311: 		String[] restored = restore(result, ah.getReverse());
tikhomirov@311: 		Assert.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@311: 		ah.sort(result = initial.clone());
tikhomirov@311: 		restored = restore(result, ah.getReverse());
tikhomirov@311: 		Assert.assertArrayEquals(initial, restored);
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@311: 			rebuilt[indexInOriginal-1] = sorted[i];
tikhomirov@311: 		}
tikhomirov@311: 		return rebuilt;
tikhomirov@311: 	}
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@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@312: 		class InspectorImplementsCancel implements HgChangelog.Inspector, CancelSupport {
tikhomirov@312: 			public final int when2stop;
tikhomirov@312: 			public int lastVisitet = 0;
tikhomirov@312: 			private final CancelImpl cancelImpl = new CancelImpl(); 
tikhomirov@312: 
tikhomirov@312: 			public InspectorImplementsCancel(int limit) {
tikhomirov@312: 				when2stop = limit;
tikhomirov@312: 			}
tikhomirov@312: 			
tikhomirov@312: 			public void next(int revisionNumber, Nodeid nodeid, RawChangeset cset) {
tikhomirov@312: 				lastVisitet = revisionNumber;
tikhomirov@312: 				if (revisionNumber == when2stop) {
tikhomirov@312: 					cancelImpl.stop();
tikhomirov@312: 				}
tikhomirov@312: 			}
tikhomirov@312: 
tikhomirov@312: 			public void checkCancelled() throws CancelledException {
tikhomirov@312: 				cancelImpl.checkCancelled();
tikhomirov@312: 			}
tikhomirov@312: 		};
tikhomirov@312: 		class InspectorImplementsAdaptable implements HgChangelog.Inspector, Adaptable {
tikhomirov@312: 			public final int when2stop;
tikhomirov@312: 			public int lastVisitet = 0;
tikhomirov@312: 			private final CancelImpl cancelImpl = new CancelImpl();
tikhomirov@312: 			
tikhomirov@312: 			public InspectorImplementsAdaptable(int limit) {
tikhomirov@312: 				when2stop = limit;
tikhomirov@312: 			}
tikhomirov@312: 			
tikhomirov@312: 			public void next(int revisionNumber, Nodeid nodeid, RawChangeset cset) {
tikhomirov@312: 				lastVisitet = revisionNumber;
tikhomirov@312: 				if (revisionNumber == when2stop) {
tikhomirov@312: 					cancelImpl.stop();
tikhomirov@312: 				}
tikhomirov@312: 			}
tikhomirov@312: 			public <T> T getAdapter(Class<T> 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: 		//
tikhomirov@312: 		InspectorImplementsCancel insp1;
tikhomirov@312: 		repository.getChangelog().all(insp1= new InspectorImplementsCancel(2));
tikhomirov@312: 		Assert.assertEquals(insp1.when2stop, insp1.lastVisitet);
tikhomirov@312: 		repository.getChangelog().all(insp1 = new InspectorImplementsCancel(12));
tikhomirov@312: 		Assert.assertEquals(insp1.when2stop, insp1.lastVisitet);
tikhomirov@312: 		//
tikhomirov@312: 		InspectorImplementsAdaptable insp2;
tikhomirov@312: 		repository.getChangelog().all(insp2= new InspectorImplementsAdaptable(3));
tikhomirov@312: 		Assert.assertEquals(insp2.when2stop, insp2.lastVisitet);
tikhomirov@312: 		repository.getChangelog().all(insp2 = new InspectorImplementsAdaptable(10));
tikhomirov@312: 		Assert.assertEquals(insp2.when2stop, insp2.lastVisitet);
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@313: 		class InspectorImplementsAdaptable implements HgManifest.Inspector2, Adaptable {
tikhomirov@313: 			public final int when2stop;
tikhomirov@313: 			public int lastVisitet = 0;
tikhomirov@313: 			private final CancelImpl cancelImpl = new CancelImpl(); 
tikhomirov@313: 
tikhomirov@313: 			public InspectorImplementsAdaptable(int limit) {
tikhomirov@313: 				when2stop = limit;
tikhomirov@313: 			}
tikhomirov@313: 
tikhomirov@313: 			public boolean begin(int mainfestRevision, Nodeid nid, int changelogRevision) {
tikhomirov@313: 				if (++lastVisitet == when2stop) {
tikhomirov@313: 					cancelImpl.stop();
tikhomirov@313: 				}
tikhomirov@313: 				return true;
tikhomirov@313: 			}
tikhomirov@313: 
tikhomirov@313: 			public boolean next(Nodeid nid, String fname, String flags) {
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> T getAdapter(Class<T> 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@313: 		Assert.assertEquals(insp1.when2stop, insp1.lastVisitet);
tikhomirov@313: 		repository.getManifest().walk(0, TIP, insp1 = new InspectorImplementsAdaptable(10));
tikhomirov@313: 		Assert.assertEquals(insp1.when2stop, insp1.lastVisitet);
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@327: 	public void testRevlogInspectors() throws Exception { // FIXME move to better place
tikhomirov@327: 		HgRepository repository = Configuration.get().find("branches-1"); // any repo
tikhomirov@327: 		repository.getChangelog().walk(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@327: 		fileNode.walk(0, TIP, new HgDataFile.RevisionInspector() {
tikhomirov@327: 			int i = 0;
tikhomirov@327: 
tikhomirov@327: 			public void next(int localRevision, Nodeid revision, int linkedRevision) {
tikhomirov@327: 				Assert.assertEquals(i++, localRevision);
tikhomirov@327: 				Assert.assertEquals(fileNode.getChangesetLocalRevision(localRevision), linkedRevision);
tikhomirov@327: 				Assert.assertEquals(fileNode.getRevision(localRevision), revision);
tikhomirov@327: 			}
tikhomirov@327: 		});
tikhomirov@327: 		fileNode.walk(0, TIP, new HgDataFile.ParentInspector() {
tikhomirov@327: 			int i = 0;
tikhomirov@327: 			Nodeid[] all = new Nodeid[fileNode.getRevisionCount()];
tikhomirov@327: 
tikhomirov@327: 			public void next(int localRevision, Nodeid revision, int parent1, int parent2, Nodeid nidParent1, Nodeid nidParent2) {
tikhomirov@327: 				Assert.assertEquals(i++, localRevision);
tikhomirov@327: 				all[localRevision] = revision;
tikhomirov@327: 				Assert.assertNotNull(revision);
tikhomirov@327: 				Assert.assertFalse(localRevision == 0 && (parent1 != -1 || parent2 != -1));
tikhomirov@327: 				Assert.assertFalse(localRevision > 0 && parent1 == -1 && parent2 == -1);
tikhomirov@327: 				if (parent1 != -1) {
tikhomirov@327: 					Assert.assertNotNull(nidParent1);
tikhomirov@327: 					// deliberately ==, not asserEquals to ensure same instance
tikhomirov@327: 					Assert.assertTrue(nidParent1 == all[parent1]);  
tikhomirov@327: 				}
tikhomirov@327: 				if (parent2 != -1) {
tikhomirov@327: 					Assert.assertNotNull(nidParent2);
tikhomirov@327: 					Assert.assertTrue(nidParent2 == all[parent2]);  
tikhomirov@327: 				}
tikhomirov@327: 			}
tikhomirov@327: 		});
tikhomirov@327: 	}
tikhomirov@311: }