文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

使用Mockito测试Callback回调,你学会了吗?

2024-11-30 11:01

关注

我们将探索两种解决方案,首先使用ArgumentCaptor,然后使用直观的doAnswer()方法。

Callback回调简介

Callback回调函数是作为参数传递给方法,该方法应在给定时间执行回调参数。

使用回调的一个常见场景是在服务交互期间,当我们需要处理来自服务调用的响应时。比如:

public interface Service {
    void doAction(String request, Callback callback);
}

在回调参数中,我们传递一个类,该类将使用reply(T response)方法处理响应:

public interface Callback {
    void reply(T response);
}

我们将使用一个简单的服务示例来演示如何传递和调用回调:

public void doAction() {
    service.doAction("our-request", new Callback() {
        @Override
        public void reply(Response response) {
            handleResponse(response);
        }
    });
}

private void handleResponse(Response response) {
    if (response.isValid()) {
        response.setData(new Data("Successful data response"));
    }
}

我们选择使用Java Lamda表达式写得更简洁:

service.doAction("our-request", response -> handleResponse(response));

使用ArgumentCaptor

现在,让我们看看Mockito如何使用ArgumentCaptor来获取回调对象:

@Test
public void givenServiceWithValidResponse_whenCallbackReceived_thenProcessed() {
    ActionHandler handler = new ActionHandler(service);
    handler.doAction();

    verify(service).doAction(anyString(), callbackCaptor.capture());

    Callback callback = callbackCaptor.getValue();
    Response response = new Response();
    callback.reply(response);

    String expectedMessage = "Successful data response";
    Data data = response.getData();
    assertEquals(
      "Should receive a successful message: ", 
      expectedMessage, data.getMessage());
}

在本例中,我们首先创建一个ActionHandler,然后调用该处理程序的doAction方法。这只是service.doAction方法调用的包装器,我们在这里调用回调。

接下来,我们验证doAction是否在mock服务实例上被调用,并将anyString()作为第一个参数,将callbackCaptor.capture()作为第二个参数,这就是我们捕获回调对象的地方。然后可以使用getValue()方法返回参数的捕获值。

现在我们有了回调对象,创建一个默认有效的Response对象,直接调用reply方法并断言响应数据具有正确值。

使用doAnswer()方法

现在,我们将使用Mockito的Answer对象和doAnswer方法来存根void方法doAction:

@Test
public void givenServiceWithInvalidResponse_whenCallbackReceived_thenNotProcessed() {
    Response response = new Response();
    response.setIsValid(false);

    doAnswer((Answer) invocation -> {
        Callback callback = invocation.getArgument(1);
        callback.reply(response);

        Data data = response.getData();
        assertNull("No data in invalid response: ", data);
        return null;
    }).when(service)
        .doAction(anyString(), any(Callback.class));

    ActionHandler handler = new ActionHandler(service);
    handler.doAction();
}

在第二个示例中,我们首先创建一个无效的Response对象,该对象将在稍后的测试中使用。

接下来,我们在mock服务上设置Answer,以便在调用doAction时,我们拦截调用并使用invocation.getArgument(1)获取方法参数以获取回调参数。

最后一步是创建ActionHandler并调用doAction,从而调用Answer。

结论

本文介绍了在使用Mockito进行测试时处理测试回调的两种不同方法,相比较使用doAnswer方式比较直观。

来源:今日头条内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