当前位置: 首页 > 资讯 > 正文

单元测试与Mockito

来源:博客园    时间:2023-04-23 00:28:05

系列文章目录和关于我

零丶背景

最近在新公司第一次上手写代码,写了一个不是很难的业务逻辑代码,但是在我写单元测试的时候,发现自己对单元测试的理解的就是一坨,整个过程写得慢,还写得臭。造成这种局面我认为是因为:

对Mockito api是不是很熟悉没有自己单元测试方法论,不知道怎样写好单元测试。

now,我将从这两个部分来学习一下单元测试,如何写,如何写好单元测试?


(相关资料图)

一丶为什么需要单元测试

在上一份工作,我基本上不咋写单元测试,觉得很麻烦,不如直接postman,swagger开冲,这种显然不容易覆盖到所有的case。

单元测试的好处:

增强信心

单元测试覆盖率越高,我们越对自己的代码有信心。

揭示意图

写单元测试的时候,我们是明确自己的代码到底是出于什么目的写的

安全重构

不只是重构,哪怕后续在原有功能上进行添加,通过执行之前存在单元测试有助于我们验证,我们没有影响到原有功能。

快速反馈

写单元测试的过程,我们其实有可能发现自己代码存在的缺陷,通过单元测试直白的报错,我们可以很快得到反馈,这个反馈速度是测试滴滴你所不具备的。

定位缺陷

单元测试并不能帮我们找出所有存在的bug(测试同事:没事,我会出手),但是我们发现bug后,可以将输入放在单元测试中进行回放,直到可以重现并定位到问题,然后使用这种情况的case来补充单元测试用例。

二丶引入依赖&这些依赖的作用
   junit   junit   4.13.2   test   org.mockito   mockito-core   5.3.1   test   org.mockito   mockito-inline   3.7.7   test

junit

提供了许多方便使用的注解,标注在方法上

Mockito

Mockito 是一种 Java Mock 框架,主要就是用来做 Mock 测试的,可以模拟出一个对象、模拟方法的返回值、模拟抛出异常,模拟静态方法等等,同时也会记录调用这些模拟方法的参数、调用顺序,从而可以校验出这个 Mock 对象是否有被正确的顺序调用,以及按照期望的参数被调用。

Mock 测试:比如我们的Service依赖其他的服务提供的接口方法,使用mock可以模拟出这个接口的表现(正常返回,抛出异常等到)从而让单元测试不那么依赖外部的服务。

powermock

可以看作是mock增强版本,提供模拟私有方法等功能,我们这里没有进行引入。

三丶Mockito 常用功能0.从一个例子开始

如上图,我们的MyService依赖于OtherClient,这个OtherClient可能由于网络原因会出现错误,或者其他情况抛出异常,我们的MyService需要进行处理。

1.@InjectMocks & @Mock &MockitoAnnotations.openMocks@InjectMocks:标记应进行注射的字段,类似于spring的依赖注入,但是这里会使用Mock产生的对象@Mock :将字段标记为模拟字段,我们可以使用Mockito提供的方法来 打桩。MockitoAnnotations.openMocks:开启Mockito注解的功能2.打桩

打桩可以理解为 mock 对象规定它的行为,使其按照我们的要求来执行具体的操作。

2.1 指定入参让mock对象返回指定对象——thenReturn
//让client在query入参为1的时候,返回100为key,aaa为value的单键值对的mapMockito.when(client.query(1)).thenReturn(new HashMap<>(Collections.singletonMap(100, "aaaa")));Map res = client.query(1);Assert.assertEquals(1, res.size());Assert.assertEquals(res.get(100), "aaaa");
2.2 指定入参让mock对象抛出异常——thenThrow
Mockito.when(client.query(2)).thenThrow(new RuntimeException("222"));Assert.assertThrows("222", RuntimeException.class, () -> client.query(2));
2.3 指定任何参数都执行指定操作——Mockito.anyInt()
Mockito.when(client.query(Mockito.anyInt())).thenReturn(new HashMap<>());Assert.assertEquals(0, client.query(-1).size());
2.4 参数匹配器——ArgumentMatcher

有时候,我们希望入参入参符合要的时候,mock对象进行什么操作。

如下,我们要求mock对象在输入参数是1, 2, 3的时候返回空map

HashSet integers = new HashSet<>(Arrays.asList(1, 2, 3));Mockito.when(client.query(Mockito.argThat(new ArgumentMatcher() {    @Override    public boolean matches(Integer argument) {        return integers.contains(argument);    }}))).thenReturn(Collections.emptyMap());Assert.assertEquals(0,client.query(2).size());
2.5 控制mock对象返回结果——thenAnswer

有时候我们希望mock对象可以根据输出的不同返回不同的结果,符合我们要求的结果。

如下,我们使用thenAnswer根据入参返回不同的结果。

Mockito.when(client.query(Mockito.anyInt())).thenAnswer(new Answer() {    @Override    public Object answer(InvocationOnMock invocation) throws Throwable {        Integer argument = invocation.getArgument(0);        String str = argument%2==0?"偶数":"奇数";        return new HashMap(Collections.singletonMap(argument,str));    }});Assert.assertEquals("偶数", client.query(2).get(2));2.6 让mock对象调用真实方法——thenCallRealMethod

上面都是说mock对象如何去控制输出,thenCallRealMethod可以让mock对象执行真实的逻辑。

Mockito.when(client.query(-1)).thenCallRealMethod();
2.7 验证——verify

verify可以让我们验证当前mock对象,比如下面验证client至少执行了四次query

//验证 client.query最起码调用了4次Mockito.verify(client,Mockito.atLeast(4)).query(Mockito.anyInt());
2.8 mock静态方法——mockStatic

有时候静态方法也需要进行mock控制,可以使用

四丶一个有依赖的单元测试0.还是这个例子

如上图,我们的MyService依赖于OtherClient,这个OtherClient可能由于网络原因会出现错误,或者其他情况抛出异常,我们的MyService需要进行处理。

1.确认需要mock什么

上面这个例子中,OtherClient是外部提供给我们的接口,它存在一定的机率失败,在单元测试的过程我们需要mock它的行为,而不是真的去调用外部接口。

2.定义对象,前置准备

这里我们得明确 MyService是我们需要测试的,那就别mock它,OtherClient是外部依赖,需要进行mock控制其行为。

3.1mock方法->调用方法->验证方法3.1 模拟OtherClient抛出异常3.2 模拟OtherClient返回空Map3.3模拟OtherClient返回非空Map

X 关闭

最近更新

Copyright ©  2015-2022 南非字画网版权所有  备案号:沪ICP备2022005074号-13   联系邮箱:58 55 97 3@qq.com