Search in sources :

Example 6 with Header

use of io.aeron.logbuffer.Header in project Aeron by real-logic.

the class FragmentAssemblerTest method shouldAssembleFourPartMessage.

@Test
public void shouldAssembleFourPartMessage() {
    when(header.flags()).thenReturn(FrameDescriptor.BEGIN_FRAG_FLAG).thenReturn((byte) 0).thenReturn((byte) 0).thenReturn(FrameDescriptor.END_FRAG_FLAG);
    final UnsafeBuffer srcBuffer = new UnsafeBuffer(new byte[1024]);
    final int offset = 0;
    final int length = srcBuffer.capacity() / 4;
    for (int i = 0; i < 4; i++) {
        srcBuffer.setMemory(i * length, length, (byte) (65 + i));
    }
    adapter.onFragment(srcBuffer, offset, length, header);
    adapter.onFragment(srcBuffer, offset + length, length, header);
    adapter.onFragment(srcBuffer, offset + (length * 2), length, header);
    adapter.onFragment(srcBuffer, offset + (length * 3), length, header);
    final ArgumentCaptor<UnsafeBuffer> bufferArg = ArgumentCaptor.forClass(UnsafeBuffer.class);
    final ArgumentCaptor<Header> headerArg = ArgumentCaptor.forClass(Header.class);
    verify(delegateFragmentHandler, times(1)).onFragment(bufferArg.capture(), eq(offset), eq(length * 4), headerArg.capture());
    final UnsafeBuffer capturedBuffer = bufferArg.getValue();
    for (int i = 0; i < srcBuffer.capacity(); i++) {
        assertEquals(srcBuffer.getByte(i), capturedBuffer.getByte(i), "same at i=" + i);
    }
    final Header capturedHeader = headerArg.getValue();
    assertEquals(SESSION_ID, capturedHeader.sessionId());
    assertEquals(FrameDescriptor.END_FRAG_FLAG, capturedHeader.flags());
}
Also used : Header(io.aeron.logbuffer.Header) UnsafeBuffer(org.agrona.concurrent.UnsafeBuffer) Test(org.junit.jupiter.api.Test)

Example 7 with Header

use of io.aeron.logbuffer.Header in project Aeron by real-logic.

the class EgressAdapterTest method onFragmentIsANoOpIfSessionIdDoesNotMatchOnNewLeader.

@Test
void onFragmentIsANoOpIfSessionIdDoesNotMatchOnNewLeader() {
    final int offset = 0;
    final long clusterSessionId = -100;
    final long leadershipTermId = 6;
    final int leaderMemberId = 9999;
    final String ingressEndpoints = "ingress endpoints ...";
    newLeaderEventEncoder.wrapAndApplyHeader(buffer, offset, messageHeaderEncoder).leadershipTermId(leadershipTermId).clusterSessionId(clusterSessionId).leaderMemberId(leaderMemberId).ingressEndpoints(ingressEndpoints);
    final EgressListener egressListener = mock(EgressListener.class);
    final Header header = new Header(1, 3);
    final EgressAdapter adapter = new EgressAdapter(egressListener, 0, mock(Subscription.class), 10);
    adapter.onFragment(buffer, offset, newLeaderEventEncoder.encodedLength(), header);
    verifyNoInteractions(egressListener);
}
Also used : Header(io.aeron.logbuffer.Header) Subscription(io.aeron.Subscription) Test(org.junit.jupiter.api.Test)

Example 8 with Header

use of io.aeron.logbuffer.Header in project Aeron by real-logic.

the class EgressAdapterTest method onFragmentShouldInvokeOnNewLeaderCallbackIfSessionIdMatches.

@Test
void onFragmentShouldInvokeOnNewLeaderCallbackIfSessionIdMatches() {
    final int offset = 0;
    final long clusterSessionId = 0;
    final long leadershipTermId = 6;
    final int leaderMemberId = 9999;
    final String ingressEndpoints = "ingress endpoints ...";
    newLeaderEventEncoder.wrapAndApplyHeader(buffer, offset, messageHeaderEncoder).leadershipTermId(leadershipTermId).clusterSessionId(clusterSessionId).leaderMemberId(leaderMemberId).ingressEndpoints(ingressEndpoints);
    final EgressListener egressListener = mock(EgressListener.class);
    final Header header = new Header(1, 3);
    final EgressAdapter adapter = new EgressAdapter(egressListener, clusterSessionId, mock(Subscription.class), 10);
    adapter.onFragment(buffer, offset, newLeaderEventEncoder.encodedLength(), header);
    verify(egressListener).onNewLeader(clusterSessionId, leadershipTermId, leaderMemberId, ingressEndpoints);
    verifyNoMoreInteractions(egressListener);
}
Also used : Header(io.aeron.logbuffer.Header) Subscription(io.aeron.Subscription) Test(org.junit.jupiter.api.Test)

