Replace JSON return object with Kotlin version
[Sone.git] / src / test / kotlin / net / pterodactylus / sone / web / ajax / CreateReplyAjaxPageTest.kt
index 6b110ae..6931b69 100644 (file)
@@ -28,8 +28,8 @@ class CreateReplyAjaxPageTest : JsonPageTest("createReply.ajax", pageSupplier =
                whenever(core.createReply(currentSone, post, "")).thenReturn(reply)
            addRequestParameter("post", "post-id")
                assertThat(json.isSuccess, equalTo(true))
-               assertThat(json["reply"].asText(), equalTo("reply-id"))
-               assertThat(json["sone"].asText(), equalTo("soneId"))
+               assertThat(json["reply"]?.asText(), equalTo("reply-id"))
+               assertThat(json["sone"]?.asText(), equalTo("soneId"))
        }
 
        @Test
@@ -42,14 +42,14 @@ class CreateReplyAjaxPageTest : JsonPageTest("createReply.ajax", pageSupplier =
                addRequestParameter("text", "Text with http://127.0.0.1:8888/KSK@foo.bar link")
                addRequestHeader("Host", "127.0.0.1:8888")
                assertThat(json.isSuccess, equalTo(true))
-               assertThat(json["reply"].asText(), equalTo("reply-id"))
-               assertThat(json["sone"].asText(), equalTo("soneId"))
+               assertThat(json["reply"]?.asText(), equalTo("reply-id"))
+               assertThat(json["sone"]?.asText(), equalTo("soneId"))
        }
 
        @Test
        fun `sender can be chosen from local sones`() {
            val sone = mock<Sone>().apply { whenever(id).thenReturn("local-sone") }
-               addLocalSone("local-sone", sone)
+               addLocalSone(sone)
                val post = mock<Post>()
                addPost(post, "post-id")
                val reply = mock<PostReply>().apply { whenever(id).thenReturn("reply-id") }
@@ -58,8 +58,8 @@ class CreateReplyAjaxPageTest : JsonPageTest("createReply.ajax", pageSupplier =
                addRequestParameter("text", "Text")
                addRequestParameter("sender", "local-sone")
                assertThat(json.isSuccess, equalTo(true))
-               assertThat(json["reply"].asText(), equalTo("reply-id"))
-               assertThat(json["sone"].asText(), equalTo("local-sone"))
+               assertThat(json["reply"]?.asText(), equalTo("reply-id"))
+               assertThat(json["sone"]?.asText(), equalTo("local-sone"))
        }
 
 }