X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FNewElementsTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FNewElementsTest.kt;h=2b4436dbfceb5e3b95762a2ee9a4b3f5bd8ed7aa;hb=5c1acc9f42d039acb0174e3ab1266d76122675ef;hp=ccda1c7850f2414f9279d8189d1ced1b950cb8f0;hpb=fda2239d4bd30e90a397d4cc3726250270876856;p=Sone.git

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/NewElementsTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/NewElementsTest.kt
index ccda1c7..2b4436d 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/NewElementsTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/NewElementsTest.kt
@@ -20,11 +20,14 @@ import com.google.inject.Guice
 import com.google.inject.name.Names.named
 import net.pterodactylus.sone.data.Post
 import net.pterodactylus.sone.data.PostReply
+import net.pterodactylus.sone.data.Sone
 import net.pterodactylus.sone.notify.ListNotification
+import net.pterodactylus.sone.notify.PostVisibilityFilter
 import net.pterodactylus.sone.notify.matchThisPost
 import net.pterodactylus.sone.notify.matchThisReply
 import net.pterodactylus.sone.notify.showAllPosts
 import net.pterodactylus.sone.notify.showAllReplies
+import net.pterodactylus.sone.test.createLocalSone
 import net.pterodactylus.sone.test.createPost
 import net.pterodactylus.sone.test.createPostReply
 import net.pterodactylus.sone.test.getInstance
@@ -84,9 +87,19 @@ class NewElementsTest {
 		assertThat(newElements.newReplies(), containsInAnyOrder(reply2))
 	}
 
+	@Test
+	fun `new posts are filtered using the given Sone`() {
+		val postVisibilityFilter = object : PostVisibilityFilter {
+			override fun isPostVisible(sone: Sone?, post: Post) = (sone == localSone) && (post == post2)
+		}
+		val newElements = NewElements(newPostNotification, newReplyNotification, localPostNotification, localReplyNotification, postVisibilityFilter, showAllReplies)
+		assertThat(newElements.newPosts(localSone), contains(post2))
+	}
+
 }
 
 private val post1 = createPost()
 private val post2 = createPost()
 private val reply1 = createPostReply()
 private val reply2 = createPostReply()
+private val localSone = createLocalSone()