| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| using System.Threading; |
| using Apache.NMS.Test; |
| using NUnit.Framework; |
| |
| namespace Apache.NMS.ActiveMQ.Test |
| { |
| [TestFixture] |
| public class VirtualTopicTest : NMSTestSupport |
| { |
| protected static string DESTINATION_NAME = "TEST.VTopicDestination"; |
| protected static string PRODUCER_DESTINATION_NAME = "VirtualTopic." + DESTINATION_NAME; |
| protected static string CONSUMER_A_DESTINATION_NAME = "Consumer.A." + PRODUCER_DESTINATION_NAME; |
| protected static string CONSUMER_B_DESTINATION_NAME = "Consumer.B." + PRODUCER_DESTINATION_NAME; |
| protected static string TEST_CLIENT_ID = "VirtualTopicTestClientId"; |
| |
| protected const int totalMsgs = 5; |
| |
| [Test] |
| public void SendReceiveVirtualTopicMessage( |
| [Values(AcknowledgementMode.AutoAcknowledge, AcknowledgementMode.ClientAcknowledge, |
| AcknowledgementMode.DupsOkAcknowledge, AcknowledgementMode.Transactional)] |
| AcknowledgementMode ackMode, |
| [Values(MsgDeliveryMode.NonPersistent, MsgDeliveryMode.Persistent)] |
| MsgDeliveryMode deliveryMode) |
| { |
| using(IConnection connection = CreateConnection(TEST_CLIENT_ID)) |
| { |
| connection.Start(); |
| using(ISession session = connection.CreateSession(ackMode)) |
| { |
| using(IMessageConsumer consumerA = session.CreateConsumer(session.GetQueue(CONSUMER_A_DESTINATION_NAME))) |
| using(IMessageConsumer consumerB = session.CreateConsumer(session.GetQueue(CONSUMER_B_DESTINATION_NAME))) |
| using(IMessageProducer producer = session.CreateProducer(session.GetTopic(PRODUCER_DESTINATION_NAME))) |
| { |
| producer.DeliveryMode = deliveryMode; |
| |
| for(int index = 0; index < totalMsgs; index++) |
| { |
| string msgText = "Message #" + index; |
| Tracer.Info("Sending: " + msgText); |
| producer.Send(session.CreateTextMessage(msgText)); |
| } |
| |
| if(AcknowledgementMode.Transactional == ackMode) |
| { |
| session.Commit(); |
| } |
| |
| for(int index = 0; index < totalMsgs; index++) |
| { |
| string msgText = "Message #" + index; |
| ITextMessage messageA = consumerA.Receive(receiveTimeout) as ITextMessage; |
| Assert.IsNotNull(messageA, "Did not receive message for consumer A."); |
| messageA.Acknowledge(); |
| Tracer.Info("Received A: " + msgText); |
| |
| ITextMessage messageB = consumerB.Receive(receiveTimeout) as ITextMessage; |
| Assert.IsNotNull(messageB, "Did not receive message for consumer B."); |
| messageB.Acknowledge(); |
| Tracer.Info("Received B: " + msgText); |
| |
| Assert.AreEqual(msgText, messageA.Text, "Message text A does not match."); |
| Assert.AreEqual(msgText, messageB.Text, "Message text B does not match."); |
| } |
| |
| if(AcknowledgementMode.Transactional == ackMode) |
| { |
| session.Commit(); |
| } |
| } |
| |
| // Give the Broker some time to remove the subscriptions. |
| Thread.Sleep(2000); |
| |
| try |
| { |
| ((Session) session).DeleteDestination(session.GetQueue(CONSUMER_A_DESTINATION_NAME)); |
| ((Session) session).DeleteDestination(session.GetQueue(CONSUMER_B_DESTINATION_NAME)); |
| } |
| catch |
| { |
| } |
| } |
| } |
| } |
| |
| protected int receivedA; |
| protected int receivedB; |
| |
| [Test] |
| // Do not use listeners with transactional processing. |
| public void AsyncSendReceiveVirtualTopicMessage( |
| [Values(AcknowledgementMode.AutoAcknowledge, AcknowledgementMode.ClientAcknowledge, AcknowledgementMode.DupsOkAcknowledge)] |
| AcknowledgementMode ackMode, |
| [Values(MsgDeliveryMode.NonPersistent, MsgDeliveryMode.Persistent)] |
| MsgDeliveryMode deliveryMode) |
| { |
| receivedA = 0; |
| receivedB = 0; |
| |
| using(IConnection connection = CreateConnection(TEST_CLIENT_ID)) |
| { |
| connection.Start(); |
| using(ISession session = connection.CreateSession(ackMode)) |
| { |
| using(IMessageConsumer consumerA = session.CreateConsumer(session.GetQueue(CONSUMER_A_DESTINATION_NAME))) |
| using(IMessageConsumer consumerB = session.CreateConsumer(session.GetQueue(CONSUMER_B_DESTINATION_NAME))) |
| using(IMessageProducer producer = session.CreateProducer(session.GetTopic(PRODUCER_DESTINATION_NAME))) |
| { |
| producer.DeliveryMode = deliveryMode; |
| |
| consumerA.Listener += MessageListenerA; |
| consumerB.Listener += MessageListenerB; |
| |
| for(int index = 0; index < totalMsgs; index++) |
| { |
| string msgText = "Message #" + index; |
| Tracer.Info("Sending: " + msgText); |
| producer.Send(session.CreateTextMessage(msgText)); |
| } |
| |
| int waitCount = 0; |
| while(receivedA < totalMsgs && receivedB < totalMsgs) |
| { |
| if(waitCount++ > 50) |
| { |
| Assert.Fail("Timed out waiting for message consumers. A = " + receivedA + ", B = " + receivedB); |
| } |
| |
| Tracer.Info("Waiting... Received A = " + receivedA + ", Received B = " + receivedB); |
| Thread.Sleep(250); |
| } |
| } |
| |
| // Give the Broker some time to remove the subscriptions. |
| Thread.Sleep(2000); |
| |
| try |
| { |
| ((Session) session).DeleteDestination(session.GetQueue(CONSUMER_A_DESTINATION_NAME)); |
| ((Session) session).DeleteDestination(session.GetQueue(CONSUMER_B_DESTINATION_NAME)); |
| } |
| catch |
| { |
| } |
| } |
| } |
| } |
| |
| private void MessageListenerA(IMessage message) |
| { |
| message.Acknowledge(); |
| ITextMessage messageA = message as ITextMessage; |
| string msgText = "Message #" + receivedA; |
| Assert.AreEqual(msgText, messageA.Text, "Message text A does not match."); |
| Tracer.Info("Received Listener A: " + msgText); |
| receivedA++; |
| } |
| |
| private void MessageListenerB(IMessage message) |
| { |
| message.Acknowledge(); |
| ITextMessage messageB = message as ITextMessage; |
| string msgText = "Message #" + receivedB; |
| Assert.AreEqual(msgText, messageB.Text, "Message text B does not match."); |
| Tracer.Info("Received Listener B: " + msgText); |
| receivedB++; |
| } |
| } |
| } |