java-tutorials/persistence-modules/blaze-persistence/src/test/java/com/baeldung/PostUnitTest.java

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());
    }
}