X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FKnownSonesPageTest.kt;h=23987b64b3a4c3f28fcce46b6d70765ef97f8dd0;hp=a816d23405d6a27aafc3829ec297d35cf1bf2f24;hb=HEAD;hpb=a76956e389fcfe6282ad4ca7156bbf76327bb0c0

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/pages/KnownSonesPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/pages/KnownSonesPageTest.kt
index a816d23..23987b6 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/KnownSonesPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/KnownSonesPageTest.kt
@@ -1,6 +1,7 @@
 package net.pterodactylus.sone.web.pages
 
 import net.pterodactylus.sone.data.*
+import net.pterodactylus.sone.data.impl.*
 import net.pterodactylus.sone.freenet.wot.*
 import net.pterodactylus.sone.test.*
 import net.pterodactylus.sone.utils.*
@@ -13,7 +14,7 @@ import org.junit.*
 /**
  * Unit test for [KnownSonesPage].
  */
-class KnownSonesPageTest: WebPageTest(::KnownSonesPage) {
+class KnownSonesPageTest : WebPageTest(::KnownSonesPage) {
 
 	private val sones = listOf(
 			createSone(1000, 4, 7, 2, "sone2", true, true),
@@ -37,11 +38,12 @@ class KnownSonesPageTest: WebPageTest(::KnownSonesPage) {
 		whenever(this.time).thenReturn(time)
 		whenever(this.posts).thenReturn((0..(posts - 1)).map { mock<Post>() })
 		whenever(this.replies).thenReturn((0..(replies - 1)).map { mock<PostReply>() }.toSet())
-		val album = mock<Album>()
-		whenever(album.images).thenReturn(((0..(images - 1)).map { mock<Image>() }))
-		val rootAlbum = mock<Album>().apply {
-			whenever(albums).thenReturn(listOf(album))
+		val album = AlbumImpl(this)
+		repeat(images) {
+			ImageImpl().modify().setSone(this).update()
+					.also(album::addImage)
 		}
+		val rootAlbum = AlbumImpl(this).also { it.addAlbum(album) }
 		whenever(this.rootAlbum).thenReturn(rootAlbum)
 		whenever(this.profile).thenReturn(mock())
 		whenever(id).thenReturn(name.toLowerCase())
@@ -73,7 +75,7 @@ class KnownSonesPageTest: WebPageTest(::KnownSonesPage) {
 
 	@Test
 	fun `page returns correct title`() {
-		whenever(l10n.getString("Page.KnownSones.Title")).thenReturn("known sones page title")
+		addTranslation("Page.KnownSones.Title", "known sones page title")
 		assertThat(page.getPageTitle(soneRequest), equalTo("known sones page title"))
 	}
 
@@ -237,17 +239,17 @@ class KnownSonesPageTest: WebPageTest(::KnownSonesPage) {
 
 	@Test
 	fun `page can be created by dependency injection`() {
-	    assertThat(baseInjector.getInstance<KnownSonesPage>(), notNullValue())
+		assertThat(baseInjector.getInstance<KnownSonesPage>(), notNullValue())
 	}
 
 	@Test
 	fun `page is annotated with the correct menuname`() {
-	    assertThat(page.menuName, equalTo("KnownSones"))
+		assertThat(page.menuName, equalTo("KnownSones"))
 	}
 
 	@Test
 	fun `page is annotated with corrrect template path`() {
-	    assertThat(page.templatePath, equalTo("/templates/knownSones.html"))
+		assertThat(page.templatePath, equalTo("/templates/knownSones.html"))
 	}
 
 }