有赞测试 有赞单元测试实践

有赞测试团队 · July 05, 2019 · Last by xiaomingpapapa replied at July 06, 2019 · 1612 hits

作者:苏国庆


一、概述


单元测试是指对软件中的最小可测试单元进行检查和验证。单元在质量保证中是非常重要的环节,根据测试金字塔原理,越往上层的测试,所需的测试投入比例越大,效果也越差,而单元测试的成本要小的多,也更容易发现问题。

1.1 有赞单元测试1.0架构


以有赞中台某应用为例,应用部署是微服务架构,对外提供dubbo服务,当前的单元测试,采用了分层测试框架,根据代码的分层,分为Service 层测试,Biz层测试,外部服务访问层测试,DAO 测试,Redis 访问层测试,每一层均使用mock 框架屏蔽下层的具体实现。

1.2 单元测试的过程


单元测试的编写,主要包含以下几个阶段:

  1. 数据准备:在编写测试用例前,需要依赖到一些数据,数据来源一般是数据库,而构造数据,又不能依赖DAO层的代码,需要使用原生jdbc 去插入数据,测试代码编写效率低。
  2. 构造参数及打桩(stub):调用方法需要传递入参,有时候一个入参十几个参数需要set,set 方法写完,代码已经写了十来行了。
  3. 执行测试:这一步比较简单,直接调用被测方法即可。
  4. 结果验证:这里除了验证被测方法的返回值外,还需要验证插入到数据库中的数据是否正确,某外部方法被调用过n次或未调用过。
  5. 必要的清理:对打桩进行清理,对数据库脏数据进行清理。 # 二、 痛点 ## 1. 重构代码需要改写大量单元测试用例 对外的Service接口在不变的情况下,对内部实现进行重构,这时候头痛的问题来了,大量的Service 层单元测试,biz层单元测试都要重写;有时候Service调用biz层接口时,参数传错了,而由于开发人员编写单元测试时不规范,参数匹配使用了 anyxxx(),导致参数传错的bug未被发现。 ## 2. 测试库数据随意修改导致的单元测试不稳定 DAO层单元测试直连测试库,由于测试库的数据可以被任意修改,从而导致测试依赖的数据被更改,单元测试不通过,另外开发在编写单元测试时,没有清理意识,导致测试库大量垃圾数据。 ## 3. 单元测试结果校验缺失 例如一个SaveItem()接口,执行完成后除了要验证执行成功以外,还应该验证落库数据的正确性,而编写这部分测试代码需要大量的使用原生jdbc 接口查询sql,并逐字段验证正确性,代码编写效率低下。 # 三、几个常用的测试框架的简介 ## 1. 数据层单元测试框架 DbUnit 可以优雅的构造DB层的初始化数据,例如:
<?xml version='1.0' encoding='UTF-8'?>  
<dataset>
<employee employee_uid='1'
start_date='2001-11-01'
first_name='Andrew'
ssn='xxx-xx-xxxx'
last_name='Glover' />
</dataset>
12345678

其中employee 是要构造数据的表名,后面的键值对是列名及对应的值,需要注意的是,第一行必须包含完整的字段名,否则加载的数据中全部会缺失某些字段。

2. 嵌入式的内存数据库H2

非常适合在测试程序中使用,程序关闭时自动清理数据,H2 数据库的表结构初始化是通过 jdbc:initialize-database 标签实现的,单元测试中使用h2数据库非常简单,仅需修改jdbc连接即可。 引入依赖:

<dependency>  
<groupId>com.h2database</groupId>
<artifactId>h2</artifactId>
<version>1.4.191</version>
<scope>test</scope>
</dependency>
123456

数据源连接:

spring.datasource.url=jdbc:h2:mem:test  
spring.datasource.driver-class-name=org.hsqldb.jdbcDriver
spring.datasource.username=root
spring.datasource.password=
1234

schema 初始化:

<jdbc:initialize-database data-source="dataSource" ignore-failures="NONE">  
<jdbc:script location="classpath:h2/schema.sql" encoding="UTF-8"/>
</jdbc:initialize-database>
123

