如何检查函数在测试中被调用了多少次？

Question

我正在使用一个调用函数的API开发一个web服务器。这个函数的任务很繁重，并且会缓存结果。我的设计方式是，如果没有缓存，并且多个用户同时使用相同的参数调用此API，服务器只为第一个请求调用该函数一次，所有其他请求等待完成作业并从缓存返回响应。

我用这种方式写了一个测试：

func TestConcurentRequests(t *testing.T) {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            // do the request
            wg.Done()
        }
    }
    wg.Wait()
}

我可以通过在这个沉重的函数中打印一个值来检查我的代码是否正常工作，并检查控制台，看看它是否只出现一次，但是如果这个函数被调用了更多次，我正在寻找一种方法来失败测试。比如：

if n := NumberOfCalls(MyHeavyFunction); n > 1 {
    t.Fatalf("Expected heavy function to run only once but called %d times", n)
}

Answer 1

我建议把你的重功能分为缓存部分和实际处理部分。在这里，缓存部分将是并发客户端的门面。这里测试的系统实际上是缓存层，它决定是转发请求、保持还是从缓存返回。因此，为了测试这一点，您需要模拟进行处理的实际函数，并且模拟应该记录它服务了多少个请求。

以下是带有某些假设的演示：

package main

import "fmt"

func CreateCachedProcessor(processor Processing) func(string) string {
    //cache here
    return func(req string) string {
        //Assume you have some logic to call the actual processor based on your conditions
        return processor.ProcessRequest(req)
    }
}

type Processing interface {
    ProcessRequest(string) string
}

type HeavyProcessing struct{}

func (p *HeavyProcessing) ProcessRequest(request string) string {
    return "response"
}

type HeavyProcessingRecorder struct {
    Count int
}

func (p *HeavyProcessingRecorder) ProcessRequest(request string) string {
    p.Count += 1
    return "response recorded"
}

func main() {

    //normal
    hp := &HeavyProcessing{}
    c := CreateCachedProcessor(hp)
    fmt.Println(c("req1"))

    //Test with recorder
    hpr := &HeavyProcessingRecorder{}
    cr := CreateCachedProcessor(hpr)
    cr("req1")
    cr("req1")
    cr("req1")
    fmt.Println("No. of times Called ", hpr.Count)

}