src/test/java/net/pterodactylus/sone/core/SoneInserterTest.java

   1 package net.pterodactylus.sone.core;
   2
   3 import static com.google.common.base.Optional.absent;
   4 import static com.google.common.base.Optional.of;
   5 import static com.google.common.util.concurrent.MoreExecutors.sameThreadExecutor;
   6 import static org.hamcrest.MatcherAssert.assertThat;
   7 import static org.hamcrest.Matchers.containsInAnyOrder;
   8 import static org.hamcrest.Matchers.instanceOf;
   9 import static org.hamcrest.Matchers.is;
  10 import static org.mockito.Matchers.any;
  11 import static org.mockito.Matchers.anyString;
  12 import static org.mockito.Matchers.argThat;
  13 import static org.mockito.Matchers.eq;
  14 import static org.mockito.Mockito.doAnswer;
  15 import static org.mockito.Mockito.mock;
  16 import static org.mockito.Mockito.never;
  17 import static org.mockito.Mockito.times;
  18 import static org.mockito.Mockito.verify;
  19 import static org.mockito.Mockito.when;
  20
  21 import java.util.HashMap;
  22
  23 import net.pterodactylus.sone.core.SoneInserter.InsertInformation;
  24 import net.pterodactylus.sone.core.event.InsertionDelayChangedEvent;
  25 import net.pterodactylus.sone.core.event.SoneEvent;
  26 import net.pterodactylus.sone.core.event.SoneInsertAbortedEvent;
  27 import net.pterodactylus.sone.core.event.SoneInsertedEvent;
  28 import net.pterodactylus.sone.core.event.SoneInsertingEvent;
  29 import net.pterodactylus.sone.data.Album;
  30 import net.pterodactylus.sone.data.Sone;
  31
  32 import freenet.keys.FreenetURI;
  33
  34 import com.google.common.base.Optional;
  35 import com.google.common.eventbus.AsyncEventBus;
  36 import com.google.common.eventbus.EventBus;
  37 import org.junit.Before;
  38 import org.junit.Test;
  39 import org.mockito.ArgumentCaptor;
  40 import org.mockito.invocation.InvocationOnMock;
  41 import org.mockito.stubbing.Answer;
  42
  43 /**
  44  * Unit test for {@link SoneInserter} and its subclasses.
  45  *
  46  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  47  */
  48 public class SoneInserterTest {
  49
  50         private final Core core = mock(Core.class);
  51         private final EventBus eventBus = mock(EventBus.class);
  52         private final FreenetInterface freenetInterface = mock(FreenetInterface.class);
  53
  54         @Before
  55         public void setupCore() {
  56                 UpdateChecker updateChecker = mock(UpdateChecker.class);
  57                 when(core.getUpdateChecker()).thenReturn(updateChecker);
  58                 when(core.getSone(anyString())).thenReturn(Optional.<Sone>absent());
  59         }
  60
  61         @Test
  62         public void insertionDelayIsForwardedToSoneInserter() {
  63                 EventBus eventBus = new AsyncEventBus(sameThreadExecutor());
  64                 eventBus.register(new SoneInserter(core, eventBus, freenetInterface, "SoneId"));
  65                 eventBus.post(new InsertionDelayChangedEvent(15));
  66                 assertThat(SoneInserter.getInsertionDelay().get(), is(15));
  67         }
  68
  69         @Test
  70         /* this test is hilariously bad. */
  71         public void manifestEntriesAreCreated() {
  72                 FreenetURI insertUri = mock(FreenetURI.class);
  73                 String fingerprint = "fingerprint";
  74                 Sone sone = createSone(insertUri, fingerprint);
  75                 SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId");
  76                 InsertInformation insertInformation = soneInserter.new InsertInformation(sone);
  77                 HashMap<String, Object> manifestEntries = insertInformation.generateManifestEntries();
  78                 assertThat(manifestEntries.keySet(), containsInAnyOrder("index.html", "sone.xml"));
  79                 assertThat(insertInformation.getFingerprint(), is(fingerprint));
  80         }
  81
  82         private Sone createSone(FreenetURI insertUri, String fingerprint) {
  83                 Sone sone = mock(Sone.class);
  84                 when(sone.getInsertUri()).thenReturn(insertUri);
  85                 when(sone.getFingerprint()).thenReturn(fingerprint);
  86                 when(sone.getRootAlbum()).thenReturn(mock(Album.class));
  87                 when(core.getSone(anyString())).thenReturn(of(sone));
  88                 return sone;
  89         }
  90
  91         @Test
  92         public void isModifiedIsTrueIfModificationDetectorSaysSo() {
  93                 SoneModificationDetector soneModificationDetector = mock(SoneModificationDetector.class);
  94                 when(soneModificationDetector.isModified()).thenReturn(true);
  95                 SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId", soneModificationDetector, 1);
  96                 assertThat(soneInserter.isModified(), is(true));
  97         }
  98
  99         @Test
 100         public void isModifiedIsFalseIfModificationDetectorSaysSo() {
 101                 SoneModificationDetector soneModificationDetector = mock(SoneModificationDetector.class);
 102                 SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId", soneModificationDetector, 1);
 103                 assertThat(soneInserter.isModified(), is(false));
 104         }
 105
 106         @Test
 107         public void lastFingerprintIsStoredCorrectly() {
 108                 SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId");
 109                 soneInserter.setLastInsertFingerprint("last-fingerprint");
 110                 assertThat(soneInserter.getLastInsertFingerprint(), is("last-fingerprint"));
 111         }
 112
 113         @Test
 114         public void soneInserterStopsWhenItShould() {
 115                 SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId");
 116                 soneInserter.stop();
 117                 soneInserter.serviceRun();
 118         }
 119
 120         @Test
 121         public void soneInserterInsertsASoneIfItIsEligible() throws SoneException {
 122                 FreenetURI insertUri = mock(FreenetURI.class);
 123                 final FreenetURI finalUri = mock(FreenetURI.class);
 124                 String fingerprint = "fingerprint";
 125                 Sone sone = createSone(insertUri, fingerprint);
 126                 SoneModificationDetector soneModificationDetector = mock(SoneModificationDetector.class);
 127                 when(soneModificationDetector.isEligibleForInsert()).thenReturn(true);
 128                 when(freenetInterface.insertDirectory(eq(insertUri), any(HashMap.class), eq("index.html"))).thenReturn(finalUri);
 129                 final SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId", soneModificationDetector, 1);
 130                 doAnswer(new Answer<Void>() {
 131                         @Override
 132                         public Void answer(InvocationOnMock invocation) throws Throwable {
 133                                 soneInserter.stop();
 134                                 return null;
 135                         }
 136                 }).when(core).touchConfiguration();
 137                 soneInserter.serviceRun();
 138                 ArgumentCaptor<SoneEvent> soneEvents = ArgumentCaptor.forClass(SoneEvent.class);
 139                 verify(freenetInterface).insertDirectory(eq(insertUri), any(HashMap.class), eq("index.html"));
 140                 verify(eventBus, times(2)).post(soneEvents.capture());
 141                 assertThat(soneEvents.getAllValues().get(0), instanceOf(SoneInsertingEvent.class));
 142                 assertThat(soneEvents.getAllValues().get(0).sone(), is(sone));
 143                 assertThat(soneEvents.getAllValues().get(1), instanceOf(SoneInsertedEvent.class));
 144                 assertThat(soneEvents.getAllValues().get(1).sone(), is(sone));
 145         }
 146
 147         @Test
 148         public void soneInserterBailsOutIfItIsStoppedWhileInserting() throws SoneException {
 149                 FreenetURI insertUri = mock(FreenetURI.class);
 150                 final FreenetURI finalUri = mock(FreenetURI.class);
 151                 String fingerprint = "fingerprint";
 152                 Sone sone = createSone(insertUri, fingerprint);
 153                 SoneModificationDetector soneModificationDetector = mock(SoneModificationDetector.class);
 154                 when(soneModificationDetector.isEligibleForInsert()).thenReturn(true);
 155                 final SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId", soneModificationDetector, 1);
 156                 when(freenetInterface.insertDirectory(eq(insertUri), any(HashMap.class), eq("index.html"))).thenAnswer(new Answer<FreenetURI>() {
 157                         @Override
 158                         public FreenetURI answer(InvocationOnMock invocation) throws Throwable {
 159                                 soneInserter.stop();
 160                                 return finalUri;
 161                         }
 162                 });
 163                 soneInserter.serviceRun();
 164                 ArgumentCaptor<SoneEvent> soneEvents = ArgumentCaptor.forClass(SoneEvent.class);
 165                 verify(freenetInterface).insertDirectory(eq(insertUri), any(HashMap.class), eq("index.html"));
 166                 verify(eventBus, times(2)).post(soneEvents.capture());
 167                 assertThat(soneEvents.getAllValues().get(0), instanceOf(SoneInsertingEvent.class));
 168                 assertThat(soneEvents.getAllValues().get(0).sone(), is(sone));
 169                 assertThat(soneEvents.getAllValues().get(1), instanceOf(SoneInsertedEvent.class));
 170                 assertThat(soneEvents.getAllValues().get(1).sone(), is(sone));
 171                 verify(core, never()).touchConfiguration();
 172         }
 173
 174         @Test
 175         public void soneInserterDoesNotInsertSoneIfItIsNotEligible() throws SoneException {
 176                 FreenetURI insertUri = mock(FreenetURI.class);
 177                 String fingerprint = "fingerprint";
 178                 Sone sone = createSone(insertUri, fingerprint);
 179                 SoneModificationDetector soneModificationDetector = mock(SoneModificationDetector.class);
 180                 final SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId", soneModificationDetector, 1);
 181                 new Thread(new Runnable() {
 182                         @Override
 183                         public void run() {
 184                                 try {
 185                                         Thread.sleep(500);
 186                                 } catch (InterruptedException ie1) {
 187                                         throw new RuntimeException(ie1);
 188                                 }
 189                                 soneInserter.stop();
 190                         }
 191                 }).start();
 192                 soneInserter.serviceRun();
 193                 verify(freenetInterface, never()).insertDirectory(eq(insertUri), any(HashMap.class), eq("index.html"));
 194                 verify(eventBus, never()).post(argThat(org.hamcrest.Matchers.any(SoneEvent.class)));
 195         }
 196
 197         @Test
 198         public void soneInserterPostsAbortedEventIfAnExceptionOccurs() throws SoneException {
 199                 FreenetURI insertUri = mock(FreenetURI.class);
 200                 String fingerprint = "fingerprint";
 201                 Sone sone = createSone(insertUri, fingerprint);
 202                 SoneModificationDetector soneModificationDetector = mock(SoneModificationDetector.class);
 203                 when(soneModificationDetector.isEligibleForInsert()).thenReturn(true);
 204                 final SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId", soneModificationDetector, 1);
 205                 final SoneException soneException = new SoneException(new Exception());
 206                 when(freenetInterface.insertDirectory(eq(insertUri), any(HashMap.class), eq("index.html"))).thenAnswer(new Answer<FreenetURI>() {
 207                         @Override
 208                         public FreenetURI answer(InvocationOnMock invocation) throws Throwable {
 209                                 soneInserter.stop();
 210                                 throw soneException;
 211                         }
 212                 });
 213                 soneInserter.serviceRun();
 214                 ArgumentCaptor<SoneEvent> soneEvents = ArgumentCaptor.forClass(SoneEvent.class);
 215                 verify(freenetInterface).insertDirectory(eq(insertUri), any(HashMap.class), eq("index.html"));
 216                 verify(eventBus, times(2)).post(soneEvents.capture());
 217                 assertThat(soneEvents.getAllValues().get(0), instanceOf(SoneInsertingEvent.class));
 218                 assertThat(soneEvents.getAllValues().get(0).sone(), is(sone));
 219                 assertThat(soneEvents.getAllValues().get(1), instanceOf(SoneInsertAbortedEvent.class));
 220                 assertThat(soneEvents.getAllValues().get(1).sone(), is(sone));
 221                 verify(core, never()).touchConfiguration();
 222         }
 223
 224 }