/*
 * 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 NUnit.Framework;

namespace Apache.NMS.Test
{
	[TestFixture]
	public class ConnectionTest : NMSTestSupport
	{
		/// <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();
				}
			}
		}

        /// <summary>
        /// Verify disposing a connection after a consumer has been created and disposed.
        /// </summary>
        [Test]
        public void DisposeConnectionAfterDisposingConsumer()
        {
            CreateAndDisposeWithConsumer(true);
        }

        /// <summary>
        /// Verify disposing a connection after a consumer has been created but not disposed.
        /// </summary>
        [Test]
        public void DisposeConnectionWithoutDisposingConsumer()
        {
            CreateAndDisposeWithConsumer(false);
        }

        /// <summary>
        /// Verify disposing a connection after a producer has been created and disposed.
        /// </summary>
        [Test]
        public void DisposeConnectionAfterDisposingProducer()
        {
            CreateAndDisposeWithProducer(true);
        }

        /// <summary>
        /// Verify disposing a connection after a producer has been created but not disposed.
        /// </summary>
        [Test]
        public void DisposeConnectionWithoutDisposingProducer()
        {
            CreateAndDisposeWithProducer(false);
        }

        private void CreateAndDisposeWithConsumer(bool disposeConsumer)
        {
            IConnection connection = CreateConnection("DisposalTestConnection");
            connection.Start();

            ISession session = connection.CreateSession();
            IQueue queue = session.GetQueue("DisposalTestQueue");

            IMessageConsumer consumer = session.CreateConsumer(queue);

            connection.Stop();

            if (disposeConsumer)
                consumer.Dispose();

            session.Dispose();
            connection.Dispose();
        }

        private void CreateAndDisposeWithProducer(bool disposeProducer)
        {
            IConnection connection = CreateConnection("DisposalTestConnection");
            connection.Start();

            ISession session = connection.CreateSession();
            IQueue queue = session.GetQueue("DisposalTestQueue");

            IMessageProducer producer = session.CreateProducer(queue);

            connection.Stop();

            if (disposeProducer)
                producer.Dispose();

            session.Dispose();
            connection.Dispose();
        }
	}
}