X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/ce0eacc2fe5090810cf22e022af3ab83006068a4..80e348724041a694002fc38f57e8868b9b9a2be9:/summarizer_helper_test.go diff --git a/summarizer_helper_test.go b/summarizer_helper_test.go index 94c8154..6544b74 100644 --- a/summarizer_helper_test.go +++ b/summarizer_helper_test.go @@ -21,8 +21,13 @@ type FakeSummarizer struct { tasks []Task } +func newFakeSummarizer() *FakeSummarizer { + return &FakeSummarizer{} +} + 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 +36,7 @@ func (f *FakeSummarizer) Summarize(ctx context.Context, task Task) (string, erro defer func() { f.mutex.Lock() + f.active-- f.mutex.Unlock() }() @@ -88,7 +94,3 @@ func fakeLine(task Task) string { return line } - -func newFakeSummarizer() *FakeSummarizer { - return &FakeSummarizer{} -}