Require a recipient when creating a post.

[Sone.git] / src / test / java / net / pterodactylus / sone / fcp / AbstractSoneCommandTest.java

diff --git a/src/test/java/net/pterodactylus/sone/fcp/AbstractSoneCommandTest.java b/src/test/java/net/pterodactylus/sone/fcp/AbstractSoneCommandTest.java
index 90f78cc..62285ea 100644 (file)
--- a/src/test/java/net/pterodactylus/sone/fcp/AbstractSoneCommandTest.java
+++ b/src/test/java/net/pterodactylus/sone/fcp/AbstractSoneCommandTest.java
@@ -17,6 +17,7 @@
 
 package net.pterodactylus.sone.fcp;
 
+import static com.google.common.base.Optional.fromNullable;
 import static com.google.common.base.Optional.of;
 import static java.util.Arrays.asList;
 import static java.util.UUID.randomUUID;
@@ -34,9 +35,11 @@ import static org.mockito.Mockito.when;
 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 +59,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 +67,10 @@ public class AbstractSoneCommandTest {
                }
        };
 
+       public AbstractSoneCommandTest() {
+               when(core.getDatabase()).thenReturn(database);
+       }
+
        @Test
        public void testStringEncoding() {
                String testString = prepareStringToBeEncoded();
@@ -315,4 +323,70 @@ public class AbstractSoneCommandTest {
                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.<String>any())).thenReturn(Optional.<Post>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.<String>any())).thenReturn(Optional.<PostReply>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");
+       }
+
 }