Add test for DI constructability of DistrustAjaxPage

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

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/ajax/DistrustAjaxPage.kt b/src/main/kotlin/net/pterodactylus/sone/web/ajax/DistrustAjaxPage.kt
index 1c3506b..10384df 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/web/ajax/DistrustAjaxPage.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/ajax/DistrustAjaxPage.kt
@@ -6,22 +6,24 @@ import net.pterodactylus.sone.utils.let
 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 distrust a Sone.
  *
  * @see Core.distrustSone(Sone, Sone)
  */
-class DistrustAjaxPage(webInterface: WebInterface) : LoggedInJsonPage("distrustSone.ajax", webInterface) {
+class DistrustAjaxPage @Inject constructor(webInterface: WebInterface) :
+               LoggedInJsonPage("distrustSone.ajax", webInterface) {
 
        override fun createJsonObject(currentSone: Sone, request: FreenetRequest) =
                        request.parameters["sone"]
-                                       .let(webInterface.core::getSone)
+                                       ?.let(core::getSone)
                                        ?.let { sone ->
                                                createSuccessJsonObject()
-                                                               .put("trustValue", webInterface.core.preferences.negativeTrust)
+                                                               .put("trustValue", core.preferences.negativeTrust)
                                                                .also {
-                                                                       webInterface.core.distrustSone(currentSone, sone)
+                                                                       core.distrustSone(currentSone, sone)
                                                                }
                                        } ?: createErrorJsonObject("invalid-sone-id")