Springboot单元测试mysql_springboot + jpa + h2 单元测试实战
最近需要开发一个组件，里面会有数据库操作，但由于该组件是单独开发没有实际的数据库，数据库后期会创建到集成该组件的业务系统里面。所以想到用内存数据来做测试，也避免了拿真实数据库作测试时留下垃圾数据，而且速度还比较快。
经历了一顿操作，终于搭好了单元测试环境。
¶POM文件
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
org.projectlombok
lombok
org.springframework.boot
spring-boot-starter-data-jpa
org.springframework.boot
spring-boot-starter-test
test
com.h2database
h2
test
org.springframework.boot
spring-boot-dependencies
2.1.2.RELEASE
pom
import
¶配置文件application.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25spring:
h2:
console:
enabled: true
path: /console
settings:
web-allow-others: true
trace: false
datasource:
name: dataSource
platform: h2
driver-class-name: org.h2.Driver
url: jdbc:h2:mem:test;MODE=PostgreSQL;DB_CLOSE_DELAY=-1;DATABASE_TO_UPPER=false;
username: sa
password: 123456
schema:
- classpath:sql/create/01.CS_SCHEMA.sql
- classpath:sql/create/02.CT_TABLE.sql
data:
- classpath:sql/data/02.I_T_TEST.sql
jpa:
show-sql: true
hibernate:
ddl-auto: none
database-platform: org.hibernate.dialect.H2Dialect
¶创建DO层
Entity
1
2
3
4
5
6
7
8
9
10
11
12
13@Entity
@Table(name = "t_test", schema = "test")
@Data
public class TestDO implements Serializable{
private static final long serialVersionUID = 1L;
@Id
@Column(name = "id")
private String id;
// columns定义
// ···
}
Repository
1
2
3
4@Repository
public interface TestRepository extends JpaRepository{
}
JPA的使用和配置参考淋哥的：Spring data jpa使用总结
¶单元测试
1
2
3
4
5
6
7
8
9
10
11
12@RunWith(SpringRunner.class)
@SpringBootTest(classes =TestApplication.class)
public class TestRepositoryTest{
@Resource
private TestRepository testRepository;
@Test
public void test_table_should_have_one_record(){
Assert.assertEquals(1, testRepository.count());
}
}
TimeLimitTestApplication代码如下：
1
2
3
4
5@SpringBootApplication
@EnableJpaRepositories(basePackages = "com.teddy.repository")
@EntityScan(basePackages = "com.teddy.entity")
public class TestApplication{
}
¶遇到的问题
table not found
h2数据库初始化的时候，已经执行了初始化脚本，但是执行单元测试时找不到表，经过排查，发现创建数据库脚本如下：
1
2
3
4
5
6
7set search_path to test;
commit;
drop table if exists t_test;
create table t_test(
···
);
该脚本在真实数据库中执行没有任何问题，但是在h2数据库中，并不能将表正确的创建到test模式中，而是创建到PUBLIC模式中了，导致找不到表。
修改脚本如下：
1
2
3
4
5
6
7set search_path to test;
commit;
drop table if exists test.t_test;
create table test.t_test(
···
);
即可解决该问题。
schema not found
由于JPA不区分大小写，注解里面配置的都会转成小写(可以通过配置：
spring.jpa.hibernate.naming.physical-strategy=org.hibernate.boot.model.naming.PhysicalNamingStrategyStandardImpl即可解决表名，但是schema都是小写)。而H2数据库是区分大小写的，因此可能导致schema not found。
只要统一大小写即可处理。