From 5c3d55750f4d521ae15f9ad394468d8faefcd828 Mon Sep 17 00:00:00 2001 From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?= Date: Tue, 10 Dec 2019 21:35:58 +0100 Subject: [PATCH] =?utf8?q?=F0=9F=9A=A7=20Add=20new=20handler=20for=20local?= =?utf8?q?=20posts?= MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit --- .../sone/web/notification/LocalPostHandler.kt | 58 +++++++++++ .../sone/web/notification/LocalPostHandlerTest.kt | 115 +++++++++++++++++++++ 2 files changed, 173 insertions(+) create mode 100644 src/main/kotlin/net/pterodactylus/sone/web/notification/LocalPostHandler.kt create mode 100644 src/test/kotlin/net/pterodactylus/sone/web/notification/LocalPostHandlerTest.kt 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 +} -- 2.7.4