Example 9 with Header

use of io.aeron.logbuffer.Header in project Aeron by real-logic.

the class EgressAdapterTest method onFragmentIsANoOpIfSessionIdDoesNotMatchOnAdminResponse.

@Test
void onFragmentIsANoOpIfSessionIdDoesNotMatchOnAdminResponse() {
    final int offset = 24;
    final long clusterSessionId = 18;
    final long correlationId = 3274239749237498239L;
    final AdminRequestType type = AdminRequestType.SNAPSHOT;
    final AdminResponseCode responseCode = AdminResponseCode.OK;
    final String message = "Unauthorised access detected!";
    final byte[] payload = new byte[] { 0x1, 0x2, 0x3 };
    adminResponseEncoder.wrapAndApplyHeader(buffer, offset, messageHeaderEncoder).clusterSessionId(clusterSessionId).correlationId(correlationId).requestType(type).responseCode(responseCode).message(message);
    adminResponseEncoder.putPayload(payload, 0, payload.length);
    final EgressListener egressListener = mock(EgressListener.class);
    final Header header = new Header(1, 3);
    final EgressAdapter adapter = new EgressAdapter(egressListener, -clusterSessionId, mock(Subscription.class), 10);
    adapter.onFragment(buffer, offset, adminResponseEncoder.encodedLength(), header);
    verifyNoInteractions(egressListener);
}
Also used : Header(io.aeron.logbuffer.Header) Subscription(io.aeron.Subscription) Test(org.junit.jupiter.api.Test)

Example 10 with Header

use of io.aeron.logbuffer.Header in project Aeron by real-logic.

the class EgressAdapterTest method onFragmentShouldInvokeOnSessionEventCallbackIfSessionIdMatches.

@Test
void onFragmentShouldInvokeOnSessionEventCallbackIfSessionIdMatches() {
    final int offset = 8;
    final long clusterSessionId = 42;
    final long correlationId = 777;
    final long leadershipTermId = 6;
    final int leaderMemberId = 3;
    final EventCode eventCode = EventCode.REDIRECT;
    final int version = 18;
    final String eventDetail = "Event details";
    sessionEventEncoder.wrapAndApplyHeader(buffer, offset, messageHeaderEncoder).clusterSessionId(clusterSessionId).correlationId(correlationId).leadershipTermId(leadershipTermId).leaderMemberId(leaderMemberId).code(eventCode).version(version).detail(eventDetail);
    final EgressListener egressListener = mock(EgressListener.class);
    final Header header = new Header(1, 3);
    final EgressAdapter adapter = new EgressAdapter(egressListener, clusterSessionId, mock(Subscription.class), 10);
    adapter.onFragment(buffer, offset, sessionEventEncoder.encodedLength(), header);
    verify(egressListener).onSessionEvent(correlationId, clusterSessionId, leadershipTermId, leaderMemberId, eventCode, eventDetail);
    verifyNoMoreInteractions(egressListener);
}
Also used : Header(io.aeron.logbuffer.Header) Subscription(io.aeron.Subscription) Test(org.junit.jupiter.api.Test)

Aggregations

Header (io.aeron.logbuffer.Header)63 Test (org.junit.jupiter.api.Test)44 DirectBuffer (org.agrona.DirectBuffer)33 ClientSession (io.aeron.cluster.service.ClientSession)21 StubClusteredService (io.aeron.test.cluster.StubClusteredService)20 ExpandableArrayBuffer (org.agrona.ExpandableArrayBuffer)20 ClusteredService (io.aeron.cluster.service.ClusteredService)18 ClusteredServiceContainer (io.aeron.cluster.service.ClusteredServiceContainer)18 MediaDriver (io.aeron.driver.MediaDriver)18 CloseHelper (org.agrona.CloseHelper)17 Subscription (io.aeron.Subscription)16 Archive (io.aeron.archive.Archive)16 ArchiveThreadingMode (io.aeron.archive.ArchiveThreadingMode)16 AeronCluster (io.aeron.cluster.client.AeronCluster)16 ThreadingMode (io.aeron.driver.ThreadingMode)16 ExtendWith (org.junit.jupiter.api.extension.ExtendWith)16 UnsafeBuffer (org.agrona.concurrent.UnsafeBuffer)15 EgressListener (io.aeron.cluster.client.EgressListener)14 BeforeEach (org.junit.jupiter.api.BeforeEach)14 CLUSTER_MEMBERS (io.aeron.cluster.ClusterTestConstants.CLUSTER_MEMBERS)12