Add test for DI constructability of TrustAjaxPage

[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / ajax / MarkAsKnownAjaxPage.kt

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/ajax/MarkAsKnownAjaxPage.kt b/src/main/kotlin/net/pterodactylus/sone/web/ajax/MarkAsKnownAjaxPage.kt
index 9e4e4a3..4ab6d18 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/web/ajax/MarkAsKnownAjaxPage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/ajax/MarkAsKnownAjaxPage.kt
@@ -1,16 +1,16 @@
 package net.pterodactylus.sone.web.ajax
 
-import com.google.common.base.Optional
-import net.pterodactylus.sone.utils.mapPresent
 import net.pterodactylus.sone.utils.parameters
 import net.pterodactylus.sone.web.WebInterface
 import net.pterodactylus.sone.web.page.FreenetRequest
+import javax.inject.Inject
 
 /**
  * AJAX page that lets the user mark a number of [net.pterodactylus.sone.data.Sone]s,
  * [net.pterodactylus.sone.data.Post]s, or [net.pterodactylus.sone.data.Reply]s as known.
  */
-class MarkAsKnownAjaxPage(webInterface: WebInterface) : JsonPage("markAsKnown.ajax", webInterface) {
+class MarkAsKnownAjaxPage @Inject constructor(webInterface: WebInterface) :
+               JsonPage("markAsKnown.ajax", webInterface) {
 
        override val requiresLogin = false
 
@@ -21,10 +21,10 @@ class MarkAsKnownAjaxPage(webInterface: WebInterface) : JsonPage("markAsKnown.aj
                else -> createErrorJsonObject("invalid-type")
        }
 
-       private fun <T> processIds(request: FreenetRequest, getter: (String) -> Optional<T>, marker: (T) -> Unit) =
+       private fun <T : Any> processIds(request: FreenetRequest, getter: (String) -> T?, marker: (T) -> Unit) =
                        request.parameters["id"]
                                        ?.split(Regex(" +"))
-                                       ?.mapPresent(getter)
+                                       ?.mapNotNull(getter)
                                        ?.onEach(marker)
                                        .let { createSuccessJsonObject() }