/**
 * 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.
 */
package org.apache.activemq.bugs;

import java.io.File;
import javax.jms.Connection;
import javax.jms.DeliveryMode;
import javax.jms.Destination;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.MessageProducer;
import javax.jms.Session;
import javax.jms.TextMessage;
import junit.framework.TestCase;
import org.apache.activemq.ActiveMQConnectionFactory;
import org.apache.activemq.RedeliveryPolicy;
import org.apache.activemq.broker.BrokerPlugin;
import org.apache.activemq.broker.BrokerService;
import org.apache.activemq.broker.TransportConnector;
import org.apache.activemq.broker.region.policy.RedeliveryPolicyMap;
import org.apache.activemq.broker.util.RedeliveryPlugin;
import org.apache.activemq.util.IOHelper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Testing if the the broker "sends" the message as expected after the redeliveryPlugin has redelivered the
 * message previously.
 */

public class RedeliveryPluginHeaderTest extends TestCase {

    private static final String TEST_QUEUE_ONE = "TEST_QUEUE_ONE";
    private static final String TEST_QUEUE_TWO = "TEST_QUEUE_TWO";
    private static final Logger LOG = LoggerFactory
            .getLogger(RedeliveryPluginHeaderTest.class);
    private String transportURL;
    private BrokerService broker;

    /**
     * Test
     * - consumes message from Queue1
     * - rolls back message to Queue1 and message is scheduled for redelivery to Queue1 by brokers plugin
     * - consumes message from Queue1 again
     * - sends same message to Queue2
     * - expects to consume message from Queue2 immediately
     */

    public void testSendAfterRedelivery() throws Exception {
        broker = this.createBroker(false);
        broker.start();
        broker.waitUntilStarted();

        LOG.info("***Broker started...");

        //pushed message to broker

        ActiveMQConnectionFactory factory = new ActiveMQConnectionFactory(
                transportURL + "?trace=true&jms.redeliveryPolicy.maximumRedeliveries=0");

        Connection connection = factory.createConnection();
        connection.start();

        try {

            Session session = connection.createSession(true,
                    Session.SESSION_TRANSACTED);

            Destination destinationQ1 = session.createQueue(TEST_QUEUE_ONE);
            Destination destinationQ2 = session.createQueue(TEST_QUEUE_TWO);

            MessageProducer producerQ1 = session.createProducer(destinationQ1);
            producerQ1.setDeliveryMode(DeliveryMode.PERSISTENT);

            Message m = session.createTextMessage("testMessage");
            LOG.info("*** send message to broker...");
            producerQ1.send(m);
            session.commit();

            //consume message from Q1 and rollback to get it redelivered
            MessageConsumer consumerQ1 = session.createConsumer(destinationQ1);

            LOG.info("*** consume message from Q1 and rolled back..");

            TextMessage textMessage = (TextMessage) consumerQ1.receive();
            LOG.info("got redelivered: " + textMessage);
            assertFalse("JMSRedelivered flag is not set", textMessage.getJMSRedelivered());
            session.rollback();

            LOG.info("*** consumed message from Q1 again and sending to Q2..");
            TextMessage textMessage2 = (TextMessage) consumerQ1.receive();
            LOG.info("got: " + textMessage2);
            session.commit();
            assertTrue("JMSRedelivered flag is set", textMessage2.getJMSRedelivered());

            //send message to Q2 and consume from Q2
            MessageConsumer consumerQ2 = session.createConsumer(destinationQ2);
            MessageProducer producer_two = session.createProducer(destinationQ2);
            producer_two.send(textMessage2);
            session.commit();

            //Message should be available straight away on the queue_two
            Message textMessage3 = consumerQ2.receive(1000);
            assertNotNull("should have consumed a message from TEST_QUEUE_TWO", textMessage3);
            assertFalse("JMSRedelivered flag is not set", textMessage3.getJMSRedelivered());
            session.commit();

        } finally {

            if (connection != null) {
                connection.close();
            }

            if (broker != null) {
                broker.stop();
            }

        }

    }

    protected BrokerService createBroker(boolean withJMX) throws Exception {
        File schedulerDirectory = new File("target/scheduler");
        IOHelper.mkdirs(schedulerDirectory);
        IOHelper.deleteChildren(schedulerDirectory);

        BrokerService answer = new BrokerService();
        answer.setAdvisorySupport(false);
        answer.setDataDirectory("target");
        answer.setSchedulerDirectoryFile(schedulerDirectory);
        answer.setSchedulerSupport(true);
        answer.setPersistent(true);
        answer.setDeleteAllMessagesOnStartup(true);
        answer.setUseJmx(withJMX);

        RedeliveryPlugin redeliveryPlugin = new RedeliveryPlugin();
        RedeliveryPolicyMap redeliveryPolicyMap = new RedeliveryPolicyMap();
        RedeliveryPolicy defaultEntry = new RedeliveryPolicy();
        defaultEntry.setInitialRedeliveryDelay(5000);
        defaultEntry.setMaximumRedeliveries(5);
        redeliveryPolicyMap.setDefaultEntry(defaultEntry);
        redeliveryPlugin.setRedeliveryPolicyMap(redeliveryPolicyMap);

        answer.setPlugins(new BrokerPlugin[] {redeliveryPlugin});
        TransportConnector transportConnector =
                answer.addConnector("tcp://localhost:0");

        transportURL = transportConnector.getConnectUri().toASCIIString();

        return answer;
    }

}