1 package net.pterodactylus.sone.web
3 import net.pterodactylus.sone.data.Album
4 import net.pterodactylus.sone.data.Image
5 import net.pterodactylus.sone.data.Sone
6 import net.pterodactylus.sone.test.mock
7 import net.pterodactylus.sone.test.whenever
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.Mockito.verify
17 * Unit test for [DeleteImagePage].
19 class DeleteImagePageTest: WebPageTest() {
21 private val page = DeleteImagePage(template, webInterface)
22 private val image = mock<Image>()
23 private val sone = mock<Sone>()
25 override fun getPage() = page
29 val album = mock<Album>()
30 whenever(album.id).thenReturn("album-id")
31 whenever(image.id).thenReturn("image-id")
32 whenever(image.sone).thenReturn(sone)
33 whenever(image.album).thenReturn(album)
34 whenever(sone.isLocal).thenReturn(true)
38 fun `page returns correct path`() {
39 assertThat(page.path, equalTo("deleteImage.html"))
43 fun `page requires login`() {
44 assertThat(page.requiresLogin(), equalTo(true))
48 fun `get request with invalid image redirects to invalid page`() {
50 verifyRedirect("invalid.html")
54 fun `get request with image from non-local sone redirects to no permissions page`() {
56 whenever(sone.isLocal).thenReturn(false)
57 addImage("image-id", image)
58 addHttpRequestParameter("image", "image-id")
59 verifyRedirect("noPermission.html")
63 fun `get request with image from local sone sets image in template context`() {
65 addImage("image-id", image)
66 addHttpRequestParameter("image", "image-id")
67 page.processTemplate(freenetRequest, templateContext)
68 assertThat(templateContext["image"], equalTo<Any>(image))
72 fun `post request with abort delete flag set redirects to image browser`() {
74 addImage("image-id", image)
75 addHttpRequestParameter("image", "image-id")
76 addHttpRequestParameter("abortDelete", "true")
77 verifyRedirect("imageBrowser.html?image=image-id")
81 fun `post request deletes image and redirects to image browser`() {
83 addImage("image-id", image)
84 addHttpRequestParameter("image", "image-id")
85 verifyRedirect("imageBrowser.html?album=album-id") {
86 verify(webInterface.core).deleteImage(image)