Load bookmarked posts in configuration loader, too.

[Sone.git] / src / test / java / net / pterodactylus / sone / database / memory / ConfigurationLoaderTest.java

diff --git a/src/test/java/net/pterodactylus/sone/database/memory/ConfigurationLoaderTest.java b/src/test/java/net/pterodactylus/sone/database/memory/ConfigurationLoaderTest.java
index 6449b43..de357e9 100644 (file)
--- a/src/test/java/net/pterodactylus/sone/database/memory/ConfigurationLoaderTest.java
+++ b/src/test/java/net/pterodactylus/sone/database/memory/ConfigurationLoaderTest.java
@@ -35,4 +35,29 @@ public class ConfigurationLoaderTest {
                assertThat(knownPosts, containsInAnyOrder("Post1", "Post2"));
        }
 
+       @Test
+       public void loaderCanLoadKnownPostReplies() {
+               when(configuration.getStringValue("KnownReplies/0/ID"))
+                               .thenReturn(new TestValue<String>("PostReply2"));
+               when(configuration.getStringValue("KnownReplies/1/ID"))
+                               .thenReturn(new TestValue<String>("PostReply1"));
+               when(configuration.getStringValue("KnownReplies/2/ID"))
+                               .thenReturn(new TestValue<String>(null));
+               Set<String> knownPosts = configurationLoader.loadKnownPostReplies();
+               assertThat(knownPosts,
+                               containsInAnyOrder("PostReply1", "PostReply2"));
+       }
+
+       @Test
+       public void loaderCanLoadBookmarkedPosts() {
+               when(configuration.getStringValue("Bookmarks/Post/0/ID"))
+                               .thenReturn(new TestValue<String>("Post2"));
+               when(configuration.getStringValue("Bookmarks/Post/1/ID"))
+                               .thenReturn(new TestValue<String>("Post1"));
+               when(configuration.getStringValue("Bookmarks/Post/2/ID"))
+                               .thenReturn(new TestValue<String>(null));
+               Set<String> knownPosts = configurationLoader.loadBookmarkedPosts();
+               assertThat(knownPosts, containsInAnyOrder("Post1", "Post2"));
+       }
+
 }