Search in sources :

Example 1 with FooterMatcher

use of org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher in project qpid-protonj2 by apache.

the class StreamSenderTest method testStreamSenderWritesFooterAfterStreamClosed.

@Test
public void testStreamSenderWritesFooterAfterStreamClosed() throws Exception {
    try (ProtonTestServer peer = new ProtonTestServer()) {
        peer.expectSASLAnonymousConnect();
        peer.expectOpen().respond();
        peer.expectBegin().respond();
        peer.expectAttach().ofSender().respond();
        peer.remoteFlow().withLinkCredit(10).queue();
        peer.start();
        URI remoteURI = peer.getServerURI();
        LOG.info("Sender test started, peer listening on: {}", remoteURI);
        Client container = Client.create();
        Connection connection = container.connect(remoteURI.getHost(), remoteURI.getPort()).openFuture().get();
        StreamSender sender = connection.openStreamSender("test-queue");
        StreamSenderMessage message = sender.beginMessage();
        final byte[] payload = new byte[] { 0, 1, 2, 3, 4, 5 };
        // First frame should include only the bits up to the body
        HeaderMatcher headerMatcher = new HeaderMatcher(true);
        headerMatcher.withDurable(true);
        headerMatcher.withPriority((byte) 1);
        headerMatcher.withTtl(65535);
        headerMatcher.withFirstAcquirer(true);
        headerMatcher.withDeliveryCount(2);
        ApplicationPropertiesMatcher apMatcher = new ApplicationPropertiesMatcher(true);
        apMatcher.withEntry("ap1", Matchers.equalTo(1));
        apMatcher.withEntry("ap2", Matchers.equalTo(2));
        apMatcher.withEntry("ap3", Matchers.equalTo(3));
        FooterMatcher footerMatcher = new FooterMatcher(false);
        footerMatcher.withEntry("f1", Matchers.equalTo(1));
        footerMatcher.withEntry("f2", Matchers.equalTo(2));
        footerMatcher.withEntry("f3", Matchers.equalTo(3));
        EncodedDataMatcher bodyMatcher = new EncodedDataMatcher(payload, true);
        TransferPayloadCompositeMatcher payloadMatcher = new TransferPayloadCompositeMatcher();
        payloadMatcher.setHeadersMatcher(headerMatcher);
        payloadMatcher.setApplicationPropertiesMatcher(apMatcher);
        payloadMatcher.setMessageContentMatcher(bodyMatcher);
        payloadMatcher.setFootersMatcher(footerMatcher);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
        peer.expectTransfer().withPayload(payloadMatcher).withMore(false).accept();
        // Populate all Header values
        message.durable(true);
        message.priority((byte) 1);
        message.timeToLive(65535);
        message.firstAcquirer(true);
        message.deliveryCount(2);
        // Populate message application properties
        message.property("ap1", 1);
        message.property("ap2", 2);
        message.property("ap3", 3);
        // Populate message footers
        assertFalse(message.hasFooters());
        assertFalse(message.hasFooter("f1"));
        message.footer("f1", 1);
        message.footer("f2", 2);
        message.footer("f3", 3);
        assertTrue(message.hasFooter("f1"));
        assertTrue(message.hasFooters());
        OutputStreamOptions bodyOptions = new OutputStreamOptions().completeSendOnClose(true);
        OutputStream stream = message.body(bodyOptions);
        assertThrows(ClientUnsupportedOperationException.class, () -> message.encode(Collections.emptyMap()));
        stream.write(payload);
        stream.close();
        assertThrows(ClientIllegalStateException.class, () -> message.footer(new Footer(Collections.emptyMap())));
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
        peer.expectDetach().respond();
        peer.expectEnd().respond();
        peer.expectClose().respond();
        assertNotNull(message.tracker());
        Wait.assertTrue(() -> message.tracker().settlementFuture().isDone());
        assertTrue(message.tracker().settlementFuture().get().settled());
        sender.closeAsync().get(10, TimeUnit.SECONDS);
        connection.closeAsync().get(10, TimeUnit.SECONDS);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
    }
}
Also used : HeaderMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.HeaderMatcher) ProtonTestServer(org.apache.qpid.protonj2.test.driver.ProtonTestServer) ByteArrayOutputStream(java.io.ByteArrayOutputStream) OutputStream(java.io.OutputStream) Connection(org.apache.qpid.protonj2.client.Connection) EncodedDataMatcher(org.apache.qpid.protonj2.test.driver.matchers.types.EncodedDataMatcher) TransferPayloadCompositeMatcher(org.apache.qpid.protonj2.test.driver.matchers.transport.TransferPayloadCompositeMatcher) URI(java.net.URI) ApplicationPropertiesMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.ApplicationPropertiesMatcher) OutputStreamOptions(org.apache.qpid.protonj2.client.OutputStreamOptions) Footer(org.apache.qpid.protonj2.types.messaging.Footer) StreamSender(org.apache.qpid.protonj2.client.StreamSender) StreamSenderMessage(org.apache.qpid.protonj2.client.StreamSenderMessage) Client(org.apache.qpid.protonj2.client.Client) FooterMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher) Test(org.junit.jupiter.api.Test)

