commit 9f383347d57df6a026f610d60f236a33077e5f32
author srikanth <srikanthvavila@gmail.com> Thu Jun 12 11:49:00 2014 -0700
committer Gerrit Code Review <gerrit2@gerrit.onos.onlab.us> Thu Jun 12 13:07:19 2014 -0700
tree 30424ffcab9bfd305f8409509160d9300bcf530c
parent ee69895eef2d83bd22fa318bbb18a870ca50ae84

Adding Unit Testcase for LinkDiscoverManager.sendDiscoveryMessage()

Change-Id: I6cef56bc3e6c9d321da406be474856c3809e4300

src/test/java/net/onrc/onos/core/linkdiscovery/internal/LinkDiscoveryManagerTest.java

diff --git a/src/test/java/net/onrc/onos/core/linkdiscovery/internal/LinkDiscoveryManagerTest.java b/src/test/java/net/onrc/onos/core/linkdiscovery/internal/LinkDiscoveryManagerTest.java
index af29d76..c343631 100644
--- a/src/test/java/net/onrc/onos/core/linkdiscovery/internal/LinkDiscoveryManagerTest.java
+++ b/src/test/java/net/onrc/onos/core/linkdiscovery/internal/LinkDiscoveryManagerTest.java
@@ -22,11 +22,13 @@
 import static org.easymock.EasyMock.replay;
 import static org.easymock.EasyMock.verify;
 
+import java.io.IOException;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
 
+import net.floodlightcontroller.core.FloodlightContext;
 import net.floodlightcontroller.core.IFloodlightProviderService;
 import net.floodlightcontroller.core.IOFSwitch;
 import net.floodlightcontroller.core.module.FloodlightModuleContext;
@@ -41,8 +43,11 @@
 import net.onrc.onos.core.linkdiscovery.LinkInfo;
 import net.onrc.onos.core.linkdiscovery.NodePortTuple;
 
+import org.easymock.EasyMock;
 import org.junit.Before;
 import org.junit.Test;
+import org.openflow.protocol.OFMessage;
+import org.openflow.protocol.OFPhysicalPort;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -379,4 +384,41 @@
         assertTrue(topology.portBroadcastDomainLinks.get(srcNpt).contains(lt));
         assertTrue(topology.portBroadcastDomainLinks.get(dstNpt).contains(lt));
     }
+
+    /**
+     * This test case verifies that LinkDiscoveryManager.sendDiscoveryMessage()
+     * performs "write" operation on the specified IOFSwitch object 
+     * with a LLDP packet
+     * 
+     * @throws IOException
+     */
+    @Test
+    public void testSendDiscoveryMessage() throws IOException{
+        LinkDiscoveryManager topology = getTopology();
+
+        // Mock up our expected behavior
+        IOFSwitch swTest = createMockSwitch(3L);
+        getMockFloodlightProvider().getSwitches().put(3L, swTest);
+        
+        short portNum = 1;
+        OFPhysicalPort ofpPort = new OFPhysicalPort();
+        ofpPort.setPortNumber(portNum);
+        
+        /* sendDiscoverMessage() should perform the following actions on 
+         * IOFSwitch object
+         * 	- getPort() with argument as "1"
+         * 	- write() with OFPacketOut
+         * 	- flush()
+         */
+        expect(swTest.getPort(portNum)).andReturn(ofpPort).atLeastOnce();
+        swTest.write(EasyMock.anyObject(OFMessage.class), EasyMock.anyObject(FloodlightContext.class));
+        EasyMock.expectLastCall().times(1);
+        swTest.flush();
+        EasyMock.expectLastCall().once();
+        replay(swTest);
+        
+        topology.sendDiscoveryMessage(3L, portNum, true, false);
+        
+        verify(swTest);
+    }
 }