Function TestAddAddition_WgBalance

core/pool/pool_test.go:306–332

Regression: old code had a `default` branch that spawned an async goroutine. If the goroutine sent successfully, wg.Done() was never called — wg leak. This test verifies wg stays balanced after many addAddition calls.

(t *testing.T)

// If the goroutine sent successfully, wg.Done() was never called — wg leak.
// This test verifies wg stays balanced after many addAddition calls.
func TestAddAddition_WgBalance(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	pool := newTestBasePool(ctx, cancel)

	const N = 100
	// drain concurrently so addAddition never blocks on a full buffer
	var received int32
	go func() {
		for range pool.additionCh {
			atomic.AddInt32(&received, 1)
			pool.wg.Done()
		}
	}()

	for i := 0; i < N; i++ {
		pool.addAddition(&Unit{path: "/x", source: parsers.WordSource})
	}

	mustFinish(t, 2*time.Second, "wg.Wait hung — wg counter leaked", func() {
		pool.wg.Wait()
	})
	close(pool.additionCh) // stop drain goroutine
	if r := atomic.LoadInt32(&received); r != N {
		t.Fatalf("received %d items, want %d", r, N)
	}
}

func TestAddAddition_FullBufferUnblocksOnCancel(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())