X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FDeleteImagePageTest.kt;h=eccf9491d3687f866a0cdc5e26e6c492b3ac053e;hp=1c5c2748ecb9a6f36ec4197ab8a9bc12a2f1effb;hb=2ab560633c24940fa49bdaf569635dd673ad7b19;hpb=a76956e389fcfe6282ad4ca7156bbf76327bb0c0

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/pages/DeleteImagePageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/pages/DeleteImagePageTest.kt
index 1c5c274..eccf949 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/DeleteImagePageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/DeleteImagePageTest.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.test.*
 import net.pterodactylus.sone.web.*
 import net.pterodactylus.sone.web.page.*
@@ -13,18 +14,14 @@ import org.mockito.Mockito.*
 /**
  * Unit test for [DeleteImagePage].
  */
-class DeleteImagePageTest: WebPageTest(::DeleteImagePage) {
+class DeleteImagePageTest : WebPageTest(::DeleteImagePage) {
 
-	private val image = mock<Image>()
 	private val sone = mock<Sone>()
+	private val image = ImageImpl("image-id").modify().setSone(sone).update()!!
 
 	@Before
 	fun setupImage() {
-		val album = mock<Album>()
-		whenever(album.id).thenReturn("album-id")
-		whenever(image.id).thenReturn("image-id")
-		whenever(image.sone).thenReturn(sone)
-		whenever(image.album).thenReturn(album)
+		AlbumImpl(sone, "album-id").also { it.addImage(image) }
 		whenever(sone.isLocal).thenReturn(true)
 	}
 
@@ -80,12 +77,12 @@ class DeleteImagePageTest: WebPageTest(::DeleteImagePage) {
 
 	@Test
 	fun `page can be created by dependency injection`() {
-	    assertThat(baseInjector.getInstance<DeleteImagePage>(), notNullValue())
+		assertThat(baseInjector.getInstance<DeleteImagePage>(), notNullValue())
 	}
 
 	@Test
 	fun `page is annotated with correct template path`() {
-	    assertThat(page.templatePath, equalTo("/templates/deleteImage.html"))
+		assertThat(page.templatePath, equalTo("/templates/deleteImage.html"))
 	}
 
 }