/protocols/smpp/src/test/java/org/mobicents/protocols/smpp/event/TaskExecutorEventDispatcherTest.java

 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}