X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/ce0eacc2fe5090810cf22e022af3ab83006068a4..5a93b6124068c2a8748fc0bfbb2a2d151815bcf0:/summarizer_helper_test.go 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() }()