X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FImageAccessorTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FImageAccessorTest.kt;h=21a3d00ab828f51983b793f5c35055720a0b3804;hp=f7da592afcb4379396dd69255b3b1610aad71c0c;hb=5c5bee980f9cab5792e34d1c9840f73b8b191830;hpb=faf66247a34f64946990a985d2ea3003465969cb

diff --git a/src/test/kotlin/net/pterodactylus/sone/template/ImageAccessorTest.kt b/src/test/kotlin/net/pterodactylus/sone/template/ImageAccessorTest.kt
index f7da592..21a3d00 100644
--- a/src/test/kotlin/net/pterodactylus/sone/template/ImageAccessorTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/template/ImageAccessorTest.kt
@@ -1,13 +1,9 @@
 package net.pterodactylus.sone.template
 
-import net.pterodactylus.sone.data.Album
-import net.pterodactylus.sone.data.Image
-import net.pterodactylus.sone.test.mock
-import net.pterodactylus.sone.test.whenever
+import net.pterodactylus.sone.data.impl.*
 import org.hamcrest.MatcherAssert.assertThat
 import org.hamcrest.Matchers.equalTo
 import org.hamcrest.Matchers.nullValue
-import org.junit.Before
 import org.junit.Test
 
 /**
@@ -16,16 +12,6 @@ import org.junit.Test
 class ImageAccessorTest {
 
 	private val accessor = ImageAccessor()
-	private val album = mock<Album>()
-	private val images = listOf(mock<Image>(), mock())
-
-	@Before
-	fun setupImages() {
-		whenever(album.images).thenReturn(images)
-		images.forEach {
-			whenever(it.album).thenReturn(album)
-		}
-	}
 
 	@Test
 	fun `accessor returns next image for first image`() {
@@ -53,3 +39,7 @@ class ImageAccessorTest {
 	}
 
 }
+
+private val sone = IdOnlySone("sone")
+private val album = AlbumImpl(sone)
+private val images = listOf(ImageImpl().modify().setSone(sone).update(), ImageImpl().modify().setSone(sone).update()).onEach(album::addImage)