java单元测试Mockito、PowerMockito相关整理

原创已于 2024-08-09 00:09:31 修改 · 532 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#单元测试 #mock #mockito

于 2024-07-27 17:04:57 首次发布

引入的依赖

<dependency>
    <groupId>junit</groupId>
    <artifactId>junit</artifactId>
    <version>4.12</version> <!-- 版本号根据实际需要选择 -->
    <scope>test</scope> <!-- 作用范围为测试阶段 -->
</dependency>

<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-core</artifactId>
    <version>3.11.2</version> <!-- 版本号根据实际需要选择 -->
    <scope>test</scope> <!-- 作用范围为测试阶段 -->
</dependency>

基础的操作

在需要写的项目中看下有没有对应的目录:在src目录下面有没有test/java和test/resource的文件夹，没有手动创建下就好了

1.以idea创建测试类为例,在你需要写的则试类的类里面按住快捷键: alt+Insert, 就会出现一个Test...选项，选中就会出现一个窗口， Testting Library一般选择JUnit4即可;

Generate这里根据自己的需要勾选setUp/@Before这个我们比较常用;

最后再把所有需要写的方法勾选，点击0K就好了

2.在创建好的测试类上面加入注解: @Runwith(PowerMockRunner.class)

3.在测试类中写入需要将目标类mock下:

例如目标类为A类，则需要写成下面的代码:

@InjectMocks

private A a;

4.目标类中应该会有一些oc容器的对象，通过aRes ouc e或@Autowir ed主解获取，那需要将这些都要放入到测试类中

例如目标A类中引入了b和c两个对象，原代码如下:

@Autowired

private B b;

@Resource

private C c;

测试类的代码则需要写成

@Mock

private B b;

@Mock

private C C;

对于方法中new的对象怎么mock

先加个@PrepareForTest({TestTarget.class}) TestTarget 是测试的类名

然后写入下面的代码

@Before

public void setUp(){

MockitoAnnotations.initMocks(this);

}

最后根据情况 mock

//方法中的 new 的一个有参实体类
TargetClazz clazz = new TargetClazz("111",111,...);

//写成下面的代码
TargetClazz clazz = Powermockito.mock(TargetClazz.class);
PowerMockito.whenNew(clazz.class).withAnyArguments().thenReturn(clazz);

//方法中的 new 的一个无参实体类
TargetClazz clazz = new TargetClazz();

//写成下面的代码
TargetClazz clazz = Powermockito.mock(TargetClazz.class);
PowerMockito.whenNew(clazz.class).withNoArguments().thenReturn(clazz);

对于目标类中有私有的对象怎么 mock

通过@Value 注解获取值的

@Value("${test.status}")

private String status;

//需要在测试类中写入以下代码
ReflectionTestUtils.setField(impl，"status","1");

//impl是测试目标类，写在了测试类中
@InjectMocks
private TestImpl impl;

//第二个参数就是目标类中的参数名称
//第三个参数就是对应的值，这个值跟参数的类型有关，值可以随便写

只是单纯的字段，怎么 mock？

private String status;

通过反射去获取字段的属性
1.当前测试的类：TargetClazz


TargetClazz clazz = new TargetClazz();

TargetClazz spy = Mockito.spy(clazz);

Field statusField = TargetClazz..class.getDeclaredField("status");

FieldSetter.setField(spy,statusField,"1");

如果这个参数是写在目标类里面，作为一个私有参数并且new一个对象出来怎么 mock

//目标代码的参数：
private final Map map = new HashMap;

//可以写成下面的代码
private Map mp:

@Before
public void setUp(){
    map = PowerMockito.mock(HashMap.Class);
    Whitebox.setInternalState(controller, "map", map) ;
}

对于静态方法怎么mock