Example 2 with FooterMatcher

use of org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher in project qpid-protonj2 by apache.

the class MessageSendTest method testSendMessageWithFootersPopulated.

@Test
public void testSendMessageWithFootersPopulated() throws Exception {
    try (ProtonTestServer peer = new ProtonTestServer()) {
        peer.expectSASLAnonymousConnect();
        peer.expectOpen().respond();
        peer.expectBegin().respond();
        peer.expectAttach().ofSender().respond();
        peer.remoteFlow().withLinkCredit(10).queue();
        // Open a receiver to ensure sender link has processed
        peer.expectAttach().respond();
        // the inbound flow frame we sent previously before send.
        peer.expectFlow();
        peer.start();
        URI remoteURI = peer.getServerURI();
        LOG.info("Sender test started, peer listening on: {}", remoteURI);
        Client container = Client.create();
        Connection connection = container.connect(remoteURI.getHost(), remoteURI.getPort()).openFuture().get();
        Session session = connection.openSession().openFuture().get();
        SenderOptions options = new SenderOptions().deliveryMode(DeliveryMode.AT_MOST_ONCE);
        Sender sender = session.openSender("test-qos", options);
        // Gates send on remote flow having been sent and received
        session.openReceiver("dummy").openFuture().get();
        FooterMatcher footerMatcher = new FooterMatcher(false);
        footerMatcher.withEntry("f1", Matchers.equalTo(1));
        footerMatcher.withEntry("f2", Matchers.equalTo(2));
        footerMatcher.withEntry("f3", Matchers.equalTo(3));
        EncodedAmqpValueMatcher bodyMatcher = new EncodedAmqpValueMatcher("Hello World", true);
        TransferPayloadCompositeMatcher payloadMatcher = new TransferPayloadCompositeMatcher();
        payloadMatcher.setMessageContentMatcher(bodyMatcher);
        payloadMatcher.setFootersMatcher(footerMatcher);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
        peer.expectTransfer().withPayload(payloadMatcher).accept();
        peer.expectDetach().respond();
        peer.expectClose().respond();
        final Message<String> message = Message.create("Hello World");
        // Populate message footers
        message.footer("f1", 1);
        message.footer("f2", 2);
        message.footer("f3", 3);
        final Tracker tracker = sender.send(message);
        assertNotNull(tracker);
        assertNotNull(tracker.settlementFuture().isDone());
        assertNotNull(tracker.settlementFuture().get().settled());
        sender.closeAsync().get(10, TimeUnit.SECONDS);
        connection.closeAsync().get(10, TimeUnit.SECONDS);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
    }
}
Also used : Tracker(org.apache.qpid.protonj2.client.Tracker) ProtonTestServer(org.apache.qpid.protonj2.test.driver.ProtonTestServer) SenderOptions(org.apache.qpid.protonj2.client.SenderOptions) Connection(org.apache.qpid.protonj2.client.Connection) EncodedAmqpValueMatcher(org.apache.qpid.protonj2.test.driver.matchers.types.EncodedAmqpValueMatcher) TransferPayloadCompositeMatcher(org.apache.qpid.protonj2.test.driver.matchers.transport.TransferPayloadCompositeMatcher) URI(java.net.URI) Sender(org.apache.qpid.protonj2.client.Sender) Client(org.apache.qpid.protonj2.client.Client) Session(org.apache.qpid.protonj2.client.Session) FooterMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher) Test(org.junit.jupiter.api.Test)

