2 * Sone - IdentityLoaderTest.java - Copyright © 2013–2019 David Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.sone.freenet.wot
20 import org.hamcrest.MatcherAssert.*
21 import org.hamcrest.Matchers.*
25 * Unit test for [IdentityLoader].
27 class IdentityLoaderTest {
29 private val ownIdentities = createOwnIdentities()
30 private val webOfTrustConnector = object : TestWebOfTrustConnector() {
31 override fun loadAllOwnIdentities() = ownIdentities.toSet()
32 override fun loadTrustedIdentities(ownIdentity: OwnIdentity, context: String?) =
34 ownIdentities[0] -> createTrustedIdentitiesForFirstOwnIdentity()
35 ownIdentities[1] -> createTrustedIdentitiesForSecondOwnIdentity()
36 ownIdentities[2] -> createTrustedIdentitiesForThirdOwnIdentity()
37 ownIdentities[3] -> createTrustedIdentitiesForFourthOwnIdentity()
38 else -> throw RuntimeException()
43 fun loadingIdentities() {
44 val identityLoader = IdentityLoader(webOfTrustConnector, Context("Test"))
45 val identities = identityLoader.loadIdentities()
46 assertThat(identities.keys, hasSize(4))
47 assertThat(identities.keys, containsInAnyOrder(ownIdentities[0], ownIdentities[1], ownIdentities[2], ownIdentities[3]))
48 verifyIdentitiesForOwnIdentity(identities, ownIdentities[0], createTrustedIdentitiesForFirstOwnIdentity())
49 verifyIdentitiesForOwnIdentity(identities, ownIdentities[1], createTrustedIdentitiesForSecondOwnIdentity())
50 verifyIdentitiesForOwnIdentity(identities, ownIdentities[2], emptySet())
51 verifyIdentitiesForOwnIdentity(identities, ownIdentities[3], createTrustedIdentitiesForFourthOwnIdentity())
55 fun loadingIdentitiesWithoutContext() {
56 val identityLoaderWithoutContext = IdentityLoader(webOfTrustConnector)
57 val identities = identityLoaderWithoutContext.loadIdentities()
58 assertThat(identities.keys, hasSize(4))
59 assertThat(identities.keys, containsInAnyOrder(ownIdentities[0], ownIdentities[1], ownIdentities[2], ownIdentities[3]))
60 verifyIdentitiesForOwnIdentity(identities, ownIdentities[0], createTrustedIdentitiesForFirstOwnIdentity())
61 verifyIdentitiesForOwnIdentity(identities, ownIdentities[1], createTrustedIdentitiesForSecondOwnIdentity())
62 verifyIdentitiesForOwnIdentity(identities, ownIdentities[2], createTrustedIdentitiesForThirdOwnIdentity())
63 verifyIdentitiesForOwnIdentity(identities, ownIdentities[3], createTrustedIdentitiesForFourthOwnIdentity())
66 private fun verifyIdentitiesForOwnIdentity(identities: Map<OwnIdentity, Collection<Identity>>, ownIdentity: OwnIdentity, trustedIdentities: Set<Identity>) {
67 assertThat(identities[ownIdentity], equalTo<Collection<Identity>>(trustedIdentities))
72 private fun createOwnIdentities() = listOf(
73 createOwnIdentity("O1", "ON1", "OR1", "OI1", setOf("Test", "Test2"), mapOf("KeyA" to "ValueA", "KeyB" to "ValueB")),
74 createOwnIdentity("O2", "ON2", "OR2", "OI2", setOf("Test"), mapOf("KeyC" to "ValueC")),
75 createOwnIdentity("O3", "ON3", "OR3", "OI3", setOf("Test2"), mapOf("KeyE" to "ValueE", "KeyD" to "ValueD")),
76 createOwnIdentity("O4", "ON4", "OR$", "OI4", setOf("Test"), mapOf("KeyA" to "ValueA", "KeyD" to "ValueD"))
79 private fun createTrustedIdentitiesForFirstOwnIdentity() = setOf(
80 createIdentity("I11", "IN11", "IR11", setOf("Test"), mapOf("KeyA" to "ValueA"))
83 private fun createTrustedIdentitiesForSecondOwnIdentity() = setOf(
84 createIdentity("I21", "IN21", "IR21", setOf("Test", "Test2"), mapOf("KeyB" to "ValueB"))
87 private fun createTrustedIdentitiesForThirdOwnIdentity() = setOf(
88 createIdentity("I31", "IN31", "IR31", setOf("Test", "Test3"), mapOf("KeyC" to "ValueC"))
91 private fun createTrustedIdentitiesForFourthOwnIdentity(): Set<Identity> = emptySet()
93 private fun createOwnIdentity(id: String, nickname: String, requestUri: String, insertUri: String, contexts: Set<String>, properties: Map<String, String>): OwnIdentity =
94 DefaultOwnIdentity(id, nickname, requestUri, insertUri).apply {
96 this.properties = properties
99 private fun createIdentity(id: String, nickname: String, requestUri: String, contexts: Set<String>, properties: Map<String, String>): Identity =
100 DefaultIdentity(id, nickname, requestUri).apply {
101 setContexts(contexts)
102 this.properties = properties
105 private open class TestWebOfTrustConnector : WebOfTrustConnector {
107 override fun loadAllOwnIdentities() = emptySet<OwnIdentity>()
108 override fun loadTrustedIdentities(ownIdentity: OwnIdentity, context: String?) = emptySet<Identity>()
109 override fun addContext(ownIdentity: OwnIdentity, context: String) = Unit
110 override fun removeContext(ownIdentity: OwnIdentity, context: String) = Unit
111 override fun setProperty(ownIdentity: OwnIdentity, name: String, value: String) = Unit
112 override fun removeProperty(ownIdentity: OwnIdentity, name: String) = Unit
113 override fun ping() = Unit