| package io.teknek.gossip; |
| |
| import io.teknek.tunit.TUnit; |
| |
| import java.net.UnknownHostException; |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.concurrent.Callable; |
| import java.util.concurrent.TimeUnit; |
| |
| |
| import org.junit.Assert; |
| import org.junit.Test; |
| |
| import com.google.code.gossip.GossipMember; |
| import com.google.code.gossip.GossipService; |
| import com.google.code.gossip.GossipSettings; |
| import com.google.code.gossip.RemoteGossipMember; |
| import com.google.code.gossip.event.GossipListener; |
| import com.google.code.gossip.event.GossipState; |
| |
| public class TenNodeThreeSeedTest { |
| |
| @Test |
| public void test() throws UnknownHostException, InterruptedException{ |
| abc(); |
| } |
| |
| @Test |
| public void testAgain() throws UnknownHostException, InterruptedException{ |
| abc(); |
| } |
| |
| public void abc() throws InterruptedException, UnknownHostException{ |
| GossipSettings settings = new GossipSettings(); |
| int seedNodes = 3; |
| List<GossipMember> startupMembers = new ArrayList<>(); |
| for (int i = 1; i < seedNodes+1; ++i) { |
| startupMembers.add(new RemoteGossipMember("127.0.0." + i, 2000, i + "")); |
| } |
| final List<GossipService> clients = new ArrayList<>(); |
| final int clusterMembers = 5; |
| for (int i = 1; i < clusterMembers+1; ++i) { |
| GossipService gossipService = new GossipService("127.0.0." + i, 2000, i + "", |
| startupMembers, settings, |
| new GossipListener(){ |
| @Override |
| public void gossipEvent(GossipMember member, GossipState state) { |
| System.out.println(member+" "+ state); |
| } |
| }); |
| clients.add(gossipService); |
| gossipService.start(); |
| } |
| TUnit.assertThat(new Callable<Integer> (){ |
| public Integer call() throws Exception { |
| int total = 0; |
| for (int i = 0; i < clusterMembers; ++i) { |
| total += clients.get(i).get_gossipManager().getMemberList().size(); |
| } |
| return total; |
| }}).afterWaitingAtMost(10, TimeUnit.SECONDS).isEqualTo(20); |
| |
| for (int i = 0; i < clusterMembers; ++i) { |
| clients.get(i).shutdown(); |
| } |
| } |
| } |