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 `get request with invalid image redirects to invalid page`() {
40 verifyRedirect("invalid.html")
44 fun `get request with image from non-local sone redirects to no permissions page`() {
46 whenever(sone.isLocal).thenReturn(false)
47 addImage("image-id", image)
48 addHttpRequestParameter("image", "image-id")
49 verifyRedirect("noPermission.html")
53 fun `get request with image from local sone sets image in template context`() {
55 addImage("image-id", image)
56 addHttpRequestParameter("image", "image-id")
57 page.processTemplate(freenetRequest, templateContext)
58 assertThat(templateContext["image"], equalTo<Any>(image))
62 fun `post request with abort delete flag set redirects to image browser`() {
64 addImage("image-id", image)
65 addHttpRequestParameter("image", "image-id")
66 addHttpRequestParameter("abortDelete", "true")
67 verifyRedirect("imageBrowser.html?image=image-id")
71 fun `post request deletes image and redirects to image browser`() {
73 addImage("image-id", image)
74 addHttpRequestParameter("image", "image-id")
75 verifyRedirect("imageBrowser.html?album=album-id") {
76 verify(webInterface.core).deleteImage(image)