Function verifyRLSRequest

balancer/rls/helpers_test.go:283–299

verifyRLSRequest is a test helper which listens on a channel to see if an RLS request was received by the fake RLS server. Based on whether the test expects a request to be sent out or not, it uses a different timeout.

(t *testing.T, ch chan struct{}, wantRequest bool)

// request was received by the fake RLS server. Based on whether the test
// expects a request to be sent out or not, it uses a different timeout.
func verifyRLSRequest(t *testing.T, ch chan struct{}, wantRequest bool) {
	t.Helper()

	if wantRequest {
		select {
		case <-time.After(defaultTestTimeout):
			t.Fatalf("Timeout when waiting for an RLS request to be sent out")
		case <-ch:
		}
	} else {
		select {
		case <-time.After(defaultTestShortTimeout):
		case <-ch:
			t.Fatalf("RLS request sent out when not expecting one")
		}
	}
}