Java用正确的姿势写单元测试以及mock

心已赠人 2024-03-16 16:48 16阅读 0赞

## 1. 前言 ##

对于一些简单的功能或业务，我们也许可以通过前端调试、postman等接口工具、main函数调用进行测试。但这每次改动代码都要人力测试，耗费大量的人力资源且不高效，真正的项目中单元测试是必不可少的。

## 2. 要点 ##

单元测试的三步走：

1、组装方法入参

2、执行方法

3、对方法的执行结果进行断言(Assert)比对

建议把所有实际操作数据的测试方法上面加上事务注解。

对于会抛异常的情况，需要用try包住，在catch中断言异常文本，以及在try下面加上Assert.fail()确保一定会抛异常。

## 3. 入门级知识点 ##

### 3.1 常用注解 ###

@SpringBootTest：把当前类标记为测试类并交给spring管理，通常和@RunWith(SpringRunner.class)一起出现。另如果需要初始化环境变量，建议先定义一个父测试类，加上上面两个注解，写一个静态内部类System.setProperty("key","value")。其他测试类继承即可。

@Test：用在方法上，标记为测试方法。注意必须是public。

@Before：所有单元测试前都会执行一遍。注意：启动单元测试有两种方式，启动方法和启动类，启动类会自动执行测试类下的所有测试方法

@After：所有单元测试后都会执行一遍。

测试方法的执行顺序是：构造方法->@Before->@Test->@After

上面两个方法在Junit5种被换为@BeforeEach和@AfterEach。

另外Junit5还新加了几个注解，比如：@DisplayName("对测试方法重命名")；@BeforeAll、@AfterAll。@BeforeAll会放在构造方法前，@AfterAll会放在@After后；且两个都是静态方法，整个测试类只会执行一此。

### 3.2 测试私有方法 ###

ReflectionTestUtils.invokeMethod(类对象,"私有方法名",.....私有方法入参);

另外如果此私有方法里面有spring的bean比如类对象，需要在调用invokeMethod之前注入类对象：

ReflectionTestUtils.setField(类对象,"类对象内需要注入的类名",需要注入的类);

## 4. mock ##

对于一些外部接口，可能无法控制本地环境能收到结果，或者干脆就没必要真实去调用，就可以采用mock的方法，模拟出接口返回值。

本文主要用org.mockito包进行实例说明。

### 4.1 基本写法： ###

#### 4.1.1 mock常见的写法有两种： ####

Mockito.doReturn(1)when(demoService).demoMethod("a");

when(demoService.demoMethod("a")).thenReturn(1);

表示同一个意思：当执行demoService类的demoMethod方法，且入参是a时，将会返回1。生效时间是这行代码到本测试方法结束。

#### 4.1.2 任意入参 ####

可以mock任意入参时都会返回设定的返回值。

any():任意字符串

anyLong():任意Long类型

更多略

#### 4.1.3 mock返回值 ####

除了thenReturn，还有thenThrow，另外.thenCallRealMethod代表返回真实结果。

### 4.2 mock注入 ###

如果要对一个类使用mock，需要先改变类的注入方式，常见的注解有：

@Mock：对函数的调用均使用mock，不会调用真实方法

@Spy：对函数的调用是真实调用，即代码会真实走到方法内部。

@InjectMocks：可以调用真实代码的方法，其余用@Mock（或@Spy）注解创建的mock将被注入到用该实例中。

@MockBean：spring封装的@mock，会加入spring容器

@SpyBean：spring封装的@Spy，会加入spring容器

### 4.3 Mock高级用法 ###

#### 4.3.1 捕获入参 ####

定义：

private ArgumentCaptor<DTO> ar = ArgumentCaptor.forClass(DTO.class);

使用：

Mockito.doReturn(true).when(demoClient).demoMethod(ar.capture());

校验：

List<DTO> dtoList = ar.getAllValues(); // 这里的值是方法入参

