1 package net.pterodactylus.sone.core;
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;
21 import java.util.HashMap;
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;
32 import freenet.keys.FreenetURI;
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;
44 * Unit test for {@link SoneInserter} and its subclasses.
46 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
48 public class SoneInserterTest {
50 private final Core core = mock(Core.class);
51 private final EventBus eventBus = mock(EventBus.class);
52 private final FreenetInterface freenetInterface = mock(FreenetInterface.class);
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());
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));
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));
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));
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));
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));
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"));
114 public void soneInserterStopsWhenItShould() {
115 SoneInserter soneInserter = new SoneInserter(core, eventBus, freenetInterface, "SoneId");
117 soneInserter.serviceRun();
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>() {
132 public Void answer(InvocationOnMock invocation) throws Throwable {
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));
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>() {
158 public FreenetURI answer(InvocationOnMock invocation) throws Throwable {
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();
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() {
186 } catch (InterruptedException ie1) {
187 throw new RuntimeException(ie1);
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)));
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>() {
208 public FreenetURI answer(InvocationOnMock invocation) throws Throwable {
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();