使用H2数据库来模拟进行单元测试
背景说明
环境说明:Java、Eclipse、Maven、SpringMVC、MyBatis、MySQL、H2。
在写DAO层的单元测试时,我们往往会遇到一个问题,测试用例所依赖的数据库数据被修改或删除了,或者在一个新的环境下所依赖的数据库不存在,导致单元测试无法通过,进而构建失败。
使用步骤
(注意下面的步骤基于前文提到的环境说明)
1) 在pom.xml中添加h2database的依赖
com.h2database
h2 artifactId>
1.4.192 version>
2) 修改jdbc.properties的数据库驱动和url
#h2
jdbc.driverClassName = org.h2.Driver
jdbc.url= jdbc:h2:mem:testdb;MODE=MYSQL;DB_CLOSE_DELAY=-1
jdbc.username =root
jdbc.password =123456
对于jdbc.properties文件,这里有一个技巧。首先项目正式运行的配置文件是放在src/main/resources目录的conf目录下。因为单元测试的jdbc配置和正式运行环境的配置不一致,我们只需要在单元测试的包src/test/java下配置一份相同目录的conf/jdbc.properties。这样在运行单元测试时,最近的配置会覆盖掉原来的配置。注意,这里是整个文件覆盖,而不是文件中的属性覆盖。如果你在test包下的jdbc.properties少配置了什么内容,并不会去resources目录下读取,会引起报错。
3) 配置数据库初始化SQL
在test包的conf下面新建sql文件夹,用于存放初始化数据库的SQL,也就是单元测试需要依赖的表结构及数据。一般我们可以将数据库初始化分为表结构初始化schema.sql和数据初始化data.sql两部分。但这并不是强制要求,你可以根据你的业务逻辑将SQL语句的存放进行划分,便于管理。
4) 编写一个BaseDaoTest基类
该类用于初始化H2数据库。其他单元测试类需要继承自该基类。
package com.szyciov.dao;
import java.sql.Connection;
import java.sql.Statement;
import org.apache.commons.dbcp.BasicDataSource;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.te