[Springboot 单元测试笔记] - Mock 和 spy的使用

Springboot单元测试 - 依赖类mock测试

通常单元测试中，我们会隔离依赖对于测试类的影响，也就是假设所有依赖的一定会输出理想结果，在测试中可以通过Mock方法来确保输出结果，这也就引入另一个测试框架Mockito。

Mockito框架的作用就是模拟接口功能，并不运行模拟接口的实际逻辑，而是直接输出一个假定结果。

Mockito常见注释及方法

@Mock	被注释的对象会作为Mock对象
@InjectMocks	被注释的对象依赖于@Mock的对象，通常是测试对象
when(<MockObject>.<method>).thenReturn()	对@Mock的对象进行模拟输出
doThrow(<Exception>).when(<mockObject>).<Method>	对Mock的对象模拟抛出一个异常
spy()	也是建立一个模拟对象，但可以

待测试类 - AController

public class AController {

    @Autowired
    MongoService mongoService;

    public boolean testBean() {
        boolean res = mongoService.testMethod2(4);
        return res;
    }

}

依赖类 - MongoService

@Service
public class MongoService {

    public void output() {
        System.out.println("This is a bean!!");
    }

    public boolean testMethod1(int num) {
        return num > 0;
    }

    public boolean testMethod2(int num) {
        int val = getVal();
        return val > num;
    }

    private int getVal() {
        return 2;
    }
}

Mockito测试类 - RunWith需要运行MockitoJUnitRunner.class才能使用Mockito相关注解

方法1: 使用@Mock，更为简洁只需加入注解就可以mock对应成员变量 (该成员变量可以是非依赖注入)

@RunWith(MockitoJUnitRunner.class)
public class MockTest {

    @Mock
    MongoService mongoService;

    @InjectMocks
    AController aController;

    @Test
    public void testMock1() {
        when(mongoService.testMethod2(Mockito.anyInt())).thenReturn(true);
        Assert.assertEquals(aController.testBean(), true);
    }

    @Test(expected = Exception.class)
    public void testMock2() {
        doThrow(Exception.class).when(mongoService).testMethod2(Mockito.anyInt());
        aController.testBean();
    }
}

方法2: 使用Mock函数

public class MockTest {

    MongoService mongoService;

    AController aController;

    @BeforeEach
    public void init() {
        mongoService = Mock(MongoService.class);
        aController = new AController();
        /* 假设 AController class 中有setter用于加载mongoService */
        aController.setMongoService(mongoService);
        
    }

    @Test
    public void testMock1() {
        when(mongoService.testMethod2(Mockito.anyInt())).thenReturn(true);
        Assert.assertEquals(aController.testBean(), true);
    }

    @Test(expected = Exception.class)
    public void testMock2() {
        doThrow(Exception.class).when(mongoService).testMethod2(Mockito.anyInt());
        aController.testBean();
    }
}

 

@Mock 和 @Spy区别

上面两种注释都是建立一个模拟对象，但是区别是@Mock注释的对象是完全虚拟，不能调用这个Mock对象的任何方法，只能通过when().thenReturn() 来进行模拟输出；但是@Spy是建立一个真实的Mock对象，可以调用对象中任何方法也可以模拟某些方法的输出。@Spy大大增加了测试的自由度，比如如果某个测试类你只想模拟某个方法但是想要某些方法的真实输出，这是@Spy就可以解决这个问题。

举例 - 模拟MongoService其中一个方法，并测试另一个方法真实输出

    @Spy
    MongoService mongoServiceSpy;

    @Test
    public void testSpy() {
        /** 模拟MongoService中 getVal并测试testMethod2*/
        when(mongoServiceSpy.getVal()).thenReturn(3);
        Assert.assertEquals(mongoServiceSpy.testMethod2(2), true);
    }

Reference

• SpringBoot Test及注解详解 - codedot - 博客园
• 单元测试之Mockito与PowerMock - 简书