X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryBookmarkDatabaseTest.java;h=188b74019ea484a707695b40672084f6dd81f5a4;hp=93efe116730fd7d582c38ac1d3ad9af6e3fbde58;hb=2bacfa78f3191fd9847574a6c8b218a4882844a4;hpb=aba6bd1d58ef244050703e1b99d7389e024809c6

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 93efe11..188b740 100644
--- a/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java
+++ b/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java
@@ -1,11 +1,14 @@
 package net.pterodactylus.sone.database.memory;
 
-import static com.google.common.base.Optional.fromNullable;
+import static net.pterodactylus.sone.test.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.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;
 import static org.mockito.Mockito.when;
 
 import java.util.HashMap;
@@ -15,7 +18,6 @@ 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;
@@ -23,44 +25,49 @@ 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 ConfigurationLoader configurationLoader =
+			mock(ConfigurationLoader.class);
 	private final MemoryBookmarkDatabase bookmarkDatabase =
-			new MemoryBookmarkDatabase(memoryDatabase);
-	private final Map<String, Post> posts = new HashMap<String, Post>();
+			new MemoryBookmarkDatabase(memoryDatabase, configurationLoader);
+	private final Map<String, Post> posts = new HashMap<>();
 
 	@Before
 	public void setupMemoryDatabase() {
 		when(memoryDatabase.getPost(anyString())).thenAnswer(
-				new Answer<Optional<Post>>() {
+				new Answer<Post>() {
 					@Override
-					public Optional<Post> answer(
+					public Post answer(
 							InvocationOnMock invocation) {
-						return fromNullable(
-								posts.get(invocation.getArguments()[0]));
+						return posts.get(invocation.getArguments()[0]);
 					}
 				});
 	}
 
 	@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
 	public void bookmarkDatabaseRetainsBookmarkedPosts() {
-		Set<Post> allPosts = new HashSet<Post>(posts.values());
+		Set<Post> allPosts = new HashSet<>(posts.values());
 		for (Post post : allPosts) {
 			bookmarkDatabase.bookmarkPost(post);
 		}
@@ -71,8 +78,27 @@ public class MemoryBookmarkDatabaseTest {
 	}
 
 	@Test
+	public void bookmarkingAPostSavesTheDatabase() {
+		for (Post post : posts.values()) {
+			bookmarkDatabase.bookmarkPost(post);
+		}
+		verify(configurationLoader, times(posts.size()))
+				.saveBookmarkedPosts(any(Set.class));
+	}
+
+	@Test
+	public void unbookmarkingAPostSavesTheDatabase() {
+		for (Post post : posts.values()) {
+			bookmarkDatabase.bookmarkPost(post);
+			bookmarkDatabase.unbookmarkPost(post);
+		}
+		verify(configurationLoader, times(posts.size() * 2))
+				.saveBookmarkedPosts(any(Set.class));
+	}
+
+	@Test
 	public void removingABookmarkRemovesTheCorrectBookmark() {
-		Set<Post> allPosts = new HashSet<Post>(posts.values());
+		Set<Post> allPosts = new HashSet<>(posts.values());
 		for (Post post : allPosts) {
 			bookmarkDatabase.bookmarkPost(post);
 		}
@@ -87,11 +113,25 @@ public class MemoryBookmarkDatabaseTest {
 	}
 
 	@Test
-	public void addingABookmarkByIdBookmarksTheCorrectPost() {
-		Post randomPost = posts.values().iterator().next();
-		bookmarkDatabase.bookmarkPost(randomPost.getId());
-		assertThat(bookmarkDatabase.getBookmarkedPosts(),
-				contains(randomPost));
+	public void startingTheDatabaseLoadsBookmarkedPosts() {
+		bookmarkDatabase.start();
+		verify(configurationLoader).loadBookmarkedPosts();
+	}
+
+	@Test
+	public void stoppingTheDatabaseSavesTheBookmarkedPosts() {
+		bookmarkDatabase.stop();
+		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")));
 	}
 
 }