Example 3 with FooterMatcher

use of org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher in project qpid-protonj2 by apache.

the class MessageSendTest method doTestSendOfExternalMessage.

private void doTestSendOfExternalMessage(boolean allowAdvancedConversion, boolean trySend) throws Exception {
    try (ProtonTestServer peer = new ProtonTestServer()) {
        peer.expectSASLAnonymousConnect();
        peer.expectOpen().respond();
        peer.expectBegin().respond();
        peer.expectAttach().ofSender().respond();
        peer.remoteFlow().withLinkCredit(10).queue();
        // Open a receiver to ensure sender link has processed
        peer.expectAttach().respond();
        // the inbound flow frame we sent previously before send.
        peer.expectFlow();
        peer.start();
        URI remoteURI = peer.getServerURI();
        LOG.info("Sender test started, peer listening on: {}", remoteURI);
        Client container = Client.create();
        Connection connection = container.connect(remoteURI.getHost(), remoteURI.getPort()).openFuture().get();
        Session session = connection.openSession().openFuture().get();
        SenderOptions options = new SenderOptions().deliveryMode(DeliveryMode.AT_MOST_ONCE);
        Sender sender = session.openSender("test-qos", options);
        // Gates send on remote flow having been sent and received
        session.openReceiver("dummy").openFuture().get();
        HeaderMatcher headerMatcher = new HeaderMatcher(true);
        headerMatcher.withDurable(true);
        headerMatcher.withPriority((byte) 1);
        headerMatcher.withTtl(65535);
        headerMatcher.withFirstAcquirer(true);
        headerMatcher.withDeliveryCount(2);
        PropertiesMatcher propertiesMatcher = new PropertiesMatcher(true);
        propertiesMatcher.withMessageId("ID:12345");
        propertiesMatcher.withUserId("user".getBytes(StandardCharsets.UTF_8));
        propertiesMatcher.withTo("the-management");
        propertiesMatcher.withSubject("amqp");
        propertiesMatcher.withReplyTo("the-minions");
        propertiesMatcher.withCorrelationId("abc");
        propertiesMatcher.withContentEncoding("application/json");
        propertiesMatcher.withContentEncoding("gzip");
        propertiesMatcher.withAbsoluteExpiryTime(123);
        propertiesMatcher.withCreationTime(1);
        propertiesMatcher.withGroupId("disgruntled");
        propertiesMatcher.withGroupSequence(8192);
        propertiesMatcher.withReplyToGroupId("/dev/null");
        DeliveryAnnotationsMatcher daMatcher = new DeliveryAnnotationsMatcher(true);
        daMatcher.withEntry("da1", Matchers.equalTo(1));
        daMatcher.withEntry("da2", Matchers.equalTo(2));
        daMatcher.withEntry("da3", Matchers.equalTo(3));
        MessageAnnotationsMatcher maMatcher = new MessageAnnotationsMatcher(true);
        maMatcher.withEntry("ma1", Matchers.equalTo(1));
        maMatcher.withEntry("ma2", Matchers.equalTo(2));
        maMatcher.withEntry("ma3", Matchers.equalTo(3));
        ApplicationPropertiesMatcher apMatcher = new ApplicationPropertiesMatcher(true);
        apMatcher.withEntry("ap1", Matchers.equalTo(1));
        apMatcher.withEntry("ap2", Matchers.equalTo(2));
        apMatcher.withEntry("ap3", Matchers.equalTo(3));
        FooterMatcher footerMatcher = new FooterMatcher(false);
        footerMatcher.withEntry("f1", Matchers.equalTo(1));
        footerMatcher.withEntry("f2", Matchers.equalTo(2));
        footerMatcher.withEntry("f3", Matchers.equalTo(3));
        EncodedAmqpValueMatcher bodyMatcher = new EncodedAmqpValueMatcher("Hello World", true);
        TransferPayloadCompositeMatcher payloadMatcher = new TransferPayloadCompositeMatcher();
        payloadMatcher.setHeadersMatcher(headerMatcher);
        payloadMatcher.setDeliveryAnnotationsMatcher(daMatcher);
        payloadMatcher.setMessageAnnotationsMatcher(maMatcher);
        payloadMatcher.setPropertiesMatcher(propertiesMatcher);
        payloadMatcher.setApplicationPropertiesMatcher(apMatcher);
        payloadMatcher.setMessageContentMatcher(bodyMatcher);
        payloadMatcher.setFootersMatcher(footerMatcher);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
        peer.expectTransfer().withPayload(payloadMatcher).accept();
        peer.expectDetach().respond();
        peer.expectClose().respond();
        // Populate delivery annotations
        Map<String, Object> deliveryAnnotations = new HashMap<>();
        deliveryAnnotations.put("da1", 1);
        deliveryAnnotations.put("da2", 2);
        deliveryAnnotations.put("da3", 3);
        final Message<String> message = new ExternalMessage<>(allowAdvancedConversion);
        message.body("Hello World");
        // Populate all Header values
        message.durable(true);
        message.priority((byte) 1);
        message.timeToLive(65535);
        message.firstAcquirer(true);
        message.deliveryCount(2);
        // Populate message annotations
        message.annotation("ma1", 1);
        message.annotation("ma2", 2);
        message.annotation("ma3", 3);
        // Populate all Properties values
        message.messageId("ID:12345");
        message.userId("user".getBytes(StandardCharsets.UTF_8));
        message.to("the-management");
        message.subject("amqp");
        message.replyTo("the-minions");
        message.correlationId("abc");
        message.contentEncoding("application/json");
        message.contentEncoding("gzip");
        message.absoluteExpiryTime(123);
        message.creationTime(1);
        message.groupId("disgruntled");
        message.groupSequence(8192);
        message.replyToGroupId("/dev/null");
        // Populate message application properties
        message.property("ap1", 1);
        message.property("ap2", 2);
        message.property("ap3", 3);
        // Populate message footers
        message.footer("f1", 1);
        message.footer("f2", 2);
        message.footer("f3", 3);
        // Check preconditions that should affect the send operation
        if (allowAdvancedConversion) {
            assertNotNull(message.toAdvancedMessage());
        } else {
            assertThrows(UnsupportedOperationException.class, () -> message.toAdvancedMessage());
        }
        final Tracker tracker;
        if (trySend) {
            tracker = sender.trySend(message, deliveryAnnotations);
        } else {
            tracker = sender.send(message, deliveryAnnotations);
        }
        assertNotNull(tracker);
        assertNotNull(tracker.settlementFuture().isDone());
        assertNotNull(tracker.settlementFuture().get().settled());
        sender.closeAsync().get(10, TimeUnit.SECONDS);
        connection.closeAsync().get(10, TimeUnit.SECONDS);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
    }
}
Also used : HeaderMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.HeaderMatcher) Tracker(org.apache.qpid.protonj2.client.Tracker) ProtonTestServer(org.apache.qpid.protonj2.test.driver.ProtonTestServer) SenderOptions(org.apache.qpid.protonj2.client.SenderOptions) DeliveryAnnotationsMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.DeliveryAnnotationsMatcher) MessageAnnotationsMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.MessageAnnotationsMatcher) HashMap(java.util.HashMap) Connection(org.apache.qpid.protonj2.client.Connection) EncodedAmqpValueMatcher(org.apache.qpid.protonj2.test.driver.matchers.types.EncodedAmqpValueMatcher) TransferPayloadCompositeMatcher(org.apache.qpid.protonj2.test.driver.matchers.transport.TransferPayloadCompositeMatcher) URI(java.net.URI) Sender(org.apache.qpid.protonj2.client.Sender) ApplicationPropertiesMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.ApplicationPropertiesMatcher) ExternalMessage(org.apache.qpid.protonj2.client.util.ExternalMessage) Client(org.apache.qpid.protonj2.client.Client) PropertiesMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.PropertiesMatcher) ApplicationPropertiesMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.ApplicationPropertiesMatcher) Session(org.apache.qpid.protonj2.client.Session) FooterMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher)

