如何在使用 Spring 依赖注入时模拟在其他方法中调用的方法？

Question

我有这样的事情：

public class Class1 {
    @Autowired
    Class2 class2;

    public Object class1Method {
        // Do something
        Object obj = class2.class2Method(arg);
        // Do something with obj
        return somthing;
    }
}

我正在对 Class1 进行单元测试，我想模拟 class2Method。我试图找到解决方案，我发现了这个：

Test class with a new() call in it with Mockito

我知道，如果我在 class1Method 中创建新的 Class2 实例，则此解决方案有效，但在我的情况下，我使用 Spring Dependency Injection，但我不知道该怎么做。

Answer 1

根据你是如何用 spring 构建你的 DI 的，如果你有 @Autowired 在字段上，你可以使用这样的东西

@RunWith(SpringRunner.class)
public Class1Test {

    @InjectMocks
    @Autwired
    private Class1 class1;

    @Mock
    Class2 class2;

    @BeforeMethod
    public void initMocks(){
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public Object testClass1Method {
        when(class2.class2Method(anyString()).thenReturn(expectedReturn);
        //Now test whatever you need with mocked class2.class2Method 

    }
}

如果你有构造函数 DI（@Autowired 在构造函数上），这会容易得多，并且你不需要为测试设置 spring 上下文

public Class1Test {

    private Class1 class1;

    private Class2 class2;

    @BeforeMethod
    public void initMocks(){
        class2 = mock(Class2.class);
        class1 = new Class1(class2);
    }
   //then use the same approach to mock any methods that you need
}

Answer 2

你应该去

模拟

，并让您的 JUnit 类在顶部包含此 @RunWith(MockitoJUnitRunner.class)。所以在你的情况下，你可以像这样拥有你的 Junit。

@RunWith(MockitoJUnitRunner.class)
public Class1Test {

    @InjectMocks
    private Class1 class1 = new Class1();

    @Mock
    Class2 class2;

    @Test
    public Object testClass1Method {
        // Now this will get the class2 mock instance when executed
        Object obj = class1.class1Method(arg);

    }
}

Answer 3

这是您可以使用Mockito 为Class1 编写单元测试的方法：

public Class1Test {

    @InjectMocks
    private Class1 class1;

    @Mock
    private Class2 class2;

    @BeforeMethod
    public void initMocks(){
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void test_method1() {

        Object obj = //instantiate it here...,

        Mockito.doReturn(obj).when(class2.class2Method(anyString());

        Assert.assertEquals(class1.class1Method(), something);
    }

}