X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ffreenet%2Fwot%2FIdentityLoaderTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ffreenet%2Fwot%2FIdentityLoaderTest.kt;h=780288d15056156364a9a89f33d9762b34141489;hp=0000000000000000000000000000000000000000;hb=d50730f6a330439e0e7ef97ca9329dffe72d5640;hpb=97fe04482ebb8a08e43294acde041c2975cbd8ee

diff --git a/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityLoaderTest.kt b/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityLoaderTest.kt
new file mode 100644
index 0000000..780288d
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityLoaderTest.kt
@@ -0,0 +1,118 @@
+/*
+ * 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 <http://www.gnu.org/licenses/>.
+ */
+
+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, Collection<Identity>>, ownIdentity: OwnIdentity, trustedIdentities: Set<Identity>) {
+		assertThat(identities[ownIdentity], equalTo<Collection<Identity>>(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<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 getTrust(ownIdentity: OwnIdentity, identity: Identity) = Trust(null, null, null)
+	override fun setTrust(ownIdentity: OwnIdentity, identity: Identity, trust: Int, comment: String) = Unit
+	override fun removeTrust(ownIdentity: OwnIdentity, identity: Identity) = Unit
+	override fun ping() = Unit
+
+}