3. Spring小扩展springockito

它简化了在集成测试的相关上下文XML文件中创建mockito mocks的方法。

<beans xmlns="http://www.springframework.org/schema/beans"  
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xmlns:mockito="http://www.mockito.org/spring/mockito"
xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-3.0.xsd
http://www.mockito.org/spring/mockito http://www.mockito.org/spring/mockito.xsd"
>
...
<mockito:mock id="accountService" class="org.kubek2k.account.DefaultAccountService" />
..
</beans>
123456789

4. spring 官方测试框架 spring-test

目前主流的开发框架都在使用spring 框架管理bean,在测试代码中,我们通用期望能够使用spring 框架,spring-test 框架帮助我们解决bean 的注入问题。

@ContextConfiguration(locations = "/test-context.xml", 
loader = SpringockitoContextLoader.class)
public class CustomLoaderXmlApplicationContextTests {
// class body...
}
12345

"/test-context.xml" 指定了测试类运行需要加载的spring 配置文件路径,SpringockitoContextLoader 指定了加载配置的类,这两个一起用可以支持在使用spring xml 配置的同时可以将mockito生成的mock 对象bean 注入spring 上下文中。

5.支持静态方法mock的mock框架 powermock

支持静态方法mock,同时兼容mockito,powermock示例:

@RunWith(PowerMockRunner.class)
@PrepareForTest( { YourClassWithEgStaticMethod.class })
public class YourTestCase {
...
}
12345

四、有赞基于springtest+ut+powermock 的测试框架


有赞单元测试框架,数据库层使用h2数据库代替测试库,隔离单元测试数据与测试库数据,在单元测试结束后自动清理数据,避免污染测试库数据及被测试库数据影响,基于DbUnit 可以通过xml构造DB层初始化数据,实现测试代码与测试数据分离,依赖spring jdbc的初始化脚本初始化h2数据库的表结构。

1. 数据准备

单测依赖的Db数据,通过添加测试方法监听器,在Junit 执行前通过DbUnit 工具类,加载初始化文件,写入H2数据库;单测的入参,通过param.json 文件,以json格式编写入参数据,利用工具类读取文件并json 反序列化为目标Class 实例。
H2数据库的表结构,则是通过上文提到的 jdbc:initialize-database 初始化的,开发同学必须保证此schema 与线上结构的一致性,否则会导致单测失败。
添加方法监听器
@TestExecutionListeners({ JunitMethodListener.class})
这是自定义的监听器,在执行前后执行自定义逻辑,包括数据准备、验证和清理。

public class JunitMethodListener extends AbstractTestExecutionListener {
@Override
public void beforeTestMethod(TestContext testContext) throws Exception {
Method jdkMethod = testContext.getTestMethod();
if (jdkMethod == null) {
return;
}
Object classInstance = testContext.getTestInstance();
if (!(classInstance instanceof JunitRunner)) {
return;
}
TestMethod testMethod = jdkMethod.getAnnotation(TestMethod.class);
if (testMethod == null) {
return;
}
JunitRunner runner = (JunitRunner) classInstance;
runner.init();
if (testMethod.enablePrepare()) {
TestRunnerTool.prepare(testMethod, runner);
}
}
@Override
public void afterTestMethod(TestContext testContext) throws Exception {
boolean hasException = (testContext.getTestException() != null) ? true : false;
Method jdkMethod = testContext.getTestMethod();
if (jdkMethod == null) {
return;
}
Object classInstance = testContext.getTestInstance();
if (!(classInstance instanceof JunitRunner)) {
return;
}
TestMethod testMethod = jdkMethod.getAnnotation(TestMethod.class);
if (testMethod == null) {
return;
}
JunitRunner runner = (JunitRunner) classInstance;
if (!hasException && testMethod.enableCheck()) {
TestRunnerTool.check(testMethod, runner);
}
if (testMethod.enablePrepare()) {
//清理数据
TestRunnerTool.clean(testMethod, runner);
}
}
}
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657

