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=d6eae1072ce7c7fe906eaeb71e3264ef3af377ca;hp=ce61ade1c67f9f667c82e50884b6de5461e3c71f;hb=308b94c1c75e55637644142e2553aec8d7ce01ad;hpb=36b8e20651c14970ca30a1b8ed43af219b367bc9

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 ce61ade..d6eae10 100644
--- a/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java
+++ b/src/test/java/net/pterodactylus/sone/database/memory/MemoryBookmarkDatabaseTest.java
@@ -2,10 +2,12 @@ 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.any;
 import static org.mockito.Matchers.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;
@@ -29,7 +31,8 @@ import org.mockito.stubbing.Answer;
 public class MemoryBookmarkDatabaseTest {
 
 	private final MemoryDatabase memoryDatabase = mock(MemoryDatabase.class);
-	private final ConfigurationLoader configurationLoader = mock(ConfigurationLoader.class);
+	private final ConfigurationLoader configurationLoader =
+			mock(ConfigurationLoader.class);
 	private final MemoryBookmarkDatabase bookmarkDatabase =
 			new MemoryBookmarkDatabase(memoryDatabase, configurationLoader);
 	private final Map<String, Post> posts = new HashMap<String, Post>();
@@ -72,6 +75,15 @@ 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 removingABookmarkRemovesTheCorrectBookmark() {
 		Set<Post> allPosts = new HashSet<Post>(posts.values());
 		for (Post post : allPosts) {
@@ -88,11 +100,15 @@ 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));
 	}
 
 }