44 lines
879 B
Go
44 lines
879 B
Go
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)
|
|
}
|
|
}
|