From 67dfe49ae613d32a0d174bb95b9ab7481a4b32f0 Mon Sep 17 00:00:00 2001
From: Larry Safran <lsafran@google.com>
Date: Thu, 15 Feb 2024 20:07:29 -0800
Subject: [PATCH] Fix compile failure

---
 .../grpc/util/MultiChildLoadBalancerTest.java | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/util/src/test/java/io/grpc/util/MultiChildLoadBalancerTest.java b/util/src/test/java/io/grpc/util/MultiChildLoadBalancerTest.java
index fd47261539..d90c5eab92 100644
--- a/util/src/test/java/io/grpc/util/MultiChildLoadBalancerTest.java
+++ b/util/src/test/java/io/grpc/util/MultiChildLoadBalancerTest.java
@@ -337,14 +337,27 @@ public class MultiChildLoadBalancerTest {
   }
 
   private class TestLb extends MultiChildLoadBalancer {
-
     protected TestLb(Helper mockHelper) {
       super(mockHelper);
     }
 
     @Override
-    protected SubchannelPicker getSubchannelPicker(Map<Object, SubchannelPicker> childPickers) {
-      return new TestSubchannelPicker(childPickers);
+    protected void updateOverallBalancingState() {
+      ConnectivityState overallState = null;
+      final Map<Object, SubchannelPicker> childPickers = new HashMap<>();
+      for (ChildLbState childLbState : getChildLbStates()) {
+        if (childLbState.isDeactivated()) {
+          continue;
+        }
+        childPickers.put(childLbState.getKey(), childLbState.getCurrentPicker());
+        overallState = aggregateState(overallState, childLbState.getCurrentState());
+      }
+
+      if (overallState != null) {
+        getHelper().updateBalancingState(overallState, new TestSubchannelPicker(childPickers));
+        currentConnectivityState = overallState;
+      }
+
     }
 
     private class TestSubchannelPicker extends SubchannelPicker {