package net.pterodactylus.sone.freenet.wot
-import org.hamcrest.MatcherAssert.*
-import org.hamcrest.Matchers.*
-import org.junit.*
+import net.pterodactylus.sone.test.createIdentity
+import net.pterodactylus.sone.test.createOwnIdentity
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.equalTo
+import org.junit.Test
/**
* Unit test for [IdentityLoader].
*/
class IdentityLoaderTest {
- 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()
- }
- }
-
- @Test
- fun loadingIdentities() {
- 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())
- verifyIdentitiesForOwnIdentity(identities, ownIdentities[1], createTrustedIdentitiesForSecondOwnIdentity())
- verifyIdentitiesForOwnIdentity(identities, ownIdentities[2], emptySet())
- verifyIdentitiesForOwnIdentity(identities, ownIdentities[3], createTrustedIdentitiesForFourthOwnIdentity())
- }
-
@Test
- fun loadingIdentitiesWithoutContext() {
- val identityLoaderWithoutContext = IdentityLoader(webOfTrustConnector)
- val identities = identityLoaderWithoutContext.loadTrustedIdentities()
- assertThat(identities.keys, hasSize(4))
- 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, Collection<Identity>>, ownIdentity: OwnIdentity, trustedIdentities: Set<Identity>) {
- assertThat(identities[ownIdentity], equalTo<Collection<Identity>>(trustedIdentities))
+ fun `loading all identities merges remote identities’ trust values`() {
+ val ownIdentity1 = createOwnIdentity("o1")
+ val ownIdentity2 = createOwnIdentity("o2")
+ val webOfTrustConnector = dummyWebOfTrustConnector
+ .overrideLoadAllOwnIdentities { setOf(ownIdentity1, ownIdentity2) }
+ .overrideLoadAllIdentities { ownIdentity, _ ->
+ when (ownIdentity) {
+ ownIdentity1 -> setOf(createIdentity().setTrust(ownIdentity1, Trust(100, 50, 2)))
+ else -> setOf(createIdentity().setTrust(ownIdentity2, Trust(80, 40, 2)))
+ }
+ }
+ val identityLoader = IdentityLoader(webOfTrustConnector)
+ val allIdentities = identityLoader.loadAllIdentities()
+ assertThat(allIdentities[ownIdentity1]!!.first().trust[ownIdentity2], equalTo(Trust(80, 40, 2)))
+ assertThat(allIdentities[ownIdentity2]!!.first().trust[ownIdentity1], equalTo(Trust(100, 50, 2)))
}
}
-
-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<Identity> = emptySet()
-
-private fun createOwnIdentity(id: String, nickname: String, requestUri: String, insertUri: String, contexts: Set<String>, properties: Map<String, String>): OwnIdentity =
- DefaultOwnIdentity(id, nickname, requestUri, insertUri).apply {
- setContexts(contexts)
- this.properties = properties
- }
-
-private fun createIdentity(id: String, nickname: String, requestUri: String, contexts: Set<String>, properties: Map<String, String>): Identity =
- DefaultIdentity(id, nickname, requestUri).apply {
- setContexts(contexts)
- this.properties = properties
- }
-
-private open class TestWebOfTrustConnector : WebOfTrustConnector {
-
- override fun loadAllOwnIdentities() = emptySet<OwnIdentity>()
- override fun loadTrustedIdentities(ownIdentity: OwnIdentity, context: String?) = emptySet<Identity>()
- 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
-
-}