From e409df7786c9d4556c8c754a447e494fbd95d32a Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@pterodactylus.net>
Date: Tue, 20 Jun 2017 14:34:52 +0200
Subject: [PATCH] Convert bookmarks page test to use new web page test base

---
 .../sone/web/pages/BookmarksPageTest.kt             | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/pages/BookmarksPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/pages/BookmarksPageTest.kt
index f0a9d49..04446a2 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/BookmarksPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/BookmarksPageTest.kt
@@ -13,9 +13,8 @@ import org.junit.Test
 /**
  * Unit test for [BookmarksPage].
  */
-class BookmarksPageTest: WebPageTest() {
+class BookmarksPageTest: WebPageTest2(::BookmarksPage) {
 
-	private val page = BookmarksPage(template, webInterface)
 	private val post1 = createLoadedPost(1000)
 	private val post2 = createLoadedPost(3000)
 	private val post3 = createLoadedPost(2000)
@@ -39,20 +38,22 @@ class BookmarksPageTest: WebPageTest() {
 	@Test
 	@Suppress("UNCHECKED_CAST")
 	fun `page sets correct posts in template context`() {
-		page.processTemplate(freenetRequest, templateContext)
-		assertThat(templateContext["posts"] as Collection<Post>, contains(post2, post3, post1))
-		assertThat((templateContext["pagination"] as Pagination<Post>).items, contains(post2, post3, post1))
-		assertThat(templateContext["postsNotLoaded"], equalTo<Any>(false))
+		verifyNoRedirect {
+			assertThat(templateContext["posts"] as Collection<Post>, contains(post2, post3, post1))
+			assertThat((templateContext["pagination"] as Pagination<Post>).items, contains(post2, post3, post1))
+			assertThat(templateContext["postsNotLoaded"], equalTo<Any>(false))
+		}
 	}
 
 	@Test
 	@Suppress("UNCHECKED_CAST")
 	fun `page does not put unloaded posts in template context but sets a flag`() {
 		whenever(post3.isLoaded).thenReturn(false)
-		page.processTemplate(freenetRequest, templateContext)
-		assertThat(templateContext["posts"] as Collection<Post>, contains(post2, post1))
-		assertThat((templateContext["pagination"] as Pagination<Post>).items, contains(post2, post1))
-		assertThat(templateContext["postsNotLoaded"], equalTo<Any>(true))
+		verifyNoRedirect {
+			assertThat(templateContext["posts"] as Collection<Post>, contains(post2, post1))
+			assertThat((templateContext["pagination"] as Pagination<Post>).items, contains(post2, post1))
+			assertThat(templateContext["postsNotLoaded"], equalTo<Any>(true))
+		}
 	}
 
 }
-- 
2.7.4