Search in sources :

Example 1 with MockResponseListener

use of org.webpieces.httpclient.api.mocks.MockResponseListener in project webpieces by deanhiller.

the class TestCancelStream method testClientCancelWithKeepAlive.

// @Test
// public void testRequestResponseXFutureCancelNoKeepAlive() {
// throw new UnsupportedOperationException("not done yet");
// }
// 
// @Test
// public void testRequestResponseXFutureCancelWithKeepAlive() {
// throw new UnsupportedOperationException("not done yet");
// }
@Test
public void testClientCancelWithKeepAlive() {
    XFuture<Void> connect = httpSocket.connect(new InetSocketAddress(8555));
    MockResponseListener mockListener = new MockResponseListener();
    Http2Request req = Requests.createRequest(false);
    req.addHeader(new Http2Header(Http2HeaderName.CONNECTION, "keep-alive"));
    mockChannel.addWriteResponse(XFuture.completedFuture(null));
    RequestStreamHandle requestStream = httpSocket.openStream();
    StreamRef ref = requestStream.process(req, mockListener);
    CancelReason reason = new RstStreamFrame();
    XFuture<Void> cancelDone = ref.cancel(reason);
    Assert.assertTrue(cancelDone.isDone());
    // Assert the socket is NOT closed
    Assert.assertFalse(mockChannel.isClosed());
}
Also used : RequestStreamHandle(com.webpieces.http2.api.streaming.RequestStreamHandle) StreamRef(com.webpieces.http2.api.streaming.StreamRef) CancelReason(com.webpieces.http2.api.dto.lowlevel.CancelReason) Http2Request(com.webpieces.http2.api.dto.highlevel.Http2Request) RstStreamFrame(com.webpieces.http2.api.dto.lowlevel.RstStreamFrame) InetSocketAddress(java.net.InetSocketAddress) Http2Header(com.webpieces.http2.api.dto.lowlevel.lib.Http2Header) MockResponseListener(org.webpieces.httpclient.api.mocks.MockResponseListener) Test(org.junit.Test)

Example 2 with MockResponseListener

use of org.webpieces.httpclient.api.mocks.MockResponseListener in project webpieces by deanhiller.

the class TestCancelStream method testServerCloseSocket.

@Test
public void testServerCloseSocket() {
    XFuture<Void> connect = httpSocket.connect(new InetSocketAddress(8555));
    MockResponseListener mockListener = new MockResponseListener();
    Http2Request req = Requests.createRequest(false);
    mockChannel.addWriteResponse(XFuture.completedFuture(null));
    RequestStreamHandle requestStream = httpSocket.openStream();
    StreamRef ref = requestStream.process(req, mockListener);
    Assert.assertFalse(mockListener.isCancelled());
    mockChannel.simulateClose();
    Assert.assertTrue(mockListener.isCancelled());
}
Also used : RequestStreamHandle(com.webpieces.http2.api.streaming.RequestStreamHandle) StreamRef(com.webpieces.http2.api.streaming.StreamRef) Http2Request(com.webpieces.http2.api.dto.highlevel.Http2Request) InetSocketAddress(java.net.InetSocketAddress) MockResponseListener(org.webpieces.httpclient.api.mocks.MockResponseListener) Test(org.junit.Test)

Example 3 with MockResponseListener

use of org.webpieces.httpclient.api.mocks.MockResponseListener in project webpieces by deanhiller.

the class TestCancelStream method testClientCancelNoKeepAlive.

