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

namespace Apache.NMS.Test
{
	[TestFixture]
	public class AsyncConsumeTest : NMSTestSupport
	{
		protected static string DESTINATION_NAME = "AsyncConsumeDestination";
		protected static string TEST_CLIENT_ID = "AsyncConsumeClientId";
		protected static string RESPONSE_CLIENT_ID = "AsyncConsumeResponseClientId";
		protected AutoResetEvent semaphore;
		protected bool received;
		protected IMessage receivedMsg;

		[SetUp]
		public override void SetUp()
		{
			base.SetUp();
			semaphore = new AutoResetEvent(false);
			received = false;
			receivedMsg = null;
		}

		[TearDown]
		public override void TearDown()
		{
			receivedMsg = null;
			base.TearDown();
		}

		[RowTest]
		[Row(MsgDeliveryMode.Persistent)]
		[Row(MsgDeliveryMode.NonPersistent)]
		public void TestAsynchronousConsume(MsgDeliveryMode deliveryMode)
		{
			using(IConnection connection = CreateConnection(TEST_CLIENT_ID))
			{
				connection.Start();
				using(ISession session = connection.CreateSession(AcknowledgementMode.AutoAcknowledge))
				{
					IDestination destination = SessionUtil.GetDestination(session, DESTINATION_NAME);
					using(IMessageConsumer consumer = session.CreateConsumer(destination))
					using(IMessageProducer producer = session.CreateProducer(destination))
					{
						producer.DeliveryMode = deliveryMode;
						producer.RequestTimeout = receiveTimeout;
						consumer.Listener += new MessageListener(OnMessage);

						IMessage request = session.CreateMessage();
						request.NMSCorrelationID = "AsyncConsume";
						request.NMSType = "Test";
						producer.Send(request);

						WaitForMessageToArrive();
						Assert.AreEqual(request.NMSCorrelationID, receivedMsg.NMSCorrelationID, "Invalid correlation ID.");
					}
				}
			}
		}

		[RowTest]
		[Row(MsgDeliveryMode.Persistent)]
		[Row(MsgDeliveryMode.NonPersistent)]
		public void TestCreateConsumerAfterSend(MsgDeliveryMode deliveryMode)
		{
			using(IConnection connection = CreateConnection(TEST_CLIENT_ID))
			{
				connection.Start();
				using(ISession session = connection.CreateSession(AcknowledgementMode.AutoAcknowledge))
				{
					IDestination destination = SessionUtil.GetDestination(session, DESTINATION_NAME);
					using(IMessageProducer producer = session.CreateProducer(destination))
					{
						producer.DeliveryMode = deliveryMode;
						producer.RequestTimeout = receiveTimeout;

						IMessage request = session.CreateMessage();
						request.NMSCorrelationID = "AsyncConsumeAfterSend";
						request.NMSType = "Test";
						producer.Send(request);

						using(IMessageConsumer consumer = session.CreateConsumer(destination))
						{
							consumer.Listener += new MessageListener(OnMessage);
							WaitForMessageToArrive();
							Assert.AreEqual(request.NMSCorrelationID, receivedMsg.NMSCorrelationID, "Invalid correlation ID.");
						}
					}
				}
			}
		}

		[RowTest]
		[Row(MsgDeliveryMode.Persistent)]
		[Row(MsgDeliveryMode.NonPersistent)]
		public void TestCreateConsumerBeforeSendAddListenerAfterSend(MsgDeliveryMode deliveryMode)
		{
			using(IConnection connection = CreateConnection(TEST_CLIENT_ID))
			{
				connection.Start();
				using(ISession session = connection.CreateSession(AcknowledgementMode.AutoAcknowledge))
				{
					IDestination destination = SessionUtil.GetDestination(session, DESTINATION_NAME);
					using(IMessageConsumer consumer = session.CreateConsumer(destination))
					using(IMessageProducer producer = session.CreateProducer(destination))
					{
						producer.DeliveryMode = deliveryMode;
						producer.RequestTimeout = receiveTimeout;

						IMessage request = session.CreateMessage();
						request.NMSCorrelationID = "AsyncConsumeAfterSendLateListener";
						request.NMSType = "Test";
						producer.Send(request);

						// now lets add the listener
						consumer.Listener += new MessageListener(OnMessage);
						WaitForMessageToArrive();
						Assert.AreEqual(request.NMSCorrelationID, receivedMsg.NMSCorrelationID, "Invalid correlation ID.");
					}
				}
			}
		}

