如何对调用void方法的方法进行单元测试？

Question

我陷入了单元测试场景的一个基本问题，并将感谢帮助。

我有一个类MyService，它调用MyRemvalService在DB中将标志设置为true。

@Slf4j
@Service
@RequiredArgsConstructor
public class MyService {

    private final MyRemovalService myRemovalService;
    private final MyRepository myRepository;

    public void setFlag() {
        final List<String> records = myRepository.getData(ENUM_1, ENUM_2, ENUM_3);
        records.forEach(MyRemovalService::removeData);
    }

}

我的测试：

@ExtendWith(MockitoExtension.class)
class MyServiceTest {

    @Mock
    private MyRemovalService myRemovalService;

    @Mock
    private MyRepository myRepository;

    @InjectMocks
    private MyService myService;

    @Test
    void testMyUseCase() {
        when(myRepository.getData(any(), any(), any())).thenReturn(List.of("Test1", "Test2"));
        myService.setFlag();
    }

}

我被要求测试并检查（MyRemovalService:：removeData）是否正在使用相关数据调用。

由于返回类型无效，我如何测试它？

Answer 1

Mockito允许定义ArgumentCaptors，允许您捕获模拟对象的参数方法，并使用：

var objectRequestedForRemovalCapture = ArgumentCaptor.forClass(String.class);
verify(myRemovalService).removeData(objectRequestedForRemovalCapture.capture());
var results = objectRequestedForRemovalCapture.getAllValues();

//and do the verification on your list of objects

Answer 2

您可以使用验证方法：

verify(myRemovalService).removeData("Test1"));
verify(myRemovalService).removeData("Test2"));