| /* |
| * 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; |
| using System.Threading; |
| using NUnit.Framework; |
| using NUnit.Framework.Extensions; |
| using Apache.NMS.Test; |
| |
| namespace Apache.NMS.Stomp.Test |
| { |
| [TestFixture] |
| public class ConnectionTest : NMSTestSupport |
| { |
| protected static string TEST_CLIENT_ID = "ConnectionTestClientId"; |
| protected int postfix; |
| |
| IConnection startedConnection = null; |
| IConnection stoppedConnection = null; |
| |
| [SetUp] |
| public override void SetUp() |
| { |
| base.SetUp(); |
| |
| startedConnection = CreateConnection(null); |
| startedConnection.Start(); |
| stoppedConnection = CreateConnection(null); |
| |
| this.postfix = new Random().Next(); |
| } |
| |
| [TearDown] |
| public override void TearDown() |
| { |
| startedConnection.Close(); |
| stoppedConnection.Close(); |
| |
| base.TearDown(); |
| } |
| |
| /// <summary> |
| /// Verify that it is possible to create multiple connections to the broker. |
| /// There was a bug in the connection factory which set the clientId member which made |
| /// it impossible to create an additional connection. |
| /// </summary> |
| [Test] |
| public void TwoConnections() |
| { |
| using(IConnection connection1 = CreateConnection(null)) |
| { |
| connection1.Start(); |
| using(IConnection connection2 = CreateConnection(null)) |
| { |
| // with the bug present we'll get an exception in connection2.start() |
| connection2.Start(); |
| } |
| } |
| } |
| |
| [RowTest] |
| [Row(true)] |
| [Row(false)] |
| public void CreateAndDisposeWithConsumer(bool disposeConsumer) |
| { |
| using(IConnection connection = CreateConnection("DisposalTestConnection" + ":" + this.postfix)) |
| { |
| connection.Start(); |
| |
| using(ISession session = connection.CreateSession()) |
| { |
| IQueue queue = session.GetQueue("DisposalTestQueue"); |
| IMessageConsumer consumer = session.CreateConsumer(queue); |
| |
| connection.Stop(); |
| if(disposeConsumer) |
| { |
| consumer.Dispose(); |
| } |
| } |
| } |
| } |
| |
| [RowTest] |
| [Row(true)] |
| [Row(false)] |
| public void CreateAndDisposeWithProducer(bool disposeProducer) |
| { |
| using(IConnection connection = CreateConnection("DisposalTestConnection" + ":" + this.postfix)) |
| { |
| connection.Start(); |
| |
| using(ISession session = connection.CreateSession()) |
| { |
| IQueue queue = session.GetQueue("DisposalTestQueue"); |
| IMessageProducer producer = session.CreateProducer(queue); |
| |
| connection.Stop(); |
| if(disposeProducer) |
| { |
| producer.Dispose(); |
| } |
| } |
| } |
| } |
| |
| [RowTest] |
| [Row(MsgDeliveryMode.Persistent, DestinationType.Queue)] |
| [Row(MsgDeliveryMode.Persistent, DestinationType.Topic)] |
| [Row(MsgDeliveryMode.NonPersistent, DestinationType.Queue)] |
| [Row(MsgDeliveryMode.NonPersistent, DestinationType.Topic)] |
| public void TestStartAfterSend(MsgDeliveryMode deliveryMode, DestinationType destinationType) |
| { |
| using(IConnection connection = CreateConnection(TEST_CLIENT_ID + ":" + this.postfix)) |
| { |
| ISession session = connection.CreateSession(AcknowledgementMode.AutoAcknowledge); |
| IDestination destination = CreateDestination(session, destinationType); |
| IMessageConsumer consumer = session.CreateConsumer(destination); |
| |
| // Send the messages |
| SendMessages(session, destination, deliveryMode, 1); |
| |
| // Start the conncection after the message was sent. |
| connection.Start(); |
| |
| // Make sure only 1 message was delivered. |
| Assert.IsNotNull(consumer.Receive(TimeSpan.FromMilliseconds(1000))); |
| Assert.IsNull(consumer.ReceiveNoWait()); |
| } |
| } |
| |
| /// <summary> |
| /// Tests if the consumer receives the messages that were sent before the |
| /// connection was started. |
| /// </summary> |
| [Test] |
| public void TestStoppedConsumerHoldsMessagesTillStarted() |
| { |
| ISession startedSession = startedConnection.CreateSession(AcknowledgementMode.AutoAcknowledge); |
| ISession stoppedSession = stoppedConnection.CreateSession(AcknowledgementMode.AutoAcknowledge); |
| |
| // Setup the consumers. |
| ITopic topic = startedSession.GetTopic("ConnectionTestTopic"); |
| IMessageConsumer startedConsumer = startedSession.CreateConsumer(topic); |
| IMessageConsumer stoppedConsumer = stoppedSession.CreateConsumer(topic); |
| |
| // Send the message. |
| IMessageProducer producer = startedSession.CreateProducer(topic); |
| ITextMessage message = startedSession.CreateTextMessage("Hello"); |
| producer.Send(message); |
| |
| // Test the assertions. |
| IMessage m = startedConsumer.Receive(TimeSpan.FromMilliseconds(1000)); |
| Assert.IsNotNull(m); |
| |
| m = stoppedConsumer.Receive(TimeSpan.FromMilliseconds(1000)); |
| Assert.IsNull(m); |
| |
| stoppedConnection.Start(); |
| m = stoppedConsumer.Receive(TimeSpan.FromMilliseconds(5000)); |
| Assert.IsNotNull(m); |
| |
| startedSession.Close(); |
| stoppedSession.Close(); |
| } |
| |
| /// <summary> |
| /// Tests if the consumer is able to receive messages eveb when the |
| /// connecction restarts multiple times. |
| /// </summary> |
| [Test] |
| public void TestMultipleConnectionStops() |
| { |
| TestStoppedConsumerHoldsMessagesTillStarted(); |
| stoppedConnection.Stop(); |
| TestStoppedConsumerHoldsMessagesTillStarted(); |
| stoppedConnection.Stop(); |
| TestStoppedConsumerHoldsMessagesTillStarted(); |
| } |
| } |
| } |