SslBalancerSslServerTest.java

/*
 * TeleStax, Open Source Cloud Communications
 * Copyright 2011-2015, Telestax Inc and individual contributors
 * by the @authors tag.
 *
 * This program is free software: you can redistribute it and/or modify
 * under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation; either version 3 of
 * the License, or (at your option) any later version.
 *
 * This program 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>
 */
package org.mobicents.tools.smpp.balancer;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.util.ArrayList;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.Semaphore;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.atomic.AtomicInteger;

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.mobicents.tools.sip.balancer.BalancerRunner;
import org.mobicents.tools.smpp.multiplexer.MBalancerDispatcher;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.cloudhopper.smpp.SmppSession;
import com.cloudhopper.smpp.impl.DefaultSmppClient;
import com.cloudhopper.smpp.impl.DefaultSmppServer;
import com.cloudhopper.smpp.type.SmppChannelException;
/**
 * @author Konstantin Nosach (kostyantyn.nosach@telestax.com)
 */
public class SslBalancerSslServerTest {
	private static final Logger logger = LoggerFactory
			.getLogger(CommonTest.class);

	private static ThreadPoolExecutor executor = (ThreadPoolExecutor) Executors.newCachedThreadPool();

	private static ScheduledThreadPoolExecutor monitorExecutor = (ScheduledThreadPoolExecutor) Executors.newScheduledThreadPool(1, new ThreadFactory() {
				private AtomicInteger sequence = new AtomicInteger(0);
				public Thread newThread(Runnable r) {
					Thread t = new Thread(r);
					t.setName("MonitorPool-" + sequence.getAndIncrement());
					return t;
				}
			});

	private static int clientNumbers = 6;
	private static int serverNumbers = 3;
	private static DefaultSmppServer [] serverArray;
	private static BalancerRunner balancer;
    private static DefaultSmppServerHandler [] serverHandlerArray = new DefaultSmppServerHandler[serverNumbers];
    private static DefaultSmppClientHandler [] clientHandlerArray = new DefaultSmppClientHandler[clientNumbers];

	@BeforeClass
	public static void initialization() {
		
		boolean enableSslLbPort = true;
		boolean terminateTLSTraffic = false;
		//start lb
		balancer = new BalancerRunner();
        balancer.start(ConfigInit.getLbProperties(enableSslLbPort,terminateTLSTraffic));
		// start servers
		serverArray = new DefaultSmppServer[serverNumbers];
		for (int i = 0; i < serverNumbers; i++) {
			serverHandlerArray[i] = new DefaultSmppServerHandler();
			serverArray[i] = new DefaultSmppServer(ConfigInit.getSmppServerConfiguration(i,true),serverHandlerArray[i], executor, monitorExecutor);
			logger.info("Starting SMPP server...");
			try {
				serverArray[i].start();
			} catch (SmppChannelException e) {
				logger.info("SMPP server does not started");
				e.printStackTrace();
			}
			logger.info("SMPP server started");
		}
		try {
			Thread.sleep(2000);
		} catch (InterruptedException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
	
	// tests SSL client connection to SSL server
	@Test
	public void testSslClient() {
		Locker locker = new Locker(clientNumbers);
		// start client
		
		ArrayList<Load> customers = new ArrayList<>();
		for(int i = 0; i < clientNumbers; i++)
			customers.add(new Load(i,locker,true));
		for(int i = 0; i < clientNumbers; i++)
			customers.get(i).start();
		
		locker.waitForClients();
	    
		long smsToServers = 0;
		for(DefaultSmppServerHandler serverHandler:serverHandlerArray)
			smsToServers+=serverHandler.smsNumber;
		assertEquals(clientNumbers, smsToServers);
		for(DefaultSmppClientHandler clientHandler : clientHandlerArray)
			assertEquals(1,clientHandler.getReponsesNumber().get());
		assertTrue(((MBalancerDispatcher)balancer.smppBalancerRunner.getBalancerDispatcher()).getUserSpaces().isEmpty());
		
	}

	@AfterClass
	public static void finalization() {

		for (int i = 0; i < serverNumbers; i++) {
			logger.info("Stopping SMPP server " + i + " ...");
			serverArray[i].destroy();
			logger.info("SMPP server " + i + " stopped");
		}
		executor.shutdownNow();
		monitorExecutor.shutdownNow();
		balancer.stop();
		logger.info("Done. Exiting");
		
	}

	private class Load extends Thread {
		private int i;
		private ClientListener listener;
		private boolean isSslClient;

		Load(int i, ClientListener listener, boolean isSslClient) {
			this.listener = listener;
			this.isSslClient = isSslClient;
			this.i = i;
		}

		public void run() {
			DefaultSmppClient client = new DefaultSmppClient();
			clientHandlerArray[i] = new DefaultSmppClientHandler();
			SmppSession session = null;
			try {
				session = client.bind(ConfigInit.getSmppSessionConfiguration(1,isSslClient),clientHandlerArray[i]);
				Thread.sleep(1000);
				session.submit(ConfigInit.getSubmitSm(), 12000);
				sleep(100);
				session.unbind(5000);
				sleep(100);
			} catch (Exception e) {
				logger.error("", e);
			}
			if (session != null) {
				logger.info("Cleaning up session...");
				session.destroy();
			}
			logger.info("Shutting down client bootstrap and executors...");
			client.destroy();
			listener.clientCompleted();
		}
	}

	private class Locker implements ClientListener {
		private Semaphore clientsSemaphore;

		private Locker(int clients) {
			clientsSemaphore = new Semaphore(1 - clients);
		}

		@Override
		public void clientCompleted() {
			clientsSemaphore.release();
		}

		public void waitForClients() {
			try {
				clientsSemaphore.acquire();
			} catch (InterruptedException ex) {

			}
		}
	}
}