Search in sources :

Example 6 with UrlResponseInfoImpl

use of org.chromium.net.impl.UrlResponseInfoImpl in project grpc-java by grpc.

the class CronetClientStreamTest method streamSucceeded.

@Test
public void streamSucceeded() {
    ArgumentCaptor<BidirectionalStream.Callback> callbackCaptor = ArgumentCaptor.forClass(BidirectionalStream.Callback.class);
    verify(factory).newBidirectionalStreamBuilder(isA(String.class), callbackCaptor.capture(), isA(Executor.class));
    BidirectionalStream.Callback callback = callbackCaptor.getValue();
    callback.onStreamReady(cronetStream);
    verify(cronetStream, times(0)).write(isA(ByteBuffer.class), isA(Boolean.class));
    // Send the first data frame.
    CronetWritableBufferAllocator allocator = new CronetWritableBufferAllocator();
    String request = new String("request");
    WritableBuffer writableBuffer = allocator.allocate(request.length());
    writableBuffer.write(request.getBytes(Charset.forName("UTF-8")), 0, request.length());
    clientStream.abstractClientStreamSink().writeFrame(writableBuffer, false, true, 1);
    ArgumentCaptor<ByteBuffer> bufferCaptor = ArgumentCaptor.forClass(ByteBuffer.class);
    verify(cronetStream, times(1)).write(bufferCaptor.capture(), isA(Boolean.class));
    ByteBuffer buffer = bufferCaptor.getValue();
    ((Buffer) buffer).position(request.length());
    verify(cronetStream, times(1)).flush();
    // Receive response header
    clientStream.request(2);
    UrlResponseInfo info = new UrlResponseInfoImpl(new ArrayList<>(), 200, "", responseHeader("200"), false, "", "", 0);
    callback.onResponseHeadersReceived(cronetStream, info);
    verify(cronetStream, times(1)).read(isA(ByteBuffer.class));
    // Receive one message
    callback.onReadCompleted(cronetStream, info, createMessageFrame(new String("response").getBytes(Charset.forName("UTF-8"))), false);
    verify(clientListener, times(1)).messagesAvailable(isA(MessageProducer.class));
    verify(cronetStream, times(2)).read(isA(ByteBuffer.class));
    // Send endOfStream
    callback.onWriteCompleted(cronetStream, null, buffer, false);
    clientStream.abstractClientStreamSink().writeFrame(null, true, true, 1);
    verify(cronetStream, times(2)).write(isA(ByteBuffer.class), isA(Boolean.class));
    verify(cronetStream, times(2)).flush();
    // Receive trailer
    ((CronetClientStream.BidirectionalStreamCallback) callback).processTrailers(trailers(0));
    callback.onSucceeded(cronetStream, info);
    // Verify trailer
    ArgumentCaptor<Metadata> trailerCaptor = ArgumentCaptor.forClass(Metadata.class);
    ArgumentCaptor<Status> statusCaptor = ArgumentCaptor.forClass(Status.class);
    verify(clientListener).closed(statusCaptor.capture(), isA(RpcProgress.class), trailerCaptor.capture());
    // Verify recevied headers.
    Metadata trailers = trailerCaptor.getValue();
    Status status = statusCaptor.getValue();
    assertEquals("test-trailer-value", trailers.get(Metadata.Key.of("test-trailer-key", Metadata.ASCII_STRING_MARSHALLER)));
    assertEquals("application/grpc", trailers.get(Metadata.Key.of("content-type", Metadata.ASCII_STRING_MARSHALLER)));
    assertTrue(status.isOk());
}
Also used : ByteBuffer(java.nio.ByteBuffer) Buffer(java.nio.Buffer) WritableBuffer(io.grpc.internal.WritableBuffer) Status(io.grpc.Status) Metadata(io.grpc.Metadata) UrlResponseInfo(org.chromium.net.UrlResponseInfo) ByteBuffer(java.nio.ByteBuffer) BidirectionalStream(org.chromium.net.BidirectionalStream) ExperimentalBidirectionalStream(org.chromium.net.ExperimentalBidirectionalStream) WritableBuffer(io.grpc.internal.WritableBuffer) Executor(java.util.concurrent.Executor) UrlResponseInfoImpl(org.chromium.net.impl.UrlResponseInfoImpl) RpcProgress(io.grpc.internal.ClientStreamListener.RpcProgress) MessageProducer(io.grpc.internal.StreamListener.MessageProducer) Test(org.junit.Test)

Example 7 with UrlResponseInfoImpl

use of org.chromium.net.impl.UrlResponseInfoImpl in project grpc-java by grpc.

the class CronetClientStreamTest method reportTrailersWhenTrailersReceivedBeforeReadClosed.

