Add paginate() extension method and pagination matcher

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

diff --git a/src/test/kotlin/net/pterodactylus/sone/utils/PaginationTest.kt b/src/test/kotlin/net/pterodactylus/sone/utils/PaginationTest.kt
index 9a1153c..dce14fe 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/utils/PaginationTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/utils/PaginationTest.kt
@@ -1,5 +1,6 @@
 package net.pterodactylus.sone.utils
 
+import net.pterodactylus.sone.test.hasPages
 import org.hamcrest.MatcherAssert.assertThat
 import org.hamcrest.Matchers.contains
 import org.hamcrest.Matchers.equalTo
@@ -14,6 +15,12 @@ class PaginationTest {
        private val pagination = Pagination<Int>(items, 2)
 
        @Test
+       fun `pagination can be created from iterable`() {
+               val pagination = listOf(1, 2, 3, 4, 5).asIterable().paginate(2)
+               assertThat(pagination, hasPages(3).isOnPage(0))
+       }
+
+       @Test
        fun `new pagination is at page 0`() {
                assertThat(pagination.page, equalTo(0))
        }