X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FImageBrowserPageTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FImageBrowserPageTest.kt;h=38ecce46ddc3459272cf4ae1643e96d33bd876ea;hp=1533f4badcd2a035e2c0ce7d9f102c2b6a104506;hb=5473139d890c40c85712f59ba534583961c2772e;hpb=509b81185b3a1e82bb78308640d0d7b6b741d3e0

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/pages/ImageBrowserPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/pages/ImageBrowserPageTest.kt
index 1533f4b..38ecce4 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/ImageBrowserPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/ImageBrowserPageTest.kt
@@ -1,6 +1,8 @@
 package net.pterodactylus.sone.web.pages
 
 import net.pterodactylus.sone.data.*
+import net.pterodactylus.sone.data.impl.AlbumImpl
+import net.pterodactylus.sone.data.impl.ImageImpl
 import net.pterodactylus.sone.test.*
 import net.pterodactylus.sone.web.*
 import net.pterodactylus.sone.web.page.*
@@ -105,16 +107,13 @@ class ImageBrowserPageTest : WebPageTest(::ImageBrowserPage) {
 
 	private fun createSone(firstAlbumTitle: String, secondAlbumTitle: String): Sone {
 		return mock<Sone>().apply {
-			val rootAlbum = mock<Album>()
-			val firstAlbum = mock<Album>()
-			val firstImage = mock<Image>().run { whenever(isInserted).thenReturn(true); this }
-			whenever(firstAlbum.images).thenReturn(listOf(firstImage))
-			val secondAlbum = mock<Album>()
-			val secondImage = mock<Image>().run { whenever(isInserted).thenReturn(true); this }
-			whenever(secondAlbum.images).thenReturn(listOf(secondImage))
-			whenever(firstAlbum.title).thenReturn(firstAlbumTitle)
-			whenever(secondAlbum.title).thenReturn(secondAlbumTitle)
-			whenever(rootAlbum.albums).thenReturn(listOf(firstAlbum, secondAlbum))
+			val rootAlbum = AlbumImpl(this)
+			val firstAlbum = AlbumImpl(this).modify().setTitle(firstAlbumTitle).update()
+			firstAlbum.addImage(ImageImpl("1").modify().setSone(this).setKey("key").update())
+			val secondAlbum = AlbumImpl(this).modify().setTitle(secondAlbumTitle).update()
+			secondAlbum.addImage(ImageImpl("2").modify().setSone(this).setKey("key").update())
+			rootAlbum.addAlbum(firstAlbum)
+			rootAlbum.addAlbum(secondAlbum)
 			whenever(this.rootAlbum).thenReturn(rootAlbum)
 		}
 	}