diff --git a/src/main/java/net/floodlightcontroller/core/module/FloodlightModuleContext.java b/src/main/java/net/floodlightcontroller/core/module/FloodlightModuleContext.java index 207995cbef23d737f89b1bd176e3ce6752c9466b..9b8533d835bb349f32a74b394202f0479c73d22b 100644 --- a/src/main/java/net/floodlightcontroller/core/module/FloodlightModuleContext.java +++ b/src/main/java/net/floodlightcontroller/core/module/FloodlightModuleContext.java @@ -72,6 +72,7 @@ public class FloodlightModuleContext implements IFloodlightModuleContext { public void createConfigMaps(Set<IFloodlightModule> moduleSet) { for (IFloodlightModule mod : moduleSet) { Map<String, String> moduleParams = new HashMap<String, String>(); + System.out.println(); configParams.put(mod.getClass(), moduleParams); } } diff --git a/src/test/java/net/floodlightcontroller/core/internal/ControllerTest.java b/src/test/java/net/floodlightcontroller/core/internal/ControllerTest.java index d6af6077034103987e7fcd9eda0b271e79a1aa99..e0576fa0d1d031f68276d34b7c09c596441074bc 100644 --- a/src/test/java/net/floodlightcontroller/core/internal/ControllerTest.java +++ b/src/test/java/net/floodlightcontroller/core/internal/ControllerTest.java @@ -24,6 +24,7 @@ import java.util.Date; import java.util.HashMap; import java.util.HashSet; import java.util.List; +import java.util.Set; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.Future; import java.util.concurrent.TimeUnit; @@ -37,6 +38,7 @@ import net.floodlightcontroller.core.IOFMessageListener.Command; import net.floodlightcontroller.core.IOFSwitch; import net.floodlightcontroller.core.OFMessageFilterManager; import net.floodlightcontroller.core.module.FloodlightModuleContext; +import net.floodlightcontroller.core.module.IFloodlightModule; import net.floodlightcontroller.core.test.MockFloodlightProvider; import net.floodlightcontroller.counter.CounterStore; import net.floodlightcontroller.counter.ICounterStoreService; @@ -80,17 +82,30 @@ public class ControllerTest extends FloodlightTestCase { public void setUp() throws Exception { super.setUp(); FloodlightModuleContext fmc = new FloodlightModuleContext(); + Set<IFloodlightModule> modSet = new HashSet<IFloodlightModule>(); + FloodlightProvider cm = new FloodlightProvider(); controller = (Controller)cm.getServiceImpls().get(IFloodlightProviderService.class); fmc.addService(IFloodlightProviderService.class, controller); + modSet.add(cm); + MemoryStorageSource memstorage = new MemoryStorageSource(); fmc.addService(IStorageSourceService.class, memstorage); + modSet.add(memstorage); + RestApiServer restApi = new RestApiServer(); fmc.addService(IRestApiService.class, restApi); + modSet.add(restApi); + CounterStore cs = new CounterStore(); fmc.addService(ICounterStoreService.class, cs); + modSet.add(cs); + PktInProcessingTime ppt = new PktInProcessingTime(); fmc.addService(IPktInProcessingTimeService.class, ppt); + modSet.add(ppt); + + fmc.createConfigMaps(modSet); ppt.init(fmc); restApi.init(fmc); memstorage.init(fmc); diff --git a/src/test/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImplTest.java b/src/test/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImplTest.java index 229e77dc5dadedd7ff0384f6739a75b2cd48469d..14cfe32e0c90d9a88ddc779bfcc3f0c454ebd39c 100644 --- a/src/test/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImplTest.java +++ b/src/test/java/net/floodlightcontroller/devicemanager/internal/DeviceManagerImplTest.java @@ -19,7 +19,9 @@ package net.floodlightcontroller.devicemanager.internal; import java.util.Date; import java.util.HashMap; +import java.util.HashSet; import java.util.Map; +import java.util.Set; import static org.easymock.EasyMock.createMock; import static org.easymock.EasyMock.createNiceMock; @@ -31,6 +33,7 @@ import static org.easymock.EasyMock.verify; import net.floodlightcontroller.core.IFloodlightProviderService; import net.floodlightcontroller.core.IOFSwitch; import net.floodlightcontroller.core.module.FloodlightModuleContext; +import net.floodlightcontroller.core.module.IFloodlightModule; import net.floodlightcontroller.core.test.MockFloodlightProvider; import net.floodlightcontroller.devicemanager.Device; import net.floodlightcontroller.devicemanager.DeviceAttachmentPoint; @@ -74,14 +77,20 @@ public class DeviceManagerImplTest extends FloodlightTestCase { super.setUp(); FloodlightModuleContext fmc = new FloodlightModuleContext(); + Set<IFloodlightModule> modSet = new HashSet<IFloodlightModule>(); RestApiServer restApi = new RestApiServer(); + modSet.add(restApi); mockFloodlightProvider = getMockFloodlightProvider(); + modSet.add(mockFloodlightProvider); deviceManager = new DeviceManagerImpl(); + modSet.add(deviceManager); fmc.addService(IDeviceManagerService.class, deviceManager); storageSource = new MemoryStorageSource(); + modSet.add(storageSource); fmc.addService(IStorageSourceService.class, storageSource); fmc.addService(IFloodlightProviderService.class, mockFloodlightProvider); fmc.addService(IRestApiService.class, restApi); + fmc.createConfigMaps(modSet); restApi.init(fmc); storageSource.init(fmc); deviceManager.init(fmc); diff --git a/src/test/java/net/floodlightcontroller/staticflowentry/StaticFlowTests.java b/src/test/java/net/floodlightcontroller/staticflowentry/StaticFlowTests.java index 65c88294846203f066eca5c0242fa02f425482ee..b9c06a4aaf9637ce38cbeadfadd577cb8d42ad35 100644 --- a/src/test/java/net/floodlightcontroller/staticflowentry/StaticFlowTests.java +++ b/src/test/java/net/floodlightcontroller/staticflowentry/StaticFlowTests.java @@ -23,7 +23,9 @@ import org.openflow.util.HexString; import net.floodlightcontroller.core.FloodlightContext; import net.floodlightcontroller.core.IOFSwitch; +import net.floodlightcontroller.core.module.FloodlightModuleContext; import net.floodlightcontroller.core.module.FloodlightModuleException; +import net.floodlightcontroller.core.module.IFloodlightModule; import net.floodlightcontroller.core.test.MockFloodlightProvider; import net.floodlightcontroller.test.FloodlightTestCase; import net.floodlightcontroller.restserver.RestApiServer; @@ -176,6 +178,9 @@ public class StaticFlowTests extends FloodlightTestCase { staticFlowEntryPusher.setStorageSource(storage); + FloodlightModuleContext fmc = new FloodlightModuleContext(); + Set<IFloodlightModule> modSet = new HashSet<IFloodlightModule>(); + MockFloodlightProvider mockFloodlightProvider = getMockFloodlightProvider(); Map<Long, IOFSwitch> switchMap = new HashMap<Long, IOFSwitch>(); switchMap.put(dpid, mockSwitch); @@ -183,8 +188,10 @@ public class StaticFlowTests extends FloodlightTestCase { mockFloodlightProvider.setSwitches(switchMap); staticFlowEntryPusher.setFloodlightProvider(mockFloodlightProvider); RestApiServer restApi = new RestApiServer(); + modSet.add(restApi); + fmc.createConfigMaps(modSet); try { - restApi.init(null); + restApi.init(fmc); } catch (FloodlightModuleException e) { e.printStackTrace(); } diff --git a/src/test/java/net/floodlightcontroller/storage/memory/tests/MemoryStorageTest.java b/src/test/java/net/floodlightcontroller/storage/memory/tests/MemoryStorageTest.java index c2500662218deb316fcb2cb620f766c04012c3a6..127a9f0f7f7e3d3f7e76d54d53141848cc15bd39 100644 --- a/src/test/java/net/floodlightcontroller/storage/memory/tests/MemoryStorageTest.java +++ b/src/test/java/net/floodlightcontroller/storage/memory/tests/MemoryStorageTest.java @@ -17,7 +17,11 @@ package net.floodlightcontroller.storage.memory.tests; +import java.util.HashSet; +import java.util.Set; + import net.floodlightcontroller.core.module.FloodlightModuleContext; +import net.floodlightcontroller.core.module.IFloodlightModule; import net.floodlightcontroller.restserver.IRestApiService; import net.floodlightcontroller.restserver.RestApiServer; import net.floodlightcontroller.storage.memory.MemoryStorageSource; @@ -32,6 +36,10 @@ public class MemoryStorageTest extends StorageTest { restApi = new RestApiServer(); FloodlightModuleContext fmc = new FloodlightModuleContext(); fmc.addService(IRestApiService.class, restApi); + Set<IFloodlightModule> modSet = new HashSet<IFloodlightModule>(); + modSet.add(restApi); + modSet.add(storageSource); + fmc.createConfigMaps(modSet); restApi.init(fmc); storageSource.init(fmc); restApi.startUp(fmc);