|
46 | 46 | import io.grpc.NameResolver;
|
47 | 47 | import io.grpc.NameResolver.ConfigOrError;
|
48 | 48 | import io.grpc.Status;
|
| 49 | +import io.grpc.internal.ManagedChannelImplBuilder.ChannelBuilderDefaultPortProvider; |
| 50 | +import io.grpc.internal.ManagedChannelImplBuilder.ClientTransportFactoryBuilder; |
49 | 51 | import java.net.SocketAddress;
|
50 | 52 | import java.net.URI;
|
51 | 53 | import java.util.ArrayList;
|
@@ -153,7 +155,18 @@ public ConfigOrError parseLoadBalancingPolicyConfig(
|
153 | 155 | private ObjectPool<Executor> balancerRpcExecutorPool;
|
154 | 156 | @Mock
|
155 | 157 | private Executor blockingExecutor;
|
156 |
| - private ChannelBuilder channelBuilder; |
| 158 | + |
| 159 | + private ManagedChannelImplBuilder channelBuilder = new ManagedChannelImplBuilder(TARGET, |
| 160 | + new ClientTransportFactoryBuilder() { |
| 161 | + @Override public ClientTransportFactory buildClientTransportFactory() { |
| 162 | + throw new UnsupportedOperationException(); |
| 163 | + } |
| 164 | + }, |
| 165 | + new ChannelBuilderDefaultPortProvider() { |
| 166 | + @Override public int getDefaultPort() { |
| 167 | + return DEFAULT_PORT; |
| 168 | + } |
| 169 | + }); |
157 | 170 |
|
158 | 171 | private void createChannel(ClientInterceptor... interceptors) {
|
159 | 172 | checkState(channel == null);
|
@@ -199,14 +212,12 @@ public void setUp() throws Exception {
|
199 | 212 | .thenReturn(timer.getScheduledExecutorService());
|
200 | 213 | when(executorPool.getObject()).thenReturn(executor.getScheduledExecutorService());
|
201 | 214 |
|
202 |
| - channelBuilder = |
203 |
| - new ChannelBuilder() |
204 |
| - .nameResolverFactory(new FakeNameResolverFactory.Builder(expectedUri).build()) |
205 |
| - .defaultLoadBalancingPolicy(MOCK_POLICY_NAME) |
206 |
| - .userAgent(USER_AGENT) |
207 |
| - .idleTimeout( |
208 |
| - AbstractManagedChannelImplBuilder.IDLE_MODE_MAX_TIMEOUT_DAYS, TimeUnit.DAYS) |
209 |
| - .offloadExecutor(blockingExecutor); |
| 215 | + channelBuilder |
| 216 | + .nameResolverFactory(new FakeNameResolverFactory.Builder(expectedUri).build()) |
| 217 | + .defaultLoadBalancingPolicy(MOCK_POLICY_NAME) |
| 218 | + .userAgent(USER_AGENT) |
| 219 | + .idleTimeout(ManagedChannelImplBuilder.IDLE_MODE_MAX_TIMEOUT_DAYS, TimeUnit.DAYS) |
| 220 | + .offloadExecutor(blockingExecutor); |
210 | 221 | channelBuilder.executorPool = executorPool;
|
211 | 222 | channelBuilder.binlog = null;
|
212 | 223 | channelBuilder.channelz = channelz;
|
@@ -529,22 +540,6 @@ public void validConfig_thenNoConfig_withDefaultConfig() throws Exception {
|
529 | 540 | verify(mockLoadBalancer, never()).handleNameResolutionError(any(Status.class));
|
530 | 541 | }
|
531 | 542 |
|
532 |
| - private static final class ChannelBuilder |
533 |
| - extends AbstractManagedChannelImplBuilder<ChannelBuilder> { |
534 |
| - |
535 |
| - ChannelBuilder() { |
536 |
| - super(TARGET); |
537 |
| - } |
538 |
| - |
539 |
| - @Override protected ClientTransportFactory buildTransportFactory() { |
540 |
| - throw new UnsupportedOperationException(); |
541 |
| - } |
542 |
| - |
543 |
| - @Override protected int getDefaultPort() { |
544 |
| - return DEFAULT_PORT; |
545 |
| - } |
546 |
| - } |
547 |
| - |
548 | 543 | private static final class FakeBackoffPolicyProvider implements BackoffPolicy.Provider {
|
549 | 544 | @Override
|
550 | 545 | public BackoffPolicy get() {
|
|
0 commit comments