X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ffreenet%2Fwot%2FIdentityManagerTest.kt;h=90e010b7d9d39d9bc29b3020b372f21c933752e9;hp=3768efdc4634fef26fdcb3d64da29e482891a6e2;hb=1fae41d3e01d0281794d85b2b9b4cacabb4e3091;hpb=bf509980d5097e67d1a32f6b53bef052b396137f

diff --git a/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityManagerTest.kt b/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityManagerTest.kt
index 3768efd..90e010b 100644
--- a/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityManagerTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityManagerTest.kt
@@ -1,33 +1,32 @@
 package net.pterodactylus.sone.freenet.wot
 
-import com.google.common.eventbus.*
-import net.pterodactylus.sone.freenet.plugin.*
-import net.pterodactylus.sone.test.*
-import org.hamcrest.MatcherAssert.*
-import org.hamcrest.Matchers.*
-import org.junit.*
-import org.mockito.Mockito.*
+import com.google.common.eventbus.EventBus
+import net.pterodactylus.sone.freenet.plugin.PluginException
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.equalTo
+import org.junit.Test
 
 /**
  * Unit test for [IdentityManagerImpl].
  */
 class IdentityManagerTest {
 
-	private val eventBus = mock<EventBus>()
-	private val webOfTrustConnector = mock<WebOfTrustConnector>()
-	private val identityManager = IdentityManagerImpl(eventBus, webOfTrustConnector, IdentityLoader(webOfTrustConnector, Context("Test")))
+	private val eventBus = EventBus()
 
 	@Test
 	fun identityManagerPingsWotConnector() {
+		var pingCalled = false
+		val webOfTrustConnector = dummyWebOfTrustConnector.overridePing { Unit.also { pingCalled = true } }
+		val identityManager = IdentityManagerImpl(eventBus, webOfTrustConnector, IdentityLoader(webOfTrustConnector, Context("Test")))
 		assertThat(identityManager.isConnected, equalTo(true))
-		verify(webOfTrustConnector).ping()
+		assertThat(pingCalled, equalTo(true))
 	}
 
 	@Test
 	fun disconnectedWotConnectorIsRecognized() {
-		doThrow(PluginException::class.java).whenever(webOfTrustConnector).ping()
+		val webOfTrustConnector = dummyWebOfTrustConnector.overridePing { throw PluginException() }
+		val identityManager = IdentityManagerImpl(eventBus, webOfTrustConnector, IdentityLoader(webOfTrustConnector, Context("Test")))
 		assertThat(identityManager.isConnected, equalTo(false))
-		verify(webOfTrustConnector).ping()
 	}
 
 }