Mockito 与工厂进行条件验证

发布于 2024-10-06 17:52:01 字数 735 浏览 2 评论 0原文

在mockito中是否可以根据模拟是否实际在被测单元中使用来验证模拟上调用的方法？

举一个简单的例子，我向我的被测单元提供一个模拟工厂（FooFactory），当调用 Foo.create() 时，它返回一个模拟（Foo），供被测单元中的某些方法使用。测试。如何验证仅当被测单元调用 Foo.create() 时才调用 Foo.method() ？

我设想代码看起来像这样：

@Before
public void init() {
  Foo mockFoo = mock(Foo.class);
  when(fooFactory.create()).thenReturn(mockFoo);
  test = new UnitUnderTest(fooFactory);
}

@Test
... may or may not create a foo ...

@After
public void cleanup() {
  if (verify(fooFactory).create()) {  // Here's the 'conditional verification'
    Foo mockFoo = fooFactory.create();
    verify(mockFoo).close();
  }
}

举一个更具体的例子，我的工厂返回一个我想确保关闭的 Reader 对象，但并不是类中的每个方法实际上都构造了一个 Reader。显然，我可以将验证添加到我知道需要 Reader 的每个测试中，但这似乎需要大量重复工作。

原文

Is it possible in mockito to verify a method was called on a mock based on whether the mock was actually used in the unit-under-test?

For a simple example, I supply a mock factory (FooFactory) to my unit-under-test, and when Foo.create() is called, it returns a mock (Foo) to be used by certain methods in the unit-under-test. How can I verify that Foo.method() is called only if Foo.create() was called by the unit-under-test?

I'm envisioning that the code would look something like this:

@Before
public void init() {
  Foo mockFoo = mock(Foo.class);
  when(fooFactory.create()).thenReturn(mockFoo);
  test = new UnitUnderTest(fooFactory);
}

@Test
... may or may not create a foo ...

@After
public void cleanup() {
  if (verify(fooFactory).create()) {  // Here's the 'conditional verification'
    Foo mockFoo = fooFactory.create();
    verify(mockFoo).close();
  }
}

For a little more concrete example, my factory returns a Reader object that I want to ensure is closed, but not every method in the class actually constructs a Reader. I could obviously add the verification to every test where I know the Reader is needed, but that seems like a lot of duplicated effort.

分享到QQ

分享到微博