使用传递给模拟对象的函数进行测试

问题描述

我有一个简单的功能,最近被一些附加的逻辑包装。我正在努力更新测试逻辑,因为突然之间,方法主体被包裹在一个模拟中。

让我给你举个例子。

以前的逻辑和测试:

// logic
public void doSomething(Transaction t,int a) {
  myService.foo(t,a);
}

我的测试:

// test
TestedService service;

@Mock
MyService myService;

@Mock
Transaction t;

@Test
public void testSomething() {
   testedService.doSomething(t,10);
   Mockito.verify(myService).foo(t,10);
}

发生的事情是我们将逻辑包装在一些附加功能

public void doSomething(Transaction t,int a) {
   model.runInEnhancedTransaction(t,t2 -> { myService.foo(t2,a) });
}

我的问题是,当逻辑突然用model方法包装时(该模型在我的测试中是模拟的),该如何测试?

我基本上需要验证t2 -> { myService.foo(t2,a) }对象是模拟对象时是否调用model

编辑:我只是出于测试目的而实现了自定义版本model,但是仍然想知道是否还有更优雅的方法

解决方法

测试这些lambda调用有点困难。我要做的是执行两项测试:一项称为model.runInEnhancedTransaction()的测试,另一项针对model.runInEnhancedTransaction()本身的测试。例如

@Test
void doSomethingCallsModelEnhancedTransaction() {
    testedService.doSomething(t,10);
    verify(model).runInEnhancedTransaction(eq(t),any());
}
@Test
void modelRunInEnhancedTransaction() {
    Transaction t = mock(Transaction.class);
    BiConsumer<Transaction,Integer> consumer = mock(Biconsumer.class);
    model.runInEnhancedTransaction(t,consumer);
    verify(consumer).accept(...);
}