Add unit test for index page

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/IndexPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/IndexPageTest.kt
new file mode 100644 (file)
index 0000000..414c0f2
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/web/IndexPageTest.kt
@@ -0,0 +1,80 @@
+package net.pterodactylus.sone.web
+
+import com.google.common.base.Optional.fromNullable
+import com.google.common.base.Predicate
+import net.pterodactylus.sone.data.Post
+import net.pterodactylus.sone.data.Sone
+import net.pterodactylus.sone.notify.PostVisibilityFilter
+import net.pterodactylus.sone.test.mock
+import net.pterodactylus.sone.test.whenever
+import net.pterodactylus.util.web.Method.GET
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.contains
+import org.junit.Before
+import org.junit.Test
+import org.mockito.ArgumentMatchers
+
+/**
+ * Unit test for [IndexPage].
+ */
+class IndexPageTest : WebPageTest() {
+
+       private val postVisibilityFilter = mock<PostVisibilityFilter>()
+       private val page = IndexPage(template, webInterface, postVisibilityFilter)
+
+       @Before
+       fun setupPostVisibilityFilter() {
+               whenever(postVisibilityFilter.isVisible(ArgumentMatchers.eq(currentSone))).thenReturn(object : Predicate<Post> {
+                       override fun apply(input: Post?) = true
+               })
+       }
+
+       private fun createPost(time: Long, directed: Boolean = false) = mock<Post>().apply {
+               whenever(this.time).thenReturn(time)
+               whenever(recipient).thenReturn(fromNullable(if (directed) currentSone else null))
+       }
+
+       @Test
+       fun `index page shows all posts of current sone`() {
+               val posts = listOf(createPost(3000), createPost(2000), createPost(1000))
+               whenever(currentSone.posts).thenReturn(posts)
+               request("", GET)
+               page.handleRequest(freenetRequest, templateContext)
+               @Suppress("UNCHECKED_CAST")
+               assertThat(templateContext["posts"] as Iterable<Post>, contains(*posts.toTypedArray()))
+       }
+
+       @Test
+       fun `index page shows posts directed at current sone from non-followed sones`() {
+               val posts = listOf(createPost(3000), createPost(2000), createPost(1000))
+               whenever(currentSone.posts).thenReturn(posts)
+               val notFollowedSone = mock<Sone>()
+               val notFollowedPosts = listOf(createPost(2500, true), createPost(1500))
+               whenever(notFollowedSone.posts).thenReturn(notFollowedPosts)
+               addSone("notfollowed1", notFollowedSone)
+               request("", GET)
+               page.handleRequest(freenetRequest, templateContext)
+               @Suppress("UNCHECKED_CAST")
+               assertThat(templateContext["posts"] as Iterable<Post>, contains(
+                               posts[0], notFollowedPosts[0], posts[1], posts[2]
+               ))
+       }
+
+       @Test
+       fun `index page does not show duplicate posts`() {
+               val posts = listOf(createPost(3000), createPost(2000), createPost(1000))
+               whenever(currentSone.posts).thenReturn(posts)
+               val followedSone = mock<Sone>()
+               val followedPosts = listOf(createPost(2500, true), createPost(1500))
+               whenever(followedSone.posts).thenReturn(followedPosts)
+               whenever(currentSone.friends).thenReturn(listOf("followed1", "followed2"))
+               addSone("followed1", followedSone)
+               request("", GET)
+               page.handleRequest(freenetRequest, templateContext)
+               @Suppress("UNCHECKED_CAST")
+               assertThat(templateContext["posts"] as Iterable<Post>, contains(
+                               posts[0], followedPosts[0], posts[1], followedPosts[1], posts[2]
+               ))
+       }
+
+}