Don’t return null to function not allowing it

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / database / memory / MemoryPostTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/database/memory/MemoryPostTest.kt b/src/test/kotlin/net/pterodactylus/sone/database/memory/MemoryPostTest.kt
index 611ad70..90344e1 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/database/memory/MemoryPostTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/database/memory/MemoryPostTest.kt
@@ -9,10 +9,17 @@ import java.util.*
 
 class MemoryPostTest {
 
+       private val memoryDatabase = mock<MemoryDatabase>()
+
        @Test
        fun `memory post returns empty optional for post without recipient`() {
-               val postDatabase = mock<MemoryDatabase>()
-               val memoryPost = MemoryPost(postDatabase, postDatabase, UUID.randomUUID().toString(), "soneId", null, 123, "text")
+               val memoryPost = MemoryPost(memoryDatabase, memoryDatabase, UUID.randomUUID().toString(), "soneId", null, 123, "text")
+               assertThat(memoryPost.recipient, equalTo(absent()))
+       }
+
+       @Test
+       fun `empty optional is returned if recipient is set but non-existent`() {
+               val memoryPost = MemoryPost(memoryDatabase, memoryDatabase, UUID.randomUUID().toString(), "soneId", "recipientId", 123, "text")
                assertThat(memoryPost.recipient, equalTo(absent()))
        }