Add unit test for get post command

diff --git a/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java b/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
index b2264a5..1723b69 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
+++ b/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
@@ -114,7 +114,7 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
         * @return The encoded text
         */
        protected static String encodeString(String text) {
-               return text.replaceAll("\\\\", "\\\\").replaceAll("\n", "\\\\n").replaceAll("\r", "\\\\r");
+               return text.replaceAll("\\\\", "\\\\\\\\").replaceAll("\n", "\\\\n").replaceAll("\r", "\\\\r");
        }
 
        /**

diff --git a/src/test/kotlin/net/pterodactylus/sone/fcp/GetPostCommandTest.kt b/src/test/kotlin/net/pterodactylus/sone/fcp/GetPostCommandTest.kt
new file mode 100644 (file)
index 0000000..dff812b
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/fcp/GetPostCommandTest.kt
@@ -0,0 +1,136 @@
+package net.pterodactylus.sone.fcp
+
+import freenet.support.SimpleFieldSet
+import net.pterodactylus.sone.core.Core
+import net.pterodactylus.sone.data.Post
+import net.pterodactylus.sone.data.PostReply
+import net.pterodactylus.sone.data.Sone
+import net.pterodactylus.sone.test.asOptional
+import net.pterodactylus.sone.test.mock
+import net.pterodactylus.sone.test.whenever
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.containsInAnyOrder
+import org.hamcrest.Matchers.equalTo
+import org.hamcrest.Matchers.nullValue
+import org.junit.Before
+import org.junit.Test
+
+/**
+ * Unit test for [GetPostCommand].
+ */
+class GetPostCommandTest : SoneCommandTest() {
+
+       private val sone = mock<Sone>().apply {
+               whenever(id).thenReturn("SoneId")
+       }
+       private val post = createPost()
+
+       override fun createCommand(core: Core) = GetPostCommand(core)
+
+       @Before
+       fun setupPostWithLikesAndReplies() {
+               whenever(core.getPost("ValidPostId")).thenReturn(post.asOptional())
+               val sone1 = mock<Sone>().apply { whenever(id).thenReturn("Sone1") }
+               val sone2 = mock<Sone>().apply { whenever(id).thenReturn("Sone2") }
+               whenever(core.getLikes(post)).thenReturn(setOf(sone1, sone2))
+               val replies = listOf(
+                               createPostReply("ReplyId1", sone1, 1000, "Reply 1"),
+                               createPostReply("ReplyId2", sone2, 2000, "Reply 2")
+               )
+               whenever(core.getReplies("ValidPostId")).thenReturn(replies)
+       }
+
+       @Test
+       fun `command does not require write access`() {
+               assertThat(command.requiresWriteAccess(), equalTo(false))
+       }
+
+       @Test
+       fun `request without any parameter results in fcp exception`() {
+               requestWithoutAnyParameterResultsInFcpException()
+       }
+
+       @Test
+       fun `request with invalid post parameter results in fcp exception`() {
+               parameters.putSingle("Post", "InvalidPostId")
+               executeCommandAndExpectFcpException()
+       }
+
+       private fun createPostReply(id: String, sone: Sone, time: Long, text: String) = mock<PostReply>().apply {
+               whenever(this.id).thenReturn(id)
+               whenever(this.sone).thenReturn(sone)
+               whenever(this.time).thenReturn(time)
+               whenever(this.text).thenReturn(text)
+       }
+
+       private fun createPost() = mock<Post>().apply {
+               whenever(id).thenReturn("ValidPostId")
+               whenever(this.sone).thenReturn(this@GetPostCommandTest.sone)
+               whenever(recipientId).thenReturn(null.asOptional())
+               whenever(time).thenReturn(1000)
+               whenever(text).thenReturn("Post Text\r\nSecond \\Line")
+       }
+
+       private fun verifyPostWithLikes(replyParameters: SimpleFieldSet) {
+               assertThat(replyParameters["Message"], equalTo("Post"))
+               assertThat(replyParameters["Post.ID"], equalTo("ValidPostId"))
+               assertThat(replyParameters["Post.Sone"], equalTo("SoneId"))
+               assertThat(replyParameters["Post.Time"], equalTo("1000"))
+               assertThat(replyParameters["Post.Text"], equalTo("Post Text\\r\\nSecond \\\\Line"))
+               assertThat(replyParameters["Post.Likes.Count"], equalTo("2"))
+               assertThat((0..1).map { replyParameters["Post.Likes.$it.ID"] }, containsInAnyOrder("Sone1", "Sone2"))
+       }
+
+       private fun verifyReplies(replyParameters: SimpleFieldSet) {
+               assertThat(replyParameters["Post.Replies.Count"], equalTo("2"))
+               assertThat(replyParameters["Post.Replies.0.ID"], equalTo("ReplyId1"))
+               assertThat(replyParameters["Post.Replies.0.Sone"], equalTo("Sone1"))
+               assertThat(replyParameters["Post.Replies.0.Time"], equalTo("1000"))
+               assertThat(replyParameters["Post.Replies.0.Text"], equalTo("Reply 1"))
+               assertThat(replyParameters["Post.Replies.1.ID"], equalTo("ReplyId2"))
+               assertThat(replyParameters["Post.Replies.1.Sone"], equalTo("Sone2"))
+               assertThat(replyParameters["Post.Replies.1.Time"], equalTo("2000"))
+               assertThat(replyParameters["Post.Replies.1.Text"], equalTo("Reply 2"))
+       }
+
+       @Test
+       fun `request with valid post parameter returns post response`() {
+               parameters.putSingle("Post", "ValidPostId")
+               val replyParameters = command.execute(parameters).replyParameters
+               verifyPostWithLikes(replyParameters)
+               assertThat(replyParameters["Post.Recipient"], nullValue())
+               verifyReplies(replyParameters)
+       }
+
+       @Test
+       fun `request with valid post parameter without replies returns post response without replies`() {
+               parameters.putSingle("Post", "ValidPostId")
+               parameters.putSingle("IncludeReplies", "false")
+               val replyParameters = command.execute(parameters).replyParameters
+               verifyPostWithLikes(replyParameters)
+               assertThat(replyParameters["Post.Recipient"], nullValue())
+               assertThat(replyParameters["Post.Replies.Count"], nullValue())
+       }
+
+       @Test
+       fun `request with valid post parameter returns post response with recipient`() {
+               parameters.putSingle("Post", "ValidPostId")
+               whenever(post.recipientId).thenReturn("Sone2".asOptional())
+               val replyParameters = command.execute(parameters).replyParameters
+               verifyPostWithLikes(replyParameters)
+               assertThat(replyParameters["Post.Recipient"], equalTo("Sone2"))
+               verifyReplies(replyParameters)
+       }
+
+       @Test
+       fun `request with valid post parameter without replies returns post response without replies but with recipient`() {
+               parameters.putSingle("Post", "ValidPostId")
+               parameters.putSingle("IncludeReplies", "false")
+               whenever(post.recipientId).thenReturn("Sone2".asOptional())
+               val replyParameters = command.execute(parameters).replyParameters
+               verifyPostWithLikes(replyParameters)
+               assertThat(replyParameters["Post.Recipient"], equalTo("Sone2"))
+               assertThat(replyParameters["Post.Replies.Count"], nullValue())
+       }
+
+}

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/Guava.kt b/src/test/kotlin/net/pterodactylus/sone/test/Guava.kt
new file mode 100644 (file)
index 0000000..076962b
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/test/Guava.kt
@@ -0,0 +1,7 @@
+package net.pterodactylus.sone.test
+
+import com.google.common.base.Optional
+import com.google.common.base.Optional.absent
+import com.google.common.base.Optional.of
+
+fun <T> T?.asOptional(): Optional<T> = if (this == null) absent<T>() else of(this)