由此可以得到mock时的入参，一般用于定时任务时校验深层方法的入参是否正确。

#### 4.3.2 mock静态方法 ####

MockedStatic<Util> theMock = Mockito.mockStatic(Util.class);

theMock.when(()->Util.getTime(any())).thenReturn(new Date());

另外因为是mock静态方法，如果一个测试类中静态方法被多个方法mock，会抛异常，解决办法有两种：

1、手动调用close()。

2、把第一行包在try with resources块的小括号里面，把第二行写在try()\{\}的大括号里面，代表只有块里面的需要mock。

#### 4.3.3 mock私有方法 ####

需要引入PowerMockito包，具体使用可另行百度。

发表评论取消回复

表情：

评论列表（有 0 条评论，16人围观）

还没有评论，来说两句吧...

相关阅读

相关单元测试 -- mock模拟测试总结

前言世面上测试的工具还是比较多的。本人用mock模拟测试用的比较多。所以一种用好了，其实都是大同小异。可以自测就可以了 **关于单元测试的命名** **测试类的...

桃扇骨/ 2024年04月19日 09:17/ 0 赞/ 90 阅读

相关 Service层代码单元测试以及单元测试如何Mock

一、背景接着上一篇文章：单元测试入门篇，本篇文章作为单元测试的进阶篇，主要介绍如何对Springboot Service层代码做单元测试，以及单元测试中涉及外调服务时，

怼烎@/ 2024年04月01日 19:23/ 0 赞/ 107 阅读

相关 Java用正确的姿势写单元测试以及mock

1. 前言对于一些简单的功能或业务，我们也许可以通过前端调试、postman等接口工具、main函数调用进行测试。但这每次改动代码都要人力测试，耗费大量的人力资源且不高

心已赠人/ 2024年03月16日 16:48/ 0 赞/ 17 阅读

相关单元测试：Mock对象

为什么要引入mock对象单元测试的目标是一次只验证一个方法，小步的前进，细粒度的测试，但是假如某个方法依赖于其他一些难以操控的东东，比如说网络连接，数据库连接，或者是S

谁借莪１个温暖的怀抱￠/ 2023年01月01日 06:57/ 0 赞/ 247 阅读

相关 Mock完成单元测试

转自：[https://zhuanlan.zhihu.com/p/98074553][https_zhuanlan.zhihu.com_p_98074553] 1、什么是M

心已赠人/ 2022年11月01日 04:23/ 0 赞/ 271 阅读

相关 java groovy + spock 单元测试实例 powermock mock静态方法以及Flink单元测试实例

具体的介绍看这篇介绍的很全面。 [https://blog.csdn.net/u012129558/article/details/78677357][https_blog.

迷南。/ 2022年10月10日 12:50/ 0 赞/ 331 阅读

相关快速入门JAVA单元测试——mock

背景为了确保代码的质量，对编写的代码进行单元测试是非常有必要的。在JAVA项目中，一般的项目结构比较复杂、依赖众多。在微服务与spring boot大行其道的今天，

我就是我/ 2022年09月08日 00:17/ 0 赞/ 456 阅读

相关单元测试mock

单元测试传统方式与mock方式添加依赖  <dependency> <grou

朴灿烈づ我的快乐病毒、/ 2022年08月31日 09:49/ 0 赞/ 260 阅读

相关单元测试——Mock

[【背景】][Link 1] 单元测试的目标是一次只验证一个方法，但是倘若遇到这样的情况：某个方法依赖于其他一些难以操控的东西，诸如网络、数据库，甚至是微

青旅半醒/ 2022年08月07日 03:56/ 0 赞/ 379 阅读

相关单元测试之Mock

Mock测试就是在测试过程中，对于某些不容易构造或者不容易获取的对象，用一个虚拟的对象来创建以便测试的测试方法。引入Mock最大的优势在于：Mock的行为固定，它确保当你

忘是亡心i/ 2022年08月05日 02:51/ 0 赞/ 364 阅读