unit-testing - 如何在 go 中测试 pubsub 消息确认？

Question

如何编写一个单元测试来验证消息是否真的被确认了？

我想模拟 pubsub 消息并验证 Ack被调用但不知道如何调用。

func processMessage(ctx context.Context, msg *pubsub.Message) {
    log.Printf("Processing message, data: %s", msg.Data)
    msg.Ack()
}

Answer 1

您可以通过使用接口(interface)和模拟来做到这一点，如下所示。

// Create an interface that'll be implemented by *pubsub.Message
type message interface{
    Ack()
}

// Accept the interface instead of the concrete struct
func processMessage(ctx context.Context, msg message) {
    log.Printf("Processing message, data: %s", msg.Data)
    msg.Ack()
}

现在，在测试文件中，创建一个模拟消息并确认 Ack正在被调用。您可以使用 testify/mock为了那个原因。