Add test for storing IDs on known post replies

[Sone.git] / src / main / java / net / pterodactylus / sone / database / memory / MemoryDatabase.kt

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
index fda4227..d7e84e5 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.kt
@@ -1,5 +1,5 @@
 /*
- * Sone - MemoryDatabase.java - Copyright © 2013–2016 David Roden
+ * Sone - MemoryDatabase.java - Copyright © 2013–2019 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -287,6 +287,7 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
                                        knownPosts.add(post.id)
                                else
                                        knownPosts.remove(post.id)
+                               saveKnownPosts()
                        }
 
        protected fun isPostReplyKnown(postReply: PostReply) = readLock.withLock { postReply.id in knownPostReplies }
@@ -297,6 +298,7 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
                                        knownPostReplies.add(postReply.id)
                                else
                                        knownPostReplies.remove(postReply.id)
+                               saveKnownPostReplies()
                        }
 
        private fun loadKnownPosts() =
@@ -312,7 +314,7 @@ class MemoryDatabase @Inject constructor(private val configuration: Configuratio
                        try {
                                readLock.withLock {
                                        knownPosts.forEachIndexed { index, knownPostId ->
-                                               configuration.getStringValue("KnowsPosts/$index/ID").value = knownPostId
+                                               configuration.getStringValue("KnownPosts/$index/ID").value = knownPostId
                                        }
                                        configuration.getStringValue("KnownPosts/${knownPosts.size}/ID").value = null
                                }