From: David ‘Bombe’ Roden Date: Tue, 10 Dec 2019 20:35:58 +0000 (+0100) Subject: 🚧 Add new handler for local posts X-Git-Tag: v81^2~5^2~64 X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=commitdiff_plain;h=5c3d55750f4d521ae15f9ad394468d8faefcd828 🚧 Add new handler for local posts --- diff --git a/src/main/kotlin/net/pterodactylus/sone/web/notification/LocalPostHandler.kt b/src/main/kotlin/net/pterodactylus/sone/web/notification/LocalPostHandler.kt new file mode 100644 index 0000000..1ff4ea2 --- /dev/null +++ b/src/main/kotlin/net/pterodactylus/sone/web/notification/LocalPostHandler.kt @@ -0,0 +1,58 @@ +/** + * Sone - NewLocalPostHandler.kt - Copyright © 2019 David ‘Bombe’ Roden + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +package net.pterodactylus.sone.web.notification + +import com.google.common.eventbus.* +import net.pterodactylus.sone.core.event.* +import net.pterodactylus.sone.data.* +import net.pterodactylus.sone.notify.* +import net.pterodactylus.util.notify.* + +/** + * Handler for local posts. + */ +class LocalPostHandler(private val notificationManager: NotificationManager, private val notification: ListNotification) { + + @Subscribe + fun newPostFound(newPostFoundEvent: NewPostFoundEvent) { + newPostFoundEvent.post.onLocal { post -> + notification.add(post) + } + if (!notificationManager.hasFirstStartNotification()) { + notificationManager.addNotification(notification) + } + } + + @Subscribe + fun postRemoved(postRemovedEvent: PostRemovedEvent) { + postRemovedEvent.post.onLocal { post -> + notification.remove(post) + } + } + + @Subscribe + fun postMarkedAsKnown(markPostKnownEvent: MarkPostKnownEvent) { + markPostKnownEvent.post.onLocal { post -> + notification.remove(post) + } + } + +} + +private fun Post.onLocal(action: (Post) -> Unit) = + if (sone.isLocal) action(this) else Unit diff --git a/src/test/kotlin/net/pterodactylus/sone/web/notification/LocalPostHandlerTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/notification/LocalPostHandlerTest.kt new file mode 100644 index 0000000..d435101 --- /dev/null +++ b/src/test/kotlin/net/pterodactylus/sone/web/notification/LocalPostHandlerTest.kt @@ -0,0 +1,115 @@ +/** + * Sone - NewLocalPostHandlerTest.kt - Copyright © 2019 David ‘Bombe’ Roden + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +package net.pterodactylus.sone.web.notification + +import com.google.common.eventbus.* +import net.pterodactylus.sone.core.event.* +import net.pterodactylus.sone.data.* +import net.pterodactylus.sone.data.impl.* +import net.pterodactylus.sone.notify.* +import net.pterodactylus.util.notify.* +import net.pterodactylus.util.template.* +import org.hamcrest.MatcherAssert.* +import org.hamcrest.Matchers.* +import java.io.* +import kotlin.test.* + +/** + * Unit test for [LocalPostHandler]. + */ +class LocalPostHandlerTest { + + private val eventBus = EventBus() + private val notificationManager = NotificationManager() + private val notification = ListNotification("", "", Template()) + + init { + eventBus.register(LocalPostHandler(notificationManager, notification)) + } + + @Test + fun `handler adds post by local sone to notification`() { + eventBus.post(NewPostFoundEvent(localPost)) + assertThat(notification.elements, contains(localPost)) + } + + @Test + fun `handler does not add post by remote sone to notification`() { + eventBus.post(NewPostFoundEvent(remotePost)) + assertThat(notification.elements, emptyIterable()) + } + + @Test + fun `handler adds notification to manager`() { + eventBus.post(NewPostFoundEvent(remotePost)) + assertThat(notificationManager.notifications, contains(notification)) + } + + @Test + fun `handler does not add notification during first start`() { + notificationManager.addNotification(object : AbstractNotification("first-start-notification") { + override fun render(writer: Writer?) = Unit + }) + eventBus.post(NewPostFoundEvent(remotePost)) + assertThat(notificationManager.notifications, not(hasItem(notification))) + } + + @Test + fun `handler removes post from notification when post is removed`() { + notification.add(localPost) + notificationManager.addNotification(notification) + eventBus.post(PostRemovedEvent(localPost)) + assertThat(notification.elements, emptyIterable()) + } + + @Test + fun `handler does not remove remote post from notification when post is removed`() { + notification.add(remotePost) + notificationManager.addNotification(notification) + eventBus.post(PostRemovedEvent(remotePost)) + assertThat(notification.elements, contains(remotePost)) + } + + @Test + fun `handler removes post from notification when post is marked as known`() { + notification.add(localPost) + notificationManager.addNotification(notification) + eventBus.post(MarkPostKnownEvent(localPost)) + assertThat(notification.elements, emptyIterable()) + } + + @Test + fun `handler does not remove remote post from notification when post is marked as known`() { + notification.add(remotePost) + notificationManager.addNotification(notification) + eventBus.post(MarkPostKnownEvent(remotePost)) + assertThat(notification.elements, contains(remotePost)) + } + +} + +private val localSone: Sone = object : IdOnlySone("local-sone") { + override fun isLocal() = true +} +private val localPost: Post = object : Post.EmptyPost("local-post") { + override fun getSone() = localSone +} +private val remoteSone: Sone = IdOnlySone("remote-sone") +private val remotePost: Post = object : Post.EmptyPost("remote-post") { + override fun getSone() = remoteSone +}