barrier_test.go

package syncx

import (
	"context"
	"errors"
	"sync"
	"testing"
	"time"
)

func TestBarrierSynchronizes(t *testing.T) {
	b := NewBarrier(3)
	var wg sync.WaitGroup
	wg.Add(3)

	errCh := make(chan error, 3)
	for i := 0; i < 3; i++ {
		go func() {
			defer wg.Done()
			errCh <- b.Wait(context.Background())
		}()
	}

	wg.Wait()
	close(errCh)
	for err := range errCh {
		if err != nil {
			t.Fatalf("unexpected barrier err: %v", err)
		}
	}
}

func TestBarrierCancellationBreaksRound(t *testing.T) {
	b := NewBarrier(3)
	errOther := make(chan error, 1)
	go func() {
		errOther <- b.Wait(context.Background())
	}()

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Millisecond)
	defer cancel()

	err := b.Wait(ctx)
	if !errors.Is(err, context.DeadlineExceeded) {
		t.Fatalf("expected cancellation error, got %v", err)
	}

	err2 := <-errOther
	if !errors.Is(err2, ErrBarrierBroken) {
		t.Fatalf("expected broken barrier, got %v", err2)
	}
}