alfresco-community-repo/source/java/org/alfresco/repo/cluster/HazelcastTest.java

/*
 * Copyright (C) 2005-2012 Alfresco Software Limited.
 *
 * This file is part of Alfresco
 *
 * Alfresco 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 3 of the License, or
 * (at your option) any later version.
 *
 * Alfresco 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 Alfresco. If not, see <http://www.gnu.org/licenses/>.
 */

package org.alfresco.repo.cluster;


import org.alfresco.repo.cluster.MessengerTestHelper.TestMessageReceiver;
import org.alfresco.util.ApplicationContextHelper;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Ignore;
import org.junit.Test;
import org.springframework.context.ApplicationContext;

import com.hazelcast.core.Hazelcast;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.core.ITopic;
import com.hazelcast.core.MessageListener;

/**
 * Tests for Hazelcast implementations of {@link Messenger} and related classes.
 * These are integration tests and configured through a Spring test context file.
 * 
 * @author Matt Ward
 */
public class HazelcastTest implements MessageListener<String>
{
    private static ApplicationContext ctx;
    private MessengerTestHelper helper;
    private HazelcastInstanceFactory hiFactory;
    private HazelcastInstance hi;
    
    @BeforeClass
    public static void setUpClass()
    {
        ctx = ApplicationContextHelper.
                getApplicationContext(new String[] { "cluster-test/hazelcast-messenger-test.xml" });
    }
    
    @AfterClass
    public static void tearDownClass()
    {
        ApplicationContextHelper.closeApplicationContext();
        Hazelcast.shutdownAll();
    }
    
    @Before
    public void setUp()
    {
        helper = new MessengerTestHelper();
        hiFactory = ctx.getBean(HazelcastInstanceFactory.class);
        hi = hiFactory.getInstance();
    }
 
    
    @Test
    public void canSendWithHazelcastMessengerFactory() throws InterruptedException
    {
        ITopic<String> topic = hi.getTopic("testregion");
        
        topic.addMessageListener(this);
        
        MessengerFactory messengerFactory = (MessengerFactory) ctx.getBean("messengerFactory");
        Messenger<String> messenger = messengerFactory.createMessenger("testregion");
        messenger.send("Full test including spring.");
        
        helper.checkMessageReceivedWas("Full test including spring.");
    }

    @Ignore("Behaviour not yet implemented.")
    @Test
    public void messengerWillNotReceiveMessagesFromSelf() throws InterruptedException
    {
        MessengerFactory messengerFactory = (MessengerFactory) ctx.getBean("messengerFactory");
        Messenger<String> m1 = messengerFactory.createMessenger("testregion");
        TestMessageReceiver r1 = new TestMessageReceiver();
        m1.setReceiver(r1);
     
        ITopic<String> topic2 = hi.getTopic("testregion");
        String address2 = hi.getCluster().getLocalMember().getInetSocketAddress().toString();
        Messenger<String> m2 = new HazelcastMessenger<String>(topic2, address2);
        TestMessageReceiver r2 = new TestMessageReceiver();
        m2.setReceiver(r2);
        
        m1.send("This should be received by r2 but not r1");
        
        r2.helper.checkMessageReceivedWas("This should be received by r2 but not r1");
        r1.helper.checkNoMessageReceived();
    }

    @Override
    public void onMessage(String message)
    {
        helper.setReceivedMsg(message);
    }
}