Example 4 with FooterMatcher

use of org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher in project qpid-protonj2 by apache.

the class MessageSendTest method testSendMessageWithMultipleSectionsPopulated.

@Test
public void testSendMessageWithMultipleSectionsPopulated() throws Exception {
    try (ProtonTestServer peer = new ProtonTestServer()) {
        peer.expectSASLAnonymousConnect();
        peer.expectOpen().respond();
        peer.expectBegin().respond();
        peer.expectAttach().ofSender().respond();
        peer.remoteFlow().withLinkCredit(10).queue();
        // Open a receiver to ensure sender link has processed
        peer.expectAttach().respond();
        // the inbound flow frame we sent previously before send.
        peer.expectFlow();
        peer.start();
        URI remoteURI = peer.getServerURI();
        LOG.info("Sender test started, peer listening on: {}", remoteURI);
        Client container = Client.create();
        Connection connection = container.connect(remoteURI.getHost(), remoteURI.getPort()).openFuture().get();
        Session session = connection.openSession().openFuture().get();
        SenderOptions options = new SenderOptions().deliveryMode(DeliveryMode.AT_MOST_ONCE);
        Sender sender = session.openSender("test-qos", options);
        // Gates send on remote flow having been sent and received
        session.openReceiver("dummy").openFuture().get();
        HeaderMatcher headerMatcher = new HeaderMatcher(true);
        headerMatcher.withDurable(true);
        headerMatcher.withPriority((byte) 1);
        headerMatcher.withTtl(65535);
        headerMatcher.withFirstAcquirer(true);
        headerMatcher.withDeliveryCount(2);
        PropertiesMatcher propertiesMatcher = new PropertiesMatcher(true);
        propertiesMatcher.withMessageId("ID:12345");
        propertiesMatcher.withUserId("user".getBytes(StandardCharsets.UTF_8));
        propertiesMatcher.withTo("the-management");
        propertiesMatcher.withSubject("amqp");
        propertiesMatcher.withReplyTo("the-minions");
        propertiesMatcher.withCorrelationId("abc");
        propertiesMatcher.withContentEncoding("application/json");
        propertiesMatcher.withContentEncoding("gzip");
        propertiesMatcher.withAbsoluteExpiryTime(123);
        propertiesMatcher.withCreationTime(1);
        propertiesMatcher.withGroupId("disgruntled");
        propertiesMatcher.withGroupSequence(8192);
        propertiesMatcher.withReplyToGroupId("/dev/null");
        DeliveryAnnotationsMatcher daMatcher = new DeliveryAnnotationsMatcher(true);
        daMatcher.withEntry("da1", Matchers.equalTo(1));
        daMatcher.withEntry("da2", Matchers.equalTo(2));
        daMatcher.withEntry("da3", Matchers.equalTo(3));
        MessageAnnotationsMatcher maMatcher = new MessageAnnotationsMatcher(true);
        maMatcher.withEntry("ma1", Matchers.equalTo(1));
        maMatcher.withEntry("ma2", Matchers.equalTo(2));
        maMatcher.withEntry("ma3", Matchers.equalTo(3));
        ApplicationPropertiesMatcher apMatcher = new ApplicationPropertiesMatcher(true);
        apMatcher.withEntry("ap1", Matchers.equalTo(1));
        apMatcher.withEntry("ap2", Matchers.equalTo(2));
        apMatcher.withEntry("ap3", Matchers.equalTo(3));
        EncodedAmqpValueMatcher bodyMatcher = new EncodedAmqpValueMatcher("Hello World", true);
        FooterMatcher footerMatcher = new FooterMatcher(false);
        footerMatcher.withEntry("f1", Matchers.equalTo(1));
        footerMatcher.withEntry("f2", Matchers.equalTo(2));
        footerMatcher.withEntry("f3", Matchers.equalTo(3));
        TransferPayloadCompositeMatcher payloadMatcher = new TransferPayloadCompositeMatcher();
        payloadMatcher.setHeadersMatcher(headerMatcher);
        payloadMatcher.setDeliveryAnnotationsMatcher(daMatcher);
        payloadMatcher.setMessageAnnotationsMatcher(maMatcher);
        payloadMatcher.setPropertiesMatcher(propertiesMatcher);
        payloadMatcher.setApplicationPropertiesMatcher(apMatcher);
        payloadMatcher.setMessageContentMatcher(bodyMatcher);
        payloadMatcher.setFootersMatcher(footerMatcher);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
        peer.expectTransfer().withPayload(payloadMatcher).accept();
        peer.expectDetach().respond();
        peer.expectClose().respond();
        // Populate delivery annotations
        Map<String, Object> deliveryAnnotations = new HashMap<>();
        deliveryAnnotations.put("da1", 1);
        deliveryAnnotations.put("da2", 2);
        deliveryAnnotations.put("da3", 3);
        final Message<String> message = Message.create("Hello World");
        // Populate all Header values
        message.durable(true);
        message.priority((byte) 1);
        message.timeToLive(65535);
        message.firstAcquirer(true);
        message.deliveryCount(2);
        // Populate message annotations
        message.annotation("ma1", 1);
        message.annotation("ma2", 2);
        message.annotation("ma3", 3);
        // Populate all Properties values
        message.messageId("ID:12345");
        message.userId("user".getBytes(StandardCharsets.UTF_8));
        message.to("the-management");
        message.subject("amqp");
        message.replyTo("the-minions");
        message.correlationId("abc");
        message.contentEncoding("application/json");
        message.contentEncoding("gzip");
        message.absoluteExpiryTime(123);
        message.creationTime(1);
        message.groupId("disgruntled");
        message.groupSequence(8192);
        message.replyToGroupId("/dev/null");
        // Populate message application properties
        message.property("ap1", 1);
        message.property("ap2", 2);
        message.property("ap3", 3);
        // Populate message footers
        message.footer("f1", 1);
        message.footer("f2", 2);
        message.footer("f3", 3);
        final Tracker tracker = sender.send(message, deliveryAnnotations);
        assertNotNull(tracker);
        assertNotNull(tracker.settlementFuture().isDone());
        assertNotNull(tracker.settlementFuture().get().settled());
        sender.closeAsync().get(10, TimeUnit.SECONDS);
        connection.closeAsync().get(10, TimeUnit.SECONDS);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
    }
}
Also used : HeaderMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.HeaderMatcher) Tracker(org.apache.qpid.protonj2.client.Tracker) ProtonTestServer(org.apache.qpid.protonj2.test.driver.ProtonTestServer) SenderOptions(org.apache.qpid.protonj2.client.SenderOptions) DeliveryAnnotationsMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.DeliveryAnnotationsMatcher) MessageAnnotationsMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.MessageAnnotationsMatcher) HashMap(java.util.HashMap) Connection(org.apache.qpid.protonj2.client.Connection) EncodedAmqpValueMatcher(org.apache.qpid.protonj2.test.driver.matchers.types.EncodedAmqpValueMatcher) TransferPayloadCompositeMatcher(org.apache.qpid.protonj2.test.driver.matchers.transport.TransferPayloadCompositeMatcher) URI(java.net.URI) Sender(org.apache.qpid.protonj2.client.Sender) ApplicationPropertiesMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.ApplicationPropertiesMatcher) Client(org.apache.qpid.protonj2.client.Client) PropertiesMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.PropertiesMatcher) ApplicationPropertiesMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.ApplicationPropertiesMatcher) Session(org.apache.qpid.protonj2.client.Session) FooterMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher) Test(org.junit.jupiter.api.Test)

