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

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)

}