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 "rongrong@qq.com";

                } else if ("jqj".equals(arg)) {

                    return "jiuqujian@qq.com";

                }

                throw new NullPointerException();

            }

        });

        try {

            PowerMockito.whenNew(StudentAnswerService.class).withAnyArguments().thenReturn(studentAnswerService);

            StudentController studentController = new StudentController();

            String email = studentController.getEmail("rr");

            assertEquals("rongrong@qq.com",email);

            email = studentController.getEmail("jqj");

            assertEquals("jiuqujian@qq.com",email);

            email = studentController.getEmail("tony");

            assertEquals("jiuqujian@qq.com",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的使用介绍完，有兴趣的同学可以自己从上到下自己敲一遍。

巴特西