+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));
+ }
+
+}