如何告诉我的测试等待 goroutine 中的回调？

Question

我正在使用 dockerclient https://github.com/samalba/dockerclient，它有一个基于通道的 API 来监听事件 client.MonitorEvents() 和一个方便的回调方法 client.StartMonitorEvents(callbackHandler)。

我想测试处理程序是否被调用。当然，dockerclient 在 goroutine 中处理事件。

目前，我的处理程序只是吐出一个日志。如果我在测试中等待，一切都会处理。如果我不这样做，它会在处理任何事情之前退出：

func eventCallback(event *dockerclient.Event, ec chan error, args ...interface{}) {
  log.Printf("Received event: %#v\n", *event)
}

我的测试看起来很简单：

func TestReceiveEvent(t *testing.T) {
   createAndMonitorEvents(server.URL)
   <- eventReady
   eventWriter.Write([]byte(someEvent))
   // test for something here
}

当然，由于 goroutine 的原因，除非我输入 time.Sleep()，否则它不起作用。

我如何告诉我的测试，“在运行测试之前等待另一个例程完成它的工作”，而不是进入一些任意的睡眠？我正在测试我的处理程序是否正确处理了该事件。

备用接口client.MonitorEvents() 返回一个频道，这给了我更大的控制权，但从频道接收会吐出无限的nil 事件。

更新：

根据要求，createAndMonitorEvents 是：

func createAndMonitorEvents(url string) {
  // Init the client
  docker, _ := dockerclient.NewDockerClient(url, nil)

  // Listen to events
  stopchan := make(chan struct{})

  go func() {
    eventErrChan, err := docker.MonitorEvents(nil, stopchan)
    if err != nil {
        return
    }

    for e := range eventErrChan {
        if e.Error != nil {
            return
        }
        eventCallback(&e.Event, nil)
    }
    fmt.Println("monitor in place")
  }()
}

Answer 1

我认为当您使用MonitorEvents 获得nils 时，您只是看到事件通道已关闭（source of MonitorEvents 包含close(eventOrErrorChan)，支持这一点）。 evt, ok := <-c 让您直接检查是否（ok 关闭时为假），for evt := range c 将在关闭后停止。一般来说，从关闭的通道接收是specified 到“[yield] 在收到任何先前发送的值之后元素类型的零值”

关于等待回调的问题：回调可以关闭通道。 （或发送给它。）然后您的测试可以使用select 等待指定的时间长度：

select {
case <-c:
        /* ...success... */
case <-time.After(5 * time.Second):
        /* timed out */
}

如果您知道某些错误情况会导致处理程序无法完成或无法运行，它可能会在不同的通道上发出这些情况的信号，或者通过向c 发送不同的值。

Answer 2

我认为这会有所帮助

WaitGroup 等待一组 goroutine 完成。主 goroutine 调用 Add 来设置要等待的 goroutine 的数量。然后每个 goroutine 运行并在完成时调用 Done。同时，Wait 可以用来阻塞，直到所有的 goroutine 都完成。

http://golang.org/pkg/sync/#example_WaitGroup

Answer 3

“我如何告诉我的测试，“在运行测试之前等待另一个例程完成它的工作”，而不是进入一些任意的睡眠？”

你要么在一个频道上发送，要么关闭一个。调用者正在接收执行阻塞，直到信号发生。我真的不知道如何将你的代码变成有意义的东西......你不能在函数中分配停止通道，你必须将它传递给函数，以便调用者可以监听它。比如这甚至是如何编译的？

func TestReceiveEvent(t *testing.T) {
   createAndMonitorEvents(server.URL)
   <- eventReady // undeclared variable, and discarding the event you recieved?
   eventWriter.Write([]byte(someEvent)) //and another variable that is not declared
   // test for something here
}

也许和想法会帮助你开始......

func createAndMonitorEvents(url string, done chan bool) {
      //the codes
      close(done)
}


func TestReceiveEvent(t *testing.T) {
  eventReady := make(chan bool)
    createAndMonitorEvents(server.URL, eventReady)
    <- eventReady
    eventWriter.Write([]byte(someEvent)) // dis still don't exist though
    // test for something here

}