单元测试mock之mockito使用

先来一个简单的例子来感受一下

外部接口类：TestService.java

package com.yzl.mock;

/**
 * 测试用服务
 *
 * @author yangzhilong
 */
public interface TestService {
    /**
     * 
     * 简单的测试服务，实现两个数相加
     *
     * @param a
     * @param b
     * @return
     */
    public int add(int a, int b);
}

正常流程的JUnit单元测试方法类似如下：

@Test
public void test0(){
    //从spring容器里取到该接口的实现类
    TestService service = new ClassPathXmlApplicationContext("xxx.xml").getBean(TestService.class);
    int a=2, b=3;
    int result = service.add(a, b);
    Assert.assertSame("计算结果预期", 5, result);
    
    //拿到这个result然后调用其他的业务逻辑
}

但是目前该接口的实现类(服务)目前是调用不通的，或者是压根还没做，但是你又不想因为这个接口而影响自己的代码的测试，就可以使用Mockito来模拟接口的调用。

首先引入mockito的jar

<dependency>
    <groupId>org.mockito</groupId>
    <artifactId>mockito-core</artifactId>
    <version>2.1.0-RC.1</version>
    <scope>test</scope>
</dependency>

然后编写代码如下：

package com.yzl.mock;

import org.junit.Assert;
import org.junit.Test;
import org.mockito.Mockito;

public class MockTest {
    /*
     * 使用mockito来模拟测试
     */
    @Test
    public void test1(){
        TestService service = Mockito.mock(TestService.class);
        int a=2, b=3;
        //使用mockito模拟接口返回
        Mockito.when(service.add(a, b)).thenReturn(5);
        int result = service.add(a, b);
        System.out.println(a + "+" + b + "=" + result);
        Assert.assertSame("计算结果预期", 5, service.add(a, b));
        //使用result继续做后面的业务
    }
}

上面的代码中Mockito的静态方法可以通过静态Import的形式导入，然后使用里面的静态方法就可以不带类名了，例如：

//非静态import的写法
import org.mockito.Mockito;
Mockito.when(service.add(a, b)).thenReturn(5);

//静态import的写法
import static org.mockito.Mockito.*;
when(service.add(a, b)).thenReturn(5);

运行junit单元测试方法将输出：

2+3=5