X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ffreenet%2Fwot%2FIdentityLoaderTest.kt;h=0e0e336551834dfeda39011b5f4fac4ce3dcd12c;hp=5c363fca9d6d0e1cdea7f41e09ac0ec9ec7da8d3;hb=6ebf42a54cc1ef44bdd762ad06e9b1a84bd7ad25;hpb=46d3d75e26cda2484932fcc2607d6768bc99e810 diff --git a/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityLoaderTest.kt b/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityLoaderTest.kt index 5c363fc..0e0e336 100644 --- a/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityLoaderTest.kt +++ b/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityLoaderTest.kt @@ -1,5 +1,5 @@ /* - * Sone - IdentityLoaderTest.java - Copyright © 2013–2019 David Roden + * Sone - IdentityLoaderTest.kt - Copyright © 2013–2020 David 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 @@ -17,77 +17,32 @@ package net.pterodactylus.sone.freenet.wot -import com.google.common.base.Optional.* -import net.pterodactylus.sone.test.* import org.hamcrest.MatcherAssert.* import org.hamcrest.Matchers.* import org.junit.* -import org.mockito.ArgumentMatchers.any -import org.mockito.ArgumentMatchers.eq -import org.mockito.Mockito.never -import org.mockito.Mockito.verify /** * Unit test for [IdentityLoader]. */ class IdentityLoaderTest { - private val webOfTrustConnector = mock() - private val identityLoader = IdentityLoader(webOfTrustConnector, of(Context("Test"))) - private val identityLoaderWithoutContext = IdentityLoader(webOfTrustConnector) - - @Before - fun setup() { - val ownIdentities = createOwnIdentities() - whenever(webOfTrustConnector.loadAllOwnIdentities()).thenReturn(ownIdentities.toSet()) - whenever(webOfTrustConnector.loadTrustedIdentities(eq(ownIdentities[0]), any())).thenReturn(createTrustedIdentitiesForFirstOwnIdentity()) - whenever(webOfTrustConnector.loadTrustedIdentities(eq(ownIdentities[1]), any())).thenReturn(createTrustedIdentitiesForSecondOwnIdentity()) - whenever(webOfTrustConnector.loadTrustedIdentities(eq(ownIdentities[2]), any())).thenReturn(createTrustedIdentitiesForThirdOwnIdentity()) - whenever(webOfTrustConnector.loadTrustedIdentities(eq(ownIdentities[3]), any())).thenReturn(createTrustedIdentitiesForFourthOwnIdentity()) + private val ownIdentities = createOwnIdentities() + private val webOfTrustConnector = object : TestWebOfTrustConnector() { + override fun loadAllOwnIdentities() = ownIdentities.toSet() + override fun loadTrustedIdentities(ownIdentity: OwnIdentity, context: String?) = + when (ownIdentity) { + ownIdentities[0] -> createTrustedIdentitiesForFirstOwnIdentity() + ownIdentities[1] -> createTrustedIdentitiesForSecondOwnIdentity() + ownIdentities[2] -> createTrustedIdentitiesForThirdOwnIdentity() + ownIdentities[3] -> createTrustedIdentitiesForFourthOwnIdentity() + else -> throw RuntimeException() + } } - private fun createOwnIdentities() = listOf( - createOwnIdentity("O1", "ON1", "OR1", "OI1", listOf("Test", "Test2"), mapOf("KeyA" to "ValueA", "KeyB" to "ValueB")), - createOwnIdentity("O2", "ON2", "OR2", "OI2", listOf("Test"), mapOf("KeyC" to "ValueC")), - createOwnIdentity("O3", "ON3", "OR3", "OI3", listOf("Test2"), mapOf("KeyE" to "ValueE", "KeyD" to "ValueD")), - createOwnIdentity("O4", "ON4", "OR$", "OI4", listOf("Test"), mapOf("KeyA" to "ValueA", "KeyD" to "ValueD")) - ) - - private fun createTrustedIdentitiesForFirstOwnIdentity() = setOf( - createIdentity("I11", "IN11", "IR11", listOf("Test"), mapOf("KeyA" to "ValueA")) - ) - - private fun createTrustedIdentitiesForSecondOwnIdentity() = setOf( - createIdentity("I21", "IN21", "IR21", listOf("Test", "Test2"), mapOf("KeyB" to "ValueB")) - ) - - private fun createTrustedIdentitiesForThirdOwnIdentity() = setOf( - createIdentity("I31", "IN31", "IR31", listOf("Test", "Test3"), mapOf("KeyC" to "ValueC")) - ) - - private fun createTrustedIdentitiesForFourthOwnIdentity(): Set = emptySet() - - private fun createOwnIdentity(id: String, nickname: String, requestUri: String, insertUri: String, contexts: List, properties: Map): OwnIdentity = - DefaultOwnIdentity(id, nickname, requestUri, insertUri).apply { - setContexts(contexts) - this.properties = properties - } - - private fun createIdentity(id: String, nickname: String, requestUri: String, contexts: List, properties: Map): Identity = - DefaultIdentity(id, nickname, requestUri).apply { - setContexts(contexts) - this.properties = properties - } - @Test fun loadingIdentities() { - val ownIdentities = createOwnIdentities() - val identities = identityLoader.loadIdentities() - verify(webOfTrustConnector).loadAllOwnIdentities() - verify(webOfTrustConnector).loadTrustedIdentities(eq(ownIdentities[0]), eq(of("Test"))) - verify(webOfTrustConnector).loadTrustedIdentities(eq(ownIdentities[1]), eq(of("Test"))) - verify(webOfTrustConnector, never()).loadTrustedIdentities(eq(ownIdentities[2]), any()) - verify(webOfTrustConnector).loadTrustedIdentities(eq(ownIdentities[3]), eq(of("Test"))) + val identityLoader = IdentityLoader(webOfTrustConnector, Context("Test")) + val identities = identityLoader.loadTrustedIdentities() assertThat(identities.keys, hasSize(4)) assertThat(identities.keys, containsInAnyOrder(ownIdentities[0], ownIdentities[1], ownIdentities[2], ownIdentities[3])) verifyIdentitiesForOwnIdentity(identities, ownIdentities[0], createTrustedIdentitiesForFirstOwnIdentity()) @@ -98,23 +53,14 @@ class IdentityLoaderTest { @Test fun loadingIdentitiesWithoutContext() { - val ownIdentities = createOwnIdentities() - val identities = identityLoaderWithoutContext.loadIdentities() - verify(webOfTrustConnector).loadAllOwnIdentities() - verify(webOfTrustConnector).loadTrustedIdentities(eq(ownIdentities[0]), eq(absent())) - verify(webOfTrustConnector).loadTrustedIdentities(eq(ownIdentities[1]), eq(absent())) - verify(webOfTrustConnector).loadTrustedIdentities(eq(ownIdentities[2]), eq(absent())) - verify(webOfTrustConnector).loadTrustedIdentities(eq(ownIdentities[3]), eq(absent())) + val identityLoaderWithoutContext = IdentityLoader(webOfTrustConnector) + val identities = identityLoaderWithoutContext.loadTrustedIdentities() assertThat(identities.keys, hasSize(4)) - val firstOwnIdentity = ownIdentities[0] - val secondOwnIdentity = ownIdentities[1] - val thirdOwnIdentity = ownIdentities[2] - val fourthOwnIdentity = ownIdentities[3] - assertThat(identities.keys, containsInAnyOrder(firstOwnIdentity, secondOwnIdentity, thirdOwnIdentity, fourthOwnIdentity)) - verifyIdentitiesForOwnIdentity(identities, firstOwnIdentity, createTrustedIdentitiesForFirstOwnIdentity()) - verifyIdentitiesForOwnIdentity(identities, secondOwnIdentity, createTrustedIdentitiesForSecondOwnIdentity()) - verifyIdentitiesForOwnIdentity(identities, thirdOwnIdentity, createTrustedIdentitiesForThirdOwnIdentity()) - verifyIdentitiesForOwnIdentity(identities, fourthOwnIdentity, createTrustedIdentitiesForFourthOwnIdentity()) + assertThat(identities.keys, containsInAnyOrder(ownIdentities[0], ownIdentities[1], ownIdentities[2], ownIdentities[3])) + verifyIdentitiesForOwnIdentity(identities, ownIdentities[0], createTrustedIdentitiesForFirstOwnIdentity()) + verifyIdentitiesForOwnIdentity(identities, ownIdentities[1], createTrustedIdentitiesForSecondOwnIdentity()) + verifyIdentitiesForOwnIdentity(identities, ownIdentities[2], createTrustedIdentitiesForThirdOwnIdentity()) + verifyIdentitiesForOwnIdentity(identities, ownIdentities[3], createTrustedIdentitiesForFourthOwnIdentity()) } private fun verifyIdentitiesForOwnIdentity(identities: Map>, ownIdentity: OwnIdentity, trustedIdentities: Set) { @@ -122,3 +68,49 @@ class IdentityLoaderTest { } } + +private fun createOwnIdentities() = listOf( + createOwnIdentity("O1", "ON1", "OR1", "OI1", setOf("Test", "Test2"), mapOf("KeyA" to "ValueA", "KeyB" to "ValueB")), + createOwnIdentity("O2", "ON2", "OR2", "OI2", setOf("Test"), mapOf("KeyC" to "ValueC")), + createOwnIdentity("O3", "ON3", "OR3", "OI3", setOf("Test2"), mapOf("KeyE" to "ValueE", "KeyD" to "ValueD")), + createOwnIdentity("O4", "ON4", "OR$", "OI4", setOf("Test"), mapOf("KeyA" to "ValueA", "KeyD" to "ValueD")) +) + +private fun createTrustedIdentitiesForFirstOwnIdentity() = setOf( + createIdentity("I11", "IN11", "IR11", setOf("Test"), mapOf("KeyA" to "ValueA")) +) + +private fun createTrustedIdentitiesForSecondOwnIdentity() = setOf( + createIdentity("I21", "IN21", "IR21", setOf("Test", "Test2"), mapOf("KeyB" to "ValueB")) +) + +private fun createTrustedIdentitiesForThirdOwnIdentity() = setOf( + createIdentity("I31", "IN31", "IR31", setOf("Test", "Test3"), mapOf("KeyC" to "ValueC")) +) + +private fun createTrustedIdentitiesForFourthOwnIdentity(): Set = emptySet() + +private fun createOwnIdentity(id: String, nickname: String, requestUri: String, insertUri: String, contexts: Set, properties: Map): OwnIdentity = + DefaultOwnIdentity(id, nickname, requestUri, insertUri).apply { + setContexts(contexts) + this.properties = properties + } + +private fun createIdentity(id: String, nickname: String, requestUri: String, contexts: Set, properties: Map): Identity = + DefaultIdentity(id, nickname, requestUri).apply { + setContexts(contexts) + this.properties = properties + } + +private open class TestWebOfTrustConnector : WebOfTrustConnector { + + override fun loadAllOwnIdentities() = emptySet() + override fun loadTrustedIdentities(ownIdentity: OwnIdentity, context: String?) = emptySet() + override fun loadAllIdentities(ownIdentity: OwnIdentity, context: String?) = emptySet() + override fun addContext(ownIdentity: OwnIdentity, context: String) = Unit + override fun removeContext(ownIdentity: OwnIdentity, context: String) = Unit + override fun setProperty(ownIdentity: OwnIdentity, name: String, value: String) = Unit + override fun removeProperty(ownIdentity: OwnIdentity, name: String) = Unit + override fun ping() = Unit + +}