src/test/kotlin/net/pterodactylus/sone/freenet/wot/IdentityLoaderTest.kt

   1 /*
   2  * Sone - IdentityLoaderTest.kt - Copyright © 2013–2020 David Roden
   3  *
   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.
   8  *
   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.
  13  *
  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/>.
  16  */
  17
  18 package net.pterodactylus.sone.freenet.wot
  19
  20 import org.hamcrest.MatcherAssert.*
  21 import org.hamcrest.Matchers.*
  22 import org.junit.*
  23
  24 /**
  25  * Unit test for [IdentityLoader].
  26  */
  27 class IdentityLoaderTest {
  28
  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?) =
  33                                 when (ownIdentity) {
  34                                         ownIdentities[0] -> createTrustedIdentitiesForFirstOwnIdentity()
  35                                         ownIdentities[1] -> createTrustedIdentitiesForSecondOwnIdentity()
  36                                         ownIdentities[2] -> createTrustedIdentitiesForThirdOwnIdentity()
  37                                         ownIdentities[3] -> createTrustedIdentitiesForFourthOwnIdentity()
  38                                         else -> throw RuntimeException()
  39                                 }
  40         }
  41
  42         @Test
  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())
  52         }
  53
  54         @Test
  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())
  64         }
  65
  66         private fun verifyIdentitiesForOwnIdentity(identities: Map<OwnIdentity, Collection<Identity>>, ownIdentity: OwnIdentity, trustedIdentities: Set<Identity>) {
  67                 assertThat(identities[ownIdentity], equalTo<Collection<Identity>>(trustedIdentities))
  68         }
  69
  70 }
  71
  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"))
  77 )
  78
  79 private fun createTrustedIdentitiesForFirstOwnIdentity() = setOf(
  80                 createIdentity("I11", "IN11", "IR11", setOf("Test"), mapOf("KeyA" to "ValueA"))
  81 )
  82
  83 private fun createTrustedIdentitiesForSecondOwnIdentity() = setOf(
  84                 createIdentity("I21", "IN21", "IR21", setOf("Test", "Test2"), mapOf("KeyB" to "ValueB"))
  85 )
  86
  87 private fun createTrustedIdentitiesForThirdOwnIdentity() = setOf(
  88                 createIdentity("I31", "IN31", "IR31", setOf("Test", "Test3"), mapOf("KeyC" to "ValueC"))
  89 )
  90
  91 private fun createTrustedIdentitiesForFourthOwnIdentity(): Set<Identity> = emptySet()
  92
  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 {
  95                         setContexts(contexts)
  96                         this.properties = properties
  97                 }
  98
  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
 103                 }
 104
 105 private open class TestWebOfTrustConnector : WebOfTrustConnector {
 106
 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
 114
 115 }