【单元测试】Spring Data JPA + H2 测试DAO层

1. 环境配置

• Springboot 2.7.8
  
• h2 2.1.214
  

2. POM文件

• 引入springboot parent pom
  

点击查看代码

1. <parent>
2.   <groupId>org.springframework.boot</groupId>
3.   <artifactId>spring-boot-starter-parent</artifactId>
4.   <version>2.7.8</version>
5.   <relativePath/>
6. </parent>

复制代码

• 引入junit , springboot-test, spring-data-jpa,  H2
  

点击查看代码

1.         <dependency>
2.             <groupId>org.springframework.boot</groupId>
3.             <artifactId>spring-boot-starter-test</artifactId>
4.             <scope>test</scope>
5.             <exclusions>
6.                 <exclusion>
7.                     <groupId>org.junit.vintage</groupId>
8.                     <artifactId>junit-vintage-engine</artifactId>
9.                 </exclusion>
10.             </exclusions>
11.         </dependency>
13.         <dependency>
14.                 <groupId>org.springframework.boot</groupId>
15.                 <artifactId>spring-boot-starter-data-jpa</artifactId>
16.         </dependency>
17.        
18.         <dependency>
19.             <groupId>junit</groupId>
20.             <artifactId>junit</artifactId>
21.             <scope>test</scope>
22.         </dependency>
23.        
24.         <dependency>
25.             <groupId>com.h2database</groupId>
26.             <artifactId>h2</artifactId>
27.             <scope>test</scope>
28.         </dependency>

复制代码

3. H2配置

H2是一款内存数据库，可以配合单测测试数据层。实现原理是将依赖的三方数据库通过内存的方式进行mock。其中H2数据库的数据表结构以及数据可以通过sql脚本文件初始化，也可以在代码中增删数据。
test/resources/h2conf.properties

1. #使用h2数据库
2. spring.datasource.url=jdbc:h2:mem:test;MODE=Oracle;
3. spring.datasource.username=sa
4. spring.datasource.password=
5. spring.datasource.driver-class-name=org.h2.Driver
7. spring.sql.init.platform=h2
8. #数据初始化脚本地址
9. spring.sql.init.schema-locations=classpath:script/schema-h2.sql
10. # jpa配置
11. spring.jpa.database=h2
12. spring.jpa.show-sql=true
13. spring.jpa.hibernate.ddl-auto=none
14. spring.jpa.generate-ddl=false
15. spring.jpa.open-in-view=true
17. spring.main.banner-mode=off

复制代码

在 test/resources/script 下创建schema-h2.sql，内容为建表语句。
4. 编写单测代码

使用 @DataJpaTest 测试spring-data-jpa 框架下的数据层，@TestPropertySource 引入指定配置文件， @ContextConfiguration 引入springboot上下文。
点击查看代码

1. @DataJpaTest
2. @ContextConfiguration(classes=Application.class)
3. @TestPropertySource(locations = {"classpath:h2conf.properties"})
4. public class ServiceImplDaoTest {
6.     private SomeServiceImpl service;
7.     @Autowired
8.     private SomeRepository repository;
9.     @Autowired
10.     private TestEntityManager entityManager;
12.     @BeforeEach
13.     public  void setUp() {
14.         service = new SomeServiceImpl();
15.         ReflectionTestUtils.setField(service, "repository", repository);
17.     }
18.     @Test
19.     public void testSelect() {
20.         insertData();
21.         List<SomeEntity> all =         service.findAll();
22.         Assertions.assertEquals(10, all.size());
23.     }
25.     /**
26.      * for test sql
27.      */
28.     public void insertData() {
29.         // mock 10 data
30.         for (int i = 0; i < 10; i++) {
31.             SomeEntity entity = new SomeEntity();
32.             entityManager.persist(entity);
33.         }
34.     }
36. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！