Function oneTimeAllowingThrottler

balancer/rls/helpers_test.go:106–111

oneTimeAllowingThrottler returns a fake throttler which does not throttle requests until the client RPC succeeds, but throttles everything that comes after. This is useful for tests which need to set up a valid cache entry before testing other cases.

(firstRPCDone *grpcsync.Event)

// after. This is useful for tests which need to set up a valid cache entry
// before testing other cases.
func oneTimeAllowingThrottler(firstRPCDone *grpcsync.Event) *fakeThrottler {
	return &fakeThrottler{
		throttleFunc: firstRPCDone.HasFired,
		throttleCh:   make(chan struct{}, 1),
	}
}

func overrideAdaptiveThrottler(t *testing.T, f *fakeThrottler) {
	origAdaptiveThrottler := newAdaptiveThrottler