X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ffcp%2FAbstractSoneCommandTest.java;h=0595a9bff176ae7362c285d6a6d011a707687536;hb=0c5f5e59b4902f2d988fc191b22b81831f2a1aad;hp=cfaf660d35065a585c47cb414a2e5bd8c0adaf13;hpb=5680eb94d4545c38e2f6bb6e9fd903838459ee56;p=Sone.git diff --git a/src/test/java/net/pterodactylus/sone/fcp/AbstractSoneCommandTest.java b/src/test/java/net/pterodactylus/sone/fcp/AbstractSoneCommandTest.java index cfaf660..0595a9b 100644 --- a/src/test/java/net/pterodactylus/sone/fcp/AbstractSoneCommandTest.java +++ b/src/test/java/net/pterodactylus/sone/fcp/AbstractSoneCommandTest.java @@ -17,9 +17,12 @@ package net.pterodactylus.sone.fcp; +import static com.google.common.base.Optional.fromNullable; import static com.google.common.base.Optional.of; +import static com.google.common.collect.FluentIterable.from; import static java.util.Arrays.asList; import static java.util.UUID.randomUUID; +import static net.pterodactylus.sone.data.Reply.FUTURE_REPLY_FILTER; import static net.pterodactylus.sone.fcp.AbstractSoneCommand.encodeSone; import static net.pterodactylus.sone.fcp.AbstractSoneCommand.encodeString; import static net.pterodactylus.sone.template.SoneAccessor.getNiceName; @@ -31,12 +34,16 @@ import static org.mockito.Matchers.eq; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; +import java.util.Collection; +import java.util.Collections; import java.util.List; import net.pterodactylus.sone.core.Core; +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.database.Database; import net.pterodactylus.sone.freenet.SimpleFieldSetBuilder; import net.pterodactylus.sone.freenet.fcp.FcpException; @@ -56,6 +63,7 @@ import org.mockito.Matchers; public class AbstractSoneCommandTest { private final Core core = mock(Core.class); + private final Database database = mock(Database.class); private final AbstractSoneCommand abstractSoneCommand = new AbstractSoneCommand(core) { @Override public Response execute(SimpleFieldSet parameters, Bucket data, AccessType accessType) throws FcpException { @@ -63,6 +71,10 @@ public class AbstractSoneCommandTest { } }; + public AbstractSoneCommandTest() { + when(core.getDatabase()).thenReturn(database); + } + @Test public void testStringEncoding() { String testString = prepareStringToBeEncoded(); @@ -222,11 +234,17 @@ public class AbstractSoneCommandTest { PostReply postReply = mock(PostReply.class); when(postReply.getId()).thenReturn(randomUUID().toString()); when(postReply.getSone()).thenReturn(sone); - when(postReply.getTime()).thenReturn((long) (Math.random() * Long.MAX_VALUE)); + when(postReply.getTime()).thenReturn(System.currentTimeMillis()); when(postReply.getText()).thenReturn(text); return postReply; } + private PostReply createFuturePostReply(Sone sone, String text) { + PostReply postReply = createPostReply(sone, text); + when(postReply.getTime()).thenReturn(System.currentTimeMillis() + 86400000); + return postReply; + } + @Test public void testEncodingLikes() throws FSParseException { List likes = prepareMultipleSones(); @@ -288,4 +306,257 @@ public class AbstractSoneCommandTest { abstractSoneCommand.getMandatoryLocalSone(soneFieldSet, "RealSone"); } + @Test + public void testParsingAnExistingOptionalSone() throws FcpException { + Sone sone = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test", "First", "M.", "Last", (long) (Math.random() * Long.MAX_VALUE)); + when(core.getSone(eq("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E"))).thenReturn(of(sone)); + SimpleFieldSet soneFieldSet = new SimpleFieldSetBuilder().put("Sone", "jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E").get(); + Optional parsedSone = abstractSoneCommand.getOptionalSone(soneFieldSet, "Sone"); + assertThat(parsedSone, notNullValue()); + assertThat(parsedSone.isPresent(), is(true)); + assertThat(parsedSone.get(), is(sone)); + } + + @Test + public void testParsingANonExistingOptionalSone() throws FcpException { + when(core.getSone(Matchers.any())).thenReturn(Optional.absent()); + SimpleFieldSet soneFieldSet = new SimpleFieldSetBuilder().put("Sone", "jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E").get(); + Optional parsedSone = abstractSoneCommand.getOptionalSone(soneFieldSet, "Sone"); + assertThat(parsedSone, notNullValue()); + assertThat(parsedSone.isPresent(), is(false)); + } + + @Test(expected = FcpException.class) + public void testParsingAnOptionalSoneFromANonExistingFieldCausesAnError() throws FcpException { + when(core.getSone(Matchers.any())).thenReturn(Optional.absent()); + SimpleFieldSet soneFieldSet = new SimpleFieldSetBuilder().put("Sone", "jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E").get(); + abstractSoneCommand.getOptionalSone(soneFieldSet, "RealSone"); + } + + @Test + public void testParsingAPost() throws FcpException { + Sone sone = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test", "First", "M.", "Last", (long) (Math.random() * Long.MAX_VALUE)); + Post post = createPost(sone, null, (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + when(database.getPost(eq(post.getId()))).thenReturn(of(post)); + SimpleFieldSet postFieldSet = new SimpleFieldSetBuilder().put("Post", post.getId()).get(); + Post parsedPost = abstractSoneCommand.getPost(postFieldSet, "Post"); + assertThat(parsedPost, notNullValue()); + assertThat(parsedPost, is(post)); + } + + private Post createPost(Sone sone, String recipient, long time, String text) { + Post post = mock(Post.class); + when(post.getId()).thenReturn(randomUUID().toString()); + when(post.getSone()).thenReturn(sone); + when(post.getRecipientId()).thenReturn(fromNullable(recipient)); + when(post.getTime()).thenReturn(time); + when(post.getText()).thenReturn(text); + return post; + } + + @Test(expected = FcpException.class) + public void testThatTryingToParseANonExistingPostCausesAnError() throws FcpException { + Sone sone = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test", "First", "M.", "Last", (long) (Math.random() * Long.MAX_VALUE)); + Post post = createPost(sone, null, (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + when(database.getPost(Matchers.any())).thenReturn(Optional.absent()); + SimpleFieldSet postFieldSet = new SimpleFieldSetBuilder().put("Post", post.getId()).get(); + abstractSoneCommand.getPost(postFieldSet, "Post"); + } + + @Test(expected = FcpException.class) + public void testThatTryingToParseAPostFromANonExistingFieldCausesAnError() throws FcpException { + SimpleFieldSet postFieldSet = new SimpleFieldSetBuilder().get(); + abstractSoneCommand.getPost(postFieldSet, "Post"); + } + + @Test + public void testParsingAReply() throws FcpException { + PostReply reply = createPostReply(); + when(database.getPostReply(eq(reply.getId()))).thenReturn(of(reply)); + SimpleFieldSet replyFieldSet = new SimpleFieldSetBuilder().put("Reply", reply.getId()).get(); + PostReply parsedReply = abstractSoneCommand.getReply(replyFieldSet, "Reply"); + assertThat(parsedReply, notNullValue()); + assertThat(parsedReply, is(reply)); + } + + private PostReply createPostReply() { + PostReply postReply = mock(PostReply.class); + when(postReply.getId()).thenReturn(randomUUID().toString()); + return postReply; + } + + @Test(expected = FcpException.class) + public void testParsingANonExistingReply() throws FcpException { + PostReply reply = createPostReply(); + when(database.getPostReply(Matchers.any())).thenReturn(Optional.absent()); + SimpleFieldSet replyFieldSet = new SimpleFieldSetBuilder().put("Reply", reply.getId()).get(); + abstractSoneCommand.getReply(replyFieldSet, "Reply"); + } + + @Test(expected = FcpException.class) + public void testParsingAReplyFromANonExistingField() throws FcpException { + SimpleFieldSet replyFieldSet = new SimpleFieldSetBuilder().get(); + abstractSoneCommand.getReply(replyFieldSet, "Reply"); + } + + @Test + public void testEncodingAPostWithoutRecipientAndReplies() throws FSParseException { + Sone sone = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test", "First", "M.", "Last", (long) (Math.random() * Long.MAX_VALUE)); + Post post = createPost(sone, null, (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePost(post, "Post."); + assertThat(postFieldSet, notNullValue()); + verifyPost(postFieldSet, "Post.", post); + } + + private void verifyPost(SimpleFieldSet postFieldSet, String prefix, Post post) throws FSParseException { + assertThat(postFieldSet.get(prefix + "ID"), is(post.getId())); + assertThat(postFieldSet.get(prefix + "Sone"), is(post.getSone().getId())); + assertThat(postFieldSet.get(prefix + "Recipient"), is(post.getRecipientId().orNull())); + assertThat(postFieldSet.getLong(prefix + "Time"), is(post.getTime())); + assertThat(postFieldSet.get(prefix + "Text"), is(post.getText())); + } + + @Test + public void testEncodingAPostWithRecipientWithoutReplies() throws FSParseException { + Sone sone = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test", "First", "M.", "Last", (long) (Math.random() * Long.MAX_VALUE)); + Post post = createPost(sone, "KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePost(post, "Post."); + assertThat(postFieldSet, notNullValue()); + verifyPost(postFieldSet, "Post.", post); + } + + @Test + public void testEncodingAPostWithoutRecipientWithReplies() throws FSParseException { + Sone sone = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test", "First", "M.", "Last", (long) (Math.random() * Long.MAX_VALUE)); + Post post = createPost(sone, null, (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + PostReply postReply = createPostReply(sone, "Reply."); + when(post.getReplies()).thenReturn(asList(postReply)); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePostWithReplies(post, "Post."); + assertThat(postFieldSet, notNullValue()); + verifyPost(postFieldSet, "Post.", post); + verifyPostReplies(postFieldSet, "Post.", asList(postReply)); + } + + private void verifyPostReplies(SimpleFieldSet postFieldSet, String prefix, Collection postReplies) throws FSParseException { + assertThat(postFieldSet.getInt(prefix + "Replies.Count"), is(from(postReplies).filter(FUTURE_REPLY_FILTER).size())); + int postReplyIndex = 0; + for (PostReply postReply : from(postReplies).filter(FUTURE_REPLY_FILTER)) { + assertThat(postFieldSet.get(prefix + "Replies." + postReplyIndex + ".ID"), is(postReply.getId())); + assertThat(postFieldSet.get(prefix + "Replies." + postReplyIndex + ".Sone"), is(postReply.getSone().getId())); + assertThat(postFieldSet.getLong(prefix + "Replies." + postReplyIndex + ".Time"), is(postReply.getTime())); + assertThat(postFieldSet.get(prefix + "Replies." + postReplyIndex + ".Text"), is(postReply.getText())); + postReplyIndex++; + } + } + + @Test + public void testEncodingAPostWithoutRecipientWithFutureReplies() throws FSParseException { + Sone sone = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test", "First", "M.", "Last", (long) (Math.random() * Long.MAX_VALUE)); + Post post = createPost(sone, null, (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + PostReply postReply = createFuturePostReply(sone, "Reply."); + when(post.getReplies()).thenReturn(asList(postReply)); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePostWithReplies(post, "Post."); + assertThat(postFieldSet, notNullValue()); + verifyPost(postFieldSet, "Post.", post); + verifyPostReplies(postFieldSet, "Post.", Collections.emptyList()); + } + + @Test + public void testEncodingAPostWithRecipientAndReplies() throws FSParseException { + Sone sone = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test", "First", "M.", "Last", (long) (Math.random() * Long.MAX_VALUE)); + Post post = createPost(sone, "KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + PostReply postReply = createPostReply(sone, "Reply."); + when(post.getReplies()).thenReturn(asList(postReply)); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePostWithReplies(post, "Post."); + assertThat(postFieldSet, notNullValue()); + verifyPost(postFieldSet, "Post.", post); + verifyPostReplies(postFieldSet, "Post.", asList(postReply)); + } + + @Test + public void testEncodingPostsWithoutRecipientAndReplies() throws FSParseException { + Sone sone1 = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test1", "Alpha", "A.", "First", (long) (Math.random() * Long.MAX_VALUE)); + Sone sone2 = createSone("KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", "Test2", "Beta", "B.", "Second", (long) (Math.random() * Long.MAX_VALUE)); + Post post1 = createPost(sone1, null, (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + Post post2 = createPost(sone2, null, (long) (Math.random() * Long.MAX_VALUE), "Some other Text."); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePosts(asList(post1, post2), "Posts."); + assertThat(postFieldSet, notNullValue()); + verifyPosts(postFieldSet, "Posts.", asList(post1, post2)); + } + + private void verifyPosts(SimpleFieldSet postFieldSet, String prefix, Collection posts) throws FSParseException { + assertThat(postFieldSet.getInt(prefix + "Count"), is(posts.size())); + int postIndex = 0; + for (Post post : posts) { + verifyPost(postFieldSet, prefix + postIndex + ".", post); + postIndex++; + } + } + + @Test + public void testEncodingPostsWithRecipientWithoutReplies() throws FSParseException { + Sone sone1 = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test1", "Alpha", "A.", "First", (long) (Math.random() * Long.MAX_VALUE)); + Sone sone2 = createSone("KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", "Test2", "Beta", "B.", "Second", (long) (Math.random() * Long.MAX_VALUE)); + Post post1 = createPost(sone1, "KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + Post post2 = createPost(sone2, "jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", (long) (Math.random() * Long.MAX_VALUE), "Some other Text."); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePosts(asList(post1, post2), "Posts."); + assertThat(postFieldSet, notNullValue()); + verifyPosts(postFieldSet, "Posts.", asList(post1, post2)); + } + + @Test + public void testEncodingPostsWithoutRecipientWithReplies() throws FSParseException { + Sone sone1 = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test1", "Alpha", "A.", "First", (long) (Math.random() * Long.MAX_VALUE)); + Sone sone2 = createSone("KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", "Test2", "Beta", "B.", "Second", (long) (Math.random() * Long.MAX_VALUE)); + Post post1 = createPost(sone1, null, (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + Post post2 = createPost(sone2, null, (long) (Math.random() * Long.MAX_VALUE), "Some other Text."); + PostReply postReply1 = createPostReply(sone2, "Reply from 2 to 1"); + PostReply postReply2 = createPostReply(sone1, "Reply from 1 to 2"); + when(post1.getReplies()).thenReturn(asList(postReply1)); + when(post2.getReplies()).thenReturn(asList(postReply2)); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePostsWithReplies(asList(post1, post2), "Posts."); + assertThat(postFieldSet, notNullValue()); + verifyPostsWithReplies(postFieldSet, "Posts.", asList(post1, post2)); + } + + private void verifyPostsWithReplies(SimpleFieldSet postFieldSet, String prefix, Collection posts) throws FSParseException { + assertThat(postFieldSet.getInt(prefix + "Count"), is(posts.size())); + int postIndex = 0; + for (Post post : posts) { + verifyPost(postFieldSet, prefix + postIndex + ".", post); + verifyPostReplies(postFieldSet, prefix + postIndex + ".", post.getReplies()); + postIndex++; + } + } + + @Test + public void testEncodingPostsWithRecipientAndReplies() throws FSParseException { + Sone sone1 = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test1", "Alpha", "A.", "First", (long) (Math.random() * Long.MAX_VALUE)); + Sone sone2 = createSone("KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", "Test2", "Beta", "B.", "Second", (long) (Math.random() * Long.MAX_VALUE)); + Post post1 = createPost(sone1, "KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + Post post2 = createPost(sone2, "jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", (long) (Math.random() * Long.MAX_VALUE), "Some other Text."); + PostReply postReply1 = createPostReply(sone2, "Reply from 2 to 1"); + PostReply postReply2 = createPostReply(sone1, "Reply from 1 to 2"); + when(post1.getReplies()).thenReturn(asList(postReply1)); + when(post2.getReplies()).thenReturn(asList(postReply2)); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePostsWithReplies(asList(post1, post2), "Posts."); + assertThat(postFieldSet, notNullValue()); + verifyPostsWithReplies(postFieldSet, "Posts.", asList(post1, post2)); + } + + @Test + public void testEncodingPostsWithRecipientAndFutureReplies() throws FSParseException { + Sone sone1 = createSone("jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", "Test1", "Alpha", "A.", "First", (long) (Math.random() * Long.MAX_VALUE)); + Sone sone2 = createSone("KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", "Test2", "Beta", "B.", "Second", (long) (Math.random() * Long.MAX_VALUE)); + Post post1 = createPost(sone1, "KpoohJSbZGltHHG-YsxKV8ojjS5gwScRv50kl3AkLXg", (long) (Math.random() * Long.MAX_VALUE), "Some Text."); + Post post2 = createPost(sone2, "jXH8d-eFdm14R69WyaCgQoSjaY0jl-Ut6etlXjK0e6E", (long) (Math.random() * Long.MAX_VALUE), "Some other Text."); + PostReply postReply1 = createPostReply(sone2, "Reply from 2 to 1"); + PostReply postReply2 = createFuturePostReply(sone1, "Reply from 1 to 2"); + when(post1.getReplies()).thenReturn(asList(postReply1)); + when(post2.getReplies()).thenReturn(asList(postReply2)); + SimpleFieldSet postFieldSet = abstractSoneCommand.encodePostsWithReplies(asList(post1, post2), "Posts."); + assertThat(postFieldSet, notNullValue()); + verifyPostsWithReplies(postFieldSet, "Posts.", asList(post1, post2)); + } + }