X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/ce0eacc2fe5090810cf22e022af3ab83006068a4..b1be0ee8530df5db8973bfe355f2f6d3149383a0:/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() }()