Examples with ResourceLimitChecks org.eclipse.hono.adapter.resourcelimits.ResourceLimitChecks used on opensource projects

Example 6 with ResourceLimitChecks

use of org.eclipse.hono.adapter.resourcelimits.ResourceLimitChecks in project hono by eclipse.

the class AbstractProtocolAdapterBaseTest method testCheckMessageLimitFailsIfMessageLimitIsReached.

/**
 * Verifies that the message limit check fails, if the maximum number of messages
 * for a tenant have been reached. Also verifies that the payload size of the incoming
 * message is calculated based on the configured minimum message size.
 *
 * @param ctx The vert.x test context.
 */
@Test
public void testCheckMessageLimitFailsIfMessageLimitIsReached(final VertxTestContext ctx) {
    // GIVEN a tenant with a minimum message size of 4kb configured
    final TenantObject tenant = TenantObject.from("my-tenant", Boolean.TRUE);
    tenant.setMinimumMessageSize(4096);
    final ArgumentCaptor<Long> payloadSizeCaptor = ArgumentCaptor.forClass(Long.class);
    // And for that tenant, the maximum messages limit has been already reached
    final ResourceLimitChecks checks = mock(ResourceLimitChecks.class);
    when(checks.isConnectionLimitReached(any(TenantObject.class), any(SpanContext.class))).thenReturn(Future.succeededFuture(Boolean.FALSE));
    when(checks.isMessageLimitReached(any(TenantObject.class), payloadSizeCaptor.capture(), any(SpanContext.class))).thenReturn(Future.succeededFuture(Boolean.TRUE));
    when(checks.isConnectionDurationLimitReached(any(TenantObject.class), any(SpanContext.class))).thenReturn(Future.succeededFuture(Boolean.FALSE));
    adapter.setResourceLimitChecks(checks);
    // WHEN a device sends a message with a payload size of 5000 bytes
    adapter.checkMessageLimit(tenant, 5000, mock(SpanContext.class)).onComplete(ctx.failing(t -> {
        // THEN the payload size used for the message limit checks is calculated based on the minimum message size.
        // In this case it should be 8kb
        assertEquals(8 * 1024, payloadSizeCaptor.getValue());
        // THEN the message limit check fails
        ctx.verify(() -> assertThat(ServiceInvocationException.extractStatusCode(t)).isEqualTo(HttpUtils.HTTP_TOO_MANY_REQUESTS));
        ctx.completeNow();
    }));
}