src/test/java/net/pterodactylus/sone/web/NewPageTest.java

   1 package net.pterodactylus.sone.web;
   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.mockito.Matchers.any;
   7 import static org.mockito.Matchers.anyBoolean;
   8 import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
   9 import static org.mockito.Mockito.mock;
  10 import static org.mockito.Mockito.when;
  11
  12 import java.util.Collections;
  13 import java.util.List;
  14
  15 import net.pterodactylus.sone.data.Post;
  16 import net.pterodactylus.sone.data.PostReply;
  17 import net.pterodactylus.sone.data.Sone;
  18 import net.pterodactylus.sone.web.page.FreenetRequest;
  19 import net.pterodactylus.util.notify.Notification;
  20 import net.pterodactylus.util.template.Template;
  21 import net.pterodactylus.util.template.TemplateContext;
  22
  23 import freenet.clients.http.ToadletContext;
  24
  25 import com.google.common.base.Optional;
  26 import org.junit.Before;
  27 import org.junit.Test;
  28
  29 /**
  30  * Unit test for {@link NewPage}.
  31  *
  32  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  33  */
  34 public class NewPageTest {
  35
  36         private final Template template = mock(Template.class);
  37         private final WebInterface webInterface = mock(WebInterface.class, RETURNS_DEEP_STUBS);
  38         private final NewPage newPage = new NewPage(template, webInterface);
  39         private final Sone currentSone = mock(Sone.class);
  40         private final TemplateContext templateContext = new TemplateContext();
  41         private final FreenetRequest freenetRequest = mock(FreenetRequest.class, RETURNS_DEEP_STUBS);
  42
  43         @Before
  44         public void setupFreenetRequest() {
  45                 when(freenetRequest.getToadletContext()).thenReturn(mock(ToadletContext.class));
  46         }
  47
  48         @Before
  49         public void setupWebInterface() {
  50                 when(webInterface.getCore().getPreferences().getPostsPerPage()).thenReturn(5);
  51                 when(webInterface.getCurrentSone(any(ToadletContext.class), anyBoolean())).thenReturn(currentSone);
  52                 when(webInterface.getNotifications(any(Sone.class))).thenReturn(Collections.<Notification>emptyList());
  53         }
  54
  55         @Test
  56         public void postsAreNotDuplicatedWhenTheyComeFromBothNewPostsAndNewRepliesNotifications() throws Exception {
  57                 // given
  58                 Post extraPost = mock(Post.class);
  59                 List<Post> posts = asList(mock(Post.class), mock(Post.class));
  60                 List<PostReply> postReplies = asList(mock(PostReply.class), mock(PostReply.class));
  61                 when(postReplies.get(0).getPost()).thenReturn(Optional.of(posts.get(0)));
  62                 when(postReplies.get(1).getPost()).thenReturn(Optional.of(extraPost));
  63                 when(webInterface.getNewPosts(currentSone)).thenReturn(posts);
  64                 when(webInterface.getNewReplies(currentSone)).thenReturn(postReplies);
  65
  66                 // when
  67                 newPage.processTemplate(freenetRequest, templateContext);
  68
  69                 // then
  70                 List<Post> renderedPosts = templateContext.get("posts", List.class);
  71                 assertThat(renderedPosts, containsInAnyOrder(posts.get(0), posts.get(1), extraPost));
  72         }
  73
  74 }