joram.amqp.PersistenceSimpleTest.java Source code

Introduction

Here is the source code for joram.amqp.PersistenceSimpleTest.java

Source

/*
 * JORAM: Java(TM) Open Reliable Asynchronous Messaging
 * Copyright (C) 2011-2012 ScalAgent Distributed Technologies
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
 * USA.
 *
 * Initial developer(s): ScalAgent Distributed Technologies
 * Contributor(s): 
 */
package joram.amqp;

import com.rabbitmq.client.AMQP.Queue.DeclareOk;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.GetResponse;
import com.rabbitmq.client.MessageProperties;
import com.rabbitmq.client.QueueingConsumer;
import com.rabbitmq.client.QueueingConsumer.Delivery;

import framework.TestCase;

public class PersistenceSimpleTest extends TestCase {

    public static void main(String[] args) {
        new PersistenceSimpleTest().run();
    }

    public void run() {
        try {
            startAgentServer((short) 0);
            recover1();

            //deleteDirectory(new File("s0"));
            //TODO
            //recover2();

        } catch (Throwable exc) {
            exc.printStackTrace();
            error(exc);
        } finally {
            System.out.println("finaly kill servers.");
            killAgentServer((short) 0);
            endTest();
        }
    }

    public void recover1() throws Exception {
        ConnectionFactory cnxFactory = new ConnectionFactory();
        Connection connection = cnxFactory.newConnection();

        Channel channel = connection.createChannel();
        DeclareOk declareOk = channel.queueDeclare("testqueue", true, false, false, null);

        channel.txSelect();
        for (int i = 0; i < 5; i++) {
            channel.basicPublish("", declareOk.getQueue(), MessageProperties.PERSISTENT_BASIC,
                    "this is a test message !!!".getBytes());
        }
        channel.txCommit();

        killAgentServer((short) 0);

        startAgentServer((short) 0);

        connection = cnxFactory.newConnection();
        channel = connection.createChannel();
        declareOk = channel.queueDeclarePassive("testqueue");

        for (int i = 0; i < 5; i++) {
            GetResponse msg = channel.basicGet("testqueue", true);
            assertNotNull(msg);
            assertEquals("this is a test message !!!", new String(msg.getBody()));
        }

        GetResponse msg = channel.basicGet("testqueue", true);
        assertNull(msg);

        channel.queueDelete(declareOk.getQueue());
    }

    public void recover2() throws Exception {
        ConnectionFactory cnxFactory = new ConnectionFactory();
        Connection connection = cnxFactory.newConnection();

        Channel channel = connection.createChannel();
        DeclareOk declareOk = channel.queueDeclare("testqueue", true, false, false, null);

        channel.txSelect();
        for (int i = 0; i < 5; i++) {
            channel.basicPublish("", declareOk.getQueue(), MessageProperties.PERSISTENT_BASIC,
                    "this is a test message !!!".getBytes());
        }
        channel.txCommit();

        killAgentServer((short) 0);

        startAgentServer((short) 0);
        Thread.sleep(500);

        connection = cnxFactory.newConnection();
        channel = connection.createChannel();
        declareOk = channel.queueDeclarePassive("testqueue");

        QueueingConsumer consumer = new QueueingConsumer(channel);
        channel.basicConsume(declareOk.getQueue(), true, consumer);

        for (int i = 0; i < 5; i++) {
            Delivery msg = consumer.nextDelivery(1000);
            assertNotNull(msg);
            assertEquals("this is a test message !!!", new String(msg.getBody()));
        }

        Delivery msg = consumer.nextDelivery(1000);
        assertNull(msg);

        channel.queueDelete(declareOk.getQueue());

    }

}