🚧 Add matcher for trust values

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / test / Matchers.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt b/src/test/kotlin/net/pterodactylus/sone/test/Matchers.kt
index c084d35..3bbdd23 100644 (file)
--- 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,20 @@ 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)
+