如何在 goroutine 中测试结果而不在测试中等待

Question

我在使用golang的时候，有时候需要在goroutine中测试一下结果，我是用time.Sleep来测试的，不知道有没有更好的测试方法。

假设我有一个这样的示例代码

func Hello() {
    go func() {
        // do something and store the result for example in db
    }()
    // do something
}

然后当我测试 func 时，我想在 goroutine 中测试两个结果， 我正在这样做：

 func TestHello(t *testing.T) {
        Hello()
        time.Sleep(time.Second) // sleep for a while so that goroutine can finish
        // test the result of goroutine
 }

有没有更好的测试方法？

基本上，在真正的逻辑中，我不关心 goroutine 中的结果，我不需要等待它完成。但在测试中，我想在它完成后检查。

Answer 1

如果你真的想检查一个 goroutine 的结果，你应该使用这样的通道：

package main

import (
    "fmt"
)

func main() {
    // in test
    c := Hello()
    if <-c != "done" {
        fmt.Println("assert error")
    }

    // not want to check result
    Hello()
}

func Hello() <-chan string {
    c := make(chan string)
    go func() {
        fmt.Println("do something")
        c <- "done"
    }()
    return c
}

https://play.golang.org/p/zUpNXg61Wn

Answer 2

大多数问题都是“我如何测试 X？”往往归结为 X 太大。

在您的情况下，最简单的解决方案是不在测试中使用 goroutine。单独测试每个功能。将您的代码更改为：

func Hello() {
    go updateDatabase()
    doSomething()
}

func updateDatabase() {
    // do something and store the result for example in db
}

func doSomething() {
    // do something
}

然后为updateDatabase 和doSomething 编写单独的测试。