grpc
/
grpc-java
mirror of https://github.com/grpc/grpc-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

testing: RpcBehaviorLoadBalancingProvider to use acceptResolvedAddresses() (#10030)

This commit is contained in:
Terry Wilson 2023-04-10 12:52:43 -07:00 committed by GitHub
parent 5201e49ce1
commit 17e1fcb393
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
interop-testing/src/main/java/io/grpc/testing/integration/RpcBehaviorLoadBalancerProvider.java
View File

@ -111,10 +111,10 @@ public class RpcBehaviorLoadBalancerProvider extends LoadBalancerProvider {
}
@Override
public void handleResolvedAddresses(ResolvedAddresses resolvedAddresses) {
public boolean acceptResolvedAddresses(ResolvedAddresses resolvedAddresses) {
helper.setRpcBehavior(
((RpcBehaviorConfig) resolvedAddresses.getLoadBalancingPolicyConfig()).rpcBehavior);
delegateLb.handleResolvedAddresses(resolvedAddresses);
return delegateLb.acceptResolvedAddresses(resolvedAddresses);
}
}

6
interop-testing/src/test/java/io/grpc/testing/integration/RpcBehaviorLoadBalancerProviderTest.java
View File

@ -79,12 +79,12 @@ public class RpcBehaviorLoadBalancerProviderTest {
}
@Test
public void handleResolvedAddressesDelegated() {
public void acceptResolvedAddressesDelegated() {
RpcBehaviorLoadBalancer lb = new RpcBehaviorLoadBalancer(new RpcBehaviorHelper(mockHelper),
mockDelegateLb);
ResolvedAddresses resolvedAddresses = buildResolvedAddresses(buildConfig());
lb.handleResolvedAddresses(resolvedAddresses);
verify(mockDelegateLb).handleResolvedAddresses(resolvedAddresses);
assertThat(lb.acceptResolvedAddresses(resolvedAddresses)).isFalse();
verify(mockDelegateLb).acceptResolvedAddresses(resolvedAddresses);
}
@Test