/*
* Sone - IdentityLoaderTest.java - Copyright © 2013–2019 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
* 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.freenet.wot
import org.hamcrest.MatcherAssert.*
import org.hamcrest.Matchers.*
import org.junit.*
/**
* 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.loadIdentities()
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.loadIdentities()
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: OwnIdentity, trustedIdentities: Set) {
assertThat(identities[ownIdentity], equalTo>(trustedIdentities))
}
}
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 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
}