[jboss-cvs] JBoss Messaging SVN: r4624 - trunk/tests/src/org/jboss/messaging/tests/unit/core/remoting/impl/mina.

[jboss-cvs-commits at lists.jboss.org]

Author: clebert.suconic at jboss.com
Date: 2008-06-27 18:21:46 -0400 (Fri, 27 Jun 2008)
New Revision: 4624

Added:
   trunk/tests/src/org/jboss/messaging/tests/unit/core/remoting/impl/mina/MinaHandlerTest.java
Log:
Unit test for MinaHandler

Added: trunk/tests/src/org/jboss/messaging/tests/unit/core/remoting/impl/mina/MinaHandlerTest.java
===================================================================
--- trunk/tests/src/org/jboss/messaging/tests/unit/core/remoting/impl/mina/MinaHandlerTest.java	                        (rev 0)
+++ trunk/tests/src/org/jboss/messaging/tests/unit/core/remoting/impl/mina/MinaHandlerTest.java	2008-06-27 22:21:46 UTC (rev 4624)
@@ -0,0 +1,157 @@
+/*
+ * JBoss, Home of Professional Open Source
+ * Copyright 2005-2008, Red Hat Middleware LLC, and individual contributors
+ * by the @authors tag. See the copyright.txt in the distribution for a
+ * full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+
+
+package org.jboss.messaging.tests.unit.core.remoting.impl.mina;
+
+import java.util.concurrent.ExecutorService;
+
+import org.apache.mina.common.IoSession;
+import org.easymock.EasyMock;
+import org.jboss.messaging.core.exception.MessagingException;
+import org.jboss.messaging.core.remoting.CleanUpNotifier;
+import org.jboss.messaging.core.remoting.PacketDispatcher;
+import org.jboss.messaging.core.remoting.impl.mina.MinaHandler;
+import org.jboss.messaging.tests.util.UnitTestCase;
+
+/**
+ * 
+ * @author <a href="mailto:clebert.suconic at jboss.com">Clebert Suconic</a>
+ *
+ */
+public class MinaHandlerTest extends UnitTestCase
+{
+   
+   // Constants -----------------------------------------------------
+   
+   // Attributes ----------------------------------------------------
+   
+   private PacketDispatcher dispatcher;
+   
+   private CleanUpNotifier failureNotifier;
+   
+   private ExecutorService executorService;
+   
+   // Static --------------------------------------------------------
+   
+   // Constructors --------------------------------------------------
+   
+   // Public --------------------------------------------------------
+   
+   
+   public void setUp()
+   {
+      dispatcher = null;
+      
+      failureNotifier = null;
+      
+      executorService = null;
+   }
+   
+   
+   public void testConstructor() throws Exception
+   {
+      newMinaHandler(true);
+      
+      try
+      {
+         new MinaHandler(null, executorService, failureNotifier, true, true);
+         fail("Supposed to throw exception");
+      }
+      catch (IllegalArgumentException e)
+      {
+         
+      }
+
+      try
+      {
+         new MinaHandler(dispatcher, null, failureNotifier, true, true);
+         fail("Supposed to throw exception");
+      }
+      catch (IllegalArgumentException e)
+      {
+         
+      }
+
+      // As the executor is not being used, it shouldn't throw an exception
+      new MinaHandler(dispatcher, null, failureNotifier, true, false);
+   }
+
+   public void testExceptionCaught() throws Exception
+   {
+      MinaHandler handler = newMinaHandler(true);
+      
+      IoSession session = EasyMock.createMock(IoSession.class);
+      
+      EasyMock.expect(session.getId()).andStubReturn(1l);
+      
+      // If the method is refactored to use the Future, this might have to be changed
+      EasyMock.expect(session.close()).andReturn(null);
+      
+      failureNotifier.fireCleanup(EasyMock.eq(1l), messagingExceptionMatch(MessagingException.INTERNAL_ERROR));
+      
+      EasyMock.replay(dispatcher, failureNotifier, executorService, session);
+      
+      handler.exceptionCaught(session, new Exception("anything"));
+
+      EasyMock.verify(dispatcher, failureNotifier, executorService, session);
+      
+   }
+   
+   public void testMessageReceived() throws Exception
+   {
+      
+   }
+
+
+   // Package protected ---------------------------------------------
+   
+   // Protected -----------------------------------------------------
+   
+   // Private -------------------------------------------------------
+   
+   private MinaHandler newMinaHandler(boolean closeSessionOnExceptionCaught)
+   {
+      dispatcher = EasyMock.createMock(PacketDispatcher.class);
+         
+      failureNotifier = EasyMock.createMock(CleanUpNotifier.class);
+      
+      if (executorService == null)
+      {
+         executorService = EasyMock.createMock(ExecutorService.class);
+      }
+      
+      dispatcher.setListener(EasyMock.isA(MinaHandler.class));
+      
+      EasyMock.replay(dispatcher, failureNotifier, executorService);
+      
+      MinaHandler handler = new MinaHandler(dispatcher, executorService, failureNotifier, closeSessionOnExceptionCaught, true);
+      
+      EasyMock.verify(dispatcher, failureNotifier, executorService);
+      
+      EasyMock.reset(dispatcher, failureNotifier, executorService);
+      return handler;
+   }
+   
+   // Inner classes -------------------------------------------------
+   
+}