Method TestEndpointUpdate

interop/orcalb_test.go:283–348

TestEndpointUpdate verifies client behavior in response to a resolver update that changes an endpoint's address. It ensures that the client disconnects from the old address and connects to the new address via pick_first. It also validates that the OOB listener for the old connection is stopped, and

(t *testing.T)

// and a new OOB listener is registered for the new connection once it becomes
// READY.
func (s) TestEndpointUpdate(t *testing.T) {
	ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
	defer cancel()

	srvA := startORCAServer(t)
	srvB := startORCAServer(t)

	r := manual.NewBuilderWithScheme("test")
	cc, err := grpc.Dial(r.Scheme()+":///test",
		grpc.WithTransportCredentials(insecure.NewCredentials()),
		grpc.WithResolvers(r),
		grpc.WithDefaultServiceConfig(orcaSvcConfig()),
	)
	if err != nil {
		t.Fatalf("Failed to dial: %v", err)
	}
	defer cc.Close()
	tc := testgrpc.NewTestServiceClient(cc)

	state := resolver.State{
		Endpoints: []resolver.Endpoint{
			{Addresses: []resolver.Address{{Addr: srvA.Address}}},
		},
	}
	r.UpdateState(state)

	orcaRes := &v3orcapb.OrcaLoadReport{}
	req := &testpb.SimpleRequest{
		OrcaPerQueryReport: &testpb.TestOrcaReport{
			CpuUtilization:    0.11,
			MemoryUtilization: 0.22,
		},
	}
	if _, err := tc.UnaryCall(contextWithORCAResult(ctx, &orcaRes), req); err != nil {
		t.Fatalf("UnaryCall to srvA failed: %v", err)
	}
	wantA := &v3orcapb.OrcaLoadReport{CpuUtilization: 0.11, MemUtilization: 0.22}
	if diff := cmp.Diff(orcaRes, wantA, protocmp.Transform()); diff != "" {
		t.Fatalf("ORCA from srvA mismatch (-got +want):\n%s", diff)
	}

	// Distinctive OOB metric on srvB to confirm the switch via OOB fallback.
	srvB.metricsRecorder.SetCPUUtilization(0.77)

	state = resolver.State{
		Endpoints: []resolver.Endpoint{
			{Addresses: []resolver.Address{{Addr: srvB.Address}}},
		},
	}
	r.UpdateState(state)
	// No per-call report forces OOB fallback. Once srvB's metric appears,
	// all three are confirmed: pick_first reconnected, new OOB listener
	// registered for srvB, old one for srvA stopped.
	wantB := &v3orcapb.OrcaLoadReport{CpuUtilization: 0.77}
	for ; ctx.Err() == nil; <-time.After(100 * time.Millisecond) {
		orcaRes := &v3orcapb.OrcaLoadReport{}
		if _, err := tc.UnaryCall(contextWithORCAResult(ctx, &orcaRes), &testpb.SimpleRequest{}); err != nil {
			continue