package com.baeldung; import static org.junit.jupiter.api.Assertions.assertEquals; import java.util.ArrayList; import java.util.List; import org.junit.jupiter.api.Test; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import com.baeldung.model.Post; import com.baeldung.repository.PostRepository; import com.baeldung.repository.PostViewRepository; import com.baeldung.view.PostView; import com.baeldung.view.PostWithAuthorView; @SpringBootTest public class PostUnitTest { @Autowired private PostViewRepository postViewRepository; @Autowired private PostRepository postRepository; @Test public void whenFindAll_thenReturnCorrectListViewSize() { final Iterable<PostWithAuthorView> listIterable = postViewRepository.findAll(); final List<PostView> list = new ArrayList<>(); listIterable.forEach(list::add); assertEquals(7, list.size()); } @Test public void givenPostIdAndAuthorName_whenFind_thenReturnCorrectResult() { final Iterable<PostWithAuthorView> listIterable = postRepository.findBy("Spring", "Peter"); final List<PostView> list = new ArrayList<>(); listIterable.forEach(list::add); assertEquals(4, list.size()); } @Test public void whenFindAll_thenReturnCorrectListSize() { final Iterable<Post> listIterable = postRepository.findAll(); final List<Post> list = new ArrayList<>(); listIterable.forEach(list::add); assertEquals(7, list.size()); } }