@Test
public void testClientCancelNoKeepAlive() {
    XFuture<Void> connect = httpSocket.connect(new InetSocketAddress(8555));
    MockResponseListener mockListener = new MockResponseListener();
    Http2Request req = Requests.createRequest(false);
    mockChannel.addWriteResponse(XFuture.completedFuture(null));
    RequestStreamHandle requestStream = httpSocket.openStream();
    StreamRef ref = requestStream.process(req, mockListener);
    CancelReason reason = new RstStreamFrame();
    XFuture<Void> cancelDone = ref.cancel(reason);
    Assert.assertTrue(cancelDone.isDone());
    // Assert the socket is NOT closed
    Assert.assertTrue(mockChannel.isClosed());
}
Also used : RequestStreamHandle(com.webpieces.http2.api.streaming.RequestStreamHandle) StreamRef(com.webpieces.http2.api.streaming.StreamRef) CancelReason(com.webpieces.http2.api.dto.lowlevel.CancelReason) Http2Request(com.webpieces.http2.api.dto.highlevel.Http2Request) RstStreamFrame(com.webpieces.http2.api.dto.lowlevel.RstStreamFrame) InetSocketAddress(java.net.InetSocketAddress) MockResponseListener(org.webpieces.httpclient.api.mocks.MockResponseListener) Test(org.junit.Test)

Example 4 with MockResponseListener

use of org.webpieces.httpclient.api.mocks.MockResponseListener in project webpieces by deanhiller.

the class TestHttp1Backpressure method testBasicBackpressure.

@Test
public void testBasicBackpressure() throws InterruptedException, ExecutionException, TimeoutException {
    MockResponseListener listener = new MockResponseListener();
    RequestStreamHandle handle = socket.openStream();
    mockChannel.addWriteResponse(XFuture.completedFuture(null));
    Http2Request request = Requests.createRequest();
    StreamRef streamRef = handle.process(request, listener);
    XFuture<StreamWriter> writer = streamRef.getWriter();
    Assert.assertTrue(writer.isDone());
    Assert.assertEquals(request, mockChannel.getLastWriteParam());
    Http2Response response1 = Requests.createResponse(1, 250);
    HttpData response2 = Requests.createHttpData(250);
    List<ByteBuffer> buffers = create3BuffersWithTwoMessags(response1, response2);
    DataListener dataListener = mockChannel.getConnectedListener();
    XFuture<StreamWriter> requestFuture = new XFuture<StreamWriter>();
    listener.addProcessResponse(requestFuture);
    XFuture<Void> fut1 = dataListener.incomingData(mockChannel, buffers.get(0));
    // resolved since data is cached in parser (client doesn't receive this data)
    Assert.assertTrue(fut1.isDone());
    // This next one is confusing BUT in http1.1 parsing terms, data is data for content length so this results
    // in a full HttpData packet actually...
    XFuture<Void> fut2 = dataListener.incomingData(mockChannel, buffers.get(1));
    // not resolved yet since client only has part of the data
    Assert.assertFalse(fut2.isDone());
    MockStreamWriter mockWriter = new MockStreamWriter();
    XFuture<Void> streamWriterFuture = new XFuture<Void>();
    mockWriter.addProcessResponse(streamWriterFuture);
    // This releases BOTH packets above to be processed!!!(not just the one)
    requestFuture.complete(mockWriter);
    Assert.assertFalse(fut2.isDone());
    streamWriterFuture.complete(null);
    // NOW it's resolved
    fut2.get(2, TimeUnit.SECONDS);
    // feed the last buffer in
    XFuture<Void> future3 = new XFuture<Void>();
    mockWriter.addProcessResponse(future3);
    XFuture<Void> fut3 = dataListener.incomingData(mockChannel, buffers.get(2));
    Assert.assertFalse(fut3.isDone());
    future3.complete(null);
    fut3.get(2, TimeUnit.SECONDS);
}
Also used : Http2Response(com.webpieces.http2.api.dto.highlevel.Http2Response) XFuture(org.webpieces.util.futures.XFuture) MockStreamWriter(org.webpieces.httpclient.api.mocks.MockStreamWriter) StreamWriter(com.webpieces.http2.api.streaming.StreamWriter) MockResponseListener(org.webpieces.httpclient.api.mocks.MockResponseListener) ByteBuffer(java.nio.ByteBuffer) RequestStreamHandle(com.webpieces.http2.api.streaming.RequestStreamHandle) StreamRef(com.webpieces.http2.api.streaming.StreamRef) Http2Request(com.webpieces.http2.api.dto.highlevel.Http2Request) HttpData(org.webpieces.httpparser.api.dto.HttpData) DataListener(org.webpieces.nio.api.handlers.DataListener) MockStreamWriter(org.webpieces.httpclient.api.mocks.MockStreamWriter) Test(org.junit.Test)

