should favorite article

src/test/java/io/spring/application/article/ArticleQueryServiceTest.java

@@ -2,10 +2,14 @@ package io.spring.application.article;
 
 import io.spring.core.article.Article;
 import io.spring.core.article.ArticleRepository;
+import io.spring.core.favorite.ArticleFavorite;
+import io.spring.core.favorite.ArticleFavoriteRepository;
 import io.spring.core.user.User;
 import io.spring.core.user.UserRepository;
 import io.spring.infrastructure.article.MyBatisArticleRepository;
+import io.spring.infrastructure.favorite.MyBatisArticleFavoriteRepository;
 import io.spring.infrastructure.user.MyBatisUserRepository;
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mybatis.spring.boot.test.autoconfigure.MybatisTest;
@@ -15,13 +19,14 @@ import org.springframework.test.context.junit4.SpringRunner;
 
 import java.util.Optional;
 
+import static org.hamcrest.CoreMatchers.anyOf;
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.CoreMatchers.notNullValue;
 import static org.junit.Assert.*;
 
 @RunWith(SpringRunner.class)
 @MybatisTest
-@Import({ArticleQueryService.class, MyBatisUserRepository.class, MyBatisArticleRepository.class})
+@Import({ArticleQueryService.class, MyBatisUserRepository.class, MyBatisArticleRepository.class, MyBatisArticleFavoriteRepository.class})
 public class ArticleQueryServiceTest {
     @Autowired
     private ArticleQueryService queryService;
@@ -32,13 +37,22 @@ public class ArticleQueryServiceTest {
     @Autowired
     private UserRepository userRepository;
 
+    @Autowired
+    private ArticleFavoriteRepository articleFavoriteRepository;
+
+    private User user;
+    private Article article;
+
+    @Before
+    public void setUp() throws Exception {
+        user = new User("aisensiy@gmail.com", "aisensiy", "123", "", "");
+        userRepository.save(user);
+        article = new Article("test", "desc", "body", new String[]{"java", "spring"}, user.getId());
+        articleRepository.save(article);
+    }
+
     @Test
     public void should_fetch_article_success() throws Exception {
-        User user = new User("aisensiy@gmail.com", "aisensiy", "123", "", "");
-        userRepository.save(user);
-
-        Article article = new Article("test", "desc", "body", new String[]{"java", "spring"}, user.getId());
-        articleRepository.save(article);
 
         Optional<ArticleData> optional = queryService.findById(article.getId(), user);
         assertThat(optional.isPresent(), is(true));
@@ -48,4 +62,15 @@ public class ArticleQueryServiceTest {
         assertThat(fetched.getCreatedAt(), notNullValue());
         assertThat(fetched.getUpdatedAt(), notNullValue());
     }
+
+    @Test
+    public void should_get_article_with_right_favorite_and_favorite_count() throws Exception {
+        User anotherUser = new User("other@test.com", "other", "123", "", "");
+        userRepository.save(anotherUser);
+        articleFavoriteRepository.save(new ArticleFavorite(article.getId(), anotherUser.getId()));
+
+        ArticleData articleData = queryService.findById(article.getId(), anotherUser).get();
+        assertThat(articleData.getFavoritesCount(), is(1));
+        assertThat(articleData.isFavorited(), is(true));
+    }
 }