java - 如何在 Mockito 和 JUnit 中模拟具有复杂请求的服务？

Question

我有一个界面:

public interface SenderService {
    String send(long amount);
}

我有这个接口(interface)的实现:

public class SenderServiceAdapter implements SenderService {

    private final ThirdPartyService thirdPartyService;

    public SenderServiceAdapter(ThirdPartyService thirdPartyService) {
        this.thirdPartyService = thirdPartyService;
    }

    @Override
    public String send(long amount) {

        ThirdPartyRequest thirdPartyRequest = new ThirdPartyRequest();

        thirdPartyRequest.setAmount(amount);
        thirdPartyRequest.setId(UUID.randomUUID().toString());
        thirdPartyRequest.setDate(new Date());

        ThirdPartyResponse thirdPartyResponse = thirdPartyService.send(thirdPartyRequest);

        String status = thirdPartyResponse.getStatus();

        if (status.equals("Error")) throw new RuntimeException("blablabla");

        return thirdPartyResponse.getMessage();
    }
}

现在我想为此服务编写单元测试。我需要模拟 ThirdPartyService 的方法 send。但我不明白怎么办。

public class SenderServiceAdapterTest {

    private ThirdPartyService thirdPartyService;
    private SenderService senderService;

    @Before
    public void setUp() throws Exception {
        thirdPartyService = Mockito.mock(ThirdPartyService.class);
        senderService = new SenderServiceAdapter(thirdPartyService);
    }

    @Test
    public void send() {

        when(thirdPartyService.send(new ThirdPartyRequest())).thenReturn(new ThirdPartyResponse());

        String message = senderService.send(100L);

    }
}

ThirdPartyRequest 在 SenderServiceAdapter 中创建。我怎样才能 mock 它？

Answer 1

试试这个:

doReturn(new ThirdPartyResponse()).when(thirdPartyService).send(any(ThirdPartyRequest.class));

此外，通过查看您的代码，您将需要在响应中设置一些内容，因此您必须执行以下操作:

ThirdPartyResponse response = new ThirdPartyResponse(); //or mock
response.setStatus(...);
response.setMessage(...);
doReturn(response).when(thirdPartyService).send(any(ThirdPartyRequest.class));