1 package net.pterodactylus.sone.web
3 import net.pterodactylus.sone.data.Album
4 import net.pterodactylus.sone.data.Sone
5 import net.pterodactylus.sone.test.mock
6 import net.pterodactylus.sone.test.whenever
7 import net.pterodactylus.sone.web.WebTestUtils.redirectsTo
8 import net.pterodactylus.util.web.Method.GET
9 import net.pterodactylus.util.web.Method.POST
10 import org.hamcrest.MatcherAssert.assertThat
11 import org.hamcrest.Matchers.equalTo
12 import org.junit.Before
14 import org.mockito.ArgumentMatchers.anyString
17 * Unit test for [DeleteAlbumPage].
19 class DeleteAlbumPageTest : WebPageTest() {
21 private val page = DeleteAlbumPage(template, webInterface)
23 private val sone = mock<Sone>()
24 private val album = mock<Album>()
25 private val parentAlbum = mock<Album>()
29 whenever(sone.id).thenReturn("sone-id")
30 whenever(sone.isLocal).thenReturn(true)
31 whenever(parentAlbum.id).thenReturn("parent-id")
32 whenever(album.id).thenReturn("album-id")
33 whenever(album.sone).thenReturn(sone)
34 whenever(album.parent).thenReturn(parentAlbum)
35 whenever(sone.rootAlbum).thenReturn(parentAlbum)
39 fun `get request with invalid album ID results in redirect to invalid page`() {
41 whenever(core.getAlbum(anyString())).thenReturn(null)
42 expectedException.expect(redirectsTo("invalid.html"))
43 page.handleRequest(freenetRequest, templateContext)
47 fun `get request with valid album ID sets album in template context`() {
49 val album = mock<Album>()
50 addAlbum("album-id", album)
51 addHttpRequestParameter("album", "album-id")
52 page.handleRequest(freenetRequest, templateContext)
53 assertThat(templateContext["album"], equalTo<Any>(album))
57 fun `post request redirects to invalid page if album is invalid`() {
59 expectedException.expect(redirectsTo("invalid.html"))
60 page.handleRequest(freenetRequest, templateContext)
64 fun `post request redirects to no permissions page if album is not local`() {
66 whenever(sone.isLocal).thenReturn(false)
67 addAlbum("album-id", album)
68 addHttpRequestParameter("album", "album-id")
69 expectedException.expect(redirectsTo("noPermission.html"))
70 page.handleRequest(freenetRequest, templateContext)
74 fun `post request with abort delete parameter set redirects to album browser`() {
76 addAlbum("album-id", album)
77 addHttpRequestParameter("album", "album-id")
78 addHttpRequestParameter("abortDelete", "true")
79 expectedException.expect(redirectsTo("imageBrowser.html?album=album-id"))
80 page.handleRequest(freenetRequest, templateContext)
84 fun `album is deleted and page redirects to sone if parent album is root album`() {
86 addAlbum("album-id", album)
87 addHttpRequestParameter("album", "album-id")
88 expectedException.expect(redirectsTo("imageBrowser.html?sone=sone-id"))
89 page.handleRequest(freenetRequest, templateContext)
93 fun `album is deleted and page redirects to album if parent album is not root album`() {
95 whenever(sone.rootAlbum).thenReturn(mock<Album>())
96 addAlbum("album-id", album)
97 addHttpRequestParameter("album", "album-id")
98 expectedException.expect(redirectsTo("imageBrowser.html?album=parent-id"))
99 page.handleRequest(freenetRequest, templateContext)