1 package net.pterodactylus.sone.web.pages
3 import net.pterodactylus.sone.data.Album
4 import net.pterodactylus.sone.data.Image
5 import net.pterodactylus.sone.data.Image.Modifier
6 import net.pterodactylus.sone.data.Sone
7 import net.pterodactylus.sone.data.TemporaryImage
8 import net.pterodactylus.sone.test.mock
9 import net.pterodactylus.sone.test.mockBuilder
10 import net.pterodactylus.sone.test.whenever
11 import net.pterodactylus.sone.web.pages.UploadImagePage
12 import net.pterodactylus.util.web.Method.POST
13 import org.hamcrest.MatcherAssert.assertThat
14 import org.hamcrest.Matchers.equalTo
16 import org.mockito.Mockito.any
17 import org.mockito.Mockito.eq
18 import org.mockito.Mockito.never
19 import org.mockito.Mockito.verify
22 * Unit test for [UploadImagePage].
24 class UploadImagePageTest : WebPageTest() {
26 private val parentAlbum = mock<Album>().apply {
27 whenever(id).thenReturn("parent-id")
28 whenever(sone).thenReturn(currentSone)
31 override fun getPage() = UploadImagePage(template, webInterface)
34 fun `get request does not redirect or upload anything`() {
35 page.handleRequest(freenetRequest, templateContext)
36 verify(core, never()).createTemporaryImage(any(), any())
37 verify(core, never()).createImage(any(), any(), any())
41 fun `post request without parent results in no permission error page`() {
43 verifyRedirect("noPermission.html")
47 fun `post request with parent that is not the current sone results in no permission error page`() {
49 addHttpRequestParameter("parent", "parent-id")
50 whenever(parentAlbum.sone).thenReturn(mock<Sone>())
51 addAlbum("parent-id", parentAlbum)
52 verifyRedirect("noPermission.html")
56 fun `post request with empty name redirects to error page`() {
58 addAlbum("parent-id", parentAlbum)
59 addHttpRequestParameter("parent", "parent-id")
60 addHttpRequestParameter("title", " ")
61 verifyRedirect("emptyImageTitle.html")
65 fun `uploading an invalid image results in no redirect and message set in template context`() {
67 addAlbum("parent-id", parentAlbum)
68 addHttpRequestParameter("parent", "parent-id")
69 addHttpRequestParameter("title", "title")
70 addUploadedFile("image", "image.png", "image/png", "no-image.png")
71 page.handleRequest(freenetRequest, templateContext)
72 verify(core, never()).createTemporaryImage(any(), any())
73 assertThat(templateContext["messages"] as String?, equalTo<String>("Page.UploadImage.Error.InvalidImage"))
77 fun `uploading a valid image uploads image and redirects to album browser`() {
79 addAlbum("parent-id", parentAlbum)
80 addHttpRequestParameter("parent", "parent-id")
81 addHttpRequestParameter("title", "Title")
82 addHttpRequestParameter("description", "Description")
83 addUploadedFile("image", "image.png", "image/png", "image.png")
84 val temporaryImage = TemporaryImage("temp-image")
85 val imageModifier = mockBuilder<Modifier>()
86 val image = mock<Image>().apply {
87 whenever(modify()).thenReturn(imageModifier)
89 whenever(core.createTemporaryImage(eq("image/png"), any())).thenReturn(temporaryImage)
90 whenever(core.createImage(currentSone, parentAlbum, temporaryImage)).thenReturn(image)
91 verifyRedirect("imageBrowser.html?album=parent-id") {
92 verify(image).modify()
93 verify(imageModifier).setWidth(2)
94 verify(imageModifier).setHeight(1)
95 verify(imageModifier).setTitle("Title")
96 verify(imageModifier).setDescription("Description")
97 verify(imageModifier).update()