		[RowTest]
		[Row(MsgDeliveryMode.Persistent)]
		[Row(MsgDeliveryMode.NonPersistent)]
		public void TestAsynchronousTextMessageConsume(MsgDeliveryMode deliveryMode)
		{
			using(IConnection connection = CreateConnection(TEST_CLIENT_ID))
			{
				connection.Start();
				using(ISession session = connection.CreateSession(AcknowledgementMode.AutoAcknowledge))
				{
					IDestination destination = SessionUtil.GetDestination(session, DESTINATION_NAME);
					using(IMessageConsumer consumer = session.CreateConsumer(destination))
					{
						consumer.Listener += new MessageListener(OnMessage);
						using(IMessageProducer producer = session.CreateProducer(destination))
						{
							producer.DeliveryMode = deliveryMode;
							producer.RequestTimeout = receiveTimeout;

							ITextMessage request = session.CreateTextMessage("Hello, World!");
							request.NMSCorrelationID = "AsyncConsumeTextMessage";
							request.Properties["NMSXGroupID"] = "cheese";
							request.Properties["myHeader"] = "James";

							producer.Send(request);

							WaitForMessageToArrive();
							Assert.AreEqual(request.NMSCorrelationID, receivedMsg.NMSCorrelationID, "Invalid correlation ID.");
							Assert.AreEqual(request.Properties["NMSXGroupID"], receivedMsg.Properties["NMSXGroupID"], "Invalid NMSXGroupID.");
							Assert.AreEqual(request.Properties["myHeader"], receivedMsg.Properties["myHeader"], "Invalid myHeader.");
							Assert.AreEqual(request.Text, ((ITextMessage) receivedMsg).Text, "Invalid text body.");
						}
					}
				}
			}
		}

		[RowTest]
		[Row(MsgDeliveryMode.Persistent)]
		[Row(MsgDeliveryMode.NonPersistent)]
		public void TestTemporaryQueueAsynchronousConsume(MsgDeliveryMode deliveryMode)
		{
			using(IConnection connection = CreateConnection(TEST_CLIENT_ID))
			{
				connection.Start();
				using(ISession session = connection.CreateSession(AcknowledgementMode.AutoAcknowledge))
				{
					IDestination destination = SessionUtil.GetDestination(session, DESTINATION_NAME);
					ITemporaryQueue tempReplyDestination = session.CreateTemporaryQueue();

					using(IMessageConsumer consumer = session.CreateConsumer(destination))
					using(IMessageConsumer tempConsumer = session.CreateConsumer(tempReplyDestination))
					using(IMessageProducer producer = session.CreateProducer(destination))
					{
						producer.DeliveryMode = deliveryMode;
						producer.RequestTimeout = receiveTimeout;
						tempConsumer.Listener += new MessageListener(OnMessage);
						consumer.Listener += new MessageListener(OnQueueMessage);

						IMessage request = session.CreateMessage();
						request.NMSCorrelationID = "TemqQueueAsyncConsume";
						request.NMSType = "Test";
						request.NMSReplyTo = tempReplyDestination;
						producer.Send(request);

						WaitForMessageToArrive();
						Assert.AreEqual("TempQueueAsyncResponse", receivedMsg.NMSCorrelationID, "Invalid correlation ID.");
					}
				}
			}
		}

		protected void OnQueueMessage(IMessage message)
		{
			Assert.AreEqual("TemqQueueAsyncConsume", message.NMSCorrelationID, "Invalid correlation ID.");
			using(IConnection connection = CreateConnection(RESPONSE_CLIENT_ID))
			{
				connection.Start();
				using(ISession session = connection.CreateSession(AcknowledgementMode.AutoAcknowledge))
				{
					using(IMessageProducer producer = session.CreateProducer(message.NMSReplyTo))
					{
						producer.DeliveryMode = message.NMSDeliveryMode;
						producer.RequestTimeout = receiveTimeout;

						ITextMessage response = session.CreateTextMessage("Asynchronous Response Message Text");
						response.NMSCorrelationID = "TempQueueAsyncResponse";
						response.NMSType = message.NMSType;
						producer.Send(response); 
					}
				}
			}
		}

		protected void OnMessage(IMessage message)
		{
			receivedMsg = message;
			received = true;
			semaphore.Set();
		}

		protected void WaitForMessageToArrive()
		{
			semaphore.WaitOne((int) receiveTimeout.TotalMilliseconds, true);
			Assert.IsTrue(received, "Should have received a message by now!");
		}
	}
}