package org.apache.catalina.tribes.test.membership;
import java.util.ArrayList;
import org.apache.catalina.tribes.ByteMessage;
import org.apache.catalina.tribes.Channel;
import org.apache.catalina.tribes.ChannelException;
import org.apache.catalina.tribes.ManagedChannel;
import org.apache.catalina.tribes.Member;
import org.apache.catalina.tribes.MembershipListener;
import org.apache.catalina.tribes.group.GroupChannel;
import org.apache.catalina.tribes.group.interceptors.TcpFailureDetector;
import junit.framework.TestCase;
/**
* <p>Title: </p>
*
* <p>Description: </p>
*
* <p>Copyright: Copyright (c) 2005</p>
*
* <p>Company: </p>
*
* @author not attributable
* @version 1.0
*/
public class TestTcpFailureDetector extends TestCase {
private TcpFailureDetector tcpFailureDetector1 = null;
private TcpFailureDetector tcpFailureDetector2 = null;
private ManagedChannel channel1 = null;
private ManagedChannel channel2 = null;
private TestMbrListener mbrlist1 = null;
private TestMbrListener mbrlist2 = null;
protected void setUp() throws Exception {
super.setUp();
channel1 = new GroupChannel();
channel2 = new GroupChannel();
channel1.getMembershipService().setPayload("Channel-1".getBytes("ASCII"));
channel2.getMembershipService().setPayload("Channel-2".getBytes("ASCII"));
mbrlist1 = new TestMbrListener("Channel-1");
mbrlist2 = new TestMbrListener("Channel-2");
tcpFailureDetector1 = new TcpFailureDetector();
tcpFailureDetector2 = new TcpFailureDetector();
channel1.addInterceptor(tcpFailureDetector1);
channel2.addInterceptor(tcpFailureDetector2);
channel1.addMembershipListener(mbrlist1);
channel2.addMembershipListener(mbrlist2);
}
public void clear() {
mbrlist1.members.clear();
mbrlist2.members.clear();
}
public void testTcpSendFailureMemberDrop() throws Exception {
System.out.println("testTcpSendFailureMemberDrop()");
clear();
channel1.start(channel1.DEFAULT);
channel2.start(channel2.DEFAULT);
//Thread.sleep(1000);
assertEquals("Expecting member count to be equal",mbrlist1.members.size(),mbrlist2.members.size());
channel2.stop(channel2.SND_RX_SEQ);
ByteMessage msg = new ByteMessage(new byte[1024]);
try {
channel1.send(channel1.getMembers(), msg, 0);
assertEquals("Message send should have failed.",true,false);
} catch ( ChannelException x ) {
}
assertEquals("Expecting member count to not be equal",mbrlist1.members.size()+1,mbrlist2.members.size());
channel1.stop(Channel.DEFAULT);
channel2.stop(Channel.DEFAULT);
}
public void testTcpFailureMemberAdd() throws Exception {
System.out.println("testTcpFailureMemberAdd()");
clear();
channel1.start(channel1.DEFAULT);
channel2.start(channel2.SND_RX_SEQ);
channel2.start(channel2.SND_TX_SEQ);
channel2.start(channel2.MBR_RX_SEQ);
channel2.stop(channel2.SND_RX_SEQ);
channel2.start(channel2.MBR_TX_SEQ);
//Thread.sleep(1000);
assertEquals("Expecting member count to not be equal",mbrlist1.members.size()+1,mbrlist2.members.size());
channel1.stop(Channel.DEFAULT);
channel2.stop(Channel.DEFAULT);
}
public void testTcpMcastFail() throws Exception {
System.out.println("testTcpMcastFail()");
clear();
channel1.start(channel1.DEFAULT);
channel2.start(channel2.DEFAULT);
//Thread.sleep(1000);
assertEquals("Expecting member count to be equal",mbrlist1.members.size(),mbrlist2.members.size());
channel2.stop(channel2.MBR_TX_SEQ);
ByteMessage msg = new ByteMessage(new byte[1024]);
try {
Thread.sleep(5000);
assertEquals("Expecting member count to be equal",mbrlist1.members.size(),mbrlist2.members.size());
channel1.send(channel1.getMembers(), msg, 0);
} catch ( ChannelException x ) {
assertEquals("Message send should have succeeded.",true,false);
}
channel1.stop(Channel.DEFAULT);
channel2.stop(Channel.DEFAULT);
}
protected void tearDown() throws Exception {
tcpFailureDetector1 = null;
tcpFailureDetector2 = null;
try { channel1.stop(Channel.DEFAULT);}catch (Exception ignore){}
channel1 = null;
try { channel2.stop(Channel.DEFAULT);}catch (Exception ignore){}
channel2 = null;
super.tearDown();
}
public class TestMbrListener implements MembershipListener {
public String name = null;
public TestMbrListener(String name) {
this.name = name;
}
public ArrayList members = new ArrayList();
public void memberAdded(Member member) {
if ( !members.contains(member) ) {
members.add(member);
try{
System.out.println(name + ":member added[" + new String(member.getPayload(), "ASCII") + "]");
}catch ( Exception x ) {
System.out.println(name + ":member added[unknown]");
}
}
}
public void memberDisappeared(Member member) {
if ( members.contains(member) ) {
members.remove(member);
try{
System.out.println(name + ":member disappeared[" + new String(member.getPayload(), "ASCII") + "]");
}catch ( Exception x ) {
System.out.println(name + ":member disappeared[unknown]");
}
}
}
}
}