Mockito：如何验证在方法中创建的对象上调用了方法？

Question

我是 Mockito 的新手。

鉴于下面的类，我如何使用 Mockito 来验证 someMethod 在 foo 被调用之后被调用了一次？

public class Foo
{
    public void foo(){
        Bar bar = new Bar();
        bar.someMethod();
    }
}

我想拨打以下验证电话，

verify(bar, times(1)).someMethod();

其中bar 是Bar 的模拟实例。

Answer 1

Dependency Injection

如果您注入 Bar 实例或用于创建 Bar 实例的工厂（或执行此操作的其他 483 种方法之一），您将拥有执行测试所需的访问权限。

工厂示例：

给定一个这样写的 Foo 类：

public class Foo {
  private BarFactory barFactory;

  public Foo(BarFactory factory) {
    this.barFactory = factory;
  }

  public void foo() {
    Bar bar = this.barFactory.createBar();
    bar.someMethod();
  }
}

在您的测试方法中，您可以像这样注入 BarFactory：

@Test
public void testDoFoo() {
  Bar bar = mock(Bar.class);
  BarFactory myFactory = new BarFactory() {
    public Bar createBar() { return bar;}
  };
  
  Foo foo = new Foo(myFactory);
  foo.foo();

  verify(bar, times(1)).someMethod();
}

奖励：这是 TDD（测试驱动开发）如何驱动代码设计的示例。

Answer 2

另一种简单的方法是在 bar.someMethod() 中添加一些日志语句，然后确定在执行测试时可以看到上述消息，请参见此处的示例：How to do a JUnit assert on a message in a logger

当您的 Bar.someMethod() 为 private 时，这尤其方便。

Answer 3

我认为 Mockito @InjectMocks 是要走的路。

根据您的意图，您可以使用：

1. 构造函数注入
2. 属性设置器注入
3. 现场注入

更多信息docs

下面是一个场注入的例子：

类：

public class Foo
{
    private Bar bar = new Bar();

    public void foo() 
    {
        bar.someMethod();
    }
}

public class Bar
{
    public void someMethod()
    {
         //something
    }
}

测试：

@RunWith(MockitoJUnitRunner.class)
public class FooTest
{
    @Mock
    Bar bar;

    @InjectMocks
    Foo foo;

    @Test
    public void FooTest()
    {
        doNothing().when( bar ).someMethod();
        foo.foo();
        verify(bar, times(1)).someMethod();
    }
}

Answer 4

使用 PowerMockito.whenNew 的示例代码解决方案

• mockito-all 1.10.8
• powermock-core 1.6.1
• powermock-module-junit4 1.6.1
• powermock-api-mockito 1.6.1
• junit 4.12

FooTest.java

package foo;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

//Both @PrepareForTest and @RunWith are needed for `whenNew` to work 
@RunWith(PowerMockRunner.class)
@PrepareForTest({ Foo.class })
public class FooTest {

    // Class Under Test
    Foo cut;

    @Mock
    Bar barMock;

    @Before
    public void setUp() throws Exception {
        cut = new Foo();

    }

    @After
    public void tearDown() {
        cut = null;

    }

    @Test
    public void testFoo() throws Exception {

        // Setup
        PowerMockito.whenNew(Bar.class).withNoArguments()
                .thenReturn(this.barMock);

        // Test
        cut.foo();

        // Validations
        Mockito.verify(this.barMock, Mockito.times(1)).someMethod();

    }

}

JUnit 输出

Answer 5

如果您不想使用 DI 或工厂。你可以用一些棘手的方式重构你的类：

public class Foo {
    private Bar bar;

    public void foo(Bar bar){
        this.bar = (bar != null) ? bar : new Bar();
        bar.someMethod();
        this.bar = null;  // for simulating local scope
    }
}

还有你的测试课：

@RunWith(MockitoJUnitRunner.class)
public class FooTest {
    @Mock Bar barMock;
    Foo foo;

    @Test
    public void testFoo() {
       foo = new Foo();
       foo.foo(barMock);
       verify(barMock, times(1)).someMethod();
    }
}

然后调用你的 foo 方法的类会这样做：

public class thirdClass {

   public void someOtherMethod() {
      Foo myFoo = new Foo();
      myFoo.foo(null);
   }
}

正如您在以这种方式调用该方法时所看到的，您不需要在任何其他调用您的 foo 方法的类中导入 Bar 类，这可能是您想要的。

当然，缺点是您允许调用者设置 Bar 对象。

希望对你有帮助。

Answer 6

是的，如果您真的想要/需要这样做，您可以使用 PowerMock。这应该被视为最后的手段。使用 PowerMock，您可以使它从对构造函数的调用中返回一个模拟。然后在模拟上进行验证。也就是说，csturtz 的答案是“正确”的。

这是Mock construction of new objects的链接

Answer 7

经典的回答是，“你不知道。”您测试的是 Foo 的公共 API，而不是其内部。

Foo 对象（或者，不太好，环境中的其他对象）是否有任何行为受到foo() 的影响？如果是这样，请测试一下。如果不是，该方法的作用是什么？