Add unit test for reply group filter

diff --git a/src/test/kotlin/net/pterodactylus/sone/template/ReplyGroupFilterTest.kt b/src/test/kotlin/net/pterodactylus/sone/template/ReplyGroupFilterTest.kt
new file mode 100644 (file)
index 0000000..4614454
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/template/ReplyGroupFilterTest.kt
@@ -0,0 +1,64 @@
+package net.pterodactylus.sone.template
+
+import com.google.common.base.Optional
+import net.pterodactylus.sone.data.Post
+import net.pterodactylus.sone.data.PostReply
+import net.pterodactylus.sone.data.Sone
+import net.pterodactylus.sone.test.mock
+import net.pterodactylus.sone.test.whenever
+import net.pterodactylus.util.template.TemplateContext
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.containsInAnyOrder
+import org.hamcrest.Matchers.equalTo
+import org.junit.Before
+import org.junit.Test
+
+/**
+ * Unit test for [ReplyGroupFilter].
+ */
+class ReplyGroupFilterTest {
+
+       private val filter = ReplyGroupFilter()
+       private val replies = mutableListOf<PostReply>()
+       private val posts = mutableListOf<Post>()
+       private val sones = mutableListOf<Sone>()
+       private val templateContext = mock<TemplateContext>()
+
+       @Before
+       fun setupReplies() {
+               (0..4).forEach {
+                       sones += mock<Sone>()
+               }
+               (0..7).forEach {
+                       posts += mock<Post>()
+                       whenever(posts[it].sone).thenReturn(sones[(it + 1) % sones.size])
+               }
+               (0..10).forEach {
+                       replies += mock<PostReply>()
+                       whenever(replies[it].sone).thenReturn(sones[it % sones.size])
+                       whenever(replies[it].post).thenReturn(Optional.of(posts[it % posts.size]))
+               }
+       }
+
+       @Test
+       @Suppress("UNCHECKED_CAST")
+       fun `replies are grouped correctly`() {
+               val groupReplies = filter.format(templateContext, replies, emptyMap()) as Map<Post, Map<String, *>>
+               assertThat(groupReplies.keys, equalTo(posts.toSet()))
+               verifyPostRepliesAndSones(groupReplies, 0, listOf(0, 3), listOf(0, 8))
+               verifyPostRepliesAndSones(groupReplies, 1, listOf(1, 4), listOf(1, 9))
+               verifyPostRepliesAndSones(groupReplies, 2, listOf(2, 0), listOf(2, 10))
+               verifyPostRepliesAndSones(groupReplies, 3, listOf(3), listOf(3))
+               verifyPostRepliesAndSones(groupReplies, 4, listOf(4), listOf(4))
+               verifyPostRepliesAndSones(groupReplies, 5, listOf(0), listOf(5))
+               verifyPostRepliesAndSones(groupReplies, 6, listOf(1), listOf(6))
+               verifyPostRepliesAndSones(groupReplies, 7, listOf(2), listOf(7))
+       }
+
+       @Suppress("UNCHECKED_CAST")
+       private fun verifyPostRepliesAndSones(groupReplies: Map<Post, Map<String, *>>, postIndex: Int, soneIndices: List<Int>, replyIndices: List<Int>) {
+               assertThat(groupReplies[posts[postIndex]]!!["sones"] as Iterable<Sone>, containsInAnyOrder(*soneIndices.map { sones[it] }.toTypedArray()))
+               assertThat(groupReplies[posts[postIndex]]!!["replies"] as Iterable<PostReply>, containsInAnyOrder(*replyIndices.map { replies[it] }.toTypedArray()))
+       }
+
+}