src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java

   1 package net.pterodactylus.sone.database.memory;
   2
   3 import static com.google.common.base.Optional.fromNullable;
   4 import static org.hamcrest.MatcherAssert.assertThat;
   5 import static org.hamcrest.Matchers.contains;
   6 import static org.hamcrest.Matchers.is;
   7 import static org.mockito.Matchers.any;
   8 import static org.mockito.Matchers.anyString;
   9 import static org.mockito.Mockito.mock;
  10 import static org.mockito.Mockito.verify;
  11 import static org.mockito.Mockito.when;
  12
  13 import java.util.HashMap;
  14 import java.util.HashSet;
  15 import java.util.Map;
  16 import java.util.Set;
  17
  18 import net.pterodactylus.sone.data.Post;
  19
  20 import com.google.common.base.Optional;
  21 import org.junit.Before;
  22 import org.junit.Test;
  23 import org.mockito.invocation.InvocationOnMock;
  24 import org.mockito.stubbing.Answer;
  25
  26 /**
  27  * Unit test for {@link MemoryBookmarkDatabase}.
  28  *
  29  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  30  */
  31 public class MemoryBookmarkDatabaseTest {
  32
  33         private final MemoryDatabase memoryDatabase = mock(MemoryDatabase.class);
  34         private final ConfigurationLoader configurationLoader = mock(ConfigurationLoader.class);
  35         private final MemoryBookmarkDatabase bookmarkDatabase =
  36                         new MemoryBookmarkDatabase(memoryDatabase, configurationLoader);
  37         private final Map<String, Post> posts = new HashMap<String, Post>();
  38
  39         @Before
  40         public void setupMemoryDatabase() {
  41                 when(memoryDatabase.getPost(anyString())).thenAnswer(
  42                                 new Answer<Optional<Post>>() {
  43                                         @Override
  44                                         public Optional<Post> answer(
  45                                                         InvocationOnMock invocation) {
  46                                                 return fromNullable(
  47                                                                 posts.get(invocation.getArguments()[0]));
  48                                         }
  49                                 });
  50         }
  51
  52         @Before
  53         public void setupPosts() {
  54                 createPost("PostId1");
  55                 createPost("PostId2");
  56         }
  57
  58         private void createPost(String postId) {
  59                 Post post = mock(Post.class);
  60                 when(post.getId()).thenReturn(postId);
  61                 posts.put(postId, post);
  62         }
  63
  64         @Test
  65         public void bookmarkDatabaseRetainsBookmarkedPosts() {
  66                 Set<Post> allPosts = new HashSet<Post>(posts.values());
  67                 for (Post post : allPosts) {
  68                         bookmarkDatabase.bookmarkPost(post);
  69                 }
  70                 assertThat(bookmarkDatabase.getBookmarkedPosts(), is(allPosts));
  71                 for (Post post : allPosts) {
  72                         assertThat(bookmarkDatabase.isPostBookmarked(post), is(true));
  73                 }
  74         }
  75
  76         @Test
  77         public void removingABookmarkRemovesTheCorrectBookmark() {
  78                 Set<Post> allPosts = new HashSet<Post>(posts.values());
  79                 for (Post post : allPosts) {
  80                         bookmarkDatabase.bookmarkPost(post);
  81                 }
  82                 Post randomPost = posts.values().iterator().next();
  83                 bookmarkDatabase.unbookmarkPost(randomPost);
  84                 allPosts.remove(randomPost);
  85                 assertThat(bookmarkDatabase.getBookmarkedPosts(), is(allPosts));
  86                 for (Post post : posts.values()) {
  87                         assertThat(bookmarkDatabase.isPostBookmarked(post),
  88                                         is(!post.equals(randomPost)));
  89                 }
  90         }
  91
  92         @Test
  93         public void startingTheDatabaseLoadsBookmarkedPosts() {
  94                 bookmarkDatabase.start();
  95                 verify(configurationLoader).loadBookmarkedPosts();
  96         }
  97
  98         @Test
  99         public void stoppingTheDatabaseSavesTheBookmarkedPosts() {
 100                 bookmarkDatabase.stop();
 101                 verify(configurationLoader).saveBookmarkedPosts(any(Set.class));
 102         }
 103
 104 }