/protocols/smpp/src/test/java/org/mobicents/protocols/smpp/event/TaskExecutorEventDispatcherTest.java
Java | 77 lines | 49 code | 7 blank | 21 comment | 0 complexity | cf19d0c818ac47d8734f1824db5c2a94 MD5 | raw file
1/* 2 * JBoss, Home of Professional Open Source 3 * Copyright 2011, Red Hat, Inc. and individual contributors 4 * by the @authors tag. See the copyright.txt in the distribution for a 5 * full listing of individual contributors. 6 * 7 * This is free software; you can redistribute it and/or modify it 8 * under the terms of the GNU Lesser General Public License as 9 * published by the Free Software Foundation; either version 2.1 of 10 * the License, or (at your option) any later version. 11 * 12 * This software is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15 * Lesser General Public License for more details. 16 * 17 * You should have received a copy of the GNU Lesser General Public 18 * License along with this software; if not, write to the Free 19 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 20 * 02110-1301 USA, or see the FSF site: http://www.fsf.org. 21 */ 22 23package org.mobicents.protocols.smpp.event; 24 25import java.util.concurrent.ExecutorService; 26import java.util.concurrent.TimeUnit; 27 28import org.easymock.EasyMock; 29import org.testng.annotations.Test; 30 31import org.mobicents.protocols.smpp.Session; 32import org.mobicents.protocols.smpp.event.ReceiverExitEvent; 33import org.mobicents.protocols.smpp.event.SMPPEvent; 34import org.mobicents.protocols.smpp.event.SessionObserver; 35import org.mobicents.protocols.smpp.event.TaskExecutorEventDispatcher; 36import org.mobicents.protocols.smpp.message.EnquireLink; 37import org.mobicents.protocols.smpp.message.SMPPPacket; 38 39@Test 40public class TaskExecutorEventDispatcherTest { 41 42 public void testNotifyEventWithDefaultExecutor() throws Exception { 43 TaskExecutorEventDispatcher dispatcher = new TaskExecutorEventDispatcher(); 44 dispatcher.setThreadCount(1); 45 dispatcher.init(); 46 SessionObserver observer1 = EasyMock.createMock(SessionObserver.class); 47 observer1.update( 48 (Session) EasyMock.isNull(), 49 EasyMock.isA(SMPPEvent.class)); 50 EasyMock.replay(observer1); 51 dispatcher.addObserver(observer1); 52 dispatcher.notifyObservers(null, new ReceiverExitEvent(null)); 53 ExecutorService executorService = 54 (ExecutorService) dispatcher.getExecutor(); 55 executorService.shutdown(); 56 executorService.awaitTermination(10, TimeUnit.SECONDS); 57 EasyMock.verify(observer1); 58 } 59 60 public void testNotifyPacketWithDefaultExecutor() throws Exception { 61 TaskExecutorEventDispatcher dispatcher = new TaskExecutorEventDispatcher(); 62 dispatcher.setThreadCount(1); 63 dispatcher.init(); 64 SessionObserver observer1 = EasyMock.createMock(SessionObserver.class); 65 observer1.packetReceived( 66 (Session) EasyMock.isNull(), 67 EasyMock.isA(SMPPPacket.class)); 68 EasyMock.replay(observer1); 69 dispatcher.addObserver(observer1); 70 dispatcher.notifyObservers(null, new EnquireLink()); 71 ExecutorService executorService = 72 (ExecutorService) dispatcher.getExecutor(); 73 executorService.shutdown(); 74 executorService.awaitTermination(10, TimeUnit.SECONDS); 75 EasyMock.verify(observer1); 76 } 77}