Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
1.7 kB · 70 lines
Go
at main
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071package pluginhost
import ( "context" "sync/atomic" "testing" "time")
type blockingGuardPluginClient struct { started chan struct{} release chan struct{} shutdown atomic.Int32}
func (c *blockingGuardPluginClient) Call(context.Context, string, []byte) ([]byte, error) { close(c.started) <-c.release return nil, nil}
func (c *blockingGuardPluginClient) Shutdown() { c.shutdown.Add(1)}
func TestGuardedPluginClientShutdownContextDetachesBlockedCall(t *testing.T) { inner := &blockingGuardPluginClient{started: make(chan struct{}), release: make(chan struct{})} guarded := newGuardedPluginClient(inner)
callDone := make(chan struct{}) go func() { _, _ = guarded.Call(context.Background(), "blocked", nil) close(callDone) }() select { case <-inner.started: case <-time.After(time.Second): t.Fatal("guarded call did not start") }
shutdownCtx, cancelShutdown := context.WithCancel(context.Background()) cancelShutdown() shutdownDone := make(chan struct{}) go func() { guarded.ShutdownContext(shutdownCtx) close(shutdownDone) }() select { case <-shutdownDone: case <-time.After(time.Second): t.Fatal("context-canceled guarded shutdown waited for the active call") } if got := inner.shutdown.Load(); got != 0 { t.Fatalf("shutdown calls before active call exits = %d, want 0", got) }
close(inner.release) select { case <-callDone: case <-time.After(time.Second): t.Fatal("guarded call did not exit") } deadline := time.Now().Add(time.Second) for inner.shutdown.Load() == 0 && time.Now().Before(deadline) { time.Sleep(time.Millisecond) } if got := inner.shutdown.Load(); got != 1 { t.Fatalf("shutdown calls after active call exits = %d, want 1", got) }}