Return mocked replies to a post.

[Sone.git] / src / test / java / net / pterodactylus / sone / data / Mocks.java

diff --git a/src/test/java/net/pterodactylus/sone/data/Mocks.java b/src/test/java/net/pterodactylus/sone/data/Mocks.java
index d5c2c08..21e5925 100644 (file)
--- a/src/test/java/net/pterodactylus/sone/data/Mocks.java
+++ b/src/test/java/net/pterodactylus/sone/data/Mocks.java
@@ -21,7 +21,6 @@ import static com.google.common.base.Optional.of;
 import static com.google.common.collect.ArrayListMultimap.create;
 import static com.google.common.collect.Ordering.from;
 import static com.google.common.collect.Sets.newHashSet;
-import static net.pterodactylus.sone.data.Post.TIME_COMPARATOR;
 import static org.mockito.Matchers.anyString;
 import static org.mockito.Matchers.eq;
 import static org.mockito.Mockito.mock;
@@ -39,6 +38,7 @@ import net.pterodactylus.sone.database.PostReplyBuilder;
 import com.google.common.base.Optional;
 import com.google.common.collect.FluentIterable;
 import com.google.common.collect.Multimap;
+import com.google.common.collect.Ordering;
 import org.mockito.invocation.InvocationOnMock;
 import org.mockito.stubbing.Answer;
 
@@ -51,6 +51,7 @@ public class Mocks {
 
        private final Multimap<Sone, Post> sonePosts = create();
        private final Collection<Sone> sones = newHashSet();
+       private final Multimap<Post, PostReply> postReplies = create();
        public final Database database;
        public final Core core;
 
@@ -133,7 +134,7 @@ public class Mocks {
                        when(mockedSone.getPosts()).then(new Answer<List<Post>>() {
                                @Override
                                public List<Post> answer(InvocationOnMock invocationOnMock) throws Throwable {
-                                       return from(TIME_COMPARATOR).sortedCopy(sonePosts.get(mockedSone));
+                                       return from(Post.TIME_COMPARATOR).sortedCopy(sonePosts.get(mockedSone));
                                }
                        });
                        when(mockedSone.toString()).thenReturn(String.format("Sone[%s]", id));
@@ -159,6 +160,12 @@ public class Mocks {
                        when(post.getSone()).thenReturn(sone);
                        when(database.getPost(eq(id))).thenReturn(of(post));
                        sonePosts.put(sone, post);
+                       when(post.getReplies()).then(new Answer<List<PostReply>>() {
+                               @Override
+                               public List<PostReply> answer(InvocationOnMock invocation) throws Throwable {
+                                       return Ordering.from(Reply.TIME_COMPARATOR).sortedCopy(postReplies.get(post));
+                               }
+                       });
                        return post;
                }