X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabaseTest.java;h=60c8c531f16cd35d85c1299870d381e3e300ac3d;hp=47c2c4c3a36c5858fa8079acebff6d33918dcfdc;hb=b798049da3b4d4a58d5ff12314bc1022808171bc;hpb=7804921fddbc1a9c7e6c739e802b724b45190cc8

diff --git a/src/test/java/net/pterodactylus/sone/database/memory/MemoryDatabaseTest.java b/src/test/java/net/pterodactylus/sone/database/memory/MemoryDatabaseTest.java
index 47c2c4c..60c8c53 100644
--- a/src/test/java/net/pterodactylus/sone/database/memory/MemoryDatabaseTest.java
+++ b/src/test/java/net/pterodactylus/sone/database/memory/MemoryDatabaseTest.java
@@ -1,5 +1,5 @@
 /*
- * Sone - MemoryDatabaseTest.java - Copyright © 2013–2016 David Roden
+ * Sone - MemoryDatabaseTest.java - Copyright © 2013–2019 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -70,8 +70,6 @@ import org.mockito.stubbing.Answer;
 
 /**
  * Tests for {@link MemoryDatabase}.
- *
- * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
 public class MemoryDatabaseTest {
 
@@ -122,7 +120,7 @@ public class MemoryDatabaseTest {
 						.withTime(4000L)
 						.withText("reply2")
 						.build();
-		Set<PostReply> postReplies = new HashSet<PostReply>(
+		Set<PostReply> postReplies = new HashSet<>(
 				asList(firstPostFirstReply, firstPostSecondReply,
 						secondPostReply));
 		when(sone.getReplies()).thenReturn(postReplies);
@@ -141,6 +139,7 @@ public class MemoryDatabaseTest {
 				"album-description3").update();
 		firstAlbum.addAlbum(thirdAlbum);
 		Album rootAlbum = mock(Album.class);
+		when(rootAlbum.getId()).thenReturn("root");
 		when(rootAlbum.getAlbums()).thenReturn(
 				asList(firstAlbum, secondAlbum));
 		when(sone.getRootAlbum()).thenReturn(rootAlbum);
@@ -215,7 +214,7 @@ public class MemoryDatabaseTest {
 
 	@Test
 	public void storedAndRemovedSoneIsNotAvailable() {
-	    storedSoneIsMadeAvailable();
+		storedSoneIsMadeAvailable();
 		memoryDatabase.removeSone(sone);
 		assertThat(memoryDatabase.getSones(), empty());
 	}
@@ -392,6 +391,22 @@ public class MemoryDatabaseTest {
 	}
 
 	@Test
+	public void soneFollowingTimeIsReturnedCorrectly() throws ConfigurationException {
+		prepareConfigurationValues();
+		configuration.getStringValue("SoneFollowingTimes/0/Sone").setValue("sone");
+		configuration.getLongValue("SoneFollowingTimes/0/Time").setValue(1000L);
+		assertThat(memoryDatabase.getFollowingTime("sone"), equalTo(1000L));
+	}
+
+	@Test
+	public void nullisReturnedWhenSoneIsNotFollowed() throws ConfigurationException {
+		prepareConfigurationValues();
+		configuration.getStringValue("SoneFollowingTimes/0/Sone").setValue("otherSone");
+		configuration.getLongValue("SoneFollowingTimes/0/Time").setValue(1000L);
+		assertThat(memoryDatabase.getFollowingTime("sone"), nullValue());
+	}
+
+	@Test
 	public void timeIsStoredInConfigurationWhenASoneIsFollowed() throws ConfigurationException {
 		prepareConfigurationValues();
 		when(sone.isLocal()).thenReturn(true);
@@ -413,4 +428,52 @@ public class MemoryDatabaseTest {
 		assertThat(configuration.getStringValue("SoneFollowingTimes/1/Sone").getValue(), nullValue());
 	}
 
+	@Test
+	public void unfollowingASoneRemovesTheFollowingTime() throws ConfigurationException {
+		prepareConfigurationValues();
+		configuration.getStringValue("Sone/sone/Friends/0/ID").setValue("Friend");
+		configuration.getStringValue("SoneFollowingTimes/0/Sone").setValue("Friend");
+		configuration.getLongValue("SoneFollowingTimes/0/Time").setValue(1000L);
+		when(sone.isLocal()).thenReturn(true);
+		memoryDatabase.removeFriend(sone, "Friend");
+		assertThat(configuration.getStringValue("SoneFollowingTimes/0/Sone").getValue(), nullValue());
+	}
+
+	@Test
+	public void unfollowingASoneDoesNotRemoveTheFollowingTimeIfAnotherLocalSoneFollowsIt() throws ConfigurationException {
+		prepareConfigurationValues();
+		configuration.getStringValue("Sone/sone/Friends/0/ID").setValue("Friend");
+		configuration.getStringValue("Sone/other-sone/Friends/0/ID").setValue("Friend");
+		configuration.getStringValue("SoneFollowingTimes/0/Sone").setValue("Friend");
+		configuration.getLongValue("SoneFollowingTimes/0/Time").setValue(1000L);
+		Sone otherSone = mock(Sone.class);
+		when(otherSone.isLocal()).thenReturn(true);
+		when(otherSone.getId()).thenReturn("other-sone");
+		memoryDatabase.getFriends(otherSone);
+		when(sone.isLocal()).thenReturn(true);
+		memoryDatabase.removeFriend(sone, "Friend");
+		assertThat(configuration.getStringValue("SoneFollowingTimes/0/Sone").getValue(), equalTo("Friend"));
+		assertThat(configuration.getLongValue("SoneFollowingTimes/0/Time").getValue(), equalTo(1000L));
+	}
+
+	@Test
+	public void markingAPostAsKnownSavesConfiguration() throws ConfigurationException {
+		prepareConfigurationValues();
+		Post post = mock(Post.class);
+		when(post.getId()).thenReturn("post-id");
+		memoryDatabase.setPostKnown(post, true);
+		assertThat(configuration.getStringValue("KnownPosts/0/ID").getValue(), equalTo("post-id"));
+		assertThat(configuration.getStringValue("KnownPosts/1/ID").getValue(), equalTo(null));
+	}
+
+	@Test
+	public void markingAPostReplyAsKnownSavesConfiguration() throws ConfigurationException {
+		prepareConfigurationValues();
+		PostReply postReply = mock(PostReply.class);
+		when(postReply.getId()).thenReturn("post-reply-id");
+		memoryDatabase.setPostReplyKnown(postReply, true);
+		assertThat(configuration.getStringValue("KnownReplies/0/ID").getValue(), equalTo("post-reply-id"));
+		assertThat(configuration.getStringValue("KnownReplies/1/ID").getValue(), equalTo(null));
+	}
+
 }