Refactoring: make id for local Sone optional

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / web / ajax / JsonPageTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt
index c950a31..73dba5a 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt
@@ -195,8 +195,8 @@ abstract class JsonPageTest(
                remoteSones += (soneId ?: sone.id) to sone
        }
 
                remoteSones += (soneId ?: sone.id) to sone
        }
 

-       protected fun addLocalSone(id: String, sone: Sone) {
-               localSones += id to sone
+       protected fun addLocalSone(sone: Sone, id: String? = null) {
+               localSones[id ?: sone.id] = sone

        }
 
        protected fun addPost(post: Post, id: String? = null) {
        }
 
        protected fun addPost(post: Post, id: String? = null) {