1 package net.pterodactylus.sone.web.pages
3 import net.pterodactylus.sone.data.Album
4 import net.pterodactylus.sone.data.Sone
5 import net.pterodactylus.sone.test.getInstance
6 import net.pterodactylus.sone.test.mock
7 import net.pterodactylus.sone.test.whenever
8 import net.pterodactylus.sone.web.baseInjector
9 import net.pterodactylus.sone.web.page.*
10 import net.pterodactylus.util.web.Method.POST
11 import org.hamcrest.MatcherAssert.assertThat
12 import org.hamcrest.Matchers.equalTo
13 import org.hamcrest.Matchers.notNullValue
14 import org.junit.Before
16 import org.mockito.ArgumentMatchers.anyString
17 import org.mockito.Mockito.verify
20 * Unit test for [DeleteAlbumPage].
22 class DeleteAlbumPageTest: WebPageTest(::DeleteAlbumPage) {
24 private val sone = mock<Sone>()
25 private val album = mock<Album>()
26 private val parentAlbum = mock<Album>()
30 whenever(sone.id).thenReturn("sone-id")
31 whenever(sone.isLocal).thenReturn(true)
32 whenever(parentAlbum.id).thenReturn("parent-id")
33 whenever(parentAlbum.isRoot).thenReturn(true)
34 whenever(album.id).thenReturn("album-id")
35 whenever(album.sone).thenReturn(sone)
36 whenever(album.parent).thenReturn(parentAlbum)
37 whenever(sone.rootAlbum).thenReturn(parentAlbum)
41 fun `page returns correct path`() {
42 assertThat(page.path, equalTo("deleteAlbum.html"))
46 fun `page requires login`() {
47 assertThat(page.requiresLogin(), equalTo(true))
51 fun `get request with invalid album ID results in redirect to invalid page`() {
52 whenever(core.getAlbum(anyString())).thenReturn(null)
53 verifyRedirect("invalid.html")
57 fun `get request with valid album ID sets album in template context`() {
58 val album = mock<Album>()
59 addAlbum("album-id", album)
60 addHttpRequestParameter("album", "album-id")
61 page.processTemplate(freenetRequest, templateContext)
62 assertThat(templateContext["album"], equalTo<Any>(album))
66 fun `post request redirects to invalid page if album is invalid`() {
68 verifyRedirect("invalid.html")
72 fun `post request redirects to no permissions page if album is not local`() {
74 whenever(sone.isLocal).thenReturn(false)
75 addAlbum("album-id", album)
76 addHttpRequestPart("album", "album-id")
77 verifyRedirect("noPermission.html")
81 fun `post request with abort delete parameter set redirects to album browser`() {
83 addAlbum("album-id", album)
84 addHttpRequestPart("album", "album-id")
85 addHttpRequestPart("abortDelete", "true")
86 verifyRedirect("imageBrowser.html?album=album-id")
90 fun `album is deleted and page redirects to sone if parent album is root album`() {
92 addAlbum("album-id", album)
93 addHttpRequestPart("album", "album-id")
94 verifyRedirect("imageBrowser.html?sone=sone-id") {
95 verify(core).deleteAlbum(album)
100 fun `album is deleted and page redirects to album if parent album is not root album`() {
102 whenever(parentAlbum.isRoot).thenReturn(false)
103 whenever(sone.rootAlbum).thenReturn(mock())
104 addAlbum("album-id", album)
105 addHttpRequestPart("album", "album-id")
106 verifyRedirect("imageBrowser.html?album=parent-id") {
107 verify(core).deleteAlbum(album)
112 fun `page can be created by dependency injection`() {
113 assertThat(baseInjector.getInstance<DeleteAlbumPage>(), notNullValue())
117 fun `page is annotated with correct template path`() {
118 assertThat(page.templatePath, equalTo("/templates/deleteAlbum.html"))