Mockito的简单使用可以参考上一篇Mockito的简单使用
一、Mock测试框架常用注解介绍
1、spy和mock生成的对象不受spring管理
2、spy调用真实方法时，其它bean是无法注入的，要使用注入，要使用SpyBean
3、SpyBean和MockBean生成的对象受spring管理，相当于自动替换对应类型bean的注入，比如@Autowired等注入。
二、Mock测试框架常用使用方式
2.1 Mock初始化
为了减少单测与spring框架的耦合，并且尽量不使用真实对象进行模拟（官方文档建议，链接见文章底部），建议使用@InjectMocks创建实例，对于测试类中其他需要注入的依赖使用@Mock。在测试之前需要对Mock初始化，之后需要关闭Mock。如果有一些公共的静态类需要在多个测试用例中使用，推荐使用这种方式，代码如下。
private AutoCloseable closeable;
@Before
public void openMocks() {
closeable = MockitoAnnotations.openMocks(this);
}
@After
public void releaseMocks() throws Exception {
closeable.close();
}如果没有公共的方法，也可以使用注解的方式直接初始化Mock
@RunWith(MockitoJUnitRunner.class)
public class MockTest {
@Test
public void test1() {
}
}2.2 简单流程单测
对于没有分支逻辑的功能，可以对结果断言，测试代码逻辑是否正确。并且需要注意，每个单测都需要进行断言，来验证代码逻辑的正确性。
public DemoTest{
@Mock
private Demo demoMock;
@Test
public void test1() {
int i = 10;
when(demoMock.Func(i)).thenReturn(false);
boolean result = demoMock.Func(i);
Assert.assertFalse(result);
}
}
class Demo {
public boolean Func(int i) {
return i > 0;
}
}2.3 分支流程单测
当逻辑中有分支逻辑时，可以通过多个测试方法构造不同的对象，测试不同分支的逻辑正确与否
public DemoTest{
@Mock
private Demo demoMock;
@Test
public void test1() {
int i = 10;
when(demoMock.Func(i)).thenReturn(false);
boolean result = demoMock.Func(i);
Assert.assertFalse(result);
}
@Test
public void test2() {
int i = 0;
when(demoMock.Func(i)).thenReturn(true);
boolean result = demoMock.Func(i);
Assert.assertTrue(result);
}
}
class Demo {
public boolean Func(int i) {
if (i > 0) {
return true;
} else {
return false;
}
}
}2.4 验证方法被执行过的次数
// 验证方法被执行的次数
Mockito.verify(className,times(1)).doSomething();
Mockito.verify(className,times(2)).doSomething();
// 验证方法是否没被执行过
Mockito.verify(className,never()).doSomething();注意：Mockito.verify()不可以验证被@InjectMocks修饰的mock对象
2.5 为调用的方法抛出异常
doThrow(new Exception()).when(className).doSomething();2.6 静态方法的mock
try (MockedStatic<Lion> lionMock = mockStatic(Lion.class)) {
lionMock.when(() -> Lion.getBoolean(anyString(), anyString(), anyBoolean())).thenReturn(true);
}注意：对静态方法的打桩，应该使用try(){}的结构包裹起来，避免不同测试方法中的静态数据相互影响
也可以将Mock的静态方法抽成函数，传入功能接口函数，供测试类调用
public abstract class BaseTest {
private AutoCloseable closeable;
public interface VoidFunc {
void execute();
}
public static void getUserInfo(VoidFunc func){
User user = new User();
user.setId(1);
try (MockedStatic<UserUtils> userUtilsMockedStatic = mockStatic(UserUtils.class)) {
userUtilsMockedStatic.when(UserUtils::getUser).thenReturn(user);
func.execute();
}
}
@Before
public void openMocks() {
closeable = MockitoAnnotations.openMocks(this);
}
@After
public void releaseMocks() throws Exception {
closeable.close();
}
}
public class ControllerTest extends BaseTest {
@Test
public void test() {
BaseTest.getUserInfo(()->{
...
});
}
}当一个静态方法依赖与另一个静态方法时，可以使用嵌套调用的结构
public abstract class BaseTest {
private AutoCloseable closeable;
public interface VoidFunc {
void execute();
}
// 线程池（ExecutorService）内部依赖于限流器（OneLimiter），所以需要Mock线程池时，需要提前Mock限流器
public static void getOneLimiter(VoidFunc func) {
OneLimiter limiter = mock(DefaultOneLimiter.class);
try (MockedStatic<Rhino> rhinoStatic = mockStatic(Rhino.class)) {
rhinoStatic.when(Rhino::newOneLimiter).thenReturn(limiter);
}
}
public static void getExecutorService(VoidFunc func) {
ExecutorService mockedThreadPool = mock(ExecutorService.class);
try (MockedStatic<ExecutorService> executorServiceMockedStatic = mockStatic(ExecutorService.class)) {
executorServiceMockedStatic.when(() -> ExecutorServices.forThreadPoolExecutor(Mockito.anyString()))
.thenReturn(mockedThreadPool);
func.execute();
}
}
@Before
public void openMocks() {
closeable = MockitoAnnotations.openMocks(this);
}
@After
public void releaseMocks() throws Exception {
closeable.close();
}
}
public class ControllerTest extends BaseTest {
@Test
public void test() {
getOneLimiter(() -> getExecutorService(() -> {
...
}));
}
}
2.7 其他情况的Mock
对于void类型的方法，可以忽略掉方法的执行逻辑
doNothing().when(className).doSomething();参考链接：
官方说明文档：https://github.com/hehonghui/mockito-doc-zh#12
SpringBoot - 单元测试利器Mockito入门_小小工匠的技术博客_51CTO博客文章来源：https://www.toymoban.com/news/detail-445815.html
Springboot单元测试:SpyBean vs MockBean - 掘金文章来源地址https://www.toymoban.com/news/detail-445815.html
到了这里，关于Mockito的简单使用（二）的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