//原代码如下：
final String originPassword = SHATools.decryptPwdStr(pwd, cryptSuffix):
测试类操作：
1.在测试类上面加注解PrepareForTest
@PrepareForTest（｛SHAToo1s.class｝），一个不需要加{}，多个的话，以,隔开
2.添加下面的代码在你的测试类的方法里面
PowerMockito.mockStatic(SHATools.class);
你也可以在@Before注解的方法里面加，这样一个测试类只需要写一次了，要不然每个调用的方法
里面有这个的话，都需要在对应的测试类的方法里面写上这个代码
@Before
public void setUp(
    MockitoAnnotations.initMocks(this);
    PowerMockito.mockStatic(SHATools.class);
}
3.然后就可以根据返回写代码了
Mockito.mock(SHATools.decryptPwdStr(Mockito.anyString(), Mockito.anyString())). thenReturn("任务字符串")
注：首先要确定好里面的参数类型，不确定可以写Mockito.any()，但是一定要保证参数是有值的，
如果是null的话，你的返回没有用，thenReturn这个是根据你的返回值类型来的，案例返回的是一个
string类型的，所以随便返回个字符串即可。
如果你原代码中的参数就是有个nul1，例如：
final String originPassword = SHATools.decryptPudStr(null, cryptSuff1x);
那测试类就需要将aryptSuffix的值确定好
Mockito.mock(SHAToo1s.decryptPwdStr(null, cryptSuffix))，thenReturn("任意字符串");

单元测试有异常怎么 mock

如果是异常，先确定是哪个代码引起的异常，可以将原代码那块异常的换成其他异常，看看报错的位置
（1）原代码报错的异常是没有返回的
public ServletOutputStream downloadById(servletoutputStream outputStream){

    InputStrean nputStream = ...;
    try{
        final byte[] buffer = new byte[BUFFER SIZE];
        int len = 0;
        while(inputSteam.available()>0){
            len =inputStream.read(buffer);
            outputStream.write(buffer, 0, len);
        }
        return outputStream;
    } catch (IOException e){
        log.error("downloadById异常,异常原因：",e);
    }
    return outputStream;
}
此处有个IOException异常，找一个个我们mock的对象，进行mock返回异常，outputStream就是一个mock的对象
ServletOutputStream outputStream = Mockito.mock(ServletOutputStream.class);
IOException ioException = Mockito.mock(IOException.class);
Mockito.doThrow(ioExcentson).when(outputstream).write(Mockito.any(),Mockito.anyint(),Mockito.anyInt());
try{
    impl.downloadEyId(outputStream);
}catch (IOException e){

}
注：需要注意的是，如果在原代码里面没有throw一个异常的话，测试类的代码是不需要try catch的

(2)原代码报错的异常是有返回的
try{
    final String originPassword = SHATools.decryptPwdstr(pwd, cryptSuffix);
    final String encodePassword = SHAToo1s.SHA(originpassword, SHATao1s.SHA512);
} catch (final NoSuchAlgorithmException e)
    throw new BusinessException(UserExceptionEnum.PWD_DECODE_ERROR) ;
} catch (final UnsupportedEncodingException e) {
    throw new BusinessException(UserExceptionEnum. PWD_DECODE_ERROR) ;
｝

此处这种代码是一个工具类，方法是static修饰的静态方法，且有两个异常，我们代码中需要怎么让他们全部走完
    SHATools.SHA -> NoSuchAlgorithmException
    SHATools.decryptPwdStr -> UnsupportedEncodingExceptjon
这是它们的异常关系，看代码顺序SHATools.decryptPwdstr这个在SHATools.SHA前，所以我们需要先走SHATools.SHA这个方法的异常，然后再走SHATools.decryptPwdStr的异常
接下来就是代码：
NoSuchAlgorithmException noSuchAlgonithmException = Mockito.mock(NoSuchAlgorithmException.class);
Mockito.when(SHATools.SHA(Mockito.anyString(),Mockito.anyString())).thenThrow(noSucnoSuchAlgonithmException);
try {
    impl.test("111","111","111111","111");
}catch (BusinessException e){

}

UnsupportedEncodingException exception = Mockito.mock(UnsupportedEncodingException.class);
Mockito.when(SHATools.decryptPwdSer(Mockito.anyString(), Mockito.anyString())). thenThrow(exception);
try {
    impl.test("111","111","111111","111");
}catch (BusinessException e){

}


由于原代码是有throw new BusinessException，所以我们则试类的代码机取是需要抓取这个BusinessException异常的，不抓取处理的话会有报错。

注：异常的是处理是写在原有正常的代码走完的情况下再去操作的

注：万物皆可mock（除了抽象类）