X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneDownloaderTest.java;h=c910233bc08186621380364810b292515dc7b37d;hp=413b8fff1e3f5795815d1213f406d361a022f24c;hb=2b47186b72e30460a6710f95a76e4a99c305909a;hpb=94088d808dfca4bf5464ee4f3b099175fb4c058a diff --git a/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java b/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java index 413b8ff..c910233 100644 --- a/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java +++ b/src/test/java/net/pterodactylus/sone/core/SoneDownloaderTest.java @@ -1,33 +1,947 @@ package net.pterodactylus.sone.core; +import static com.google.common.base.Optional.of; +import static java.lang.System.currentTimeMillis; +import static java.util.UUID.randomUUID; +import static java.util.concurrent.TimeUnit.DAYS; +import static net.pterodactylus.sone.data.Sone.SoneStatus.downloading; +import static net.pterodactylus.sone.data.Sone.SoneStatus.idle; +import static net.pterodactylus.sone.data.Sone.SoneStatus.unknown; +import static org.hamcrest.MatcherAssert.assertThat; +import static org.hamcrest.Matchers.hasSize; +import static org.hamcrest.Matchers.is; +import static org.hamcrest.Matchers.not; +import static org.hamcrest.Matchers.notNullValue; +import static org.hamcrest.Matchers.nullValue; +import static org.mockito.ArgumentCaptor.forClass; +import static org.mockito.Matchers.any; +import static org.mockito.Matchers.anyLong; +import static org.mockito.Matchers.anyString; import static org.mockito.Matchers.eq; +import static org.mockito.Mockito.doAnswer; import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.never; +import static org.mockito.Mockito.times; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; -import net.pterodactylus.sone.core.SoneDownloader.FetchSone; +import java.io.IOException; +import java.io.InputStream; +import java.net.MalformedURLException; +import java.util.ArrayList; +import java.util.HashMap; +import java.util.HashSet; +import java.util.List; +import java.util.Map; +import java.util.Set; + +import net.pterodactylus.sone.core.FreenetInterface.Fetched; +import net.pterodactylus.sone.data.Album; +import net.pterodactylus.sone.data.Album.Modifier; +import net.pterodactylus.sone.data.Client; +import net.pterodactylus.sone.data.Image; +import net.pterodactylus.sone.data.Post; +import net.pterodactylus.sone.data.PostReply; +import net.pterodactylus.sone.data.Profile; import net.pterodactylus.sone.data.Sone; +import net.pterodactylus.sone.data.Sone.SoneStatus; +import net.pterodactylus.sone.database.AlbumBuilder; +import net.pterodactylus.sone.database.ImageBuilder; +import net.pterodactylus.sone.database.PostBuilder; +import net.pterodactylus.sone.database.PostReplyBuilder; +import net.pterodactylus.sone.freenet.wot.Identity; +import freenet.client.ClientMetadata; +import freenet.client.FetchResult; +import freenet.client.async.USKCallback; import freenet.keys.FreenetURI; +import freenet.support.api.Bucket; +import com.google.common.base.Optional; +import com.google.common.collect.ArrayListMultimap; +import com.google.common.collect.ImmutableSet; +import com.google.common.collect.ListMultimap; +import org.junit.Before; import org.junit.Test; +import org.mockito.ArgumentCaptor; +import org.mockito.invocation.InvocationOnMock; +import org.mockito.stubbing.Answer; /** - * Unit test for {@link SoneDownloader} and its subclasses. + * Unit test for {@link SoneDownloaderImpl} and its subclasses. * * @author David ‘Bombe’ Roden */ public class SoneDownloaderTest { + private final Core core = mock(Core.class); + private final FreenetInterface freenetInterface = mock(FreenetInterface.class); + private final SoneDownloaderImpl soneDownloader = new SoneDownloaderImpl(core, freenetInterface); + private final FreenetURI requestUri = mock(FreenetURI.class); + private final Sone sone = mock(Sone.class); + private final PostBuilder postBuilder = mock(PostBuilder.class); + private final List createdPosts = new ArrayList(); + private Post post = mock(Post.class); + private final PostReplyBuilder postReplyBuilder = mock(PostReplyBuilder.class); + private final Set createdPostReplies = new HashSet(); + private PostReply postReply = mock(PostReply.class); + private final AlbumBuilder albumBuilder = mock(AlbumBuilder.class); + private final ListMultimap nestedAlbums = ArrayListMultimap.create(); + private final ListMultimap albumImages = ArrayListMultimap.create(); + private Album album = mock(Album.class); + private final Map albums = new HashMap(); + private final ImageBuilder imageBuilder = mock(ImageBuilder.class); + private Image image = mock(Image.class); + private final Map images = new HashMap(); + + @Before + public void setupSone() { + Identity identity = mock(Identity.class); + when(identity.getId()).thenReturn("identity"); + when(sone.getId()).thenReturn("identity"); + when(sone.getIdentity()).thenReturn(identity); + when(sone.getRequestUri()).thenReturn(requestUri); + when(sone.getTime()).thenReturn(currentTimeMillis() - DAYS.toMillis(1)); + } + + private void setupSoneAsUnknown() { + when(sone.getTime()).thenReturn(0L); + } + + @Before + public void setupRequestUri() { + when(requestUri.setKeyType(anyString())).thenReturn(requestUri); + when(requestUri.sskForUSK()).thenReturn(requestUri); + when(requestUri.setDocName(anyString())).thenReturn(requestUri); + when(requestUri.setMetaString(any(String[].class))).thenReturn(requestUri); + when(requestUri.getKeyType()).thenReturn("USK"); + } + + @Before + public void setupPost() { + when(post.getRecipientId()).thenReturn(Optional.absent()); + } + + @Before + public void setupPostBuilder() { + when(postBuilder.withId(anyString())).thenAnswer(new Answer() { + @Override + public PostBuilder answer(InvocationOnMock invocation) throws Throwable { + when(post.getId()).thenReturn((String) invocation.getArguments()[0]); + return postBuilder; + } + }); + when(postBuilder.from(anyString())).thenAnswer(new Answer() { + @Override + public PostBuilder answer(InvocationOnMock invocation) throws Throwable { + final Sone sone = mock(Sone.class); + when(sone.getId()).thenReturn((String) invocation.getArguments()[0]); + when(post.getSone()).thenReturn(sone); + return postBuilder; + } + }); + when(postBuilder.withTime(anyLong())).thenAnswer(new Answer() { + @Override + public PostBuilder answer(InvocationOnMock invocation) throws Throwable { + when(post.getTime()).thenReturn((Long) invocation.getArguments()[0]); + return postBuilder; + } + }); + when(postBuilder.withText(anyString())).thenAnswer(new Answer() { + @Override + public PostBuilder answer(InvocationOnMock invocation) throws Throwable { + when(post.getText()).thenReturn((String) invocation.getArguments()[0]); + return postBuilder; + } + }); + when(postBuilder.to(anyString())).thenAnswer(new Answer() { + @Override + public PostBuilder answer(InvocationOnMock invocation) throws Throwable { + when(post.getRecipientId()).thenReturn(of((String) invocation.getArguments()[0])); + return postBuilder; + } + }); + when(postBuilder.build()).thenAnswer(new Answer() { + @Override + public Post answer(InvocationOnMock invocation) throws Throwable { + Post post = SoneDownloaderTest.this.post; + SoneDownloaderTest.this.post = mock(Post.class); + setupPost(); + createdPosts.add(post); + return post; + } + }); + when(core.postBuilder()).thenReturn(postBuilder); + } + + @Before + public void setupPostReplyBuilder() { + when(postReplyBuilder.withId(anyString())).thenAnswer(new Answer() { + @Override + public PostReplyBuilder answer(InvocationOnMock invocation) throws Throwable { + when(postReply.getId()).thenReturn((String) invocation.getArguments()[0]); + return postReplyBuilder; + } + }); + when(postReplyBuilder.from(anyString())).thenAnswer(new Answer() { + @Override + public PostReplyBuilder answer(InvocationOnMock invocation) throws Throwable { + Sone sone = when(mock(Sone.class).getId()).thenReturn((String) invocation.getArguments()[0]).getMock(); + when(postReply.getSone()).thenReturn(sone); + return postReplyBuilder; + } + }); + when(postReplyBuilder.to(anyString())).thenAnswer(new Answer() { + @Override + public PostReplyBuilder answer(InvocationOnMock invocation) throws Throwable { + when(postReply.getPostId()).thenReturn((String) invocation.getArguments()[0]); + Post post = when(mock(Post.class).getId()).thenReturn((String) invocation.getArguments()[0]).getMock(); + when(postReply.getPost()).thenReturn(of(post)); + return postReplyBuilder; + } + }); + when(postReplyBuilder.withTime(anyLong())).thenAnswer(new Answer() { + @Override + public PostReplyBuilder answer(InvocationOnMock invocation) throws Throwable { + when(postReply.getTime()).thenReturn((Long) invocation.getArguments()[0]); + return postReplyBuilder; + } + }); + when(postReplyBuilder.withText(anyString())).thenAnswer(new Answer() { + @Override + public PostReplyBuilder answer(InvocationOnMock invocation) throws Throwable { + when(postReply.getText()).thenReturn((String) invocation.getArguments()[0]); + return postReplyBuilder; + } + }); + when(postReplyBuilder.build()).thenAnswer(new Answer() { + @Override + public PostReply answer(InvocationOnMock invocation) throws Throwable { + PostReply postReply = SoneDownloaderTest.this.postReply; + createdPostReplies.add(postReply); + SoneDownloaderTest.this.postReply = mock(PostReply.class); + return postReply; + } + }); + when(core.postReplyBuilder()).thenReturn(postReplyBuilder); + } + + @Before + public void setupAlbum() { + final Album album = SoneDownloaderTest.this.album; + when(album.getAlbumImage()).thenReturn(mock(Image.class)); + doAnswer(new Answer() { + @Override + public Void answer(InvocationOnMock invocation) { + nestedAlbums.put(album, (Album) invocation.getArguments()[0]); + return null; + } + }).when(album).addAlbum(any(Album.class)); + doAnswer(new Answer() { + @Override + public Void answer(InvocationOnMock invocation) { + albumImages.put(album, (Image) invocation.getArguments()[0]); + return null; + } + }).when(album).addImage(any(Image.class)); + when(album.getAlbums()).thenAnswer(new Answer>() { + @Override + public List answer(InvocationOnMock invocation) { + return nestedAlbums.get(album); + } + }); + when(album.getImages()).thenAnswer(new Answer>() { + @Override + public List answer(InvocationOnMock invocation) { + return albumImages.get(album); + } + }); + final Modifier albumModifier = new Modifier() { + private String title = album.getTitle(); + private String description = album.getDescription(); + private String imageId = album.getAlbumImage().getId(); + + @Override + public Modifier setTitle(String title) { + this.title = title; + return this; + } + + @Override + public Modifier setDescription(String description) { + this.description = description; + return this; + } + + @Override + public Modifier setAlbumImage(String imageId) { + this.imageId = imageId; + return this; + } + + @Override + public Album update() throws IllegalStateException { + when(album.getTitle()).thenReturn(title); + when(album.getDescription()).thenReturn(description); + Image image = mock(Image.class); + when(image.getId()).thenReturn(imageId); + when(album.getAlbumImage()).thenReturn(image); + return album; + } + }; + when(album.modify()).thenReturn(albumModifier); + } + + @Before + public void setupAlbumBuilder() { + when(albumBuilder.withId(anyString())).thenAnswer(new Answer() { + @Override + public AlbumBuilder answer(InvocationOnMock invocation) { + when(album.getId()).thenReturn((String) invocation.getArguments()[0]); + return albumBuilder; + } + }); + when(albumBuilder.randomId()).thenAnswer(new Answer() { + @Override + public AlbumBuilder answer(InvocationOnMock invocation) { + when(album.getId()).thenReturn(randomUUID().toString()); + return albumBuilder; + } + }); + when(albumBuilder.by(any(Sone.class))).thenAnswer(new Answer() { + @Override + public AlbumBuilder answer(InvocationOnMock invocation) { + when(album.getSone()).thenReturn((Sone) invocation.getArguments()[0]); + return albumBuilder; + } + }); + when(albumBuilder.build()).thenAnswer(new Answer() { + @Override + public Album answer(InvocationOnMock invocation) { + Album album = SoneDownloaderTest.this.album; + albums.put(album.getId(), album); + SoneDownloaderTest.this.album = mock(Album.class); + setupAlbum(); + return album; + } + }); + when(core.albumBuilder()).thenReturn(albumBuilder); + } + + @Before + public void setupAlbums() { + when(core.getAlbum(anyString())).thenAnswer(new Answer() { + @Override + public Album answer(InvocationOnMock invocation) throws Throwable { + return albums.get(invocation.getArguments()[0]); + } + }); + } + + @Before + public void setupImage() { + final Image image = SoneDownloaderTest.this.image; + Image.Modifier modifier = new Image.Modifier() { + private Sone sone = image.getSone(); + private long creationTime = image.getCreationTime(); + private String key = image.getKey(); + private String title = image.getTitle(); + private String description = image.getDescription(); + private int width = image.getWidth(); + private int height = image.getHeight(); + + @Override + public Image.Modifier setSone(Sone sone) { + this.sone = sone; + return this; + } + + @Override + public Image.Modifier setCreationTime(long creationTime) { + this.creationTime = creationTime; + return this; + } + + @Override + public Image.Modifier setKey(String key) { + this.key = key; + return this; + } + + @Override + public Image.Modifier setTitle(String title) { + this.title = title; + return this; + } + + @Override + public Image.Modifier setDescription(String description) { + this.description = description; + return this; + } + + @Override + public Image.Modifier setWidth(int width) { + this.width = width; + return this; + } + + @Override + public Image.Modifier setHeight(int height) { + this.height = height; + return this; + } + + @Override + public Image update() throws IllegalStateException { + when(image.getSone()).thenReturn(sone); + when(image.getCreationTime()).thenReturn(creationTime); + when(image.getKey()).thenReturn(key); + when(image.getTitle()).thenReturn(title); + when(image.getDescription()).thenReturn(description); + when(image.getWidth()).thenReturn(width); + when(image.getHeight()).thenReturn(height); + return image; + } + }; + when(image.getSone()).thenReturn(sone); + when(image.modify()).thenReturn(modifier); + } + + @Before + public void setupImageBuilder() { + when(imageBuilder.randomId()).thenAnswer(new Answer() { + @Override + public ImageBuilder answer(InvocationOnMock invocation) { + when(image.getId()).thenReturn(randomUUID().toString()); + return imageBuilder; + } + }); + when(imageBuilder.withId(anyString())).thenAnswer(new Answer() { + @Override + public ImageBuilder answer(InvocationOnMock invocation) { + when(image.getId()).thenReturn( + (String) invocation.getArguments()[0]); + return imageBuilder; + } + }); + when(imageBuilder.build()).thenAnswer(new Answer() { + @Override + public Image answer(InvocationOnMock invocation) { + Image image = SoneDownloaderTest.this.image; + images.put(image.getId(), image); + SoneDownloaderTest.this.image = mock(Image.class); + setupImage(); + return image; + } + }); + when(core.imageBuilder()).thenReturn(imageBuilder); + } + + @Before + public void setupImages() { + when(core.getImage(anyString())).thenAnswer(new Answer() { + @Override + public Image answer(InvocationOnMock invocation) + throws Throwable { + return images.get(invocation.getArguments()[0]); + } + }); + } + + @Test + public void addingASoneWillRegisterItsKey() { + soneDownloader.addSone(sone); + verify(freenetInterface).registerActiveUsk(eq(sone.getRequestUri()), any( + USKCallback.class)); + verify(freenetInterface, never()).unregisterUsk(sone); + } + + @Test + public void addingASoneTwiceWillAlsoDeregisterItsKey() { + soneDownloader.addSone(sone); + soneDownloader.addSone(sone); + verify(freenetInterface, times(2)).registerActiveUsk(eq( + sone.getRequestUri()), any(USKCallback.class)); + verify(freenetInterface).unregisterUsk(sone); + } + + @Test + public void parsingASoneFailsWhenDocumentIsNotXml() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-not-xml.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWhenDocumentHasNegativeProtocolVersion() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-negative-protocol-version.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWhenProtocolVersionIsTooLarge() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-too-large-protocol-version.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWhenThereIsNoTime() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-no-time.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWhenTimeIsNotNumeric() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-time-not-numeric.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWhenProfileIsMissing() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-no-profile.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWhenProfileFieldIsMissingAFieldName() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-profile-missing-field-name.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWhenProfileFieldNameIsEmpty() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-profile-empty-field-name.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWhenProfileFieldNameIsNotUnique() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-profile-duplicate-field-name.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneSucceedsWithoutPayload() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-no-payload.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream).getTime(), is(1407197508000L)); + } + + @Test + public void parsingASoneSucceedsWithoutProtocolVersion() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-missing-protocol-version.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), not(nullValue())); + } + + @Test + public void parsingASoneFailsWithMissingClientName() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-missing-client-name.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithMissingClientVersion() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-missing-client-version.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneSucceedsWithClientInfo() throws SoneException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-client-info.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream).getClient(), is(new Client("some-client", "some-version"))); + } + + @Test + public void parsingASoneSucceedsWithRequestUri() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-request-uri.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream).getRequestUri().toString(), is("USK@w~RyTGmv12Lg9oO91q1Untupi7my9qczT1RheGkEkVE,E8ElVfUgukSCPHxIEJp-gHMiR80wpM7sID3Jo5O7w1s,AQACAAE/Sone/0")); + } + + @Test + public void parsingASoneFailsWithInvalidRequestUri() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-invalid-request-uri.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneSucceedsWithProfile() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-profile.xml"); + final Profile profile = soneDownloader.parseSone(sone, inputStream).getProfile(); + assertThat(profile.getFirstName(), is("first")); + assertThat(profile.getMiddleName(), is("middle")); + assertThat(profile.getLastName(), is("last")); + assertThat(profile.getBirthDay(), is(18)); + assertThat(profile.getBirthMonth(), is(12)); + assertThat(profile.getBirthYear(), is(1976)); + } + + @Test + public void parsingASoneSucceedsWithoutProfileFields() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-fields.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), notNullValue()); + } + + @Test + public void parsingASoneFailsWithoutPostId() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-post-id.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutPostTime() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-post-time.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutPostText() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-post-text.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithInvalidPostTime() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-invalid-post-time.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneSucceedsWithValidPostTime() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-valid-post-time.xml"); + final List posts = soneDownloader.parseSone(sone, inputStream).getPosts(); + assertThat(posts, is(createdPosts)); + assertThat(posts.get(0).getSone().getId(), is(sone.getId())); + assertThat(posts.get(0).getId(), is("post-id")); + assertThat(posts.get(0).getTime(), is(1407197508000L)); + assertThat(posts.get(0).getRecipientId(), is(Optional.absent())); + assertThat(posts.get(0).getText(), is("text")); + } + + @Test + public void parsingASoneSucceedsWithRecipient() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-recipient.xml"); + final List posts = soneDownloader.parseSone(sone, inputStream).getPosts(); + assertThat(posts, is(createdPosts)); + assertThat(posts.get(0).getSone().getId(), is(sone.getId())); + assertThat(posts.get(0).getId(), is("post-id")); + assertThat(posts.get(0).getTime(), is(1407197508000L)); + assertThat(posts.get(0).getRecipientId(), is(of("1234567890123456789012345678901234567890123"))); + assertThat(posts.get(0).getText(), is("text")); + } + + @Test + public void parsingASoneSucceedsWithInvalidRecipient() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-invalid-recipient.xml"); + final List posts = soneDownloader.parseSone(sone, inputStream).getPosts(); + assertThat(posts, is(createdPosts)); + assertThat(posts.get(0).getSone().getId(), is(sone.getId())); + assertThat(posts.get(0).getId(), is("post-id")); + assertThat(posts.get(0).getTime(), is(1407197508000L)); + assertThat(posts.get(0).getRecipientId(), is(Optional.absent())); + assertThat(posts.get(0).getText(), is("text")); + } + + @Test + public void parsingASoneFailsWithoutPostReplyId() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-post-reply-id.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutPostReplyPostId() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-post-reply-post-id.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutPostReplyTime() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-post-reply-time.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutPostReplyText() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-post-reply-text.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithInvalidPostReplyTime() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-invalid-post-reply-time.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneSucceedsWithValidPostReplyTime() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-valid-post-reply-time.xml"); + final Set postReplies = soneDownloader.parseSone(sone, inputStream).getReplies(); + assertThat(postReplies, is(createdPostReplies)); + PostReply postReply = createdPostReplies.iterator().next(); + assertThat(postReply.getId(), is("reply-id")); + assertThat(postReply.getPostId(), is("post-id")); + assertThat(postReply.getPost().get().getId(), is("post-id")); + assertThat(postReply.getSone().getId(), is("identity")); + assertThat(postReply.getTime(), is(1407197508000L)); + assertThat(postReply.getText(), is("reply-text")); + } + + @Test + public void parsingASoneSucceedsWithoutLikedPostIds() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-liked-post-ids.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), not(nullValue())); + } + + @Test + public void parsingASoneSucceedsWithLikedPostIds() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-liked-post-ids.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream).getLikedPostIds(), is((Set) ImmutableSet.of("liked-post-id"))); + } + + @Test + public void parsingASoneSucceedsWithoutLikedPostReplyIds() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-liked-post-reply-ids.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), not(nullValue())); + } + + @Test + public void parsingASoneSucceedsWithLikedPostReplyIds() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-liked-post-reply-ids.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream).getLikedReplyIds(), is((Set) ImmutableSet.of("liked-post-reply-id"))); + } + + @Test + public void parsingASoneSucceedsWithoutAlbums() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-albums.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), not(nullValue())); + } + + @Test + public void parsingASoneFailsWithoutAlbumId() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-album-id.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutAlbumTitle() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-album-title.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneSucceedsWithNestedAlbums() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-multiple-albums.xml"); + final Sone parsedSone = soneDownloader.parseSone(sone, inputStream); + assertThat(parsedSone, not(nullValue())); + assertThat(parsedSone.getRootAlbum().getAlbums(), hasSize(1)); + Album album = parsedSone.getRootAlbum().getAlbums().get(0); + assertThat(album.getId(), is("album-id-1")); + assertThat(album.getTitle(), is("album-title")); + assertThat(album.getDescription(), is("album-description")); + assertThat(album.getAlbums(), hasSize(1)); + Album nestedAlbum = album.getAlbums().get(0); + assertThat(nestedAlbum.getId(), is("album-id-2")); + assertThat(nestedAlbum.getTitle(), is("album-title-2")); + assertThat(nestedAlbum.getDescription(), is("album-description-2")); + assertThat(nestedAlbum.getAlbums(), hasSize(0)); + } + + @Test + public void parsingASoneFailsWithInvalidParentAlbumId() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-invalid-parent-album-id.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneSucceedsWithoutImages() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-images.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), not(nullValue())); + } + + @Test + public void parsingASoneFailsWithoutImageId() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-image-id.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutImageTime() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-image-time.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutImageKey() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-image-key.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutImageTitle() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-image-title.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + @Test - public void fetchSoneDownloadsSone() { - SoneDownloader soneDownloader = mock(SoneDownloader.class); - Sone sone = mock(Sone.class); - FreenetURI soneUri = mock(FreenetURI.class); - when(sone.getRequestUri()).thenReturn(soneUri); - FetchSone fetchSone = soneDownloader.new FetchSone(sone); - fetchSone.run(); - verify(soneDownloader).fetchSone(eq(sone), eq(soneUri)); + public void parsingASoneFailsWithoutImageWidth() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-image-width.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithoutImageHeight() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-without-image-height.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithInvalidImageWidth() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-invalid-image-width.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneFailsWithInvalidImageHeight() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-invalid-image-height.xml"); + assertThat(soneDownloader.parseSone(sone, inputStream), nullValue()); + } + + @Test + public void parsingASoneSucceedsWithImage() throws SoneException, MalformedURLException { + InputStream inputStream = getClass().getResourceAsStream("sone-parser-with-image.xml"); + final Sone sone = soneDownloader.parseSone(this.sone, inputStream); + assertThat(sone, not(nullValue())); + assertThat(sone.getRootAlbum().getAlbums(), hasSize(1)); + assertThat(sone.getRootAlbum().getAlbums().get(0).getImages(), hasSize(1)); + Image image = sone.getRootAlbum().getAlbums().get(0).getImages().get(0); + assertThat(image.getId(), is("image-id")); + assertThat(image.getCreationTime(), is(1407197508000L)); + assertThat(image.getKey(), is("KSK@GPLv3.txt")); + assertThat(image.getTitle(), is("image-title")); + assertThat(image.getDescription(), is("image-description")); + assertThat(image.getWidth(), is(1920)); + assertThat(image.getHeight(), is(1080)); + assertThat(sone.getProfile().getAvatar(), is("image-id")); + } + + @Test + public void stoppingTheSoneDownloaderUnregistersTheSone() { + soneDownloader.addSone(sone); + soneDownloader.stop(); + verify(freenetInterface).unregisterUsk(sone); + } + + @Test + public void notBeingAbleToFetchAnUnknownSoneDoesNotUpdateCore() { + setupSoneAsUnknown(); + soneDownloader.fetchSoneAction(sone).run(); + verify(freenetInterface).fetchUri(requestUri); + verifyThatSoneStatusWasChangedToDownloadingAndBackTo(unknown); + verify(core, never()).updateSone(any(Sone.class)); + } + + private void verifyThatSoneStatusWasChangedToDownloadingAndBackTo(SoneStatus soneStatus) { + ArgumentCaptor soneStatuses = forClass(SoneStatus.class); + verify(sone, times(2)).setStatus(soneStatuses.capture()); + assertThat(soneStatuses.getAllValues().get(0), is(downloading)); + assertThat(soneStatuses.getAllValues().get(1), is(soneStatus)); + } + + @Test + public void notBeingAbleToFetchAKnownSoneDoesNotUpdateCore() { + soneDownloader.fetchSoneAction(sone).run(); + verify(freenetInterface).fetchUri(requestUri); + verifyThatSoneStatusWasChangedToDownloadingAndBackTo(idle); + verify(core, never()).updateSone(any(Sone.class)); + } + + @Test(expected = NullPointerException.class) + public void exceptionWhileFetchingAnUnknownSoneDoesNotUpdateCore() { + setupSoneAsUnknown(); + when(freenetInterface.fetchUri(requestUri)).thenThrow(NullPointerException.class); + try { + soneDownloader.fetchSoneAction(sone).run(); + } finally { + verify(freenetInterface).fetchUri(requestUri); + verifyThatSoneStatusWasChangedToDownloadingAndBackTo(unknown); + verify(core, never()).updateSone(any(Sone.class)); + } + } + + @Test(expected = NullPointerException.class) + public void exceptionWhileFetchingAKnownSoneDoesNotUpdateCore() { + when(freenetInterface.fetchUri(requestUri)).thenThrow(NullPointerException.class); + try { + soneDownloader.fetchSoneAction(sone).run(); + } finally { + verify(freenetInterface).fetchUri(requestUri); + verifyThatSoneStatusWasChangedToDownloadingAndBackTo(idle); + verify(core, never()).updateSone(any(Sone.class)); + } + } + + @Test + public void successfulFetchingOfSoneWithUskRequestUriUpdatesTheCoreWithASone() throws IOException { + final Fetched fetchResult = createFetchResult(requestUri, getClass().getResourceAsStream("sone-parser-no-payload.xml")); + when(freenetInterface.fetchUri(requestUri)).thenReturn(fetchResult); + soneDownloader.fetchSoneAction(sone).run(); + verifyThatParsedSoneHasTheSameIdAsTheOriginalSone(); + } + + private void verifyThatParsedSoneHasTheSameIdAsTheOriginalSone() { + ArgumentCaptor soneCaptor = forClass(Sone.class); + verify(core).updateSone(soneCaptor.capture()); + assertThat(soneCaptor.getValue().getId(), is(sone.getId())); + } + + @Test + public void successfulFetchingOfSoneWithSskRequestUriUpdatesTheCoreWithASone() throws IOException { + when(requestUri.getKeyType()).thenReturn("SSK"); + final Fetched fetchResult = createFetchResult(requestUri, getClass().getResourceAsStream("sone-parser-no-payload.xml")); + when(freenetInterface.fetchUri(requestUri)).thenReturn(fetchResult); + soneDownloader.fetchSoneAction(sone).run(); + verifyThatParsedSoneHasTheSameIdAsTheOriginalSone(); + } + + @Test + public void successfulFetchingAnUnknownSoneWithSskRequestUriUpdatesTheCoreWithASone() throws IOException { + when(requestUri.getKeyType()).thenReturn("SSK"); + final Fetched fetchResult = createFetchResult(requestUri, getClass().getResourceAsStream("sone-parser-with-zero-time.xml")); + when(freenetInterface.fetchUri(requestUri)).thenReturn(fetchResult); + soneDownloader.fetchSoneAction(sone).run(); + verifyThatParsedSoneHasTheSameIdAsTheOriginalSone(); + } + + @Test + public void fetchingSoneWithInvalidXmlWillNotUpdateTheCore() throws IOException { + final Fetched fetchResult = createFetchResult(requestUri, getClass().getResourceAsStream("sone-parser-not-xml.xml")); + when(freenetInterface.fetchUri(requestUri)).thenReturn(fetchResult); + soneDownloader.fetchSoneAction(sone).run(); + verify(core, never()).updateSone(any(Sone.class)); + } + + @Test + public void exceptionWhileFetchingSoneWillNotUpdateTheCore() throws IOException { + final Fetched fetchResult = createFetchResult(requestUri, getClass().getResourceAsStream("sone-parser-no-payload.xml")); + when(sone.getId()).thenThrow(NullPointerException.class); + when(freenetInterface.fetchUri(requestUri)).thenReturn(fetchResult); + soneDownloader.fetchSoneAction(sone).run(); + verify(core, never()).updateSone(any(Sone.class)); + } + + @Test + public void onlyFetchingASoneWillNotUpdateTheCore() throws IOException { + final Fetched fetchResult = createFetchResult(requestUri, getClass().getResourceAsStream("sone-parser-no-payload.xml")); + when(freenetInterface.fetchUri(requestUri)).thenReturn(fetchResult); + soneDownloader.fetchSone(sone, sone.getRequestUri(), true); + verify(core, never()).updateSone(any(Sone.class)); + verifyThatSoneStatusWasChangedToDownloadingAndBackTo(idle); + } + + private Fetched createFetchResult(FreenetURI uri, InputStream inputStream) throws IOException { + ClientMetadata clientMetadata = new ClientMetadata("application/xml"); + Bucket bucket = mock(Bucket.class); + when(bucket.getInputStream()).thenReturn(inputStream); + FetchResult fetchResult = new FetchResult(clientMetadata, bucket); + return new Fetched(uri, fetchResult); } }