34 lines
538 B
Go
34 lines
538 B
Go
|
package proxy
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestProxy_Stop_WaitsForSubmitDrain(t *testing.T) {
|
||
|
|
p := &Proxy{
|
||
|
|
done: make(chan struct{}),
|
||
|
|
}
|
||
|
|
p.submitCount.Store(1)
|
||
|
|
|
||
|
|
stopped := make(chan struct{})
|
||
|
|
go func() {
|
||
|
|
p.Stop()
|
||
|
|
close(stopped)
|
||
|
|
}()
|
||
|
|
|
||
|
|
select {
|
||
|
|
case <-stopped:
|
||
|
|
t.Fatalf("expected Stop to wait for pending submits")
|
||
|
|
case <-time.After(50 * time.Millisecond):
|
||
|
|
}
|
||
|
|
|
||
|
|
p.submitCount.Store(0)
|
||
|
|
|
||
|
|
select {
|
||
|
|
case <-stopped:
|
||
|
|
case <-time.After(time.Second):
|
||
|
|
t.Fatalf("expected Stop to finish after pending submits drain")
|
||
|
|
}
|
||
|
|
}
|