X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/ce0eacc2fe5090810cf22e022af3ab83006068a4..3391e4eaab1c20e3b172c443186b1947622a679e:/summarizer_helper_test.go?ds=sidebyside diff --git a/summarizer_helper_test.go b/summarizer_helper_test.go index 94c8154..e78a69e 100644 --- a/summarizer_helper_test.go +++ b/summarizer_helper_test.go @@ -23,6 +23,7 @@ type FakeSummarizer struct { func (f *FakeSummarizer) Summarize(ctx context.Context, task Task) (string, error) { f.mutex.Lock() + f.tasks = append(f.tasks, task) f.active++ f.peak = max(f.peak, f.active) @@ -31,6 +32,7 @@ func (f *FakeSummarizer) Summarize(ctx context.Context, task Task) (string, erro defer func() { f.mutex.Lock() + f.active-- f.mutex.Unlock() }()