Add bookmark database.

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

diff --git a/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java b/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java
new file mode 100644 (file)
index 0000000..93efe11
--- /dev/null
+++ b/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java
@@ -0,0 +1,97 @@
+package net.pterodactylus.sone.database.memory;
+
+import static com.google.common.base.Optional.fromNullable;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.contains;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.Matchers.anyString;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
+
+import net.pterodactylus.sone.data.Post;
+
+import com.google.common.base.Optional;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
+
+/**
+ * Unit test for {@link MemoryBookmarkDatabase}.
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+public class MemoryBookmarkDatabaseTest {
+
+       private final MemoryDatabase memoryDatabase = mock(MemoryDatabase.class);
+       private final MemoryBookmarkDatabase bookmarkDatabase =
+                       new MemoryBookmarkDatabase(memoryDatabase);
+       private final Map<String, Post> posts = new HashMap<String, Post>();
+
+       @Before
+       public void setupMemoryDatabase() {
+               when(memoryDatabase.getPost(anyString())).thenAnswer(
+                               new Answer<Optional<Post>>() {
+                                       @Override
+                                       public Optional<Post> answer(
+                                                       InvocationOnMock invocation) {
+                                               return fromNullable(
+                                                               posts.get(invocation.getArguments()[0]));
+                                       }
+                               });
+       }
+
+       @Before
+       public void setupPosts() {
+               createPost("PostId1");
+               createPost("PostId2");
+       }
+
+       private void createPost(String postId) {
+               Post post = mock(Post.class);
+               when(post.getId()).thenReturn(postId);
+               posts.put(postId, post);
+       }
+
+       @Test
+       public void bookmarkDatabaseRetainsBookmarkedPosts() {
+               Set<Post> allPosts = new HashSet<Post>(posts.values());
+               for (Post post : allPosts) {
+                       bookmarkDatabase.bookmarkPost(post);
+               }
+               assertThat(bookmarkDatabase.getBookmarkedPosts(), is(allPosts));
+               for (Post post : allPosts) {
+                       assertThat(bookmarkDatabase.isPostBookmarked(post), is(true));
+               }
+       }
+
+       @Test
+       public void removingABookmarkRemovesTheCorrectBookmark() {
+               Set<Post> allPosts = new HashSet<Post>(posts.values());
+               for (Post post : allPosts) {
+                       bookmarkDatabase.bookmarkPost(post);
+               }
+               Post randomPost = posts.values().iterator().next();
+               bookmarkDatabase.unbookmarkPost(randomPost);
+               allPosts.remove(randomPost);
+               assertThat(bookmarkDatabase.getBookmarkedPosts(), is(allPosts));
+               for (Post post : posts.values()) {
+                       assertThat(bookmarkDatabase.isPostBookmarked(post),
+                                       is(!post.equals(randomPost)));
+               }
+       }
+
+       @Test
+       public void addingABookmarkByIdBookmarksTheCorrectPost() {
+               Post randomPost = posts.values().iterator().next();
+               bookmarkDatabase.bookmarkPost(randomPost.getId());
+               assertThat(bookmarkDatabase.getBookmarkedPosts(),
+                               contains(randomPost));
+       }
+
+}