+++ /dev/null
-/*
- * Sone - DeleteImagePage.java - Copyright © 2011–2016 David Roden
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- */
-
-package net.pterodactylus.sone.web;
-
-import net.pterodactylus.sone.data.Image;
-import net.pterodactylus.sone.web.page.FreenetRequest;
-import net.pterodactylus.util.template.Template;
-import net.pterodactylus.util.template.TemplateContext;
-import net.pterodactylus.util.web.Method;
-
-/**
- * Page that lets the user delete an {@link Image}.
- *
- * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
- */
-public class DeleteImagePage extends SoneTemplatePage {
-
- /**
- * Creates a new “delete image” page.
- *
- * @param template
- * The template to render
- * @param webInterface
- * The Sone web interface
- */
- public DeleteImagePage(Template template, WebInterface webInterface) {
- super("deleteImage.html", template, "Page.DeleteImage.Title", webInterface, true);
- }
-
- //
- // SONETEMPLATEPAGE METHODS
- //
-
- /**
- * {@inheritDoc}
- */
- @Override
- protected void handleRequest(FreenetRequest request, TemplateContext templateContext) throws RedirectException {
- String imageId = (request.getMethod() == Method.POST) ? request.getHttpRequest().getPartAsStringFailsafe("image", 36) : request.getHttpRequest().getParam("image");
- Image image = webInterface.getCore().getImage(imageId, false);
- if (image == null) {
- throw new RedirectException("invalid.html");
- }
- if (!image.getSone().isLocal()) {
- throw new RedirectException("noPermission.html");
- }
- if (request.getMethod() == Method.POST) {
- if (request.getHttpRequest().isPartSet("abortDelete")) {
- throw new RedirectException("imageBrowser.html?image=" + image.getId());
- }
- webInterface.getCore().deleteImage(image);
- throw new RedirectException("imageBrowser.html?album=" + image.getAlbum().getId());
- }
- templateContext.set("image", image);
- }
-
-}
--- /dev/null
+package net.pterodactylus.sone.web
+
+import net.pterodactylus.sone.web.page.FreenetRequest
+import net.pterodactylus.util.template.Template
+import net.pterodactylus.util.template.TemplateContext
+import net.pterodactylus.util.web.Method.POST
+
+/**
+ * Page that lets the user delete an {@link Image}.
+ */
+class DeleteImagePage(template: Template, webInterface: WebInterface):
+ SoneTemplatePage("deleteImage.html", template, "Page.DeleteImage.Title", webInterface, true) {
+
+ override fun handleRequest(request: FreenetRequest, templateContext: TemplateContext) {
+ val image = webInterface.core.getImage(request.httpRequest.getPartAsStringFailsafe("image", 36)) ?: throw RedirectException("invalid.html")
+ if (!image.sone.isLocal) {
+ throw RedirectException("noPermission.html")
+ }
+ if (request.method == POST) {
+ if (request.httpRequest.isPartSet("abortDelete")) {
+ throw RedirectException("imageBrowser.html?image=${image.id}")
+ }
+ webInterface.core.deleteImage(image)
+ throw RedirectException("imageBrowser.html?album=${image.album.id}")
+ }
+ templateContext["image"] = image
+ }
+
+}
/**
* Unit test for [DeleteImagePage].
*/
-class DeleteImagePageTest : WebPageTest() {
+class DeleteImagePageTest: WebPageTest() {
private val page = DeleteImagePage(template, webInterface)
private val image = mock<Image>()
}
@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`() {
request("", GET)
verifyRedirect("invalid.html")