Example 5 with FooterMatcher

use of org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher in project qpid-protonj2 by apache.

the class StreamSenderTest method testStreamSenderWritesFooterAfterMessageCompleted.

@Test
public void testStreamSenderWritesFooterAfterMessageCompleted() throws Exception {
    try (ProtonTestServer peer = new ProtonTestServer()) {
        peer.expectSASLAnonymousConnect();
        peer.expectOpen().respond();
        peer.expectBegin().respond();
        peer.expectAttach().ofSender().respond();
        peer.remoteFlow().withLinkCredit(10).queue();
        peer.start();
        URI remoteURI = peer.getServerURI();
        LOG.info("Sender test started, peer listening on: {}", remoteURI);
        Client container = Client.create();
        Connection connection = container.connect(remoteURI.getHost(), remoteURI.getPort()).openFuture().get();
        StreamSender sender = connection.openStreamSender("test-queue");
        StreamSenderMessage message = sender.beginMessage();
        final byte[] payload = new byte[] { 0, 1, 2, 3, 4, 5 };
        // First frame should include only the bits up to the body
        HeaderMatcher headerMatcher = new HeaderMatcher(true);
        headerMatcher.withDurable(true);
        headerMatcher.withPriority((byte) 1);
        headerMatcher.withTtl(65535);
        headerMatcher.withFirstAcquirer(true);
        headerMatcher.withDeliveryCount(2);
        ApplicationPropertiesMatcher apMatcher = new ApplicationPropertiesMatcher(true);
        apMatcher.withEntry("ap1", Matchers.equalTo(1));
        apMatcher.withEntry("ap2", Matchers.equalTo(2));
        apMatcher.withEntry("ap3", Matchers.equalTo(3));
        EncodedDataMatcher bodyMatcher = new EncodedDataMatcher(payload);
        TransferPayloadCompositeMatcher payloadMatcher = new TransferPayloadCompositeMatcher();
        payloadMatcher.setHeadersMatcher(headerMatcher);
        payloadMatcher.setApplicationPropertiesMatcher(apMatcher);
        payloadMatcher.setMessageContentMatcher(bodyMatcher);
        // Second Frame should contains the appended footers
        FooterMatcher footerMatcher = new FooterMatcher(false);
        footerMatcher.withEntry("f1", Matchers.equalTo(1));
        footerMatcher.withEntry("f2", Matchers.equalTo(2));
        footerMatcher.withEntry("f3", Matchers.equalTo(3));
        TransferPayloadCompositeMatcher payloadFooterMatcher = new TransferPayloadCompositeMatcher();
        payloadFooterMatcher.setFootersMatcher(footerMatcher);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
        peer.expectTransfer().withPayload(payloadMatcher).withMore(true);
        peer.expectTransfer().withPayload(payloadFooterMatcher).withMore(false).accept();
        // Populate all Header values
        message.durable(true);
        message.priority((byte) 1);
        message.timeToLive(65535);
        message.firstAcquirer(true);
        message.deliveryCount(2);
        // Populate message application properties
        message.property("ap1", 1);
        message.property("ap2", 2);
        message.property("ap3", 3);
        OutputStreamOptions bodyOptions = new OutputStreamOptions().completeSendOnClose(false);
        OutputStream stream = message.body(bodyOptions);
        stream.write(payload);
        stream.close();
        // Populate message footers
        message.footer("f1", 1);
        message.footer("f2", 2);
        message.footer("f3", 3);
        message.complete();
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
        peer.expectDetach().respond();
        peer.expectEnd().respond();
        peer.expectClose().respond();
        assertNotNull(message.tracker());
        Wait.assertTrue(() -> message.tracker().settlementFuture().isDone());
        assertTrue(message.tracker().settlementFuture().get().settled());
        sender.closeAsync().get(10, TimeUnit.SECONDS);
        connection.closeAsync().get(10, TimeUnit.SECONDS);
        peer.waitForScriptToComplete(5, TimeUnit.SECONDS);
    }
}
Also used : HeaderMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.HeaderMatcher) ProtonTestServer(org.apache.qpid.protonj2.test.driver.ProtonTestServer) ByteArrayOutputStream(java.io.ByteArrayOutputStream) OutputStream(java.io.OutputStream) Connection(org.apache.qpid.protonj2.client.Connection) EncodedDataMatcher(org.apache.qpid.protonj2.test.driver.matchers.types.EncodedDataMatcher) TransferPayloadCompositeMatcher(org.apache.qpid.protonj2.test.driver.matchers.transport.TransferPayloadCompositeMatcher) URI(java.net.URI) ApplicationPropertiesMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.ApplicationPropertiesMatcher) OutputStreamOptions(org.apache.qpid.protonj2.client.OutputStreamOptions) StreamSender(org.apache.qpid.protonj2.client.StreamSender) StreamSenderMessage(org.apache.qpid.protonj2.client.StreamSenderMessage) Client(org.apache.qpid.protonj2.client.Client) FooterMatcher(org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher) Test(org.junit.jupiter.api.Test)

