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

   1 package net.pterodactylus.sone.database.memory;
   2
   3 import static java.util.Arrays.asList;
   4 import static org.hamcrest.MatcherAssert.assertThat;
   5 import static org.hamcrest.Matchers.containsInAnyOrder;
   6 import static org.hamcrest.Matchers.is;
   7 import static org.hamcrest.Matchers.nullValue;
   8 import static org.mockito.Matchers.anyString;
   9 import static org.mockito.Mockito.mock;
  10 import static org.mockito.Mockito.when;
  11
  12 import java.util.HashMap;
  13 import java.util.HashSet;
  14 import java.util.LinkedHashSet;
  15 import java.util.Map;
  16 import java.util.Set;
  17
  18 import net.pterodactylus.sone.TestValue;
  19 import net.pterodactylus.util.config.Configuration;
  20 import net.pterodactylus.util.config.ConfigurationException;
  21 import net.pterodactylus.util.config.Value;
  22
  23 import org.junit.Test;
  24 import org.mockito.invocation.InvocationOnMock;
  25 import org.mockito.stubbing.Answer;
  26
  27 /**
  28  * Unit test for {@link ConfigurationLoader}.
  29  *
  30  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  31  */
  32 public class ConfigurationLoaderTest {
  33
  34         private final Configuration configuration = mock(Configuration.class);
  35         private final ConfigurationLoader configurationLoader =
  36                         new ConfigurationLoader(configuration);
  37
  38         @Test
  39         public void loaderCanLoadKnownPosts() {
  40                 when(configuration.getStringValue("KnownPosts/0/ID"))
  41                                 .thenReturn(TestValue.from("Post2"));
  42                 when(configuration.getStringValue("KnownPosts/1/ID"))
  43                                 .thenReturn(TestValue.from("Post1"));
  44                 when(configuration.getStringValue("KnownPosts/2/ID"))
  45                                 .thenReturn(TestValue.<String>from(null));
  46                 Set<String> knownPosts = configurationLoader.loadKnownPosts();
  47                 assertThat(knownPosts, containsInAnyOrder("Post1", "Post2"));
  48         }
  49
  50         @Test
  51         public void loaderCanSaveKnownPosts() throws ConfigurationException {
  52                 Map<String, Value<String>> configurationValues = prepareConfigurationValues();
  53                 HashSet<String> originalPosts = new LinkedHashSet<String>(asList("Post1", "Post2"));
  54                 configurationLoader.saveKnownPosts(originalPosts);
  55                 assertThat(configurationValues.get("KnownPosts/0/ID").getValue(), is("Post1"));
  56                 assertThat(configurationValues.get("KnownPosts/1/ID").getValue(), is("Post2"));
  57                 assertThat(configurationValues.get("KnownPosts/2/ID").getValue(), nullValue());
  58         }
  59
  60         private Map<String, Value<String>> prepareConfigurationValues() {
  61                 final Map<String, Value<String>> configurationValues = new HashMap<String, Value<String>>();
  62                 when(configuration.getStringValue(anyString())).thenAnswer(new Answer<Value<String>>() {
  63                         @Override
  64                         public Value<String> answer(InvocationOnMock invocation) throws Throwable {
  65                                 Value<String> stringValue = TestValue.from(null);
  66                                 configurationValues.put((String) invocation.getArguments()[0], stringValue);
  67                                 return stringValue;
  68                         }
  69                 });
  70                 return configurationValues;
  71         }
  72
  73         @Test
  74         public void loaderCanLoadKnownPostReplies() {
  75                 when(configuration.getStringValue("KnownReplies/0/ID"))
  76                                 .thenReturn(TestValue.from("PostReply2"));
  77                 when(configuration.getStringValue("KnownReplies/1/ID"))
  78                                 .thenReturn(TestValue.from("PostReply1"));
  79                 when(configuration.getStringValue("KnownReplies/2/ID"))
  80                                 .thenReturn(TestValue.<String>from(null));
  81                 Set<String> knownPosts = configurationLoader.loadKnownPostReplies();
  82                 assertThat(knownPosts,
  83                                 containsInAnyOrder("PostReply1", "PostReply2"));
  84         }
  85
  86         @Test
  87         public void loaderCanLoadBookmarkedPosts() {
  88                 when(configuration.getStringValue("Bookmarks/Post/0/ID"))
  89                                 .thenReturn(TestValue.from("Post2"));
  90                 when(configuration.getStringValue("Bookmarks/Post/1/ID"))
  91                                 .thenReturn(TestValue.from("Post1"));
  92                 when(configuration.getStringValue("Bookmarks/Post/2/ID"))
  93                                 .thenReturn(TestValue.<String>from(null));
  94                 Set<String> knownPosts = configurationLoader.loadBookmarkedPosts();
  95                 assertThat(knownPosts, containsInAnyOrder("Post1", "Post2"));
  96         }
  97
  98         @Test
  99         public void loaderCanSaveBookmarkedPosts() throws ConfigurationException {
 100                 final Value<String> post1 = TestValue.<String>from(null);
 101                 final Value<String> post2 = TestValue.<String>from(null);
 102                 final Value<String> post3 = TestValue.<String>from(null);
 103                 when(configuration.getStringValue("Bookmarks/Post/0/ID")).thenReturn(post1);
 104                 when(configuration.getStringValue("Bookmarks/Post/1/ID")).thenReturn(post2);
 105                 when(configuration.getStringValue("Bookmarks/Post/2/ID")).thenReturn(post3);
 106                 HashSet<String> originalPosts = new HashSet<String>(asList("Post1", "Post2"));
 107                 configurationLoader.saveBookmarkedPosts(originalPosts);
 108                 HashSet<String> extractedPosts =
 109                                 new HashSet<String>(asList(post1.getValue(), post2.getValue()));
 110                 assertThat(extractedPosts, containsInAnyOrder("Post1", "Post2"));
 111                 assertThat(post3.getValue(), nullValue());
 112         }
 113
 114 }