Add paginate() extension method and pagination matcher

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

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/PaginationMatcher.kt b/src/test/kotlin/net/pterodactylus/sone/test/PaginationMatcher.kt
new file mode 100644 (file)
index 0000000..5a47ca7
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/test/PaginationMatcher.kt
@@ -0,0 +1,46 @@
+package net.pterodactylus.sone.test
+
+import net.pterodactylus.sone.utils.Pagination
+import org.hamcrest.Description
+import org.hamcrest.TypeSafeDiagnosingMatcher
+
+/**
+ * Hamcrest matcher for [Pagination]s.
+ */
+class PaginationMatcher(private val page: Int? = null, private val pages: Int? = null):
+               TypeSafeDiagnosingMatcher<Pagination<*>>() {
+
+       override fun matchesSafely(pagination: Pagination<*>, mismatchDescription: Description): Boolean {
+               page?.let {
+                       if (pagination.page != page) {
+                               mismatchDescription.appendText("page is ").appendValue(pagination.page)
+                               return false
+                       }
+               }
+               pages?.let {
+                       if (pagination.pageCount != pages) {
+                               mismatchDescription.appendText("total pages is ").appendValue(pagination.pageCount)
+                               return false
+                       }
+               }
+               return true
+       }
+
+       override fun describeTo(description: Description) {
+               page?.also {
+                       description.appendText("is on page ").appendValue(page)
+                       pages?.also {
+                                       description.appendText(" of ").appendValue(pages)
+                       }
+               } ?: pages?.also {
+                       description.appendText("has ").appendValue(pages).appendText(" pages")
+               }
+       }
+
+       fun isOnPage(page: Int) = PaginationMatcher(page = page, pages = pages)
+       fun hasPages(pages: Int) = PaginationMatcher(page = page, pages = pages)
+
+}
+
+fun isOnPage(page: Int) = PaginationMatcher(page = page)
+fun hasPages(pages: Int) = PaginationMatcher(pages = pages)