package net.tomp2p.rpc;

import net.tomp2p.connection.ChannelCreator;
import net.tomp2p.futures.FutureResponse;
import net.tomp2p.p2p.Peer;
import net.tomp2p.peers.Number160;

import org.junit.Assert;
import org.junit.Test;

public class TestReservation {
	@Test
	public void testReservationTCP() throws Exception
	{
		Peer sender = null;
		Peer recv1 = null;
		try
		{
			sender = new Peer(55, new Number160("0x9876"));
			sender.listen(2424, 2424);
			recv1 = new Peer(55, new Number160("0x1234"));
			recv1.listen(8088, 8088);
			ChannelCreator cc=recv1.getConnectionBean().getReservation().reserve(3);
			for(int i=0;i<1000;i++)
			{
				FutureResponse fr1 = sender.getHandshakeRPC().pingTCP(recv1.getPeerAddress(), cc);
				FutureResponse fr2 = sender.getHandshakeRPC().pingTCP(recv1.getPeerAddress(), cc);
				FutureResponse fr3 = sender.getHandshakeRPC().pingTCP(recv1.getPeerAddress(), cc);
				fr1.awaitUninterruptibly();
				fr2.awaitUninterruptibly();
				fr3.awaitUninterruptibly();
				System.err.println(fr1.getFailedReason());
				Assert.assertEquals(true, fr1.isSuccess());
				Assert.assertEquals(true, fr2.isSuccess());
				Assert.assertEquals(true, fr3.isSuccess());
			}
			recv1.getConnectionBean().getReservation().release(cc);
		}
		finally
		{
			if (sender != null)
				sender.shutdown();
			if (recv1 != null)
				recv1.shutdown();
		}
	}
	@Test
	public void testReservationUDP() throws Exception
	{
		Peer sender = null;
		Peer recv1 = null;
		try
		{
			sender = new Peer(55, new Number160("0x9876"));
			sender.listen(2424, 2424);
			recv1 = new Peer(55, new Number160("0x1234"));
			recv1.listen(8088, 8088);
			ChannelCreator cc=recv1.getConnectionBean().getReservation().reserve(3);
			for(int i=0;i<1000;i++)
			{
				FutureResponse fr1 = sender.getHandshakeRPC().pingUDP(recv1.getPeerAddress(), cc);
				FutureResponse fr2 = sender.getHandshakeRPC().pingUDP(recv1.getPeerAddress(), cc);
				FutureResponse fr3 = sender.getHandshakeRPC().pingUDP(recv1.getPeerAddress(), cc);
				fr1.awaitUninterruptibly();
				fr2.awaitUninterruptibly();
				fr3.awaitUninterruptibly();
				System.err.println(fr1.getFailedReason());
				Assert.assertEquals(true, fr1.isSuccess());
				Assert.assertEquals(true, fr2.isSuccess());
				Assert.assertEquals(true, fr3.isSuccess());
			}
			recv1.getConnectionBean().getReservation().release(cc);
		}
		finally
		{
			if (sender != null)
				sender.shutdown();
			if (recv1 != null)
				recv1.shutdown();
		}
	}
}
