| package net.onrc.onos.intent.runtime; |
| |
| import net.floodlightcontroller.core.module.FloodlightModuleContext; |
| import net.floodlightcontroller.core.module.FloodlightModuleException; |
| import net.onrc.onos.datagrid.IDatagridService; |
| import net.onrc.onos.datagrid.IEventChannel; |
| import net.onrc.onos.intent.ConstrainedShortestPathIntent; |
| import net.onrc.onos.intent.Intent; |
| import net.onrc.onos.intent.IntentOperation.Operator; |
| import net.onrc.onos.intent.IntentOperationList; |
| import net.onrc.onos.intent.MockNetworkGraph; |
| import net.onrc.onos.intent.PathIntent; |
| import net.onrc.onos.intent.PathIntentMap; |
| import net.onrc.onos.intent.ShortestPathIntent; |
| import net.onrc.onos.ofcontroller.networkgraph.INetworkGraphListener; |
| import net.onrc.onos.ofcontroller.networkgraph.INetworkGraphService; |
| import net.onrc.onos.ofcontroller.networkgraph.LinkEvent; |
| import net.onrc.onos.ofcontroller.networkgraph.NetworkGraph; |
| |
| import org.easymock.EasyMock; |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| /** |
| * @author Toshio Koide (t-koide@onlab.us) |
| */ |
| public class UseCaseTest { |
| private NetworkGraph g; |
| private FloodlightModuleContext modContext; |
| private IDatagridService datagridService; |
| private INetworkGraphService networkGraphService; |
| @SuppressWarnings("rawtypes") |
| private IEventChannel eventChannel; |
| |
| @SuppressWarnings("unchecked") |
| @Before |
| public void setUp() { |
| MockNetworkGraph graph = new MockNetworkGraph(); |
| graph.createSampleTopology(); |
| g = graph; |
| |
| datagridService = EasyMock.createMock(IDatagridService.class); |
| networkGraphService = EasyMock.createMock(INetworkGraphService.class); |
| modContext = EasyMock.createMock(FloodlightModuleContext.class); |
| eventChannel = EasyMock.createMock(IEventChannel.class); |
| |
| EasyMock.expect(modContext.getServiceImpl(EasyMock.eq(IDatagridService.class))) |
| .andReturn(datagridService).once(); |
| EasyMock.expect(modContext.getServiceImpl(EasyMock.eq(INetworkGraphService.class))) |
| .andReturn(networkGraphService).once(); |
| |
| EasyMock.expect(networkGraphService.getNetworkGraph()).andReturn(g).anyTimes(); |
| networkGraphService.registerNetworkGraphListener(EasyMock.anyObject(INetworkGraphListener.class)); |
| EasyMock.expectLastCall(); |
| |
| EasyMock.expect(datagridService.createChannel("onos.pathintent", String.class, IntentOperationList.class)) |
| .andReturn(eventChannel).once(); |
| |
| EasyMock.replay(datagridService); |
| EasyMock.replay(networkGraphService); |
| EasyMock.replay(modContext); |
| } |
| |
| @After |
| public void tearDown() { |
| EasyMock.verify(datagridService); |
| EasyMock.verify(networkGraphService); |
| EasyMock.verify(modContext); |
| } |
| |
| private void showResult(PathIntentMap intents) { |
| for (Intent intent: intents.getAllIntents()) { |
| PathIntent pathIntent = (PathIntent)intent; |
| System.out.println("Parent intent: " + pathIntent.getParentIntent().toString()); |
| System.out.println("Path:"); |
| for (LinkEvent linkEvent: pathIntent.getPath()) { |
| System.out.println(linkEvent); |
| } |
| } |
| } |
| |
| @Test |
| public void createShortestPaths() throws FloodlightModuleException { |
| // create shortest path intents |
| IntentOperationList opList = new IntentOperationList(); |
| opList.add(Operator.ADD, new ShortestPathIntent("1", 1L, 20L, 1L, 4L, 20L, 4L)); |
| opList.add(Operator.ADD, new ShortestPathIntent("2", 2L, 20L, 2L, 6L, 20L, 5L)); |
| opList.add(Operator.ADD, new ShortestPathIntent("3", 4L, 20L, 3L, 8L, 20L, 6L)); |
| |
| // compile high-level intent operations into low-level intent operations (calculate paths) |
| PathCalcRuntimeModule runtime1 = new PathCalcRuntimeModule(); |
| runtime1.init(modContext); |
| runtime1.startUp(modContext); |
| IntentOperationList pathIntentOpList = runtime1.executeIntentOperations(opList); |
| |
| // compile low-level intents into flow entry installation plan |
| PlanCalcRuntime runtime2 = new PlanCalcRuntime(g); |
| runtime2.addIntents(pathIntentOpList); |
| |
| // show results |
| showResult((PathIntentMap) runtime1.getPathIntents()); |
| System.out.println(runtime2.getPlan()); |
| } |
| |
| @Test |
| public void createConstrainedShortestPaths() throws FloodlightModuleException { |
| // create constrained shortest path intents |
| IntentOperationList opList = new IntentOperationList(); |
| opList.add(Operator.ADD, new ConstrainedShortestPathIntent("1", 1L, 20L, 1L, 4L, 20L, 17L, 400.0)); |
| opList.add(Operator.ADD, new ConstrainedShortestPathIntent("2", 2L, 20L, 2L, 6L, 20L, 18L, 400.0)); |
| opList.add(Operator.ADD, new ConstrainedShortestPathIntent("3", 4L, 20L, 3L, 8L, 20L, 19L, 400.0)); |
| opList.add(Operator.ADD, new ConstrainedShortestPathIntent("4", 3L, 20L, 4L, 8L, 20L, 20L, 400.0)); |
| opList.add(Operator.ADD, new ConstrainedShortestPathIntent("5", 4L, 20L, 5L, 8L, 20L, 21L, 400.0)); |
| |
| // compile high-level intent operations into low-level intent operations (calculate paths) |
| PathCalcRuntimeModule runtime1 = new PathCalcRuntimeModule(); |
| runtime1.init(modContext); |
| runtime1.startUp(modContext); |
| IntentOperationList pathIntentOpList = runtime1.executeIntentOperations(opList); |
| |
| // compile low-level intents into flow entry installation plan |
| PlanCalcRuntime runtime2 = new PlanCalcRuntime(g); |
| runtime2.addIntents(pathIntentOpList); |
| |
| // show results |
| showResult((PathIntentMap) runtime1.getPathIntents()); |
| System.out.println(runtime2.getPlan()); |
| } |
| |
| @Test |
| public void createMixedShortestPaths() throws FloodlightModuleException { |
| // create constrained & best effort shortest path intents |
| IntentOperationList opList = new IntentOperationList(); |
| opList.add(Operator.ADD, new ConstrainedShortestPathIntent("1", 1L, 20L, 1L, 4L, 20L, 6L, 600.0)); |
| opList.add(Operator.ADD, new ConstrainedShortestPathIntent("2", 2L, 20L, 2L, 6L, 20L, 7L, 600.0)); |
| opList.add(Operator.ADD, new ShortestPathIntent("3", 4L, 20L, 3L, 8L, 20L, 8L)); |
| opList.add(Operator.ADD, new ShortestPathIntent("4", 4L, 20L, 4L, 8L, 20L, 9L)); |
| opList.add(Operator.ADD, new ConstrainedShortestPathIntent("5", 4L, 20L, 5L, 8L, 20L, 10L, 600.0)); |
| |
| // compile high-level intent operations into low-level intent operations (calculate paths) |
| PathCalcRuntimeModule runtime1 = new PathCalcRuntimeModule(); |
| runtime1.init(modContext); |
| runtime1.startUp(modContext); |
| IntentOperationList pathIntentOpList = runtime1.executeIntentOperations(opList); |
| |
| // compile low-level intents into flow entry installation plan |
| PlanCalcRuntime runtime2 = new PlanCalcRuntime(g); |
| runtime2.addIntents(pathIntentOpList); |
| |
| // show results |
| showResult((PathIntentMap) runtime1.getPathIntents()); |
| System.out.println(runtime2.getPlan()); |
| } |
| |
| @Test |
| public void rerouteShortestPaths() throws FloodlightModuleException { |
| // create shortest path intents |
| IntentOperationList opList = new IntentOperationList(); |
| opList.add(Operator.ADD, new ShortestPathIntent("1", 1L, 20L, 1L, 4L, 20L, 4L)); |
| opList.add(Operator.ADD, new ShortestPathIntent("2", 2L, 20L, 2L, 6L, 20L, 5L)); |
| opList.add(Operator.ADD, new ShortestPathIntent("3", 4L, 20L, 3L, 8L, 20L, 6L)); |
| |
| // compile high-level intent operations into low-level intent operations (calculate paths) |
| PathCalcRuntimeModule runtime1 = new PathCalcRuntimeModule(); |
| runtime1.init(modContext); |
| runtime1.startUp(modContext); |
| IntentOperationList pathIntentOpList = runtime1.executeIntentOperations(opList); |
| |
| // compile low-level intents into flow entry installation plan |
| PlanCalcRuntime runtime2 = new PlanCalcRuntime(g); |
| runtime2.addIntents(pathIntentOpList); |
| |
| // show results step1 |
| showResult((PathIntentMap) runtime1.getPathIntents()); |
| System.out.println(runtime2.getPlan()); |
| |
| // link down |
| ((MockNetworkGraph)g).removeLink(1L, 2L, 9L, 1L); // This link is used by the intent "1" |
| LinkEvent linkEvent = new LinkEvent(1L, 2L, 9L, 1L); |
| runtime1.removeLinkEvent(linkEvent); |
| ((MockNetworkGraph)g).removeLink(9L, 1L, 1L, 2L); |
| linkEvent = new LinkEvent(9L, 1L, 1L, 2L); |
| runtime1.removeLinkEvent(linkEvent); |
| |
| System.out.println("Link goes down."); |
| |
| // show results step2 |
| showResult((PathIntentMap) runtime1.getPathIntents()); |
| System.out.println(runtime2.getPlan()); |
| } |
| } |