Mockito之：Spy

package com.newcrud.learn;

import lombok.Data;

@Data
public class MySpy {
    int a=100;
    public void fun(){
        System.out.println("fun");
        funOne();
        funTwo();
    }
    public void funOne(){
        System.out.println("funOne");
    }
    public void funTwo(){
        System.out.println("funTwo");
    }
    public Integer getMath(){
        return a;
    }
}

package com.newcrud.learn;

import org.mockito.MockitoAnnotations;
import org.mockito.Spy;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.testng.AbstractTestNGSpringContextTests;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import static org.mockito.Mockito.when;
import static org.testng.Assert.*;

@SpringBootTest
public class MySpyTest extends AbstractTestNGSpringContextTests {
    @Spy
    MySpy mySpy;
    @BeforeClass
    public void initMocks(){
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testFun() {
        mySpy.fun();
        System.out.println("=======");
    }

    @Test
    public void testFunOne() {
        mySpy.funOne();
        System.out.println("=======");

    }

    @Test
    public void testFunTwo() {
        mySpy.funTwo();
        System.out.println("=======");

    }

    @Test
    public void testGetMath() {
        System.out.println(mySpy.getMath());
    }
    @Test
    public void testGetMathTwo(){
        when(mySpy.getMath()).thenReturn(200);
        System.out.println(mySpy.getMath());
    }

}

结果

fun
funOne
funTwo
=======
funOne
=======
funTwo
=======
100
200

1、Mock声明的对象，对函数的调用均执行mock（即虚假函数），不执行真正部分。

2、Spy声明的对象，对函数的调用均执行真正部分。

3、相对于@Mock(answer = Answers.CALLS_REAL_METHODS)，更推荐@Spy