- public Post mockPost(Sone sone, String postId) {
- Post post = mock(Post.class);
- when(post.getId()).thenReturn(postId);
- when(post.getSone()).thenReturn(sone);
- when(database.getPost(eq(postId))).thenReturn(of(post));
- sonePosts.put(sone, post);
- return post;
+ public class PostMocker {
+
+ private final Post post = mock(Post.class);
+ private final String id;
+ private final Sone sone;
+ private Optional<String> recipientId = absent();
+ private long time;
+ private Optional<String> text = absent();
+
+ public PostMocker(String id, Sone sone) {
+ this.id = id;
+ this.sone = sone;
+ }
+
+ public PostMocker withRecipient(String recipientId) {
+ this.recipientId = fromNullable(recipientId);
+ return this;
+ }
+
+ public PostMocker withTime(long time) {
+ this.time = time;
+ return this;
+ }
+
+ public PostMocker withText(String text) {
+ this.text = fromNullable(text);
+ return this;
+ }
+
+ public Post create() {
+ when(post.getId()).thenReturn(id);
+ when(post.getSone()).thenReturn(sone);
+ when(post.getRecipientId()).thenReturn(recipientId);
+ if (recipientId.isPresent()) {
+ directedPosts.put(recipientId.get(), post);
+ }
+ when(post.getTime()).thenReturn(time);
+ if (text.isPresent()) {
+ when(post.getText()).thenReturn(text.get());
+ }
+ 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));
+ }
+ });
+ doAnswer(new Answer<Void>() {
+ @Override
+ public Void answer(InvocationOnMock invocation) throws Throwable {
+ postLikingSones.put(post, (Sone) invocation.getArguments()[0]);
+ return null;
+ }
+ }).when(post).like(Matchers.<Sone>any());
+ doAnswer(new Answer<Void>() {
+ @Override
+ public Void answer(InvocationOnMock invocation) throws Throwable {
+ postLikingSones.remove(post, (Sone) invocation.getArguments()[0]);
+ return null;
+ }
+ }).when(post).unlike(Matchers.<Sone>any());
+ when(post.getLikes()).thenAnswer(new Answer<Set<Sone>>() {
+ @Override
+ public Set<Sone> answer(InvocationOnMock invocation) throws Throwable {
+ return postLikingSones.get(post);
+ }
+ });
+ return post;
+ }
+