Example 5 with MockResponseListener

use of org.webpieces.httpclient.api.mocks.MockResponseListener in project webpieces by deanhiller.

the class TestWriteReads method testBasicReadWrite.

@Test
public void testBasicReadWrite() throws InterruptedException, ExecutionException, TimeoutException {
    MockResponseListener listener = new MockResponseListener();
    RequestStreamHandle handle = socket.openStream();
    mockChannel.addWriteResponse(XFuture.completedFuture(null));
    Http2Request request = Requests.createRequest();
    StreamRef streamRef = handle.process(request, listener);
    XFuture<StreamWriter> writer = streamRef.getWriter();
    Assert.assertTrue(writer.isDone());
    Assert.assertEquals(request, mockChannel.getLastWriteParam());
    MockResponseListener listener2 = new MockResponseListener();
    request.getHeaderLookupStruct().getHeader("serverid").setValue("2");
    mockChannel.addWriteResponse(XFuture.completedFuture(null));
    StreamRef streamRef1 = handle.process(request, listener2);
    XFuture<StreamWriter> writer2 = streamRef1.getWriter();
    Assert.assertTrue(writer2.isDone());
    Assert.assertEquals(request, mockChannel.getLastWriteParam());
    Http2Response response1 = Requests.createResponse(1, 0);
    listener.addProcessResponse(XFuture.completedFuture(null));
    XFuture<Void> fut1 = mockChannel.writeResponse(response1);
    // throws if exception occurred and ensures future resolved
    fut1.get(2, TimeUnit.SECONDS);
    Http2Response msg = listener.getIncomingMsg();
    Assert.assertEquals(response1, msg);
    Http2Response response2 = Requests.createResponse(2, 0);
    listener2.addProcessResponse(XFuture.completedFuture(null));
    XFuture<Void> fut2 = mockChannel.writeResponse(response2);
    // throws if exception occurred and ensures future resolved
    fut2.get(2, TimeUnit.SECONDS);
    Http2Response msg2 = listener2.getIncomingMsg();
    Assert.assertEquals(response2, msg2);
}
Also used : RequestStreamHandle(com.webpieces.http2.api.streaming.RequestStreamHandle) Http2Response(com.webpieces.http2.api.dto.highlevel.Http2Response) StreamRef(com.webpieces.http2.api.streaming.StreamRef) Http2Request(com.webpieces.http2.api.dto.highlevel.Http2Request) StreamWriter(com.webpieces.http2.api.streaming.StreamWriter) MockResponseListener(org.webpieces.httpclient.api.mocks.MockResponseListener) Test(org.junit.Test)

Aggregations

Http2Request (com.webpieces.http2.api.dto.highlevel.Http2Request)6 RequestStreamHandle (com.webpieces.http2.api.streaming.RequestStreamHandle)6 StreamRef (com.webpieces.http2.api.streaming.StreamRef)6 Test (org.junit.Test)6 MockResponseListener (org.webpieces.httpclient.api.mocks.MockResponseListener)6 Http2Response (com.webpieces.http2.api.dto.highlevel.Http2Response)3 StreamWriter (com.webpieces.http2.api.streaming.StreamWriter)3 InetSocketAddress (java.net.InetSocketAddress)3 CancelReason (com.webpieces.http2.api.dto.lowlevel.CancelReason)2 RstStreamFrame (com.webpieces.http2.api.dto.lowlevel.RstStreamFrame)2 ByteBuffer (java.nio.ByteBuffer)2 MockStreamWriter (org.webpieces.httpclient.api.mocks.MockStreamWriter)2 DataListener (org.webpieces.nio.api.handlers.DataListener)2 XFuture (org.webpieces.util.futures.XFuture)2 Http2Header (com.webpieces.http2.api.dto.lowlevel.lib.Http2Header)1 HttpChunk (org.webpieces.httpparser.api.dto.HttpChunk)1 HttpData (org.webpieces.httpparser.api.dto.HttpData)1 HttpLastChunk (org.webpieces.httpparser.api.dto.HttpLastChunk)1