[xds_eds_endpoint_health] check new sc addrs

grpc · Nov 4, 2019 · 623a434 · 623a434
1 parent 4c225d3
commit 623a434
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/xds/internal/balancer/edsbalancer/edsbalancer_test.go b/xds/internal/balancer/edsbalancer/edsbalancer_test.go
@@ -409,8 +409,21 @@ func TestEDS_EndpointsHealth(t *testing.T) {
 	})
 	edsb.HandleEDSResponse(clab1.build())
 
-	var readySCs []balancer.SubConn
+	var (
+		readySCs []balancer.SubConn
+
+		wantNewSubConnAddrStrs = []string{
+			testEndpointAddrs[0],
+			testEndpointAddrs[2],
+			testEndpointAddrs[6],
+			testEndpointAddrs[8],
+		}
+	)
 	for i := 0; i < 4; i++ {
+		addr := <-cc.newSubConnAddrsCh
+		if addr[0].Addr != wantNewSubConnAddrStrs[i] {
+			t.Fatalf("want newSubConn with address %q, got %v", wantNewSubConnAddrStrs[i], addr)
+		}
 		sc := <-cc.newSubConnCh
 		edsb.HandleSubConnStateChange(sc, connectivity.Connecting)
 		edsb.HandleSubConnStateChange(sc, connectivity.Ready)