X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryBookmarkDatabaseTest.java;h=c16ce9dfa27f6a0e1b992b83c0e7f030b9ec131d;hb=d338aacec01efa636f96c5f8aab11b2345147691;hp=c5ce475f8467427f0a79b6c50532bb2ad7485d25;hpb=04fbdc3d76e3b1140d178dc2600141ec92d22637;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java b/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java
index c5ce475..c16ce9d 100644
--- a/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java
+++ b/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java
@@ -1,10 +1,12 @@
 package net.pterodactylus.sone.database.memory;
 
 import static com.google.common.base.Optional.fromNullable;
+import static net.pterodactylus.sone.Matchers.isPostWithId;
 import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.contains;
 import static org.hamcrest.Matchers.is;
-import static org.mockito.Matchers.any;
-import static org.mockito.Matchers.anyString;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
@@ -52,14 +54,20 @@ public class MemoryBookmarkDatabaseTest {
 
 	@Before
 	public void setupPosts() {
-		createPost("PostId1");
-		createPost("PostId2");
+		createAndRegisterPost("PostId1");
+		createAndRegisterPost("PostId2");
 	}
 
-	private void createPost(String postId) {
+	private Post createAndRegisterPost(String postId) {
+		Post post = createPost(postId);
+		posts.put(postId, post);
+		return post;
+	}
+
+	private Post createPost(String postId) {
 		Post post = mock(Post.class);
 		when(post.getId()).thenReturn(postId);
-		posts.put(postId, post);
+		return post;
 	}
 
 	@Test
@@ -121,4 +129,14 @@ public class MemoryBookmarkDatabaseTest {
 		verify(configurationLoader).saveBookmarkedPosts(any(Set.class));
 	}
 
+	@Test
+	public void bookmarkedPostsIncludeNotYetLoadedPosts() {
+		bookmarkDatabase.bookmarkPost(posts.get("PostId1"));
+		bookmarkDatabase.bookmarkPost(createPost("PostId3"));
+		final Set<Post> bookmarkedPosts =
+				bookmarkDatabase.getBookmarkedPosts();
+		assertThat(bookmarkedPosts,
+				contains(isPostWithId("PostId1"), isPostWithId("PostId3")));
+	}
+
 }