以下是单元测试代码示例,enablePrepare 声明需要准备数据,prepareDateConfig 声明数据准备的文件路径,prepareDateType 是数据准备的类型,xml -> DB,当然也支持更多的文件类型,如csv,xls。

@TestMethod(
enablePrepare = true,
prepareDateType = PrepareDataType.XML2DB,
prepareDateConfig = {PREPARE_XML_FILE_USER}
)
@Test
public void test_updateUser(){
... 具体代码省略
}
123456789

2.桩代码相关框架

为了使被测代码能够独立运行、并控制被测代码的执行路径,我们需要对外部依赖(包括中间件、静态函数、外部服务)进行mock,mock框架依赖的是PowerMockmockito,利用spring-test集成springockito将mock 的bean注入到Spring上下文中。
使用PowerMock 运行Junit单元测试

@RunWith(PowerMockRunner.class)
@PowerMockIgnore({ "javax.management.*", "javax.net.ssl.*"})
12

PowerMock集成Spring TestContext框架

@PowerMockRunnerDelegate(SpringJUnit4ClassRunner.class)
@ContextConfiguration(
loader = SpringockitoContextLoader.class,
locations = {
"classpath:applicationContext-test.xml"})
12345

3. 结果验证

结果验证,包括两部分,一个是被测函数的返回值,这个需要编写者自行验证,另一个是写入数据库的值,这部分是通过在方法上添加注解,告诉单元测试框架要验证的语句,执行验证语句并与期望值比较。
单元测试方法示例:

@TestMethod(
enablePrepare = true,
prepareDateType = PrepareDataType.XML2DB,
prepareDateConfig = { PREPARE_XML_FILE_USER},
enableCheck = true,
checkConfigFiles = {"/saveUserCheck.json"}
)
@Test
public void test_updateUser() throws IOException {
UserParam param = MockUtil.fromFile(
"/param.json",
UserParam.class);
...
}
123456789101112131415

saveUserCheck.json 文件内容示例

{
"check.type": "DB_CHECK",
"check.desc": "检查 更新结果正确性",
"check.sql.query": "select status from user where user_id=1",
"check.expected.data": [
{
"status": 1
}
]
}
12345678910

4.以下是单元测试基类的示例代码


五、总结

第二部分提到的几个痛点,通过我们的zantest测试组件,我们完美的解决这几个问题,通过注解方式,实现了配置数据与测试代码的分离,简化测试代码编写,隔离测试环境数据库,并编写了一套测试示例进行推广。

1. 关于内部重构的痛点,我们基于有赞单元测试框架解决了这个问题。


在单元测试1.0版本时,我们分别对Service,innerBeanA,innerBeanB, UserDAO 写单元测试,当Service 层输入输出不变,内部重构时,这几个类的单元测试都要重构,而在单元测试2.0版本时,由于被测函数只有Service,通过桩代码控制Service 对innerBeanA,innerBeanB, UserDAO 的调用,从而覆盖inner层和DAO层,重构时只需要改写Service 层代码即可。

2. 测试库数据被随意改

数据准备不再依赖测试库,而是通过文件构造测试数据,例如上文的xml格式,为方便测试数据的构造,同时也支持更多的数据格式,例如csv,可以方便的将线上数据导出作为测试用例。

3. 单元测试结果校验

一方面开发仍然需要自行校验函数的返回值,校验mock函数是否被执行,另一方面对数据库数据更改的验证可以直接通过注解声明校验的sql 文件路径即可。
欢迎关注我们的公众号

共收到 2 条回复 时间 点赞

有赞真的很赞啊。
不过有一点比较疑惑的就是——我们公司团队开始也想通过 h2 作为临时的内存数据库进行单元测试,但由于我们使用的是 mysql 数据库,所以会导致一些 sql 语法在 h2 中是不支持的,请问你们团队有碰到类似的问题吗

需要 Sign In 后方可回复, 如果你还没有账号请点击这里 Sign Up