X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FMatchers.kt;h=85ed9e7ee4004557c8f2bd4828dcd40411dbda89;hp=3bbdd23f88f3373bad0b570139e42c5815a0142a;hb=faf66247a34f64946990a985d2ea3003465969cb;hpb=11b68b3a9d2437ee951e06779f337b20084f01f0

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt b/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt
index 3bbdd23..85ed9e7 100644
--- a/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt
@@ -1,11 +1,31 @@
 package net.pterodactylus.sone.test
 
+import freenet.support.*
 import net.pterodactylus.sone.freenet.wot.*
 import net.pterodactylus.sone.utils.*
 import net.pterodactylus.util.web.*
 import org.hamcrest.*
+import org.hamcrest.Matchers
 import org.hamcrest.Matchers.*
 
+/**
+ * Returns a [hamcrest matcher][Matcher] constructed from the given predicate.
+ */
+fun <T> matches(description: String? = null, predicate: (T) -> Boolean) = object : TypeSafeDiagnosingMatcher<T>() {
+
+	override fun matchesSafely(item: T, mismatchDescription: Description) =
+			predicate(item).onFalse {
+				mismatchDescription.appendValue(item).appendText(" did not match predicate")
+			}
+
+	override fun describeTo(description: Description) {
+		description.appendText("matches predicate ").appendValue(predicate)
+	}
+
+}.let { matcher ->
+	description?.let { describedAs(description, matcher) } ?: matcher
+}
+
 fun hasHeader(name: String, value: String) = object : TypeSafeDiagnosingMatcher<Header>() {
 	override fun matchesSafely(item: Header, mismatchDescription: Description) =
 			compare(item.name, { it.equals(name, ignoreCase = true) }) { mismatchDescription.appendText("name is ").appendValue(it) }
@@ -39,3 +59,106 @@ fun isTrust(trust: Int?, score: Int?, rank: Int?) =
 				.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)
+
+fun isOwnIdentity(id: String, nickname: String, requestUri: String, insertUri: String, contexts: Matcher<Iterable<String>>, properties: Matcher<Map<out String, String>>) =
+		AttributeMatcher<OwnIdentity>("own identity")
+				.addAttribute("id", id, OwnIdentity::getId)
+				.addAttribute("nickname", nickname, OwnIdentity::getNickname)
+				.addAttribute("request uri", requestUri, OwnIdentity::getRequestUri)
+				.addAttribute("insert uri", insertUri, OwnIdentity::getInsertUri)
+				.addAttribute("contexts", OwnIdentity::getContexts, contexts)
+				.addAttribute("properties", OwnIdentity::getProperties, properties)
+
+fun hasField(name: String, valueMatcher: Matcher<String>) = object : TypeSafeDiagnosingMatcher<SimpleFieldSet>() {
+	override fun matchesSafely(item: SimpleFieldSet, mismatchDescription: Description) =
+			valueMatcher.matches(item.get(name)).onFalse {
+				valueMatcher.describeMismatch(item, mismatchDescription)
+			}
+
+	override fun describeTo(description: Description) {
+		description
+				.appendText("simple field set with key ").appendValue(name)
+				.appendText(", value ").appendValue(valueMatcher)
+	}
+}
+
+/**
+ * [TypeSafeDiagnosingMatcher] implementation that aims to cut down boilerplate on verifying the attributes
+ * of typical container objects.
+ */
+class AttributeMatcher<T>(private val objectName: String) : TypeSafeDiagnosingMatcher<T>() {
+
+	private data class AttributeToMatch<T, V>(
+			val name: String,
+			val getter: (T) -> V,
+			val matcher: Matcher<out V>
+	)
+
+	private val attributesToMatch = mutableListOf<AttributeToMatch<T, *>>()
+
+	/**
+	 * Adds an attribute to check for equality, returning `this`.
+	 */
+	fun <V> addAttribute(name: String, expected: V, getter: (T) -> V): AttributeMatcher<T> = apply {
+		attributesToMatch.add(AttributeToMatch(name, getter, describedAs("$name %0", equalTo(expected), expected)))
+	}
+
+	/**
+	 * Adds an attribute to check with the given [hamcrest matcher][Matcher].
+	 */
+	fun <V> addAttribute(name: String, getter: (T) -> V, matcher: Matcher<out V>) = apply {
+		attributesToMatch.add(AttributeToMatch(name, getter, matcher))
+	}
+
+	override fun describeTo(description: Description) {
+		attributesToMatch.forEachIndexed { index, attributeToMatch ->
+			if (index == 0) {
+				description.appendText("$objectName with ")
+			} else {
+				description.appendText(", ")
+			}
+			attributeToMatch.matcher.describeTo(description)
+		}
+	}
+
+	override fun matchesSafely(item: T, mismatchDescription: Description): Boolean =
+			attributesToMatch.fold(true) { matches, attributeToMatch ->
+				if (!matches) {
+					false
+				} else {
+					if (!attributeToMatch.matcher.matches(attributeToMatch.getter(item))) {
+						mismatchDescription.appendText("but ${attributeToMatch.name} ")
+						attributeToMatch.matcher.describeMismatch(attributeToMatch.getter(item), mismatchDescription)
+						false
+					} else {
+						true
+					}
+				}
+			}
+
+}