X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FEditProfilePageTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FEditProfilePageTest.kt;h=ac909ea1e3c1b33f62e5e5aa6f56e4de9ce2b9d1;hp=74c3851321ce2f39fef3bcbca7b0849cf96d1ac7;hb=5c5bee980f9cab5792e34d1c9840f73b8b191830;hpb=faf66247a34f64946990a985d2ea3003465969cb

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/pages/EditProfilePageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/pages/EditProfilePageTest.kt
index 74c3851..ac909ea 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/EditProfilePageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/EditProfilePageTest.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.*
@@ -21,16 +22,13 @@ class EditProfilePageTest : WebPageTest(::EditProfilePage) {
 
 	@Before
 	fun setupProfile() {
-		val avatar = mock<Image>()
-		whenever(avatar.id).thenReturn("image-id")
-		whenever(avatar.sone).thenReturn(currentSone)
 		profile.firstName = "First"
 		profile.middleName = "Middle"
 		profile.lastName = "Last"
 		profile.birthDay = 31
 		profile.birthMonth = 12
 		profile.birthYear = 1999
-		profile.setAvatar(avatar)
+		profile.setAvatar(ImageImpl("image-id").modify().setSone(currentSone).update())
 		whenever(currentSone.profile).thenReturn(profile)
 	}
 
@@ -120,9 +118,7 @@ class EditProfilePageTest : WebPageTest(::EditProfilePage) {
 
 	@Test
 	fun `post request with new avatar ID and save profile saves the profile and redirects back to profile edit page`() {
-		val newAvatar = mock<Image>()
-		whenever(newAvatar.sone).thenReturn(currentSone)
-		whenever(newAvatar.id).thenReturn("avatar-id")
+		val newAvatar = ImageImpl("avatar-id").modify().setSone(currentSone).update()
 		addImage("avatar-id", newAvatar)
 		verifySingleFieldCanBeChanged("avatarId", "avatar-id") { profile.avatar }
 	}