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就可以解决这个问题。文章来源:https://www.toymoban.com/news/detail-536303.html
举例 - 模拟MongoService其中一个方法,并测试另一个方法真实输出文章来源地址https://www.toymoban.com/news/detail-536303.html
@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 - 简书
到了这里,关于[Springboot 单元测试笔记] - Mock 和 spy的使用的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!