X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FMatchers.kt;h=9d215573069a6aac03000f32153998d82e456d4c;hb=63d91fb521ea8f126efbccb6152ccc1a872479d1;hp=1f479f6ecec350fa76b599bcf43b9767535aee26;hpb=64740709990291688170ebd1f192af5eb9090618;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 1f479f6..9d21557 100644
--- a/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt
@@ -1,6 +1,7 @@
 package net.pterodactylus.sone.test
 
 import freenet.support.*
+import net.pterodactylus.sone.data.Post
 import net.pterodactylus.sone.freenet.wot.*
 import net.pterodactylus.sone.utils.*
 import net.pterodactylus.util.web.*
@@ -8,6 +9,12 @@ import org.hamcrest.*
 import org.hamcrest.Matchers.*
 
 /**
+ * A kotlin-ified version of Hamcrest’s [anything()][anything]. It matches
+ * everything and has the right type, too!
+ */
+inline fun <reified T> everything(): Matcher<T> = any(T::class.java)
+
+/**
  * Returns a [hamcrest matcher][Matcher] constructed from the given predicate.
  */
 fun <T> matches(description: String? = null, predicate: (T) -> Boolean) = object : TypeSafeDiagnosingMatcher<T>() {
@@ -37,6 +44,10 @@ fun hasHeader(name: String, value: String) = object : TypeSafeDiagnosingMatcher<
 	}
 }
 
+fun <T> handleMatcher(matcher: Matcher<T>, item: T, mismatchDescription: Description) =
+	matcher.matches(item)
+		.onFalse { matcher.describeMismatch(item, mismatchDescription) }
+
 fun <T : Any> compare(value: T, comparison: (T) -> Boolean, onError: (T) -> Unit) =
 		false.takeUnless { comparison(value) }
 				?.also { onError(value) }
@@ -95,9 +106,7 @@ fun isOwnIdentity(id: String, nickname: String, requestUri: String, insertUri: S
 
 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)
-			}
+		handleMatcher(valueMatcher, item.get(name), mismatchDescription)
 
 	override fun describeTo(description: Description) {
 		description
@@ -106,6 +115,9 @@ fun hasField(name: String, valueMatcher: Matcher<String>) = object : TypeSafeDia
 	}
 }
 
+fun isPost(isRecipientId: Matcher<String?> = any(String::class.java)) = AttributeMatcher<Post>("post")
+		.addAttribute("recipient ID", { it.recipientId.orNull() }, isRecipientId)
+
 /**
  * [TypeSafeDiagnosingMatcher] implementation that aims to cut down boilerplate on verifying the attributes
  * of typical container objects.