Aggregations

URI (java.net.URI)5 Client (org.apache.qpid.protonj2.client.Client)5 Connection (org.apache.qpid.protonj2.client.Connection)5 ProtonTestServer (org.apache.qpid.protonj2.test.driver.ProtonTestServer)5 FooterMatcher (org.apache.qpid.protonj2.test.driver.matchers.messaging.FooterMatcher)5 TransferPayloadCompositeMatcher (org.apache.qpid.protonj2.test.driver.matchers.transport.TransferPayloadCompositeMatcher)5 ApplicationPropertiesMatcher (org.apache.qpid.protonj2.test.driver.matchers.messaging.ApplicationPropertiesMatcher)4 HeaderMatcher (org.apache.qpid.protonj2.test.driver.matchers.messaging.HeaderMatcher)4 Test (org.junit.jupiter.api.Test)4 Sender (org.apache.qpid.protonj2.client.Sender)3 SenderOptions (org.apache.qpid.protonj2.client.SenderOptions)3 Session (org.apache.qpid.protonj2.client.Session)3 Tracker (org.apache.qpid.protonj2.client.Tracker)3 EncodedAmqpValueMatcher (org.apache.qpid.protonj2.test.driver.matchers.types.EncodedAmqpValueMatcher)3 ByteArrayOutputStream (java.io.ByteArrayOutputStream)2 OutputStream (java.io.OutputStream)2 HashMap (java.util.HashMap)2 OutputStreamOptions (org.apache.qpid.protonj2.client.OutputStreamOptions)2 StreamSender (org.apache.qpid.protonj2.client.StreamSender)2 StreamSenderMessage (org.apache.qpid.protonj2.client.StreamSenderMessage)2