| /* |
| * Copyright 2016-present Open Networking Foundation |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.onosproject.pcelabelstore.label; |
| |
| import static org.hamcrest.MatcherAssert.assertThat; |
| import static org.hamcrest.Matchers.is; |
| import static org.hamcrest.Matchers.notNullValue; |
| import static org.hamcrest.Matchers.nullValue; |
| import static org.onosproject.net.Link.Type.DIRECT; |
| import java.util.Iterator; |
| import java.util.List; |
| import java.util.LinkedList; |
| |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.onlab.packet.IpAddress; |
| import org.onosproject.incubator.net.resource.label.LabelResourceId; |
| import org.onosproject.incubator.net.resource.label.LabelResourceAdminService; |
| import org.onosproject.incubator.net.resource.label.LabelResourceService; |
| import org.onosproject.incubator.net.tunnel.LabelStack; |
| import org.onosproject.net.AnnotationKeys; |
| import org.onosproject.net.Annotations; |
| import org.onosproject.net.ConnectPoint; |
| import org.onosproject.net.DefaultAnnotations; |
| import org.onosproject.net.DefaultDevice; |
| import org.onosproject.net.DefaultPath; |
| import org.onosproject.net.Device; |
| import org.onosproject.net.DeviceId; |
| import org.onosproject.net.PortNumber; |
| import org.onosproject.net.Path; |
| import org.onosproject.net.provider.ProviderId; |
| import org.onosproject.pcelabelstore.api.PceLabelStore; |
| import org.onosproject.pcelabelstore.util.LabelResourceAdapter; |
| import org.onosproject.pcelabelstore.util.MockDeviceService; |
| import org.onosproject.pcelabelstore.util.MockNetConfigRegistryAdapter; |
| import org.onosproject.pcelabelstore.util.MockPcepClientController; |
| import org.onosproject.pcelabelstore.util.PceLabelStoreAdapter; |
| import org.onosproject.pcelabelstore.util.PcepClientAdapter; |
| import org.onosproject.pcep.api.DeviceCapability; |
| import org.onosproject.pcep.server.PccId; |
| import org.onosproject.pcep.server.impl.PceccSrTeBeHandler; |
| import org.onosproject.pcepio.protocol.PcepVersion; |
| import org.onosproject.net.DefaultLink; |
| import org.onosproject.net.Link; |
| |
| /** |
| * Unit tests for PceccSrTeBeHandler class. |
| */ |
| public class PceccSrTeBeHandlerTest { |
| |
| public static final long GLOBAL_LABEL_SPACE_MIN = 4097; |
| public static final long GLOBAL_LABEL_SPACE_MAX = 5121; |
| private static final String L3 = "L3"; |
| private static final String LSRID = "lsrId"; |
| |
| private PceccSrTeBeHandler srTeHandler; |
| private LabelResourceAdminService labelRsrcAdminService; |
| private LabelResourceService labelRsrcService; |
| private PceLabelStore pceStore; |
| private MockDeviceService deviceService; |
| private MockNetConfigRegistryAdapter netCfgService = new MockNetConfigRegistryAdapter(); |
| private MockPcepClientController clientController = new MockPcepClientController(); |
| private ProviderId providerId; |
| private DeviceId deviceId1, deviceId2, deviceId3, deviceId4, deviceId5; |
| private Device deviceD1; |
| private Device deviceD2; |
| private Device deviceD3; |
| private Device deviceD4; |
| private Device deviceD5; |
| private PortNumber port1; |
| private PortNumber port2; |
| private PortNumber port3; |
| private PortNumber port4; |
| private PortNumber port5; |
| private Link link1; |
| private Link link2; |
| private Link link3; |
| private Link link4; |
| private Path path1; |
| LabelResourceId labelId; |
| |
| @Before |
| public void setUp() throws Exception { |
| // Initialization of member variables |
| srTeHandler = PceccSrTeBeHandler.getInstance(); |
| labelRsrcService = new LabelResourceAdapter(); |
| labelRsrcAdminService = new LabelResourceAdapter(); |
| pceStore = new PceLabelStoreAdapter(); |
| deviceService = new MockDeviceService(); |
| |
| srTeHandler.initialize(labelRsrcAdminService, |
| labelRsrcService, |
| clientController, |
| pceStore, |
| deviceService); |
| |
| // Creates path |
| // Creates list of links |
| providerId = new ProviderId("of", "foo"); |
| |
| PccId pccId1 = PccId.pccId(IpAddress.valueOf("11.1.1.1")); |
| PccId pccId2 = PccId.pccId(IpAddress.valueOf("12.1.1.1")); |
| PccId pccId3 = PccId.pccId(IpAddress.valueOf("13.1.1.1")); |
| PccId pccId4 = PccId.pccId(IpAddress.valueOf("14.1.1.1")); |
| PccId pccId5 = PccId.pccId(IpAddress.valueOf("15.1.1.1")); |
| |
| PcepClientAdapter pc1 = new PcepClientAdapter(); |
| pc1.init(pccId1, PcepVersion.PCEP_1); |
| |
| PcepClientAdapter pc2 = new PcepClientAdapter(); |
| pc2.init(pccId2, PcepVersion.PCEP_1); |
| |
| PcepClientAdapter pc3 = new PcepClientAdapter(); |
| pc3.init(pccId3, PcepVersion.PCEP_1); |
| |
| PcepClientAdapter pc4 = new PcepClientAdapter(); |
| pc4.init(pccId4, PcepVersion.PCEP_1); |
| |
| PcepClientAdapter pc5 = new PcepClientAdapter(); |
| pc5.init(pccId5, PcepVersion.PCEP_1); |
| |
| clientController.addClient(pccId1, pc1); |
| clientController.addClient(pccId2, pc2); |
| clientController.addClient(pccId3, pc3); |
| clientController.addClient(pccId4, pc4); |
| clientController.addClient(pccId5, pc5); |
| |
| deviceId1 = DeviceId.deviceId("11.1.1.1"); |
| deviceId2 = DeviceId.deviceId("12.1.1.1"); |
| deviceId3 = DeviceId.deviceId("13.1.1.1"); |
| deviceId4 = DeviceId.deviceId("14.1.1.1"); |
| deviceId5 = DeviceId.deviceId("15.1.1.1"); |
| |
| // Devices |
| DefaultAnnotations.Builder builderDev1 = DefaultAnnotations.builder(); |
| DefaultAnnotations.Builder builderDev2 = DefaultAnnotations.builder(); |
| DefaultAnnotations.Builder builderDev3 = DefaultAnnotations.builder(); |
| DefaultAnnotations.Builder builderDev4 = DefaultAnnotations.builder(); |
| DefaultAnnotations.Builder builderDev5 = DefaultAnnotations.builder(); |
| |
| builderDev1.set(AnnotationKeys.TYPE, L3); |
| builderDev1.set(LSRID, "11.1.1.1"); |
| |
| builderDev2.set(AnnotationKeys.TYPE, L3); |
| builderDev2.set(LSRID, "12.1.1.1"); |
| |
| builderDev3.set(AnnotationKeys.TYPE, L3); |
| builderDev3.set(LSRID, "13.1.1.1"); |
| |
| builderDev4.set(AnnotationKeys.TYPE, L3); |
| builderDev4.set(LSRID, "14.1.1.1"); |
| |
| builderDev5.set(AnnotationKeys.TYPE, L3); |
| builderDev5.set(LSRID, "15.1.1.1"); |
| |
| deviceD1 = new MockDevice(deviceId1, builderDev1.build()); |
| deviceD2 = new MockDevice(deviceId2, builderDev2.build()); |
| deviceD3 = new MockDevice(deviceId3, builderDev3.build()); |
| deviceD4 = new MockDevice(deviceId4, builderDev4.build()); |
| deviceD5 = new MockDevice(deviceId5, builderDev5.build()); |
| |
| deviceService.addDevice(deviceD1); |
| deviceService.addDevice(deviceD2); |
| deviceService.addDevice(deviceD3); |
| deviceService.addDevice(deviceD4); |
| deviceService.addDevice(deviceD5); |
| |
| DeviceCapability device1Cap = netCfgService.addConfig(deviceId1, DeviceCapability.class); |
| device1Cap.setLabelStackCap(true).setLocalLabelCap(false).setSrCap(true).apply(); |
| |
| DeviceCapability device2Cap = netCfgService.addConfig(deviceId2, DeviceCapability.class); |
| device2Cap.setLabelStackCap(true).setLocalLabelCap(false).setSrCap(true).apply(); |
| |
| DeviceCapability device3Cap = netCfgService.addConfig(deviceId3, DeviceCapability.class); |
| device3Cap.setLabelStackCap(true).setLocalLabelCap(false).setSrCap(true).apply(); |
| |
| DeviceCapability device4Cap = netCfgService.addConfig(deviceId4, DeviceCapability.class); |
| device4Cap.setLabelStackCap(true).setLocalLabelCap(false).setSrCap(true).apply(); |
| |
| DeviceCapability device5Cap = netCfgService.addConfig(deviceId5, DeviceCapability.class); |
| device5Cap.setLabelStackCap(true).setLocalLabelCap(false).setSrCap(true).apply(); |
| |
| // Port Numbers |
| port1 = PortNumber.portNumber(1); |
| port2 = PortNumber.portNumber(2); |
| port3 = PortNumber.portNumber(3); |
| port4 = PortNumber.portNumber(4); |
| port5 = PortNumber.portNumber(5); |
| List<Link> linkList = new LinkedList<>(); |
| |
| link1 = DefaultLink.builder().providerId(providerId) |
| .annotations(DefaultAnnotations.builder().set("key1", "yahoo").build()) |
| .src(new ConnectPoint(deviceD1.id(), port1)).dst(new ConnectPoint(deviceD2.id(), port2)).type(DIRECT) |
| .state(Link.State.ACTIVE).build(); |
| linkList.add(link1); |
| link2 = DefaultLink.builder().providerId(providerId) |
| .annotations(DefaultAnnotations.builder().set("key2", "yahoo").build()) |
| .src(new ConnectPoint(deviceD2.id(), port2)).dst(new ConnectPoint(deviceD3.id(), port3)).type(DIRECT) |
| .state(Link.State.ACTIVE).build(); |
| linkList.add(link2); |
| link3 = DefaultLink.builder().providerId(providerId) |
| .annotations(DefaultAnnotations.builder().set("key3", "yahoo").build()) |
| .src(new ConnectPoint(deviceD3.id(), port3)).dst(new ConnectPoint(deviceD4.id(), port4)).type(DIRECT) |
| .state(Link.State.ACTIVE).build(); |
| linkList.add(link3); |
| link4 = DefaultLink.builder().providerId(providerId) |
| .annotations(DefaultAnnotations.builder().set("key4", "yahoo").build()) |
| .src(new ConnectPoint(deviceD4.id(), port4)).dst(new ConnectPoint(deviceD5.id(), port5)).type(DIRECT) |
| .state(Link.State.ACTIVE).build(); |
| linkList.add(link4); |
| |
| // Path |
| path1 = new DefaultPath(providerId, linkList, 10); |
| } |
| |
| @After |
| public void tearDown() throws Exception { |
| } |
| |
| /** |
| * Checks the operation of getInstance() method. |
| */ |
| @Test |
| public void testGetInstance() { |
| assertThat(srTeHandler, is(notNullValue())); |
| } |
| |
| /** |
| * Checks the operation of reserveGlobalPool() method. |
| */ |
| @Test |
| public void testReserveGlobalPool() { |
| assertThat(srTeHandler.reserveGlobalPool(GLOBAL_LABEL_SPACE_MIN, GLOBAL_LABEL_SPACE_MAX), is(true)); |
| } |
| |
| /** |
| * Checks the operation of allocateNodeLabel() method on node label. |
| */ |
| @Test |
| public void testAllocateNodeLabel() { |
| // Specific device D1.deviceId |
| |
| //device 1 |
| String lsrId1 = "11.1.1.1"; |
| // Allocate node label for specific device D1deviceId |
| assertThat(srTeHandler.allocateNodeLabel(deviceId1, lsrId1), is(true)); |
| // Retrieve label from store |
| LabelResourceId labelId = pceStore.getGlobalNodeLabel(deviceId1); |
| // Check whether label is generated for this device D1.deviceId() |
| assertThat(labelId, is(notNullValue())); |
| |
| // device 2 |
| String lsrId2 = "12.1.1.1"; |
| // Allocate node label for specific device D2.deviceId() |
| assertThat(srTeHandler.allocateNodeLabel(deviceId2, lsrId2), is(true)); |
| // Retrieve label from store |
| labelId = pceStore.getGlobalNodeLabel(deviceId2); |
| // Check whether label is generated for this device D2.deviceId() |
| assertThat(labelId, is(notNullValue())); |
| |
| // device 3 |
| String lsrId3 = "13.1.1.1"; |
| // Allocate node label for specific device D3.deviceId() |
| assertThat(srTeHandler.allocateNodeLabel(deviceId3, lsrId3), is(true)); |
| // Retrieve label from store |
| labelId = pceStore.getGlobalNodeLabel(deviceId3); |
| // Check whether label is generated for this device D3.deviceId() |
| assertThat(labelId, is(notNullValue())); |
| |
| // device 4 |
| String lsrId4 = "14.1.1.1"; |
| // Allocate node label for specific device D4.deviceId() |
| assertThat(srTeHandler.allocateNodeLabel(deviceId4, lsrId4), is(true)); |
| // Retrieve label from store |
| labelId = pceStore.getGlobalNodeLabel(deviceId4); |
| // Check whether label is generated for this device D4.deviceId() |
| assertThat(labelId, is(notNullValue())); |
| |
| // device 5 |
| String lsrId5 = "15.1.1.1"; |
| // Allocate node label for specific device D5.deviceId() |
| assertThat(srTeHandler.allocateNodeLabel(deviceId5, lsrId5), is(true)); |
| // Retrieve label from store |
| labelId = pceStore.getGlobalNodeLabel(deviceId5); |
| // Check whether label is generated for this device D5.deviceId() |
| assertThat(labelId, is(notNullValue())); |
| } |
| |
| /** |
| * Checks the operation of releaseNodeLabel() method on node label. |
| */ |
| @Test |
| public void testReleaseNodeLabelSuccess() { |
| testAllocateNodeLabel(); |
| // Specific device D1.deviceId() |
| |
| //device 1 |
| String lsrId1 = "11.1.1.1"; |
| // Check whether successfully released node label |
| assertThat(srTeHandler.releaseNodeLabel(deviceId1, lsrId1), is(true)); |
| // Check whether successfully removed label from store |
| LabelResourceId labelId = pceStore.getGlobalNodeLabel(deviceId1); |
| assertThat(labelId, is(nullValue())); |
| |
| //device 2 |
| String lsrId2 = "12.1.1.1"; |
| // Check whether successfully released node label |
| assertThat(srTeHandler.releaseNodeLabel(deviceId2, lsrId2), is(true)); |
| // Check whether successfully removed label from store |
| labelId = pceStore.getGlobalNodeLabel(deviceId2); |
| assertThat(labelId, is(nullValue())); |
| |
| //device 3 |
| String lsrId3 = "13.1.1.1"; |
| // Check whether successfully released node label |
| assertThat(srTeHandler.releaseNodeLabel(deviceId3, lsrId3), is(true)); |
| // Check whether successfully removed label from store |
| labelId = pceStore.getGlobalNodeLabel(deviceId3); |
| assertThat(labelId, is(nullValue())); |
| |
| //device 4 |
| String lsrId4 = "14.1.1.1"; |
| // Check whether successfully released node label |
| assertThat(srTeHandler.releaseNodeLabel(deviceId4, lsrId4), is(true)); |
| // Check whether successfully removed label from store |
| labelId = pceStore.getGlobalNodeLabel(deviceId4); |
| assertThat(labelId, is(nullValue())); |
| |
| //device 5 |
| String lsrId5 = "15.1.1.1"; |
| // Check whether successfully released node label |
| assertThat(srTeHandler.releaseNodeLabel(deviceId5, lsrId5), is(true)); |
| // Check whether successfully removed label from store |
| labelId = pceStore.getGlobalNodeLabel(deviceId5); |
| assertThat(labelId, is(nullValue())); |
| } |
| |
| @Test |
| public void testReleaseNodeLabelFailure() { |
| testAllocateNodeLabel(); |
| |
| //device 6 |
| String lsrId6 = "16.1.1.1"; |
| // Check whether successfully released node label |
| DeviceId deviceId6 = DeviceId.deviceId("foo6"); |
| assertThat(srTeHandler.releaseNodeLabel(deviceId6, lsrId6), is(false)); |
| } |
| |
| /** |
| * Checks the operation of allocateAdjacencyLabel() method on adjacency label. |
| */ |
| @Test |
| public void testAllocateAdjacencyLabel() { |
| // test link1 |
| // Check whether adjacency label is allocated successfully. |
| assertThat(srTeHandler.allocateAdjacencyLabel(link1), is(true)); |
| // Retrieve from store and check whether adjacency label is generated successfully for this device. |
| LabelResourceId labelId = pceStore.getAdjLabel(link1); |
| assertThat(labelId, is(notNullValue())); |
| |
| // test link2 |
| // Check whether adjacency label is allocated successfully. |
| assertThat(srTeHandler.allocateAdjacencyLabel(link2), is(true)); |
| // Retrieve from store and check whether adjacency label is generated successfully for this device. |
| labelId = pceStore.getAdjLabel(link2); |
| assertThat(labelId, is(notNullValue())); |
| |
| // test link3 |
| // Check whether adjacency label is allocated successfully. |
| assertThat(srTeHandler.allocateAdjacencyLabel(link3), is(true)); |
| // Retrieve from store and check whether adjacency label is generated successfully for this device. |
| labelId = pceStore.getAdjLabel(link3); |
| assertThat(labelId, is(notNullValue())); |
| |
| // test link4 |
| // Check whether adjacency label is allocated successfully. |
| assertThat(srTeHandler.allocateAdjacencyLabel(link4), is(true)); |
| // Retrieve from store and check whether adjacency label is generated successfully for this device. |
| labelId = pceStore.getAdjLabel(link4); |
| assertThat(labelId, is(notNullValue())); |
| } |
| |
| /** |
| * Checks the operation of releaseAdjacencyLabel() method on adjacency label. |
| */ |
| @Test |
| public void testReleaseAdjacencyLabel() { |
| // Test link1 |
| // Check whether adjacency label is released successfully. |
| assertThat(srTeHandler.allocateAdjacencyLabel(link1), is(true)); |
| assertThat(srTeHandler.releaseAdjacencyLabel(link1), is(true)); |
| // Retrieve from store and check whether adjacency label is removed successfully for this device. |
| LabelResourceId labelId = pceStore.getAdjLabel(link1); |
| assertThat(labelId, is(nullValue())); |
| |
| // Test link2 |
| // Check whether adjacency label is released successfully. |
| assertThat(srTeHandler.allocateAdjacencyLabel(link2), is(true)); |
| assertThat(srTeHandler.releaseAdjacencyLabel(link2), is(true)); |
| // Retrieve from store and check whether adjacency label is removed successfully for this device. |
| labelId = pceStore.getAdjLabel(link2); |
| assertThat(labelId, is(nullValue())); |
| } |
| |
| /** |
| * Checks the operation of computeLabelStack() method. |
| */ |
| @Test |
| public void testComputeLabelStack() { |
| // Allocate node labels to each devices |
| labelId = LabelResourceId.labelResourceId(4097); |
| pceStore.addGlobalNodeLabel(deviceId1, labelId); |
| labelId = LabelResourceId.labelResourceId(4098); |
| pceStore.addGlobalNodeLabel(deviceId2, labelId); |
| labelId = LabelResourceId.labelResourceId(4099); |
| pceStore.addGlobalNodeLabel(deviceId3, labelId); |
| labelId = LabelResourceId.labelResourceId(4100); |
| pceStore.addGlobalNodeLabel(deviceId4, labelId); |
| labelId = LabelResourceId.labelResourceId(4101); |
| pceStore.addGlobalNodeLabel(deviceId5, labelId); |
| |
| // Allocate adjacency labels to each devices |
| labelId = LabelResourceId.labelResourceId(5122); |
| pceStore.addAdjLabel(link1, labelId); |
| labelId = LabelResourceId.labelResourceId(5123); |
| pceStore.addAdjLabel(link2, labelId); |
| labelId = LabelResourceId.labelResourceId(5124); |
| pceStore.addAdjLabel(link3, labelId); |
| labelId = LabelResourceId.labelResourceId(5125); |
| pceStore.addAdjLabel(link4, labelId); |
| |
| // Compute label stack |
| LabelStack labelStack = srTeHandler.computeLabelStack(path1); |
| |
| List<LabelResourceId> labelList = labelStack.labelResources(); |
| Iterator<LabelResourceId> iterator = labelList.iterator(); |
| |
| // check adjacency label of D1.deviceId() |
| labelId = iterator.next(); |
| assertThat(labelId, is(LabelResourceId.labelResourceId(5122))); |
| |
| // check node-label of D2.deviceId() |
| labelId = iterator.next(); |
| assertThat(labelId, is(LabelResourceId.labelResourceId(4098))); |
| |
| // check adjacency label of D2.deviceId() |
| labelId = iterator.next(); |
| assertThat(labelId, is(LabelResourceId.labelResourceId(5123))); |
| |
| // check node-label of D3.deviceId() |
| labelId = iterator.next(); |
| assertThat(labelId, is(LabelResourceId.labelResourceId(4099))); |
| |
| // check adjacency label of D3.deviceId() |
| labelId = iterator.next(); |
| assertThat(labelId, is(LabelResourceId.labelResourceId(5124))); |
| |
| // check node-label of D4.deviceId() |
| labelId = iterator.next(); |
| assertThat(labelId, is(LabelResourceId.labelResourceId(4100))); |
| |
| // check adjacency label of D4.deviceId() |
| labelId = iterator.next(); |
| assertThat(labelId, is(LabelResourceId.labelResourceId(5125))); |
| |
| // check node-label of D5.deviceId() |
| labelId = iterator.next(); |
| assertThat(labelId, is(LabelResourceId.labelResourceId(4101))); |
| } |
| |
| private class MockDevice extends DefaultDevice { |
| MockDevice(DeviceId id, Annotations annotations) { |
| super(null, id, null, null, null, null, null, null, annotations); |
| } |
| } |
| } |