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@311: import org.junit.Assert;
tikhomirov@311: import org.junit.Test;
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@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@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: 			@SuppressWarnings("unchecked")
tikhomirov@312: 			public <T> T getAdapter(Class<T> adapterClass) {
tikhomirov@312: 				if (CancelSupport.class == adapterClass) {
tikhomirov@312: 					return (T) 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: 			@SuppressWarnings("unchecked")
tikhomirov@313: 			public <T> T getAdapter(Class<T> adapterClass) {
tikhomirov@313: 				if (CancelSupport.class == adapterClass) {
tikhomirov@313: 					return (T) 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@311: }