@Test
public void reportTrailersWhenTrailersReceivedBeforeReadClosed() {
    ArgumentCaptor<BidirectionalStream.Callback> callbackCaptor = ArgumentCaptor.forClass(BidirectionalStream.Callback.class);
    verify(factory).newBidirectionalStreamBuilder(isA(String.class), callbackCaptor.capture(), isA(Executor.class));
    BidirectionalStream.Callback callback = callbackCaptor.getValue();
    callback.onStreamReady(cronetStream);
    UrlResponseInfo info = new UrlResponseInfoImpl(new ArrayList<>(), 200, "", responseHeader("200"), false, "", "", 0);
    callback.onResponseHeadersReceived(cronetStream, info);
    // Receive trailer first
    ((CronetClientStream.BidirectionalStreamCallback) callback).processTrailers(trailers(Status.UNAUTHENTICATED.getCode().value()));
    verify(clientListener, times(0)).closed(isA(Status.class), isA(RpcProgress.class), isA(Metadata.class));
    // Receive cronet's endOfStream
    callback.onReadCompleted(cronetStream, null, ByteBuffer.allocate(0), true);
    ArgumentCaptor<Status> statusCaptor = ArgumentCaptor.forClass(Status.class);
    verify(clientListener, times(1)).closed(statusCaptor.capture(), isA(RpcProgress.class), isA(Metadata.class));
    Status status = statusCaptor.getValue();
    assertEquals(Status.UNAUTHENTICATED.getCode(), status.getCode());
}
Also used : Status(io.grpc.Status) Metadata(io.grpc.Metadata) UrlResponseInfo(org.chromium.net.UrlResponseInfo) BidirectionalStream(org.chromium.net.BidirectionalStream) ExperimentalBidirectionalStream(org.chromium.net.ExperimentalBidirectionalStream) Executor(java.util.concurrent.Executor) UrlResponseInfoImpl(org.chromium.net.impl.UrlResponseInfoImpl) RpcProgress(io.grpc.internal.ClientStreamListener.RpcProgress) Test(org.junit.Test)

Example 8 with UrlResponseInfoImpl

use of org.chromium.net.impl.UrlResponseInfoImpl in project grpc-java by grpc.

the class CronetClientStreamTest method streamSucceededWithGrpcError.

@Test
public void streamSucceededWithGrpcError() {
    ArgumentCaptor<BidirectionalStream.Callback> callbackCaptor = ArgumentCaptor.forClass(BidirectionalStream.Callback.class);
    verify(factory).newBidirectionalStreamBuilder(isA(String.class), callbackCaptor.capture(), isA(Executor.class));
    BidirectionalStream.Callback callback = callbackCaptor.getValue();
    callback.onStreamReady(cronetStream);
    verify(cronetStream, times(0)).write(isA(ByteBuffer.class), isA(Boolean.class));
    clientStream.abstractClientStreamSink().writeFrame(null, true, true, 1);
    verify(cronetStream, times(1)).write(isA(ByteBuffer.class), isA(Boolean.class));
    verify(cronetStream, times(1)).flush();
    // Receive response header
    clientStream.request(2);
    UrlResponseInfo info = new UrlResponseInfoImpl(new ArrayList<>(), 200, "", responseHeader("200"), false, "", "", 0);
    callback.onResponseHeadersReceived(cronetStream, info);
    verify(cronetStream, times(1)).read(isA(ByteBuffer.class));
    // Receive trailer
    callback.onReadCompleted(cronetStream, null, ByteBuffer.allocate(0), true);
    ((CronetClientStream.BidirectionalStreamCallback) callback).processTrailers(trailers(Status.PERMISSION_DENIED.getCode().value()));
    callback.onSucceeded(cronetStream, info);
    ArgumentCaptor<Status> statusCaptor = ArgumentCaptor.forClass(Status.class);
    verify(clientListener).closed(statusCaptor.capture(), isA(RpcProgress.class), isA(Metadata.class));
    // Verify error status.
    Status status = statusCaptor.getValue();
    assertFalse(status.isOk());
    assertEquals(Status.PERMISSION_DENIED.getCode(), status.getCode());
}
Also used : Status(io.grpc.Status) Metadata(io.grpc.Metadata) UrlResponseInfo(org.chromium.net.UrlResponseInfo) ByteBuffer(java.nio.ByteBuffer) BidirectionalStream(org.chromium.net.BidirectionalStream) ExperimentalBidirectionalStream(org.chromium.net.ExperimentalBidirectionalStream) Executor(java.util.concurrent.Executor) UrlResponseInfoImpl(org.chromium.net.impl.UrlResponseInfoImpl) RpcProgress(io.grpc.internal.ClientStreamListener.RpcProgress) Test(org.junit.Test)

Aggregations

Metadata (io.grpc.Metadata)8 Executor (java.util.concurrent.Executor)8 BidirectionalStream (org.chromium.net.BidirectionalStream)8 ExperimentalBidirectionalStream (org.chromium.net.ExperimentalBidirectionalStream)8 UrlResponseInfo (org.chromium.net.UrlResponseInfo)8 UrlResponseInfoImpl (org.chromium.net.impl.UrlResponseInfoImpl)8 Test (org.junit.Test)8 Status (io.grpc.Status)7 RpcProgress (io.grpc.internal.ClientStreamListener.RpcProgress)7 ByteBuffer (java.nio.ByteBuffer)3 CronetException (org.chromium.net.CronetException)3 MessageProducer (io.grpc.internal.StreamListener.MessageProducer)2 WritableBuffer (io.grpc.internal.WritableBuffer)1 Buffer (java.nio.Buffer)1