From 2ef2e881825e11bdfad902883c16fb2db9c5706e Mon Sep 17 00:00:00 2001
From: Alex Reimers <alex@bigswitch.com>
Date: Fri, 10 Feb 2012 13:29:48 -0800
Subject: [PATCH] Create the port channel table in DeviceManager.

---
 .../internal/DeviceManagerImpl.java              | 16 +++-------------
 .../internal/DeviceManagerImplTest.java          | 10 +++++-----
 2 files changed, 8 insertions(+), 18 deletions(-)

diff --git a/src/main/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImpl.java b/src/main/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImpl.java
index eca3a45a9..3061253db 100755
--- a/src/main/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImpl.java
+++ b/src/main/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImpl.java
@@ -1271,18 +1271,6 @@ public class DeviceManagerImpl implements IDeviceManagerService, IOFMessageListe
         // no-op
     }
 
-    /**
-     * @param deviceManagerAware the deviceManagerAware to set
-     */
-    public void setDeviceManagerAware(Set<IDeviceManagerAware> 
-                                                        deviceManagerAware) {
-        this.deviceManagerAware = deviceManagerAware;
-    }
-
-    public void setStorageSource(IStorageSource storageSource) {
-        this.storageSource = storageSource;
-    }
-
     /**
      * Process device manager aware updates.  Call without any lock held
      */
@@ -2139,7 +2127,9 @@ public class DeviceManagerImpl implements IDeviceManagerService, IOFMessageListe
         storageSource.createTable(DEVICE_NETWORK_ADDRESS_TABLE_NAME, null);
         storageSource.setTablePrimaryKeyName(
                         DEVICE_NETWORK_ADDRESS_TABLE_NAME, ID_COLUMN_NAME);
-        
+        storageSource.createTable(PORT_CHANNEL_TABLE_NAME, null);
+        storageSource.setTablePrimaryKeyName(
+                        PORT_CHANNEL_TABLE_NAME, PC_ID_COLUMN_NAME);
         storageSource.addListener(PORT_CHANNEL_TABLE_NAME, this);
 
         ScheduledExecutorService ses = floodlightProvider.getScheduledExecutor();
diff --git a/src/test/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImplTest.java b/src/test/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImplTest.java
index b03d77cd1..27a2942a1 100644
--- a/src/test/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImplTest.java
+++ b/src/test/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImplTest.java
@@ -36,7 +36,7 @@ import net.floodlightcontroller.packet.ARP;
 import net.floodlightcontroller.packet.Ethernet;
 import net.floodlightcontroller.packet.IPacket;
 import net.floodlightcontroller.packet.IPv4;
-import net.floodlightcontroller.storage.IStorageSource;
+import net.floodlightcontroller.storage.IStorageSourceService;
 import net.floodlightcontroller.storage.memory.MemoryStorageSource;
 import net.floodlightcontroller.test.FloodlightTestCase;
 import net.floodlightcontroller.topology.ITopologyService;
@@ -61,7 +61,7 @@ public class DeviceManagerImplTest extends FloodlightTestCase {
     private OFPacketIn thirdPacketIn;
     MockFloodlightProvider mockFloodlightProvider;
     DeviceManagerImpl deviceManager;
-    IStorageSource storageSource;
+    IStorageSourceService storageSource;
     
     @Before
     public void setUp() throws Exception {
@@ -70,7 +70,8 @@ public class DeviceManagerImplTest extends FloodlightTestCase {
         mockFloodlightProvider = getMockFloodlightProvider();
         deviceManager = new DeviceManagerImpl();
         deviceManager.setFloodlightProvider(mockFloodlightProvider);
-        deviceManager.storageSource = new MemoryStorageSource();
+        storageSource = new MemoryStorageSource();
+        deviceManager.storageSource = storageSource;
         deviceManager.startUp(null);
         
         // Build our test packet
@@ -415,7 +416,6 @@ public class DeviceManagerImplTest extends FloodlightTestCase {
     }
     
     private void setupPortChannel() {
-        
         storageSource.insertRow(DeviceManagerImpl.PORT_CHANNEL_TABLE_NAME, pcPort1);
         storageSource.insertRow(DeviceManagerImpl.PORT_CHANNEL_TABLE_NAME, pcPort2);
         deviceManager.readPortChannelConfigFromStorage();
@@ -443,7 +443,7 @@ public class DeviceManagerImplTest extends FloodlightTestCase {
         IOFSwitch mockSwitch = createMock(IOFSwitch.class);
         expect(mockSwitch.getId()).andReturn(1L).anyTimes();
         expect(mockSwitch.getStringId()).andReturn("00:00:00:00:00:00:00:01").anyTimes();
-        ITopology mockTopology = createMock(ITopology.class);
+        ITopologyService mockTopology = createMock(ITopologyService.class);
         expect(mockTopology.isInternal(new SwitchPortTuple(mockSwitch, 1)))
                            .andReturn(false).atLeastOnce();
         expect(mockTopology.isInternal(new SwitchPortTuple(mockSwitch, 2)))
-- 
GitLab