From 439fb3dfa1ccf157388a0e0b306013d0aa46bb50 Mon Sep 17 00:00:00 2001 From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?= Date: Thu, 12 Dec 2019 20:02:56 +0100 Subject: [PATCH] =?utf8?q?=F0=9F=9A=A7=20Change=20rescheduling=20method?= MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit --- .../kotlin/net/pterodactylus/sone/freenet/wot/WebOfTrustPinger.kt | 5 +++-- .../net/pterodactylus/sone/freenet/wot/WebOfTrustPingerTest.kt | 3 ++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/main/kotlin/net/pterodactylus/sone/freenet/wot/WebOfTrustPinger.kt b/src/main/kotlin/net/pterodactylus/sone/freenet/wot/WebOfTrustPinger.kt index 9d0522d..026389f 100644 --- a/src/main/kotlin/net/pterodactylus/sone/freenet/wot/WebOfTrustPinger.kt +++ b/src/main/kotlin/net/pterodactylus/sone/freenet/wot/WebOfTrustPinger.kt @@ -22,6 +22,7 @@ import net.pterodactylus.sone.core.event.* import net.pterodactylus.sone.freenet.plugin.* import net.pterodactylus.sone.utils.* import java.util.concurrent.atomic.* +import java.util.function.* import javax.inject.* /** @@ -32,7 +33,7 @@ import javax.inject.* class WebOfTrustPinger @Inject constructor( private val eventBus: EventBus, @Named("webOfTrustReacher") private val webOfTrustReacher: Runnable, - @Named("webOfTrustReschedule") private val reschedule: Runnable) : Runnable { + @Named("webOfTrustReschedule") private val reschedule: Consumer) : Runnable { private val lastState = AtomicBoolean(false) @@ -49,7 +50,7 @@ class WebOfTrustPinger @Inject constructor( lastState.set(false) } } - reschedule() + reschedule(this) } } diff --git a/src/test/kotlin/net/pterodactylus/sone/freenet/wot/WebOfTrustPingerTest.kt b/src/test/kotlin/net/pterodactylus/sone/freenet/wot/WebOfTrustPingerTest.kt index 9743449..cbd9172 100644 --- a/src/test/kotlin/net/pterodactylus/sone/freenet/wot/WebOfTrustPingerTest.kt +++ b/src/test/kotlin/net/pterodactylus/sone/freenet/wot/WebOfTrustPingerTest.kt @@ -24,6 +24,7 @@ import net.pterodactylus.sone.utils.* import org.hamcrest.MatcherAssert.* import org.hamcrest.Matchers.* import java.util.concurrent.atomic.* +import java.util.function.* import kotlin.test.* /** @@ -35,7 +36,7 @@ class WebOfTrustPingerTest { private val webOfTrustReachable = AtomicBoolean() private val webOfTrustReacher = Runnable { webOfTrustReachable.get().onFalse { throw PluginException() } } private val rescheduled = AtomicBoolean() - private val reschedule = Runnable { rescheduled.set(true) } + private val reschedule: Consumer = Consumer { if (it == pinger) rescheduled.set(true) } private val pinger = WebOfTrustPinger(eventBus, webOfTrustReacher, reschedule) @Test -- 2.7.4