package net.pterodactylus.sone.web

import net.pterodactylus.sone.test.mock
import net.pterodactylus.sone.test.whenever
import net.pterodactylus.sone.web.WebTestUtils.redirectsTo
import net.pterodactylus.util.notify.Notification
import net.pterodactylus.util.web.Method.GET
import org.junit.Test
import org.mockito.Mockito.never
import org.mockito.Mockito.verify
import kotlin.test.fail

/**
 * Unit test for [DismissNotificationPage].
 */
class DismissNotificationPageTest : WebPageTest() {

	private val page = DismissNotificationPage(template, webInterface)
	private val notification = mock<Notification>()

	@Test
	fun `get request with invalid notification ID redirects to return page`() {
		request("", GET)
		addHttpRequestParameter("returnPage", "return.html")
		expectedException.expect(redirectsTo("return.html"))
		page.handleRequest(freenetRequest, templateContext)
	}

	@Test
	fun `get request with dismissible notification dismisses the notification and redirects to return page`() {
		request("", GET)
		addNotification("notification-id", notification)
		addHttpRequestParameter("notification", "notification-id")
		addHttpRequestParameter("returnPage", "return.html")
		expectedException.expect(redirectsTo("return.html"))
		try {
			page.handleRequest(freenetRequest, templateContext)
			fail()
		} finally {
			verify(notification, never()).dismiss()
		}
	}

	@Test
	fun `get request with non dismissible notification redirects to return page`() {
		request("", GET)
		whenever(notification.isDismissable).thenReturn(true)
		addNotification("notification-id", notification)
		addHttpRequestParameter("notification", "notification-id")
		addHttpRequestParameter("returnPage", "return.html")
		expectedException.expect(redirectsTo("return.html"))
		try {
			page.handleRequest(freenetRequest, templateContext)
			fail()
		} finally {
			verify(notification).dismiss()
		}
	}

}