From 8c4301f5040c7acc878192be00fba47bc893aa7a Mon Sep 17 00:00:00 2001 From: Yusun Choi Date: Wed, 27 Sep 2023 20:41:56 +0900 Subject: [PATCH] test: Modify `quantityTest()` to check DB (#3) --- .../java/com/cubix/extsearchbatch/BatchTest.java | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/dev/nusuy/ext-search-batch/src/test/java/com/cubix/extsearchbatch/BatchTest.java b/dev/nusuy/ext-search-batch/src/test/java/com/cubix/extsearchbatch/BatchTest.java index 55c9fa6..11d5789 100644 --- a/dev/nusuy/ext-search-batch/src/test/java/com/cubix/extsearchbatch/BatchTest.java +++ b/dev/nusuy/ext-search-batch/src/test/java/com/cubix/extsearchbatch/BatchTest.java @@ -1,6 +1,8 @@ package com.cubix.extsearchbatch; import com.cubix.extsearchbatch.dto.NaverRawNewsItemDto; +import com.cubix.extsearchbatch.entity.NewsRepository; +import com.cubix.extsearchbatch.service.DataUpdateService; import com.cubix.extsearchbatch.util.data.NewsDataReader; import lombok.extern.slf4j.Slf4j; import org.junit.jupiter.api.DisplayName; @@ -18,11 +20,15 @@ @DisplayName("ext-search-batch Test") @SpringBootTest public class BatchTest { + private final DataUpdateService dataUpdateService; + private final NewsRepository newsRepository; private final NewsDataReader newsDataReader; final int DISPLAY_DEF = 100; @Autowired - public BatchTest(NewsDataReader newsDataReader) { + public BatchTest(DataUpdateService dataUpdateService, NewsRepository newsRepository, NewsDataReader newsDataReader) { + this.dataUpdateService = dataUpdateService; + this.newsRepository = newsRepository; this.newsDataReader = newsDataReader; } @@ -30,13 +36,13 @@ public BatchTest(NewsDataReader newsDataReader) { @DisplayName("적재 데이터량 테스트") public void quantityTest() { // given - long expectedSize = 100; + long expectedSize = 1000; // when - ArrayList items = newsDataReader.get(DISPLAY_DEF, 1).getItems(); + dataUpdateService.updateNewsData(); // then - long size = items.size(); + long size = newsRepository.count(); assertThat(size).as("[Test Failed: 1,000 건 적재 실패] - actual: " + size).isEqualTo(expectedSize); }