PowerMock学习(九)之Mock Answer的使用

关于Mock Answer

上一篇文章,有介绍过关于Arguments Matche的使用,其实 Answer的作用与其比较类似,但是它比 Arguments Matcher 更加强大。

Arguments Matche

即传入不同的参数,返回不同的结果,重在入参的判断,在入参重写方法去判断

Answer

见名知意,即返回不同的结果,但是根据传入参数去判断,在返回处重写方法去判断,返回结果

模拟场景

根据学生名字查找邮箱,controller调service层

service层

具体代码示例如下:
package com.rongrong.powermock.answers;

/**
 * @author rongrong
 * @version 1.0
 * @description:
 * @date 2019/12/4 20:24
 */
public class StudentAnswerService {

    public String getEmail(String userName){
        throw new UnsupportedOperationException();
    }
}

controller层

具体代码示例如下:

package com.rongrong.powermock.answers;

/**
 * @author rongrong
 * @version 1.0
 * @description:
 * @date 2019/12/4 20:24
 */
public class StudentController {

    public String getEmail(String userName) {
        StudentAnswerService studentAnswerService = new StudentAnswerService();
        return studentAnswerService.getEmail(userName);
    }
}

 

上面的代码的业务代码比较简单了,下面再来进行测试

具体示例代码如下:

package com.rongrong.powermock.answers;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import static junit.framework.TestCase.assertEquals;
import static junit.framework.TestCase.fail;

/**
 * @author rongrong
 * @version 1.0
 * @description:
 * @date 2019/12/4 20:34
 */
@RunWith(PowerMockRunner.class)
//准备调用层的类
@PrepareForTest(StudentController.class)
public class TestStudentAnswerService {

    @Test
    public void testStudentAnswerService() {
        StudentAnswerService studentAnswerService = PowerMockito.mock(StudentAnswerService.class);
        PowerMockito.when(studentAnswerService.getEmail(Mockito.anyString())).then(new Answer<String>() {
            @Override
            public String answer(InvocationOnMock invocation) throws Throwable {
                String arg = (String) invocation.getArguments()[0];
                if ("rr".equals(arg)) {
                    return "[email protected]";
                } else if ("jqj".equals(arg)) {
                    return "[email protected]";
                }
                throw new NullPointerException();
            }
        });
        try {
            PowerMockito.whenNew(StudentAnswerService.class).withAnyArguments().thenReturn(studentAnswerService);
            StudentController studentController = new StudentController();
            String email = studentController.getEmail("rr");
            assertEquals("[email protected]",email);
            email = studentController.getEmail("jqj");
            assertEquals("[email protected]",email);
            email = studentController.getEmail("tony");
            assertEquals("[email protected]",email);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

answer 接口中参数 InvocationOnMock使用

invocation.getArguments();(1)
invocation.callRealMethod();(2)
invocation.getMethod();(3)
invocation.getMock();(4)
(1)获取 mock 方法中传递的入参
(2)获取是那个真实的方法调用了该 mock 接口
(3)获取是那么 mock 方法被调用了
(4)获取被 mock 之后的对象

到此,关于mock中 Answer的使用介绍完,有兴趣的同学可以自己从上到下自己敲一遍。

猜你喜欢

转载自www.cnblogs.com/longronglang/p/11985991.html