教你如何測試Spring Data JPA的Repository

更新時間：2024年08月10日 14:42:13 作者：java奮斗者

Spring Data JPA 提供了一些便捷的方式來測試這種持久層的代碼,常見的兩種測試類型是集成測試和單元測試,本文通過示例代碼給大家介紹的非常詳細,感興趣的朋友跟隨小編一起看看吧

集成測試

集成測試通常意味著測試功能在真實環(huán)境中的交互情況。針對 Repository，集成測試通常需要數(shù)據(jù)庫的參與。Spring Data JPA 提供了 @DataJpaTest 注解來幫助進行集成測試。當你使用這個注解時，Spring Boot 會配置一個內(nèi)存數(shù)據(jù)庫（比如 H2）、Hibernate、Spring Data 和 DataSource，并且默認會回滾事務。

下面是一個使用 @DataJpaTest 進行集成測試的例子：

import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.TestPropertySource;
import static org.assertj.core.api.Assertions.*;
@DataJpaTest
@TestPropertySource(locations = "classpath:application-test.properties")
public class MyRepositoryTest {
    @Autowired
    private MyRepository myRepository;
    @Test
    public void testFindByLastName() {
        Person person = new Person("John", "Doe");
        entityManager.persist(person);
        List<Person> persons = myRepository.findByLastName("Doe");
        assertThat(persons).hasSize(1);
        assertThat(persons.get(0).getFirstName()).isEqualTo("John");
    }
}

在這個例子中，@DataJpaTest 提供了所需的所有配置，包含了 entityManager，以實現(xiàn)實體的持久化。此外，你還可以使用 TestPropertySource 為測試指定屬性文件，例如使用不同于生產(chǎn)的數(shù)據(jù)庫配置。

單元測試

單元測試則是更加隔離的測試，它不涉及數(shù)據(jù)庫的實際交互。在針對 Repository 的單元測試中，通常會使用 Mocking 框架（如 Mockito）來模擬 Repository 接口的行為。

下面是一個使用 Mockito 進行單元測試的示例：

import static org.mockito.Mockito.*;
import static org.junit.jupiter.api.Assertions.*;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
public class MyServiceTest {
    @Mock
    private MyRepository myRepository;
    private MyService myService;
    @BeforeEach
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        myService = new MyService(myRepository);
    }
    @Test
    public void testSomething() {
        when(myRepository.findByName("John")).thenReturn(Optional.of(new Person("John", "Doe")));
        Person person = myService.findPersonByName("John");
        assertNotNull(person);
        assertEquals("Doe", person.getLastName());
    }
}

在這個示例中，我們使用 Mockito 框架來模擬 MyRepository 的行為，即當調(diào)用 findByName 時返回一個特定的 Person 對象。這樣，我們可以測試 MyService 對這種情況的處理，而不需要配置數(shù)據(jù)庫或啟動 Spring 應用上下文。