X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fajax%2FJsonPageTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fajax%2FJsonPageTest.kt;h=73dba5a2736f8047a0b9af6f2fe45d439ce8927a;hp=c950a3121c5425b85ce47ae7a045e89396a5550d;hb=355e5656b51974303534146a3e401f3b73e4cea8;hpb=48a7e99c47320d9097ef16d822ecfd3c14374a92

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
--- 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
 	}
 
-	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) {