package net.pterodactylus.sone.web.pages import net.pterodactylus.sone.data.Album import net.pterodactylus.sone.data.Image import net.pterodactylus.sone.data.Sone import net.pterodactylus.sone.test.getInstance import net.pterodactylus.sone.test.mock import net.pterodactylus.sone.test.whenever import net.pterodactylus.sone.web.baseInjector import net.pterodactylus.sone.web.page.* import net.pterodactylus.util.web.Method.POST import org.hamcrest.MatcherAssert.assertThat import org.hamcrest.Matchers.equalTo import org.hamcrest.Matchers.notNullValue import org.junit.Before import org.junit.Test import org.mockito.Mockito.verify /** * Unit test for [DeleteImagePage]. */ class DeleteImagePageTest: WebPageTest(::DeleteImagePage) { private val image = mock() private val sone = mock() @Before fun setupImage() { val album = mock() whenever(album.id).thenReturn("album-id") whenever(image.id).thenReturn("image-id") whenever(image.sone).thenReturn(sone) whenever(image.album).thenReturn(album) whenever(sone.isLocal).thenReturn(true) } @Test fun `page returns correct path`() { assertThat(page.path, equalTo("deleteImage.html")) } @Test fun `page requires login`() { assertThat(page.requiresLogin(), equalTo(true)) } @Test fun `get request with invalid image redirects to invalid page`() { verifyRedirect("invalid.html") } @Test fun `get request with image from non-local sone redirects to no permissions page`() { whenever(sone.isLocal).thenReturn(false) addImage("image-id", image) addHttpRequestParameter("image", "image-id") verifyRedirect("noPermission.html") } @Test fun `get request with image from local sone sets image in template context`() { addImage("image-id", image) addHttpRequestParameter("image", "image-id") page.processTemplate(freenetRequest, templateContext) assertThat(templateContext["image"], equalTo(image)) } @Test fun `post request with abort delete flag set redirects to image browser`() { setMethod(POST) addImage("image-id", image) addHttpRequestPart("image", "image-id") addHttpRequestPart("abortDelete", "true") verifyRedirect("imageBrowser.html?image=image-id") } @Test fun `post request deletes image and redirects to image browser`() { setMethod(POST) addImage("image-id", image) addHttpRequestPart("image", "image-id") verifyRedirect("imageBrowser.html?album=album-id") { verify(webInterface.core).deleteImage(image) } } @Test fun `page can be created by dependency injection`() { assertThat(baseInjector.getInstance(), notNullValue()) } @Test fun `page is annotated with correct template path`() { assertThat(page.templatePath, equalTo("/templates/deleteImage.html")) } }