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