Search in sources :

Example 6 with EchoService

use of com.linkedin.r2.sample.echo.EchoService in project rest.li by linkedin.

the class TestSslTimingKey method testSslTimingKey.

@Test
public void testSslTimingKey() throws Exception {
    if (isHttp2StreamBasedChannel())
        return;
    final EchoService client = new RestEchoClient(Bootstrap.createURI(_port, Bootstrap.getEchoURI(), true), createClient());
    final String msg = "This is a simple http echo message";
    final FutureCallback<String> callback = new FutureCallback<>();
    client.echo(msg, callback);
    Assert.assertEquals(callback.get(), msg);
    RequestContext context = _clientCaptureFilter.getRequestContext();
    @SuppressWarnings("unchecked") Map<TimingKey, TimingContext> map = (Map<TimingKey, TimingContext>) context.getLocalAttr("timings");
    Assert.assertNotNull(map);
    Assert.assertTrue(map.containsKey(SslHandshakeTimingHandler.TIMING_KEY));
    TimingContext timingContext = map.get(SslHandshakeTimingHandler.TIMING_KEY);
    Assert.assertNotNull(timingContext);
}
Also used : TimingKey(com.linkedin.r2.message.timing.TimingKey) EchoService(com.linkedin.r2.sample.echo.EchoService) RestEchoClient(com.linkedin.r2.sample.echo.rest.RestEchoClient) RequestContext(com.linkedin.r2.message.RequestContext) Map(java.util.Map) FutureCallback(com.linkedin.common.callback.FutureCallback) TimingContext(com.linkedin.r2.message.timing.TimingContextUtil.TimingContext) AbstractEchoServiceTest(test.r2.integ.clientserver.providers.AbstractEchoServiceTest) Test(org.testng.annotations.Test)

Example 7 with EchoService

use of com.linkedin.r2.sample.echo.EchoService in project rest.li by linkedin.

the class TestAlpnUpgradePromise method testClearTestUpgradeFailure.

private void testClearTestUpgradeFailure() {
    final EchoService client = new RestEchoClient(Bootstrap.createURI(_port, Bootstrap.getEchoURI(), _serverProvider.isSsl()), _client);
    final String msg = "This is a simple http echo message";
    final FutureCallback<String> callback = new FutureCallback<>();
    try {
        client.echo(msg, callback);
        Assert.assertEquals(callback.get(), msg);
        Assert.fail("Should not have reached here !");
    } catch (Exception ex) {
        Throwable throwable = ExceptionUtils.getRootCause(ex);
        Assert.assertTrue(throwable instanceof IllegalStateException);
        Assert.assertEquals(throwable.getMessage(), "HTTP/2 clear text upgrade failed");
    }
}
Also used : EchoService(com.linkedin.r2.sample.echo.EchoService) RestEchoClient(com.linkedin.r2.sample.echo.rest.RestEchoClient) FutureCallback(com.linkedin.common.callback.FutureCallback)

Example 8 with EchoService

use of com.linkedin.r2.sample.echo.EchoService in project rest.li by linkedin.

the class TestGeneralEchoServiceTest method testOnExceptionEchoService.

@Test
public void testOnExceptionEchoService() throws Exception {
    final EchoService client = getEchoClient(_client, Bootstrap.getOnExceptionEchoURI());
    final String msg = "This is a simple echo message";
    final FutureCallback<String> callback = new FutureCallback<>();
    client.echo(msg, callback);
    try {
        callback.get();
        Assert.fail("Should have thrown an exception");
    } catch (ExecutionException e) {
        Assert.assertTrue(e.getCause() instanceof RemoteInvocationException);
    }
}
Also used : EchoService(com.linkedin.r2.sample.echo.EchoService) RemoteInvocationException(com.linkedin.r2.RemoteInvocationException) ExecutionException(java.util.concurrent.ExecutionException) FutureCallback(com.linkedin.common.callback.FutureCallback) AbstractEchoServiceTest(test.r2.integ.clientserver.providers.AbstractEchoServiceTest) Test(org.testng.annotations.Test)

Example 9 with EchoService

use of com.linkedin.r2.sample.echo.EchoService in project rest.li by linkedin.

the class TestGeneralEchoServiceTest method testFilterChain.

@Test
public void testFilterChain() throws Exception {
    final EchoService client = getEchoClient(_client, Bootstrap.getEchoURI());
    final String msg = "This is a simple echo message";
    final FutureCallback<String> callback = new FutureCallback<>();
    client.echo(msg, callback);
    callback.get();
    // Make sure the server got its wire attribute
    Assert.assertEquals(_serverCaptureFilter.getRequest().get(_toServerKey), _toServerValue);
    Assert.assertEquals(_serverCaptureFilter.getResponse().get(_toClientKey), _toClientValue);
    // Make sure the client got its wire attribute, but not the server's wire attribute
    Assert.assertEquals(_clientCaptureFilter.getResponse().get(_toClientKey), _toClientValue);
    Assert.assertNull(_clientCaptureFilter.getResponse().get(_toServerKey));
}
Also used : EchoService(com.linkedin.r2.sample.echo.EchoService) FutureCallback(com.linkedin.common.callback.FutureCallback) AbstractEchoServiceTest(test.r2.integ.clientserver.providers.AbstractEchoServiceTest) Test(org.testng.annotations.Test)

Example 10 with EchoService

use of com.linkedin.r2.sample.echo.EchoService in project rest.li by linkedin.

the class AbstractEchoServiceTest method testOnExceptionEchoService.

@Test
public void testOnExceptionEchoService() throws Exception {
    final EchoService client = getEchoClient(_client, Bootstrap.getOnExceptionEchoURI());
    final String msg = "This is a simple echo message";
    final FutureCallback<String> callback = new FutureCallback<String>();
    client.echo(msg, callback);
    try {
        callback.get();
        Assert.fail("Should have thrown an exception");
    } catch (ExecutionException e) {
        Assert.assertTrue(e.getCause() instanceof RemoteInvocationException);
    }
}
Also used : EchoService(com.linkedin.r2.sample.echo.EchoService) RemoteInvocationException(com.linkedin.r2.RemoteInvocationException) ExecutionException(java.util.concurrent.ExecutionException) FutureCallback(com.linkedin.common.callback.FutureCallback) Test(org.testng.annotations.Test)

Aggregations

FutureCallback (com.linkedin.common.callback.FutureCallback)20 EchoService (com.linkedin.r2.sample.echo.EchoService)20 Test (org.testng.annotations.Test)17 RemoteInvocationException (com.linkedin.r2.RemoteInvocationException)10 ExecutionException (java.util.concurrent.ExecutionException)10 AbstractEchoServiceTest (test.r2.integ.clientserver.providers.AbstractEchoServiceTest)9 RestEchoClient (com.linkedin.r2.sample.echo.rest.RestEchoClient)8 RestException (com.linkedin.r2.message.rest.RestException)6 RequestContext (com.linkedin.r2.message.RequestContext)1 TimingContext (com.linkedin.r2.message.timing.TimingContextUtil.TimingContext)1 TimingKey (com.linkedin.r2.message.timing.TimingKey)1 Map (java.util.Map)1