Return nullable Post instead of Optional

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / fcp / DeletePostCommandTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/fcp/DeletePostCommandTest.kt b/src/test/kotlin/net/pterodactylus/sone/fcp/DeletePostCommandTest.kt
index 751ad24..30fc4cb 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/fcp/DeletePostCommandTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/fcp/DeletePostCommandTest.kt
@@ -38,7 +38,7 @@ class DeletePostCommandTest : SoneCommandTest() {
        @Test
        fun `request with post from remote sone returns error response`() {
                parameters += "Post" to "RemotePostId"
-               whenever(core.getPost("RemotePostId")).thenReturn(of(postFromRemoteSone))
+               whenever(core.getPost("RemotePostId")).thenReturn(postFromRemoteSone)
                val response = command.execute(parameters)
                assertThat(response.replyParameters["Message"], equalTo("Error"))
                assertThat(response.replyParameters["ErrorCode"], equalTo("401"))
@@ -47,7 +47,7 @@ class DeletePostCommandTest : SoneCommandTest() {
        @Test
        fun `request with post from local sone deletes posts`() {
                parameters += "Post" to "LocalPostId"
-               whenever(core.getPost("LocalPostId")).thenReturn(of(postFromLocalSone))
+               whenever(core.getPost("LocalPostId")).thenReturn(postFromLocalSone)
                val response = command.execute(parameters)
                assertThat(response.replyParameters["Message"], equalTo("PostDeleted"))
                verify(core).deletePost(postFromLocalSone)