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

Implement timeout_on_sleeping_server test

This commit is contained in:
Xiao Hang 2015-08-27 14:29:32 -07:00
parent edbb237e84
commit 3c79c52427
1 changed files with 34 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
interop-testing/src/main/java/io/grpc/testing/integration/AbstractTransportTest.java
View File

@ -710,6 +710,40 @@ public abstract class AbstractTransportTest {
verifyNoMoreInteractions(responseObserver); verifyNoMoreInteractions(responseObserver);
} }
/** Sends an rpc to an unimplemented method on the server. */
@Test(timeout = 10000)
public void unimplementedMethod() {
UnimplementedServiceGrpc.UnimplementedServiceBlockingStub stub =
UnimplementedServiceGrpc.newBlockingStub(channel);
try {
stub.unimplementedCall(Empty.getDefaultInstance());
fail();
} catch (StatusRuntimeException e) {
assertEquals(Status.UNIMPLEMENTED.getCode(), e.getStatus().getCode());
}
}
/** Start a fullDuplexCall which the server will not respond, and verify the deadline expires. */
@Test(timeout = 10000)
public void timeoutOnSleepingServer() {
TestServiceGrpc.TestService stub = TestServiceGrpc.newStub(channel)
.withDeadlineAfter(1, TimeUnit.MILLISECONDS);
@SuppressWarnings("unchecked")
StreamObserver<StreamingOutputCallResponse> responseObserver = mock(StreamObserver.class);
StreamObserver<StreamingOutputCallRequest> requestObserver
= stub.fullDuplexCall(responseObserver);
requestObserver.onNext(StreamingOutputCallRequest.newBuilder()
.setPayload(Payload.newBuilder()
.setBody(ByteString.copyFrom(new byte[27182])))
.build());
ArgumentCaptor<Throwable> captor = ArgumentCaptor.forClass(Throwable.class);
verify(responseObserver, timeout(OPERATION_TIMEOUT)).onError(captor.capture());
assertEquals(Status.DEADLINE_EXCEEDED.getCode(),
Status.fromThrowable(captor.getValue()).getCode());
verifyNoMoreInteractions(responseObserver);
}
/** Sends a large unary rpc with service account credentials. */ /** Sends a large unary rpc with service account credentials. */
public void serviceAccountCreds(String jsonKey, InputStream credentialsStream, String authScope) public void serviceAccountCreds(String jsonKey, InputStream credentialsStream, String authScope)
throws Exception { throws Exception {
@ -813,19 +847,6 @@ public abstract class AbstractTransportTest {
authScope.contains(response.getOauthScope())); authScope.contains(response.getOauthScope()));
} }
/** Sends an rpc to an unimplemented method on the server. */
@Test(timeout = 10000)
public void unimplementedMethod() {
UnimplementedServiceGrpc.UnimplementedServiceBlockingStub stub =
UnimplementedServiceGrpc.newBlockingStub(channel);
try {
stub.unimplementedCall(Empty.getDefaultInstance());
fail();
} catch (StatusRuntimeException e) {
assertEquals(Status.UNIMPLEMENTED.getCode(), e.getStatus().getCode());
}
}
protected static void assertSuccess(StreamRecorder<?> recorder) { protected static void assertSuccess(StreamRecorder<?> recorder) {
if (recorder.getError() != null) { if (recorder.getError() != null) {
throw new AssertionError(recorder.getError()); throw new AssertionError(recorder.getError());