Ch1

internal/workqueue/limiter_test.go

@@ -0,0 +1,43 @@
+package workqueue
+
+import (
+	"context"
+	"errors"
+	"testing"
+	"time"
+)
+
+func TestLimiterBoundsQueue(t *testing.T) {
+	l := New(1, 1)
+	release, err := l.Acquire(context.Background())
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer release()
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+	waiterDone := make(chan error, 1)
+	go func() {
+		r, err := l.Acquire(ctx)
+		if err == nil {
+			r()
+		}
+		waiterDone <- err
+	}()
+
+	deadline := time.Now().Add(time.Second)
+	for l.Status().Waiting != 1 && time.Now().Before(deadline) {
+		time.Sleep(time.Millisecond)
+	}
+	if l.Status().Waiting != 1 {
+		t.Fatalf("expected one waiter: %+v", l.Status())
+	}
+	if _, err := l.Acquire(context.Background()); !errors.Is(err, ErrQueueFull) {
+		t.Fatalf("expected ErrQueueFull, got %v", err)
+	}
+	release()
+	if err := <-waiterDone; err != nil {
+		t.Fatalf("waiter failed: %v", err)
+	}
+}