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

protobuf: make toStatusProto return null when metadata key is absent

This commit is contained in:
Eric Gribkoff 2017-05-30 11:47:20 -07:00 committed by GitHub
parent 917f6faac6
commit 37e2131dc9
2 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
protobuf/src/main/java/io/grpc/protobuf/StatusProto.java
View File

@ -160,10 +160,12 @@ public final class StatusProto {
private static com.google.rpc.Status toStatusProto(Status status, Metadata trailers) {
if (trailers != null) {
com.google.rpc.Status statusProto = trailers.get(STATUS_DETAILS_KEY);
checkArgument(
status.getCode().value() == statusProto.getCode(),
"com.google.rpc.Status code must match gRPC status code");
return statusProto;
if (statusProto != null) {
checkArgument(
status.getCode().value() == statusProto.getCode(),
"com.google.rpc.Status code must match gRPC status code");
return statusProto;
}
}
return null;
}

9
protobuf/src/test/java/io/grpc/protobuf/StatusProtoTest.java
View File

@ -148,6 +148,15 @@ public class StatusProtoTest {
assertNull(StatusProto.fromThrowable(status.asException()));
}
@Test
public void fromThrowable_shouldReturnNullIfStatusDetailsKeyIsMissing() {
Status status = Status.fromCodeValue(0);
Metadata emptyMetadata = new Metadata();
assertNull(StatusProto.fromThrowable(status.asRuntimeException(emptyMetadata)));
assertNull(StatusProto.fromThrowable(status.asException(emptyMetadata)));
}
@Test
public void fromThrowableWithNestedStatusRuntimeException() {
StatusRuntimeException sre = StatusProto.toStatusRuntimeException(STATUS_PROTO);