[infinispan-commits] Infinispan SVN: r1406 - trunk/core/src/test/java/org/infinispan/remoting/jgroups.

[infinispan-commits at lists.jboss.org]

Author: manik.surtani at jboss.com
Date: 2010-01-21 11:18:54 -0500 (Thu, 21 Jan 2010)
New Revision: 1406

Modified:
   trunk/core/src/test/java/org/infinispan/remoting/jgroups/ChannelLookupTest.java
Log:
Deals with JGroupos 2.9.x now

Modified: trunk/core/src/test/java/org/infinispan/remoting/jgroups/ChannelLookupTest.java
===================================================================
--- trunk/core/src/test/java/org/infinispan/remoting/jgroups/ChannelLookupTest.java	2010-01-21 14:37:48 UTC (rev 1405)
+++ trunk/core/src/test/java/org/infinispan/remoting/jgroups/ChannelLookupTest.java	2010-01-21 16:18:54 UTC (rev 1406)
@@ -13,6 +13,8 @@
 import org.jgroups.Address;
 import org.jgroups.Channel;
 import org.jgroups.JChannel;
+import org.jgroups.protocols.UDP;
+import org.jgroups.stack.ProtocolStack;
 import org.testng.annotations.Test;
 
 import java.util.Properties;
@@ -42,11 +44,14 @@
     public static class DummyLookup implements JGroupsChannelLookup {
         Channel mockChannel;
         Address a = EasyMock.createNiceMock(Address.class);
+        ProtocolStack ps = EasyMock.createNiceMock(ProtocolStack.class);
 
         public DummyLookup() {
             mockChannel = EasyMock.createNiceMock(Channel.class);
             EasyMock.expect(mockChannel.getAddress()).andReturn(a);
-            EasyMock.replay(mockChannel, a);
+            EasyMock.expect(mockChannel.getProtocolStack()).andReturn(ps);
+            EasyMock.expect(ps.getTransport()).andReturn(new UDP());
+            EasyMock.replay(mockChannel, a, ps);           
         }
 
         public Channel getJGroupsChannel() {