如何对 pubsub 进行单元测试接收回调

我有以下生产代码:


func pullMessages(ctx context.Context, sub *pubsub.Subscription) {

    err := sub.Receive(ctx, func(ctx context.Context, msg *pubsub.Message) {

        log.Printf("Got message: %q\n", string(msg.Data))

        processMessage(msg)

        msg.Ack()

    })

    if err != nil {

        log.Printf("Receive: %v", err)

    }

}

我如何验证processMessage在单元测试中实际被调用?


慕丝7291255
浏览 142回答 1
1回答

繁花如伊

很简单:让你的回调成为一个命名函数而不是匿名函数:func pullMessages(ctx context.Context, sub *pubsub.Subscription) {    if err := sub.Receive(ctx, rcvCallback); err != nil {        log.Printf("Receiving message: %s",err)    }}func rcvCallback (ctx context.Context, msg *pubsub.Message) {        log.Printf("Got message: %q\n", string(msg.Data))        processMessage(msg)        msg.Ack()}现在,在您的单元测试中,您可以创建 aContext和 a的实例并将其Message传递给您的函数。但是,在这种情况下,这几乎没有意义。您只需记录和确认消息,这些功能应该由上游项目进行单元测试。因此,构造一个 message 和 unit test 的实例会更有意义processMessage。
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Go