wehub-resource-sync
44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
package plugin
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// TestWaitWithBudgetReturnsWithinBudget pins that the budgeted reap helper
|
|
// returns even when the underlying wait never completes. withStderr calls this
|
|
// while holding callMu, and a surviving grandchild can keep cmd.Wait blocked
|
|
// forever — without the budget every future call on the transport would wedge.
|
|
func TestWaitWithBudgetReturnsWithinBudget(t *testing.T) {
|
|
blocked := make(chan struct{})
|
|
t.Cleanup(func() { close(blocked) }) // let the abandoned goroutine exit
|
|
|
|
done := make(chan struct{})
|
|
go func() {
|
|
waitWithBudget(func() { <-blocked }, 100*time.Millisecond)
|
|
close(done)
|
|
}()
|
|
|
|
select {
|
|
case <-done:
|
|
case <-time.After(2 * time.Second):
|
|
t.Fatal("waitWithBudget did not return within its budget — callMu would wedge")
|
|
}
|
|
}
|
|
|
|
// TestWaitWithBudgetReturnsEarlyWhenWaitCompletes pins that a fast wait returns
|
|
// promptly rather than always paying the full budget.
|
|
func TestWaitWithBudgetReturnsEarlyWhenWaitCompletes(t *testing.T) {
|
|
done := make(chan struct{})
|
|
go func() {
|
|
waitWithBudget(func() {}, 10*time.Second)
|
|
close(done)
|
|
}()
|
|
|
|
select {
|
|
case <-done:
|
|
case <-time.After(2 * time.Second):
|
|
t.Fatal("waitWithBudget blocked on a wait that already completed")
|
|
}
|
|
}
|