[ONOS-4164] Basic PCECC and SR-TE Handlers
Change-Id: I7fbf86d845ec6b5d780f63518776fa841e9fb57c
diff --git a/apps/pce/app/src/test/java/org/onosproject/pce/pceservice/BasicPceccHandlerTest.java b/apps/pce/app/src/test/java/org/onosproject/pce/pceservice/BasicPceccHandlerTest.java
new file mode 100644
index 0000000..a5ea175
--- /dev/null
+++ b/apps/pce/app/src/test/java/org/onosproject/pce/pceservice/BasicPceccHandlerTest.java
@@ -0,0 +1,280 @@
+/*
+ * Copyright 2016-present Open Networking Laboratory
+ *
+ * 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.pce.pceservice;
+
+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.core.DefaultGroupId;
+import org.onosproject.incubator.net.tunnel.Tunnel;
+import org.onosproject.incubator.net.tunnel.TunnelEndPoint;
+import org.onosproject.incubator.net.tunnel.IpTunnelEndPoint;
+import org.onosproject.incubator.net.tunnel.TunnelName;
+import org.onosproject.incubator.net.tunnel.TunnelId;
+import org.onosproject.incubator.net.tunnel.DefaultTunnel;
+import org.onosproject.incubator.net.resource.label.LabelResourceId;
+import org.onosproject.incubator.net.resource.label.LabelResourceService;
+import org.onosproject.net.ConnectPoint;
+import org.onosproject.net.DefaultAnnotations;
+import org.onosproject.net.DefaultPath;
+import org.onosproject.net.DeviceId;
+import org.onosproject.net.PortNumber;
+import org.onosproject.net.Path;
+import org.onosproject.pce.pcestore.api.LspLocalLabelInfo;
+import org.onosproject.pce.pcestore.api.PceStore;
+import org.onosproject.pce.pcestore.PceccTunnelInfo;
+import org.onosproject.pce.pcestore.DefaultLspLocalLabelInfo;
+import org.onosproject.net.provider.ProviderId;
+import org.onosproject.pce.util.LabelResourceAdapter;
+import org.onosproject.pce.util.PceStoreAdapter;
+import org.onosproject.net.DefaultLink;
+import org.onosproject.net.Link;
+
+/**
+ * Unit tests for BasicPceccHandler class.
+ */
+public class BasicPceccHandlerTest {
+
+ public static final long LOCAL_LABEL_SPACE_MIN = 5122;
+ public static final long LOCAL_LABEL_SPACE_MAX = 9217;
+
+ private BasicPceccHandler pceccHandler;
+ protected LabelResourceService labelRsrcService;
+ protected PceStore pceStore;
+ private TunnelEndPoint src = IpTunnelEndPoint.ipTunnelPoint(IpAddress.valueOf(23423));
+ private TunnelEndPoint dst = IpTunnelEndPoint.ipTunnelPoint(IpAddress.valueOf(32421));
+ private DefaultGroupId groupId = new DefaultGroupId(92034);
+ private TunnelName tunnelName = TunnelName.tunnelName("TunnelName");
+ private TunnelId tunnelId = TunnelId.valueOf("41654654");
+ private ProviderId producerName = new ProviderId("producer1", "13");
+ private Path path;
+ private Tunnel tunnel;
+ private PceccTunnelInfo pceccTunnelInfo;
+ private DeviceId deviceId1;
+ private DeviceId deviceId2;
+ private DeviceId deviceId3;
+ private DeviceId deviceId4;
+ private DeviceId deviceId5;
+ private PortNumber port1;
+ private PortNumber port2;
+ private PortNumber port3;
+ private PortNumber port4;
+ private PortNumber port5;
+
+ @Before
+ public void setUp() throws Exception {
+ pceccHandler = BasicPceccHandler.getInstance();
+ labelRsrcService = new LabelResourceAdapter();
+ pceStore = new PceStoreAdapter();
+ pceccHandler.initialize(labelRsrcService, pceStore);
+
+ // Cretae tunnel test
+ // Link
+ ProviderId providerId = new ProviderId("of", "foo");
+ deviceId1 = DeviceId.deviceId("of:A");
+ deviceId2 = DeviceId.deviceId("of:B");
+ deviceId3 = DeviceId.deviceId("of:C");
+ deviceId4 = DeviceId.deviceId("of:D");
+ deviceId5 = DeviceId.deviceId("of:E");
+ 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<>();
+
+ Link l1 = DefaultLink.builder()
+ .providerId(providerId)
+ .annotations(DefaultAnnotations.builder().set("key1", "yahoo").build())
+ .src(new ConnectPoint(deviceId1, port1))
+ .dst(new ConnectPoint(deviceId2, port2))
+ .type(DIRECT)
+ .state(Link.State.ACTIVE)
+ .build();
+ linkList.add(l1);
+ Link l2 = DefaultLink.builder()
+ .providerId(providerId)
+ .annotations(DefaultAnnotations.builder().set("key2", "yahoo").build())
+ .src(new ConnectPoint(deviceId2, port2))
+ .dst(new ConnectPoint(deviceId3, port3))
+ .type(DIRECT)
+ .state(Link.State.ACTIVE)
+ .build();
+ linkList.add(l2);
+ Link l3 = DefaultLink.builder()
+ .providerId(providerId)
+ .annotations(DefaultAnnotations.builder().set("key3", "yahoo").build())
+ .src(new ConnectPoint(deviceId3, port3))
+ .dst(new ConnectPoint(deviceId4, port4))
+ .type(DIRECT)
+ .state(Link.State.ACTIVE)
+ .build();
+ linkList.add(l3);
+ Link l4 = DefaultLink.builder()
+ .providerId(providerId)
+ .annotations(DefaultAnnotations.builder().set("key4", "yahoo").build())
+ .src(new ConnectPoint(deviceId4, port4))
+ .dst(new ConnectPoint(deviceId5, port5))
+ .type(DIRECT)
+ .state(Link.State.ACTIVE)
+ .build();
+ linkList.add(l4);
+
+ // Path
+ path = new DefaultPath(providerId, linkList, 10);
+
+ // Tunnel
+ tunnel = new DefaultTunnel(producerName, src, dst, Tunnel.Type.VXLAN,
+ Tunnel.State.ACTIVE, groupId, tunnelId,
+ tunnelName, path);
+ }
+
+ @After
+ public void tearDown() throws Exception {
+ }
+
+ /**
+ * Checks the operation of getInstance() method.
+ */
+ @Test
+ public void testGetInstance() {
+ assertThat(pceccHandler, is(notNullValue()));
+ }
+
+ /**
+ * Checks the operation of allocateLabel() method.
+ */
+ @Test
+ public void testAllocateLabel() {
+ List<LspLocalLabelInfo> lspLocalLabelInfoList;
+ Iterator<LspLocalLabelInfo> iterator;
+ LspLocalLabelInfo lspLocalLabelInfo;
+ DeviceId deviceId;
+ LabelResourceId inLabelId;
+ LabelResourceId outLabelId;
+ PortNumber inPort;
+ PortNumber outPort;
+
+ // check allocation result
+ assertThat(pceccHandler.allocateLabel(tunnel), is(true));
+
+ // Check list of devices with IN and OUT labels whether stored properly in store
+ pceccTunnelInfo = pceStore.getTunnelInfo(tunnel.tunnelId());
+ lspLocalLabelInfoList = pceccTunnelInfo.lspLocalLabelInfoList();
+ iterator = lspLocalLabelInfoList.iterator();
+
+ // Retrieve values and check device5
+ lspLocalLabelInfo = (DefaultLspLocalLabelInfo) iterator.next();
+ deviceId = lspLocalLabelInfo.deviceId();
+ inLabelId = lspLocalLabelInfo.inLabelId();
+ outLabelId = lspLocalLabelInfo.outLabelId();
+ inPort = lspLocalLabelInfo.inPort();
+ outPort = lspLocalLabelInfo.outPort();
+
+ assertThat(deviceId, is(deviceId5));
+ assertThat(inLabelId, is(notNullValue()));
+ assertThat(outLabelId, is(nullValue()));
+ assertThat(inPort, is(port5));
+ assertThat(outPort, is(nullValue()));
+
+ // Next element check
+ // Retrieve values and check device4
+ lspLocalLabelInfo = (DefaultLspLocalLabelInfo) iterator.next();
+ deviceId = lspLocalLabelInfo.deviceId();
+ inLabelId = lspLocalLabelInfo.inLabelId();
+ outLabelId = lspLocalLabelInfo.outLabelId();
+ inPort = lspLocalLabelInfo.inPort();
+ outPort = lspLocalLabelInfo.outPort();
+
+ assertThat(deviceId, is(deviceId4));
+ assertThat(inLabelId, is(notNullValue()));
+ assertThat(outLabelId, is(notNullValue()));
+ assertThat(inPort, is(port4));
+ assertThat(outPort, is(port5));
+
+ // Next element check
+ // Retrieve values and check device3
+ lspLocalLabelInfo = (DefaultLspLocalLabelInfo) iterator.next();
+ deviceId = lspLocalLabelInfo.deviceId();
+ inLabelId = lspLocalLabelInfo.inLabelId();
+ outLabelId = lspLocalLabelInfo.outLabelId();
+ inPort = lspLocalLabelInfo.inPort();
+ outPort = lspLocalLabelInfo.outPort();
+
+ assertThat(deviceId, is(deviceId3));
+ assertThat(inLabelId, is(notNullValue()));
+ assertThat(outLabelId, is(notNullValue()));
+ assertThat(inPort, is(port3));
+ assertThat(outPort, is(port4));
+
+ // Next element check
+ // Retrieve values and check device2
+ lspLocalLabelInfo = (DefaultLspLocalLabelInfo) iterator.next();
+ deviceId = lspLocalLabelInfo.deviceId();
+ inLabelId = lspLocalLabelInfo.inLabelId();
+ outLabelId = lspLocalLabelInfo.outLabelId();
+ inPort = lspLocalLabelInfo.inPort();
+ outPort = lspLocalLabelInfo.outPort();
+
+ assertThat(deviceId, is(deviceId2));
+ assertThat(inLabelId, is(notNullValue()));
+ assertThat(outLabelId, is(notNullValue()));
+ assertThat(inPort, is(port2));
+ assertThat(outPort, is(port3));
+
+ // Next element check
+ // Retrieve values and check device1
+ lspLocalLabelInfo = (DefaultLspLocalLabelInfo) iterator.next();
+ deviceId = lspLocalLabelInfo.deviceId();
+ inLabelId = lspLocalLabelInfo.inLabelId();
+ outLabelId = lspLocalLabelInfo.outLabelId();
+ inPort = lspLocalLabelInfo.inPort();
+ outPort = lspLocalLabelInfo.outPort();
+
+ assertThat(deviceId, is(deviceId1));
+ assertThat(inLabelId, is(nullValue()));
+ assertThat(outLabelId, is(notNullValue()));
+ assertThat(inPort, is(nullValue()));
+ assertThat(outPort, is(port2));
+ }
+
+ /**
+ * Checks the operation of releaseLabel() method.
+ */
+ @Test
+ public void testReleaseLabel() {
+ // Release tunnels
+ assertThat(pceccHandler.allocateLabel(tunnel), is(true));
+ pceccHandler.releaseLabel(tunnel);
+
+ // Retrieve from store. Store should not contain this tunnel info.
+ pceccTunnelInfo = pceStore.getTunnelInfo(tunnel.tunnelId());
+ assertThat(pceccTunnelInfo, is(nullValue()));
+ }
+}
diff --git a/apps/pce/app/src/test/java/org/onosproject/pce/pceservice/PceccSrTeBeHandlerTest.java b/apps/pce/app/src/test/java/org/onosproject/pce/pceservice/PceccSrTeBeHandlerTest.java
new file mode 100644
index 0000000..93f7c69
--- /dev/null
+++ b/apps/pce/app/src/test/java/org/onosproject/pce/pceservice/PceccSrTeBeHandlerTest.java
@@ -0,0 +1,484 @@
+/*
+ * Copyright 2016-present Open Networking Laboratory
+ *
+ * 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.pce.pceservice;
+
+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.HashMap;
+import java.util.Iterator;
+import java.util.Map;
+import java.util.List;
+import java.util.LinkedList;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+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.ConnectPoint;
+import org.onosproject.net.DefaultAnnotations;
+import org.onosproject.net.DefaultPath;
+import org.onosproject.net.DeviceId;
+import org.onosproject.net.PortNumber;
+import org.onosproject.net.Path;
+import org.onosproject.pce.pcestore.api.PceStore;
+import org.onosproject.net.provider.ProviderId;
+import org.onosproject.pce.util.LabelResourceAdapter;
+import org.onosproject.pce.util.PceStoreAdapter;
+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 PceccSrTeBeHandler srTeHandler;
+ protected LabelResourceAdminService labelRsrcAdminService;
+ protected LabelResourceService labelRsrcService;
+ protected PceStore pceStore;
+ private ProviderId providerId;
+ private DeviceId deviceId1;
+ private DeviceId deviceId2;
+ private DeviceId deviceId3;
+ private DeviceId deviceId4;
+ private DeviceId deviceId5;
+ 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;
+ private LabelResourceId labelId1 = LabelResourceId.labelResourceId(4098);
+ private LabelResourceId labelId2 = LabelResourceId.labelResourceId(4099);
+ private Map<DeviceId, String> deviceIdLsrIdMap;
+
+ @Before
+ public void setUp() throws Exception {
+ // Initialization of member variables
+ srTeHandler = PceccSrTeBeHandler.getInstance();
+ labelRsrcService = new LabelResourceAdapter();
+ labelRsrcAdminService = new LabelResourceAdapter();
+ pceStore = new PceStoreAdapter();
+ srTeHandler.initialize(labelRsrcAdminService, labelRsrcService, pceStore);
+
+ // Creates path
+ // Creates list of links
+ providerId = new ProviderId("of", "foo");
+ deviceId1 = DeviceId.deviceId("of:A");
+ deviceId2 = DeviceId.deviceId("of:B");
+ deviceId3 = DeviceId.deviceId("of:C");
+ deviceId4 = DeviceId.deviceId("of:D");
+ deviceId5 = DeviceId.deviceId("of:E");
+ 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(deviceId1, port1))
+ .dst(new ConnectPoint(deviceId2, 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(deviceId2, port2))
+ .dst(new ConnectPoint(deviceId3, 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(deviceId3, port3))
+ .dst(new ConnectPoint(deviceId4, 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(deviceId4, port4))
+ .dst(new ConnectPoint(deviceId5, 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.
+ * Considered some nodes are already available before PceManager came up.
+ */
+ @Test
+ public void testAllocateNodeLabel1() {
+ // Specific device deviceId1
+
+ // Devices mapping with lsr-id
+ deviceIdLsrIdMap = new HashMap<>();
+
+ //device 1
+ String lsrId1 = "11.1.1.1";
+ deviceIdLsrIdMap.put(deviceId1, lsrId1);
+
+ // device 2
+ String lsrId2 = "12.1.1.1";
+ deviceIdLsrIdMap.put(deviceId2, lsrId2);
+
+ // device 3
+ String lsrId3 = "13.1.1.1";
+ deviceIdLsrIdMap.put(deviceId3, lsrId3);
+
+ // device 4
+ String lsrId4 = "14.1.1.1";
+ deviceIdLsrIdMap.put(deviceId4, lsrId4);
+
+ // device 5
+ String lsrId5 = "15.1.1.1";
+ deviceIdLsrIdMap.put(deviceId5, lsrId5);
+
+ // Considered devices are stored in deviceIdLsrIdMap.
+ // Creating temporary tempDeviceIdLsrIdMap to pass to allocateNodeLabel()
+ Map<DeviceId, String> tempDeviceIdLsrIdMap = new HashMap<>();
+ for (Map.Entry element : deviceIdLsrIdMap.entrySet()) {
+ DeviceId devId = (DeviceId) element.getKey();
+ String lsrId = (String) element.getValue();
+
+ // Allocate node label for specific device devId
+ assertThat(srTeHandler.allocateNodeLabel(devId, lsrId, tempDeviceIdLsrIdMap), is(true));
+
+ // Retrieve label from store
+ LabelResourceId labelId = pceStore.getGlobalNodeLabel(devId);
+
+ // Check whether label is generated for this device devId
+ assertThat(labelId, is(notNullValue()));
+
+ //Now add device to tempDeviceIdLsrIdMap
+ tempDeviceIdLsrIdMap.put(devId, lsrId);
+ }
+ }
+
+ /**
+ * Checks the operation of allocateNodeLabel() method on node label.
+ * Considered initially map is empty and keep on getting new devices from event.
+ */
+ @Test
+ public void testAllocateNodeLabel2() {
+ // Specific device deviceId1
+
+ // Device-id mapping with lsr-id
+ deviceIdLsrIdMap = new HashMap<>();
+
+ //device 1
+ String lsrId1 = "11.1.1.1";
+ // Allocate node label for specific device deviceId1
+ assertThat(srTeHandler.allocateNodeLabel(deviceId1, lsrId1, deviceIdLsrIdMap), is(true));
+ // Retrieve label from store
+ LabelResourceId labelId = pceStore.getGlobalNodeLabel(deviceId1);
+ // Check whether label is generated for this device deviceId1
+ assertThat(labelId, is(notNullValue()));
+ //Now add device to deviceIdLsrIdMap
+ deviceIdLsrIdMap.put(deviceId1, lsrId1);
+
+ // device 2
+ String lsrId2 = "12.1.1.1";
+ // Allocate node label for specific device deviceId2
+ assertThat(srTeHandler.allocateNodeLabel(deviceId2, lsrId2, deviceIdLsrIdMap), is(true));
+ // Retrieve label from store
+ labelId = pceStore.getGlobalNodeLabel(deviceId2);
+ // Check whether label is generated for this device deviceId2
+ assertThat(labelId, is(notNullValue()));
+ //Now add device to deviceIdLsrIdMap
+ deviceIdLsrIdMap.put(deviceId2, lsrId2);
+
+ // device 3
+ String lsrId3 = "13.1.1.1";
+ // Allocate node label for specific device deviceId3
+ assertThat(srTeHandler.allocateNodeLabel(deviceId3, lsrId3, deviceIdLsrIdMap), is(true));
+ // Retrieve label from store
+ labelId = pceStore.getGlobalNodeLabel(deviceId3);
+ // Check whether label is generated for this device deviceId3
+ assertThat(labelId, is(notNullValue()));
+ //Now add device to deviceIdLsrIdMap
+ deviceIdLsrIdMap.put(deviceId3, lsrId3);
+
+ // device 4
+ String lsrId4 = "14.1.1.1";
+ // Allocate node label for specific device deviceId4
+ assertThat(srTeHandler.allocateNodeLabel(deviceId4, lsrId4, deviceIdLsrIdMap), is(true));
+ // Retrieve label from store
+ labelId = pceStore.getGlobalNodeLabel(deviceId4);
+ // Check whether label is generated for this device deviceId4
+ assertThat(labelId, is(notNullValue()));
+ //Now add device to deviceIdLsrIdMap
+ deviceIdLsrIdMap.put(deviceId4, lsrId4);
+
+ // device 5
+ String lsrId5 = "15.1.1.1";
+ // Allocate node label for specific device deviceId5
+ assertThat(srTeHandler.allocateNodeLabel(deviceId5, lsrId5, deviceIdLsrIdMap), is(true));
+ // Retrieve label from store
+ labelId = pceStore.getGlobalNodeLabel(deviceId5);
+ // Check whether label is generated for this device deviceId5
+ assertThat(labelId, is(notNullValue()));
+ //Now add device to deviceIdLsrIdMap
+ deviceIdLsrIdMap.put(deviceId5, lsrId5);
+ }
+
+ /**
+ * Checks the operation of releaseNodeLabel() method on node label.
+ */
+ @Test
+ public void testReleaseNodeLabelSuccess() {
+ testAllocateNodeLabel2();
+ // Specific device deviceId1
+
+ //device 1
+ String lsrId1 = "11.1.1.1";
+ // Check whether successfully released node label
+ assertThat(srTeHandler.releaseNodeLabel(deviceId1, lsrId1, deviceIdLsrIdMap), is(true));
+ // Check whether successfully removed label from store
+ LabelResourceId labelId = pceStore.getGlobalNodeLabel(deviceId1);
+ assertThat(labelId, is(nullValue()));
+ // Remove released deviceId1
+ deviceIdLsrIdMap.remove(deviceId1);
+
+ //device 2
+ String lsrId2 = "12.1.1.1";
+ // Check whether successfully released node label
+ assertThat(srTeHandler.releaseNodeLabel(deviceId2, lsrId2, deviceIdLsrIdMap), is(true));
+ // Check whether successfully removed label from store
+ labelId = pceStore.getGlobalNodeLabel(deviceId2);
+ assertThat(labelId, is(nullValue()));
+ // Remove released deviceId2
+ deviceIdLsrIdMap.remove(deviceId2);
+
+ //device 3
+ String lsrId3 = "13.1.1.1";
+ // Check whether successfully released node label
+ assertThat(srTeHandler.releaseNodeLabel(deviceId3, lsrId3, deviceIdLsrIdMap), is(true));
+ // Check whether successfully removed label from store
+ labelId = pceStore.getGlobalNodeLabel(deviceId3);
+ assertThat(labelId, is(nullValue()));
+ // Remove released deviceId3
+ deviceIdLsrIdMap.remove(deviceId3);
+
+ //device 4
+ String lsrId4 = "14.1.1.1";
+ // Check whether successfully released node label
+ assertThat(srTeHandler.releaseNodeLabel(deviceId4, lsrId4, deviceIdLsrIdMap), is(true));
+ // Check whether successfully removed label from store
+ labelId = pceStore.getGlobalNodeLabel(deviceId4);
+ assertThat(labelId, is(nullValue()));
+ // Remove released deviceId4
+ deviceIdLsrIdMap.remove(deviceId4);
+
+ //device 5
+ String lsrId5 = "15.1.1.1";
+ // Check whether successfully released node label
+ assertThat(srTeHandler.releaseNodeLabel(deviceId5, lsrId5, deviceIdLsrIdMap), is(true));
+ // Check whether successfully removed label from store
+ labelId = pceStore.getGlobalNodeLabel(deviceId5);
+ assertThat(labelId, is(nullValue()));
+ // Remove released deviceId5
+ deviceIdLsrIdMap.remove(deviceId5);
+ }
+
+ @Test
+ public void testReleaseNodeLabelFailure() {
+ testAllocateNodeLabel2();
+
+ //device 6
+ String lsrId6 = "16.1.1.1";
+ // Check whether successfully released node label
+ DeviceId deviceId6 = DeviceId.deviceId("foo6");
+ assertThat(srTeHandler.releaseNodeLabel(deviceId6, lsrId6, deviceIdLsrIdMap), 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);
+
+ // check node-label of deviceId1
+ List<LabelResourceId> labelList = labelStack.labelResources();
+ Iterator<LabelResourceId> iterator = labelList.iterator();
+ labelId = (LabelResourceId) iterator.next();
+ assertThat(labelId, is(LabelResourceId.labelResourceId(4097)));
+
+ // check adjacency label of deviceId1
+ labelId = (LabelResourceId) iterator.next();
+ assertThat(labelId, is(LabelResourceId.labelResourceId(5122)));
+
+ // check node-label of deviceId2
+ labelId = (LabelResourceId) iterator.next();
+ assertThat(labelId, is(LabelResourceId.labelResourceId(4098)));
+
+ // check adjacency label of deviceId2
+ labelId = (LabelResourceId) iterator.next();
+ assertThat(labelId, is(LabelResourceId.labelResourceId(5123)));
+
+ // check node-label of deviceId3
+ labelId = (LabelResourceId) iterator.next();
+ assertThat(labelId, is(LabelResourceId.labelResourceId(4099)));
+
+ // check adjacency label of deviceId3
+ labelId = (LabelResourceId) iterator.next();
+ assertThat(labelId, is(LabelResourceId.labelResourceId(5124)));
+
+ // check node-label of deviceId4
+ labelId = (LabelResourceId) iterator.next();
+ assertThat(labelId, is(LabelResourceId.labelResourceId(4100)));
+
+ // check adjacency label of deviceId4
+ labelId = (LabelResourceId) iterator.next();
+ assertThat(labelId, is(LabelResourceId.labelResourceId(5125)));
+
+ // check node-label of deviceId5
+ labelId = (LabelResourceId) iterator.next();
+ assertThat(labelId, is(LabelResourceId.labelResourceId(4101)));
+ }
+}
diff --git a/apps/pce/app/src/test/java/org/onosproject/pce/util/LabelResourceAdapter.java b/apps/pce/app/src/test/java/org/onosproject/pce/util/LabelResourceAdapter.java
new file mode 100644
index 0000000..ca948b0
--- /dev/null
+++ b/apps/pce/app/src/test/java/org/onosproject/pce/util/LabelResourceAdapter.java
@@ -0,0 +1,189 @@
+/*
+ * Copyright 2016-present Open Networking Laboratory
+ *
+ * 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.pce.util;
+
+import static com.google.common.base.Preconditions.checkNotNull;
+
+import java.util.Collection;
+import java.util.LinkedList;
+import java.util.Random;
+import java.util.Set;
+
+import org.onosproject.incubator.net.resource.label.DefaultLabelResource;
+import org.onosproject.incubator.net.resource.label.LabelResource;
+import org.onosproject.incubator.net.resource.label.LabelResourceAdminService;
+import org.onosproject.incubator.net.resource.label.LabelResourceDelegate;
+import org.onosproject.incubator.net.resource.label.LabelResourceEvent;
+import org.onosproject.incubator.net.resource.label.LabelResourceId;
+import org.onosproject.incubator.net.resource.label.LabelResourceListener;
+import org.onosproject.incubator.net.resource.label.LabelResourcePool;
+import org.onosproject.incubator.net.resource.label.LabelResourceProvider;
+import org.onosproject.incubator.net.resource.label.LabelResourceProviderRegistry;
+import org.onosproject.incubator.net.resource.label.LabelResourceProviderService;
+import org.onosproject.incubator.net.resource.label.LabelResourceService;
+import org.onosproject.net.Device;
+import org.onosproject.net.DeviceId;
+import org.onosproject.net.device.DeviceEvent;
+import org.onosproject.net.device.DeviceEvent.Type;
+import org.onosproject.net.device.DeviceListener;
+import org.onosproject.net.provider.AbstractListenerProviderRegistry;
+import org.onosproject.net.provider.AbstractProviderService;
+
+import com.google.common.collect.Multimap;
+
+/**
+ * Provides test implementation of class LabelResourceService.
+ */
+public class LabelResourceAdapter
+ extends AbstractListenerProviderRegistry<LabelResourceEvent, LabelResourceListener,
+ LabelResourceProvider, LabelResourceProviderService>
+ implements LabelResourceService, LabelResourceAdminService, LabelResourceProviderRegistry {
+ public static final long GLOBAL_LABEL_SPACE_MIN = 4097;
+ public static final long GLOBAL_LABEL_SPACE_MAX = 5121;
+ public static final long LOCAL_LABEL_SPACE_MIN = 5122;
+ public static final long LOCAL_LABEL_SPACE_MAX = 9217;
+
+ private Random random = new Random();
+
+ @Override
+ public boolean createDevicePool(DeviceId deviceId,
+ LabelResourceId beginLabel,
+ LabelResourceId endLabel) {
+ return true;
+ }
+
+ @Override
+ public boolean createGlobalPool(LabelResourceId beginLabel,
+ LabelResourceId endLabel) {
+ return true;
+ }
+
+ @Override
+ public boolean destroyDevicePool(DeviceId deviceId) {
+ return true;
+ }
+
+ @Override
+ public boolean destroyGlobalPool() {
+ return true;
+ }
+
+ public long getLabelId(long min, long max) {
+ return random.nextInt((int) max - (int) min + 1) + (int) min;
+ }
+
+ public Collection<LabelResource> applyFromDevicePool(DeviceId deviceId,
+ long applyNum) {
+ Collection<LabelResource> labelList = new LinkedList<>();
+ LabelResource label = new DefaultLabelResource(deviceId,
+ LabelResourceId.labelResourceId(
+ getLabelId(LOCAL_LABEL_SPACE_MIN, LOCAL_LABEL_SPACE_MAX)));
+ labelList.add(label);
+ return labelList;
+ }
+
+ public Collection<LabelResource> applyFromGlobalPool(long applyNum) {
+ Collection<LabelResource> labelList = new LinkedList<>();
+ LabelResource label = new DefaultLabelResource(DeviceId.deviceId("foo"),
+ LabelResourceId.labelResourceId(
+ getLabelId(GLOBAL_LABEL_SPACE_MIN, GLOBAL_LABEL_SPACE_MAX)));
+ labelList.add(label);
+ return labelList;
+ }
+
+ public boolean releaseToDevicePool(Multimap<DeviceId, LabelResource> release) {
+ return true;
+ }
+
+ public boolean releaseToGlobalPool(Set<LabelResourceId> release) {
+ return true;
+ }
+
+ public boolean isDevicePoolFull(DeviceId deviceId) {
+ return false;
+ }
+
+ public boolean isGlobalPoolFull() {
+ return false;
+ }
+
+ public long getFreeNumOfDevicePool(DeviceId deviceId) {
+ return 4;
+ }
+
+ public long getFreeNumOfGlobalPool() {
+ return 4;
+ }
+
+ @Override
+ public LabelResourcePool getDeviceLabelResourcePool(DeviceId deviceId) {
+ return null;
+ }
+
+ @Override
+ public LabelResourcePool getGlobalLabelResourcePool() {
+ return null;
+ }
+
+ private class InternalLabelResourceDelegate implements LabelResourceDelegate {
+ @Override
+ public void notify(LabelResourceEvent event) {
+ post(event);
+ }
+
+ }
+
+ private class InternalDeviceListener implements DeviceListener {
+ @Override
+ public void event(DeviceEvent event) {
+ Device device = event.subject();
+ if (Type.DEVICE_REMOVED.equals(event.type())) {
+ destroyDevicePool(device.id());
+ }
+ }
+ }
+
+ private class InternalLabelResourceProviderService
+ extends AbstractProviderService<LabelResourceProvider>
+ implements LabelResourceProviderService {
+
+ protected InternalLabelResourceProviderService(LabelResourceProvider provider) {
+ super(provider);
+ }
+
+ @Override
+ public void deviceLabelResourcePoolDetected(DeviceId deviceId,
+ LabelResourceId beginLabel,
+ LabelResourceId endLabel) {
+ checkNotNull(deviceId, "deviceId is not null");
+ checkNotNull(beginLabel, "beginLabel is not null");
+ checkNotNull(endLabel, "endLabel is not null");
+ createDevicePool(deviceId, beginLabel, endLabel);
+ }
+
+ @Override
+ public void deviceLabelResourcePoolDestroyed(DeviceId deviceId) {
+ checkNotNull(deviceId, "deviceId is not null");
+ destroyDevicePool(deviceId);
+ }
+
+ }
+
+ @Override
+ protected LabelResourceProviderService createProviderService(LabelResourceProvider provider) {
+ return null;
+ }
+}