Class RlsLoadBalancerTest

rls/src/test/java/io/grpc/rls/RlsLoadBalancerTest.java:114–934

import org.mockito.junit.MockitoRule;

@RunWith(JUnit4.class)
public class RlsLoadBalancerTest {

  @Rule
  public final GrpcCleanupRule grpcCleanupRule = new GrpcCleanupRule();
  @Rule
  public final MockitoRule mocks = MockitoJUnit.rule();
  private final RlsLoadBalancerProvider provider = new RlsLoadBalancerProvider();
  private final FakeClock fakeClock = new FakeClock();
  private final SynchronizationContext syncContext =
      new SynchronizationContext(new UncaughtExceptionHandler() {
        @Override
        public void uncaughtException(Thread t, Throwable e) {
          throw new RuntimeException(e);
        }
      });
  @Mock
  private MetricRecorder mockMetricRecorder;
  @Mock
  private Registration mockGaugeRegistration;
  private final FakeHelper helperDelegate = new FakeHelper();
  private final Helper helper =
      mock(Helper.class, AdditionalAnswers.delegatesTo(helperDelegate));
  private final FakeRlsServerImpl fakeRlsServerImpl = new FakeRlsServerImpl(
      fakeClock.getScheduledExecutorService());
  private final Deque<FakeSubchannel> subchannels = new LinkedList<>();
  private final FakeThrottler fakeThrottler = new FakeThrottler();
  private final String channelTarget = "channelTarget";
  @Captor
  private ArgumentCaptor<SubchannelPicker> pickerCaptor;
  private MethodDescriptor<Void, Void> fakeSearchMethod;
  private MethodDescriptor<Void, Void> fakeRescueMethod;
  private RlsLoadBalancer rlsLb;
  private String defaultTarget = "defaultTarget";
  private PickSubchannelArgs searchSubchannelArgs;
  private PickSubchannelArgs rescueSubchannelArgs;

  @Before
  public void setUp() {
    fakeSearchMethod =
        MethodDescriptor.<Void, Void>newBuilder()
            .setFullMethodName("com.google/Search")
            .setRequestMarshaller(TestMethodDescriptors.voidMarshaller())
            .setResponseMarshaller(TestMethodDescriptors.voidMarshaller())
            .setType(MethodType.CLIENT_STREAMING)
            .build();
    fakeRescueMethod =
        MethodDescriptor.<Void, Void>newBuilder()
            .setFullMethodName("com.google/Rescue")
            .setRequestMarshaller(TestMethodDescriptors.voidMarshaller())
            .setResponseMarshaller(TestMethodDescriptors.voidMarshaller())
            .setType(MethodType.UNARY)
            .build();
    fakeRlsServerImpl.setLookupTable(
        ImmutableMap.of(
            RouteLookupRequest.create(
                ImmutableMap.of(
                "server", "fake-bigtable.googleapis.com",