测试使用 time.Ticker 的代码的正确方法？

Question

我希望您就测试使用 time.Ticker 的代码的正确方法提出建议

例如，假设我有一个如下所示的倒数计时器（只是我为这个问题而想到的一个例子）：

type TickFunc func(d time.Duration)

func Countdown(duration time.Duration, interval time.Duration, tickCallback TickFunc) {
    ticker := time.NewTicker(interval)
    for remaining := duration; remaining >= 0; remaining -= interval {
        tickCallback(remaining)
        <-ticker.C
    }

    ticker.Stop()
 }

http://play.golang.org/p/WJisY52a5L

如果我想测试这个，我想提供一个模拟，以便我可以快速和可预测地运行测试，所以我需要找到一种方法让我的模拟进入倒计时功能。

我可以想到几种方法来做到这一点：

创建一个 Ticker 接口和一个包内部的第一类函数，我可以为了测试的目的修补它：http : //play.golang.org/p/oSGY75vl0U

创建一个 Ticker 接口，并将一个实现直接传递给 Countdown 函数：http : //play.golang.org/p/i67Ko5t4qk

如果我采用后一种方式，我是否透露了太多有关 Countdown 工作原理的信息，并使潜在客户更难使用此代码？他们不必给出持续时间和间隔，而是必须构建并传入一个 Ticker。

我很想知道在测试这样的代码时最好的方法是什么？或者您将如何更改代码以保留行为，但使其更易于测试？

谢谢你的帮助！

Answer 1

由于这是一个非常简单的函数，我假设您只是使用它作为如何模拟不平凡的东西的示例。如果您确实想测试此代码，而不是模拟股票代码，为什么不只使用非常小的间隔。

恕我直言，第二个选项是两个选项中更好的一个，让用户调用：

foo(dur, NewTicker(interval)... 

看起来并没有太大的负担。

Go 中的回调也是严重的代码味道：

func Countdown(ticker Ticker, duration time.Duration) chan time.Duration {
    remainingCh := make(chan time.Duration, 1)
    go func(ticker Ticker, dur time.Duration, remainingCh chan time.Duration) {
        for remaining := duration; remaining >= 0; remaining -= ticker.Duration() {
            remainingCh <- remaining
            ticker.Tick()
        }
        ticker.Stop()
        close(remainingCh)
    }(ticker, duration, remainingCh)
    return remainingCh
}

然后您可以使用以下代码：

func main() {
    for d := range Countdown(NewTicker(time.Second), time.Minute) {
        log.Printf("%v to go", d)
    }
}

这是在操场上：http://play.golang.org/p/US0psGOvvt