+package net.pterodactylus.sone.web
+
+import net.pterodactylus.sone.data.Album
+import net.pterodactylus.sone.data.Sone
+import net.pterodactylus.sone.test.mock
+import net.pterodactylus.sone.test.whenever
+import net.pterodactylus.sone.web.WebTestUtils.redirectsTo
+import net.pterodactylus.util.web.Method.GET
+import net.pterodactylus.util.web.Method.POST
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.equalTo
+import org.junit.Before
+import org.junit.Test
+import org.mockito.ArgumentMatchers.anyString
+
+/**
+ * Unit test for [DeleteAlbumPage].
+ */
+class DeleteAlbumPageTest : WebPageTest() {
+
+ private val page = DeleteAlbumPage(template, webInterface)
+
+ private val sone = mock<Sone>()
+ private val album = mock<Album>()
+ private val parentAlbum = mock<Album>()
+
+ @Before
+ fun setupAlbums() {
+ whenever(sone.id).thenReturn("sone-id")
+ whenever(sone.isLocal).thenReturn(true)
+ whenever(parentAlbum.id).thenReturn("parent-id")
+ whenever(album.id).thenReturn("album-id")
+ whenever(album.sone).thenReturn(sone)
+ whenever(album.parent).thenReturn(parentAlbum)
+ whenever(sone.rootAlbum).thenReturn(parentAlbum)
+ }
+
+ @Test
+ fun `get request with invalid album ID results in redirect to invalid page`() {
+ request("", GET)
+ whenever(core.getAlbum(anyString())).thenReturn(null)
+ expectedException.expect(redirectsTo("invalid.html"))
+ page.handleRequest(freenetRequest, templateContext)
+ }
+
+ @Test
+ fun `get request with valid album ID sets album in template context`() {
+ request("", GET)
+ val album = mock<Album>()
+ addAlbum("album-id", album)
+ addHttpRequestParameter("album", "album-id")
+ page.handleRequest(freenetRequest, templateContext)
+ assertThat(templateContext["album"], equalTo<Any>(album))
+ }
+
+ @Test
+ fun `post request redirects to invalid page if album is invalid`() {
+ request("", POST)
+ expectedException.expect(redirectsTo("invalid.html"))
+ page.handleRequest(freenetRequest, templateContext)
+ }
+
+ @Test
+ fun `post request redirects to no permissions page if album is not local`() {
+ request("", POST)
+ whenever(sone.isLocal).thenReturn(false)
+ addAlbum("album-id", album)
+ addHttpRequestParameter("album", "album-id")
+ expectedException.expect(redirectsTo("noPermission.html"))
+ page.handleRequest(freenetRequest, templateContext)
+ }
+
+ @Test
+ fun `post request with abort delete parameter set redirects to album browser`() {
+ request("", POST)
+ addAlbum("album-id", album)
+ addHttpRequestParameter("album", "album-id")
+ addHttpRequestParameter("abortDelete", "true")
+ expectedException.expect(redirectsTo("imageBrowser.html?album=album-id"))
+ page.handleRequest(freenetRequest, templateContext)
+ }
+
+ @Test
+ fun `album is deleted and page redirects to sone if parent album is root album`() {
+ request("", POST)
+ addAlbum("album-id", album)
+ addHttpRequestParameter("album", "album-id")
+ expectedException.expect(redirectsTo("imageBrowser.html?sone=sone-id"))
+ page.handleRequest(freenetRequest, templateContext)
+ }
+
+ @Test
+ fun `album is deleted and page redirects to album if parent album is not root album`() {
+ request("", POST)
+ whenever(sone.rootAlbum).thenReturn(mock<Album>())
+ addAlbum("album-id", album)
+ addHttpRequestParameter("album", "album-id")
+ expectedException.expect(redirectsTo("imageBrowser.html?album=parent-id"))
+ page.handleRequest(freenetRequest, templateContext)
+ }
+
+}