X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FMatchers.kt;h=644f192c6ad71c3111adf3bb351da2542ac5e2e1;hb=d0a369a783e84292bbac793191a8cbf6827f395c;hp=c084d35ffdb9489f80aff75e83fb9187a2e41c09;hpb=03cec6a6772c2d836d94864adddaf544cbe9d72f;p=Sone.git

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt b/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt
index c084d35..644f192 100644
--- a/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt
@@ -1,7 +1,10 @@
 package net.pterodactylus.sone.test
 
+import net.pterodactylus.sone.freenet.wot.*
+import net.pterodactylus.sone.utils.*
 import net.pterodactylus.util.web.*
 import org.hamcrest.*
+import org.hamcrest.Matchers.*
 
 fun hasHeader(name: String, value: String) = object : TypeSafeDiagnosingMatcher<Header>() {
 	override fun matchesSafely(item: Header, mismatchDescription: Description) =
@@ -19,3 +22,46 @@ fun <T : Any> compare(value: T, comparison: (T) -> Boolean, onError: (T) -> Unit
 		false.takeUnless { comparison(value) }
 				?.also { onError(value) }
 
+fun <K, V> isEmptyMap() = object : TypeSafeDiagnosingMatcher<Map<K, V>>() {
+	override fun describeTo(description: Description) {
+		description.appendText("empty map")
+	}
+
+	override fun matchesSafely(item: Map<K, V>, mismatchDescription: Description) =
+			item.isEmpty().onFalse {
+				mismatchDescription.appendText("was ").appendValue(item)
+			}
+}
+
+fun isTrust(trust: Int?, score: Int?, rank: Int?) =
+		AttributeMatcher<Trust>("trust")
+				.addAttribute("trust", trust, Trust::explicit)
+				.addAttribute("score", score, Trust::implicit)
+				.addAttribute("rank", rank, Trust::distance)
+
+fun isTrusted(ownIdentity: OwnIdentity, trust: Matcher<Trust>) = object : TypeSafeDiagnosingMatcher<Identity>() {
+	override fun matchesSafely(item: Identity, mismatchDescription: Description) =
+			item.getTrust(ownIdentity)?.let { foundTrust ->
+				trust.matches(foundTrust).onFalse {
+					trust.describeMismatch(foundTrust, mismatchDescription)
+				}
+			} ?: {
+				mismatchDescription.appendText("not trusted")
+				false
+			}()
+
+	override fun describeTo(description: Description) {
+		description
+				.appendText("trusted by ").appendValue(ownIdentity)
+				.appendText(" with ").appendValue(trust)
+	}
+}
+
+fun isIdentity(id: String, nickname: String, requestUri: String, contexts: Matcher<out Iterable<String>>, properties: Matcher<out Map<out String, String>>) =
+		AttributeMatcher<Identity>("identity")
+				.addAttribute("id", id, Identity::getId)
+				.addAttribute("nickname", nickname, Identity::getNickname)
+				.addAttribute("requestUri", requestUri, Identity::getRequestUri)
+				.addAttribute("contexts", Identity::getContexts, contexts)
+				.addAttribute("properties", Identity::getProperties, properties)
+