Convert edit album page test to use new web page test base

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / web / pages / EditAlbumPageTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/pages/EditAlbumPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/pages/EditAlbumPageTest.kt
index b04ea8d..5ef276f 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/EditAlbumPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/EditAlbumPageTest.kt
@@ -17,17 +17,13 @@ import org.mockito.Mockito.verify
 /**
  * Unit test for [EditAlbumPage].
  */
-class EditAlbumPageTest: WebPageTest() {
-
-       private val page = EditAlbumPage(template, webInterface)
+class EditAlbumPageTest: WebPageTest2(::EditAlbumPage) {
 
        private val album = mock<Album>()
        private val parentAlbum = mock<Album>()
        private val modifier = mockBuilder<Modifier>()
        private val sone = mock<Sone>()
 
-       override fun getPage() = page
-
        @Before
        fun setup() {
                whenever(album.id).thenReturn("album-id")