Examples with InMemoryRateLimiterComponent io.apiman.gateway.engine.impl.InMemoryRateLimiterComponent used on opensource projects

Example 1 with InMemoryRateLimiterComponent

use of io.apiman.gateway.engine.impl.InMemoryRateLimiterComponent in project apiman by apiman.

the class RateLimitingPolicyTest method testApply.

/**
 * Test method for {@link io.apiman.gateway.engine.policies.RateLimitingPolicy#parseConfiguration(java.lang.String)}.
 */
@Test
public void testApply() {
    RateLimitingPolicy policy = new RateLimitingPolicy();
    String json = "{\r\n" + "  \"limit\" : 10,\r\n" + "  \"granularity\" : \"User\",\r\n" + "  \"period\" : \"Minute\",\r\n" + "  \"userHeader\" : \"X-Identity\"\r\n" + "}";
    Object config = policy.parseConfiguration(json);
    ApiRequest request = new ApiRequest();
    request.setContract(createTestContract());
    request.setType("GET");
    request.setApiKey("12345");
    request.setRemoteAddr("1.2.3.4");
    request.setDestination("/");
    // $NON-NLS-2$
    request.getHeaders().put("X-Identity", "sclause");
    IPolicyContext context = Mockito.mock(IPolicyContext.class);
    final PolicyFailure failure = new PolicyFailure();
    Mockito.when(context.getComponent(IPolicyFailureFactoryComponent.class)).thenReturn(new IPolicyFailureFactoryComponent() {

        @Override
        public PolicyFailure createFailure(PolicyFailureType type, int failureCode, String message) {
            return failure;
        }
    });
    Mockito.when(context.getComponent(IRateLimiterComponent.class)).thenReturn(new InMemoryRateLimiterComponent());
    IPolicyChain<ApiRequest> chain;
    for (int count = 0; count < 10; count++) {
        chain = Mockito.mock(IPolicyChain.class);
        policy.apply(request, context, config, chain);
        Mockito.verify(chain).doApply(request);
    }
    // Failure - only allow 10 per minute!
    chain = Mockito.mock(IPolicyChain.class);
    policy.apply(request, context, config, chain);
    Mockito.verify(chain).doFailure(failure);
}