dotnet/Qpid.Integration.Tests/interactive/SendReceiveTest.cs

/*
 *
 * 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 log4net;
using NUnit.Framework;
using Apache.Qpid.Messaging;
using Apache.Qpid.Client.Qms;
using Apache.Qpid.Client;
using Apache.Qpid.Integration.Tests.testcases;

namespace Apache.Qpid.Integration.Tests.interactive
{
    /// <summary>
    /// SendReceiveTest provides a quick interactive send-receive test, where the user is prompted to trigger each send or receive.
    ///
    /// <p><table id="crc"><caption>CRC Card</caption>
    /// <tr><th> Responsibilities <th> Collaborations
    /// <tr><td> Run an interactive send-receive loop prompting user to trigger each event.
    /// </table>
    /// </summary>
    [TestFixture, Category("Interactive")]
    public class SendReceiveTest : BaseMessagingTestFixture
    {
        /// <summary>Used for debugging purposes.</summary>
        private static ILog log = LogManager.GetLogger(typeof(SendReceiveTest));

        /// <summary>Defines the name of the test topic to use with the tests.</summary>
        public const string TEST_ROUTING_KEY = "quicktestkey";

        /// <summary>The number of consumers to test.</summary>
        private const int CONSUMER_COUNT = 5;

        /// <summary>The number of test messages to send.</summary>
        private const int MESSAGE_COUNT = 10;

        /// <summary>Monitor used to signal succesfull receipt of all test messages.</summary>
        AutoResetEvent _finishedEvent;

        /// <summary>Used to count test messages received so far.</summary>
        private int _messageReceivedCount;

        /// <summary>Used to hold the expected number of messages to receive.</summary>
        private int expectedMessageCount;

        /// <summary>Flag used to indicate that all messages really were received, and that the test did not just time out. </summary>
        private bool allReceived;

        /// <summary> Creates one producing end-point and many consuming end-points connected on a topic. </summary>
        [SetUp]
        public override void Init()
        {
            base.Init();

            // Reset all test counts and flags.
            _messageReceivedCount = 0;
            allReceived = false;
            _finishedEvent = new AutoResetEvent(false);
        }

        /// <summary> Cleans up all test end-points. </summary>
        [TearDown]
        public override void Shutdown()
        {
            try
            {
                // Close all end points for producer and consumers. 
                // Producer is on 0, and consumers on 1 .. n, so loop is from 0 to n inclusive.
                for (int i = 0; i <= CONSUMER_COUNT; i++)
                {
                    CloseEndPoint(i);
                }
            } 
            finally 
            {
                base.Shutdown();
            }
        }

        /// <summary> Check that all consumers on a topic each receive all message on it. </summary>
        [Test]
        public void AllConsumerReceiveAllMessagesOnTopic()
        {
            // Create end-points for all the consumers in the test.
            for (int i = 1; i <= CONSUMER_COUNT; i++)
            {
                SetUpEndPoint(i, false, true, TEST_ROUTING_KEY + testId, AcknowledgeMode.AutoAcknowledge, false, ExchangeNameDefaults.TOPIC,
                              true, false, null);
                testConsumer[i].OnMessage += new MessageReceivedDelegate(OnMessage);
            }

            // Create an end-point to publish to the test topic.
            SetUpEndPoint(0, true, false, TEST_ROUTING_KEY + testId, AcknowledgeMode.AutoAcknowledge, false, ExchangeNameDefaults.TOPIC,
                          true, false, null);

            expectedMessageCount = (MESSAGE_COUNT * CONSUMER_COUNT);

            PromptAndWait("Press to send...");

            for (int i = 0; i < MESSAGE_COUNT; i++)
            {
                testProducer[0].Send(testChannel[0].CreateTextMessage("A"));
            }

            _finishedEvent.WaitOne(new TimeSpan(0, 0, 0, 10), false);

            PromptAndWait("Press to complete test...");

            // Check that all messages really were received.
            Assert.IsTrue(allReceived, "All messages were not received, only got " + _messageReceivedCount + " but wanted " + expectedMessageCount);
        }

        /// <summary> Check that consumers on the same queue receive each message once accross all consumers. </summary>
        //[Test]
        public void AllConsumerReceiveAllMessagesOnDirect()
        {
            // Create end-points for all the consumers in the test.
            for (int i = 1; i <= CONSUMER_COUNT; i++)
            {
                SetUpEndPoint(i, false, true, TEST_ROUTING_KEY + testId, AcknowledgeMode.AutoAcknowledge, false, ExchangeNameDefaults.DIRECT,
                              true, false, null);
                testConsumer[i].OnMessage += new MessageReceivedDelegate(OnMessage);
            }

            // Create an end-point to publish to the test topic.
            SetUpEndPoint(0, true, false, TEST_ROUTING_KEY + testId, AcknowledgeMode.AutoAcknowledge, false, ExchangeNameDefaults.DIRECT,
                          true, false, null);

            expectedMessageCount = MESSAGE_COUNT;

            for (int i = 0; i < MESSAGE_COUNT; i++)
            {
                testProducer[0].Send(testChannel[0].CreateTextMessage("A"));
            }

            _finishedEvent.WaitOne(new TimeSpan(0, 0, 0, 10), false);

            // Check that all messages really were received.
            Assert.IsTrue(allReceived, "All messages were not received, only got: " + _messageReceivedCount + " but wanted " + expectedMessageCount);
        }

        /// <summary> Atomically increments the message count on every message, and signals once all messages in the test are received. </summary>
        public void OnMessage(IMessage m)
        {
            int newCount = Interlocked.Increment(ref _messageReceivedCount);

            if (newCount >= expectedMessageCount)
            {
                allReceived = true;
                _finishedEvent.Set();
            }
        }        

        /// <summary>Prompts the user on stdout and waits for a reply on stdin, using the specified prompt message.</summary>
        ///
        /// <param name="message">The message to prompt the user with.</param>
        private void PromptAndWait(string message)
        {
            Console.WriteLine("\n" + message);
            Console.ReadLine();
        }
    }
}