Skip to content

Commit bdcb745

Browse files
committed
Fix: test cases
1 parent f816bba commit bdcb745

1 file changed

Lines changed: 12 additions & 3 deletions

File tree

core/src/test/java/io/grpc/internal/ManagedChannelImplBuilderTest.java

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -820,8 +820,13 @@ public void configureChannelBuilder(ManagedChannelBuilder<?> builder) {
820820
NameResolver mockNameResolver = mock(NameResolver.class);
821821
when(mockNameResolver.getServiceAuthority()).thenReturn("foo.authority");
822822
ArgumentCaptor<NameResolver.Args> argsCaptor = ArgumentCaptor.forClass(NameResolver.Args.class);
823-
when(mockNameResolverFactory.newNameResolver((URI) any(),
824-
argsCaptor.capture())).thenReturn(mockNameResolver);
823+
if (enableRfc3986UrisParam) {
824+
when(mockNameResolverFactory.newNameResolver((io.grpc.Uri) any(),
825+
argsCaptor.capture())).thenReturn(mockNameResolver);
826+
} else {
827+
when(mockNameResolverFactory.newNameResolver((URI) any(),
828+
argsCaptor.capture())).thenReturn(mockNameResolver);
829+
}
825830

826831
// Use the configurator and the mock factory
827832
NameResolverRegistry registry = new NameResolverRegistry();
@@ -838,7 +843,11 @@ public void configureChannelBuilder(ManagedChannelBuilder<?> builder) {
838843
grpcCleanupRule.register(channel);
839844

840845
// Verify that newNameResolver was called
841-
verify(mockNameResolverFactory).newNameResolver((URI) any(), any());
846+
if (enableRfc3986UrisParam) {
847+
verify(mockNameResolverFactory).newNameResolver((io.grpc.Uri) any(), any());
848+
} else {
849+
verify(mockNameResolverFactory).newNameResolver((URI) any(), any());
850+
}
842851

843852
// Extract the childChannelConfigurator from Args
844853
NameResolver.Args args = argsCaptor.getValue();

0 commit comments

Comments
 (0)