Examples with Request io.gravitee.gateway.api.Request used on opensource projects

Example 21 with Request

use of io.gravitee.gateway.api.Request in project gravitee-gateway by gravitee-io.

the class ReactorHandlerResolverTest method test_uniqContextPath.

@Test
public void test_uniqContextPath() {
    Collection<ReactorHandler> handlers = new ArrayList<>(new ArrayList<ReactorHandler>() {

        {
            add(createMockHandler("/teams"));
        }
    });
    when(handlerRegistry.getReactorHandlers()).thenReturn(handlers);
    Request request = Mockito.mock(Request.class);
    when(request.path()).thenReturn("/teams");
    ReactorHandler handler = handlerResolver.resolve(request);
    Assert.assertEquals("/teams/", handler.contextPath());
}

Example 22 with Request

use of io.gravitee.gateway.api.Request in project gravitee-gateway by gravitee-io.

the class TransactionHandlerTest method shouldHaveTransactionId.

@Test
public void shouldHaveTransactionId() throws InterruptedException {
    final CountDownLatch lock = new CountDownLatch(1);
    when(request.id()).thenReturn(UUID.toString(UUID.random()));
    new TransactionHandler(request1 -> {
        assertNotNull(request1.transactionId());
        assertEquals(request1.transactionId(), request1.headers().getFirst(TransactionHandler.DEFAULT_TRANSACTIONAL_ID_HEADER));
        assertEquals(request1.transactionId(), request1.metrics().getTransactionId());
        assertEquals(request1.transactionId(), response.headers().getFirst(TransactionHandler.DEFAULT_TRANSACTIONAL_ID_HEADER));
        lock.countDown();
    }, response).handle(request);
    assertEquals(true, lock.await(10000, TimeUnit.MILLISECONDS));
}

Example 23 with Request

use of io.gravitee.gateway.api.Request in project gravitee-gateway by gravitee-io.

the class TransactionHandlerTest method shouldPropagateSameTransactionIdWithCustomHeader.

@Test
public void shouldPropagateSameTransactionIdWithCustomHeader() throws InterruptedException {
    final CountDownLatch lock = new CountDownLatch(1);
    String transactionId = UUID.toString(UUID.random());
    request.headers().set(CUSTOM_TRANSACTION_ID_HEADER, transactionId);
    new TransactionHandler(CUSTOM_TRANSACTION_ID_HEADER, request1 -> {
        assertNotNull(request1.transactionId());
        assertEquals(transactionId, request1.transactionId());
        assertEquals(transactionId, request1.headers().getFirst(CUSTOM_TRANSACTION_ID_HEADER));
        assertEquals(transactionId, request1.metrics().getTransactionId());
        assertEquals(request1.transactionId(), response.headers().getFirst(CUSTOM_TRANSACTION_ID_HEADER));
        lock.countDown();
    }, response).handle(request);
    assertEquals(true, lock.await(10000, TimeUnit.MILLISECONDS));
}

Example 24 with Request

use of io.gravitee.gateway.api.Request in project gravitee-gateway by gravitee-io.

the class RequestPolicyChainTest method doNext_multiplePolicy_throwError.

@Test
public void doNext_multiplePolicy_throwError() throws Exception {
    ExecutionContext executionContext = mock(ExecutionContext.class);
    Request request = mock(Request.class);
    Metrics metrics = Metrics.on(System.currentTimeMillis()).build();
    when(request.metrics()).thenReturn(metrics);
    PolicyChain chain = RequestPolicyChain.create(policies3(), executionContext);
    chain.setResultHandler(result -> {
    });
    chain.doNext(request, null);
    verify(request, atLeastOnce()).metrics();
    verify(policy3, atLeastOnce()).onRequest(request, null, chain, executionContext);
    verify(policy2, never()).onRequest(request, null, chain);
}

Example 25 with Request

use of io.gravitee.gateway.api.Request in project gravitee-gateway by gravitee-io.

the class ResponsePolicyChainTest method doNext_multiplePolicy_throwError.

@Test
public void doNext_multiplePolicy_throwError() throws Exception {
    ExecutionContext executionContext = mock(ExecutionContext.class);
    Request request = mock(Request.class);
    Metrics metrics = Metrics.on(System.currentTimeMillis()).build();
    when(request.metrics()).thenReturn(metrics);
    PolicyChain chain = ResponsePolicyChain.create(Arrays.asList(policy2, policy3), executionContext);
    chain.setResultHandler(result -> {
    });
    chain.doNext(request, null);
    verify(request, atLeastOnce()).metrics();
    verify(policy3, atLeastOnce()).onResponse(request, null, chain, executionContext);
    verify(policy2, never()).onResponse(request, null, chain, executionContext);
}