@@ -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