场景

公司采用的是分层开发,controller、Service、dao层分离,现在写dao层代码的人生病了,进度比较慢,现在你写的是 Service层的代码,怎么测试 Service 层代码是否正确呢?

Service层测试的重点是什么?

  1. DAO层调用的次数
  2. 以及调用的顺序
  3. 并不关心最后数据是否准确

测试代码示例

0. 准备环境

引入依赖

<!-- Junit测试包 -->
<dependency>
    <groupId>junit</groupId>
    <artifactId>junit</artifactId>
    <version>4.13</version>
</dependency>
<!-- 专门用来做业务逻辑层 或者 Controller层的测试的 -->
<dependency>
    <groupId>org.easymock</groupId>
    <artifactId>easymock</artifactId>
    <version>3.5.1</version>
</dependency>

准备Dao层测试接口,不需要方法的实现

public interface AaDao {
    void a();
    void b();
}
public interface BbDao {
    void c();
    void d();
}

service 测试代码

public interface ABService {
    void aa();
    void bb();
    void cc();
    void dd();
}

testService 在此方法中使用 mock 进行测试

public class TestService {
    private ABService abService = new ABServiceImpl();
	
}

1. 简单测试 createMock

@Override
public void aa() {
    aDao.a();
}

测试代码

@Test
public void testAA() {
    // 创建对象
    AaDao aaDao = EasyMock.createStrictMock(AaDao.class);
    // 记录
    aaDao.a();
    EasyMock.expectLastCall();
    // 使能设置
    EasyMock.replay(aaDao);
    // 接下来调用Service进行测试
    abService.setaDao(aaDao);
    abService.aa();
    // 进行校验
    EasyMock.verify(aaDao);
}

2. 测试调用顺序 createStrictMock

@Override
public void ee() {
    aDao.b();
    aDao.a();
}

测试代码1

// 测试调用顺序
@Test
public void testEE1() {
    // 创建对象
    AaDao aaDao = EasyMock.createMock(AaDao.class);
    // 进行记录
    aaDao.a();
    EasyMock.expectLastCall();
    aaDao.b();
    EasyMock.expectLastCall();
    // 使能设置,保存记录信息
    EasyMock.replay(aaDao);
    // 调用 Service 测试
    abService.setaDao(aaDao);
    abService.ee();
    // 进行校验
    EasyMock.verify(aaDao);
}

运行此代码,结果竟然是正确的,明明调用顺序不一致,却还是正确,这显然是不对的

测试代码2 使用createStrictMock

@Test
public void testEE2() {
    // 创建对象
    AaDao aaDao = EasyMock.createStrictMock(AaDao.class);
    // 进行记录
    aaDao.b();
    EasyMock.expectLastCall();
    aaDao.a();
    EasyMock.expectLastCall();
    // 使能设置,保存记录信息
    EasyMock.replay(aaDao);
    // 调用 Service 测试
    abService.setaDao(aaDao);
    abService.ee();
    // 进行校验
    EasyMock.verify(aaDao);
}

3. 测试方法多次调用以及调用顺序

@Override
public void bb() {
    aDao.a();
    aDao.b();
    aDao.b();
    aDao.a();
}

测试代码

@Test
public void testBB() {
    // 创建对象
    AaDao aaDao = EasyMock.createStrictMock(AaDao.class);
    // 记录
    aaDao.a();
    EasyMock.expectLastCall();
    aaDao.b();
    EasyMock.expectLastCall();
    aaDao.b();
    EasyMock.expectLastCall();
    aaDao.a();
    EasyMock.expectLastCall();
    // 使能设置,即保存记录信息
    EasyMock.replay(aaDao);
    // 调用 Service 测试
    abService.setaDao(aaDao);
    abService.bb();
    // 进行校验
    EasyMock.verify(aaDao);
}

如果是多次调用的话,就需要书写多次记录

4. 测试调用不同dao的方法

@Override
public void cc() {
    aDao.a();
    bDao.c();
}

测试代码1

@Test
public void testCC1() {
    // 创建对象
    AaDao aaDao = EasyMock.createStrictMock(AaDao.class);
    BbDao bbDao = EasyMock.createStrictMock(BbDao.class);
    // 进行记录
    bbDao.c();
    EasyMock.expectLastCall();
    aaDao.a();
    EasyMock.expectLastCall();
    // 使能设置,保存记录信息
    EasyMock.replay(aaDao, bbDao);
    // 调用 Service 测试
    abService.setaDao(aaDao);
    abService.setbDao(bbDao);
    abService.cc();
    // 进行校验
    EasyMock.verify(aaDao, bbDao);
}

可以看出来,createStrictMock方法并不能保证多个 dao之间的调用顺序,只能对单个的DAO的调用顺序有严格的要求

测试代码2 createStrictControl

@Test
public void testCC2() {
    // 创建一个 controller,约束这些dao是一个整体
    IMocksControl control = EasyMock.createStrictControl();
    // 创建对象
    AaDao aaDao = control.createMock(AaDao.class);
    BbDao bbDao = control.createMock(BbDao.class);
    // 进行记录
    aaDao.a();
    EasyMock.expectLastCall();
    bbDao.c();
    EasyMock.expectLastCall();
    // 使能设置,保存记录信息
    control.replay();
    // 调用 Service 测试
    abService.setaDao(aaDao);
    abService.setbDao(bbDao);
    abService.cc();
    // 进行校验
    control.verify();
}

使用 createStrictControl 定义一个整体,进行多个 dao 的测试

5. 测试不同dao调用顺序及次数

@Override
public void dd() {
    aDao.a();
    bDao.c();
    bDao.d();
    aDao.a();
}

测试代码

@Test
public void testDD() {
    // 创建一个 controller,约束这些dao是一个整体
    IMocksControl control = EasyMock.createStrictControl();
    // 创建对象
    AaDao aaDao = control.createMock(AaDao.class);
    BbDao bbDao = control.createMock(BbDao.class);
    // 进行记录
    aaDao.a();
    EasyMock.expectLastCall();
    bbDao.c();
    EasyMock.expectLastCall();
    bbDao.d();
    EasyMock.expectLastCall();
    aaDao.a();
    EasyMock.expectLastCall();
    // 使能设置,保存记录信息
    control.replay();
    // 调用 Service 测试
    abService.setaDao(aaDao);
    abService.setbDao(bbDao);
    abService.dd();
    // 进行校验
    control.verify();
}

测试此方法时,即便是调换 同一个dao的方法也会报错,满足条件

总结

  1. 测试 service 层时,可以使用 mock 框架进行测试,引入 mock依赖即可,操作简单
  2. 如果测试的 service 只包含一个dao,那么使用 EasyMock.createStrictMock() 即可进行测试
  3. 如果测试的 service 包含多个dao,那么使用 EasyMock.createStrictControl(); 创建一个 control 进行包裹即可测试