@Test
public void returnsListOfComments() {

    CommentsFeed feed = new CommentsFeed();
    Comment comment = mock(Comment.class);
    List<Comment> comments = Arrays.asList(comment);
    feed.comments = comments;
    feed.users = new HashMap<>();
    assertThat(feed.flatten(), equalTo(comments));
}