Merge "Update tox documentation link in tox-guide"
[transportpce.git] / pce / src / test / java / org / opendaylight / transportpce / pce / networkanalyzer / PceLinkTest.java
index 0dec7fd4656b44dd96882b81fae796795ac75d9d..90e53324b2f4bdabf52dc5e4bd7108823d550ea0 100644 (file)
@@ -8,14 +8,17 @@
 
 package org.opendaylight.transportpce.pce.networkanalyzer;
 
-import static org.junit.Assert.assertEquals;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 import java.math.RoundingMode;
 import java.util.HashMap;
 import java.util.Map;
-import org.junit.Assert;
-import org.junit.Before;
-import org.junit.Test;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
 import org.opendaylight.transportpce.common.NetworkUtils;
@@ -25,20 +28,20 @@ import org.opendaylight.transportpce.common.mapping.PortMapping;
 import org.opendaylight.transportpce.test.AbstractTest;
 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.link.types.rev191129.FiberPmd;
 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.link.types.rev191129.RatioDB;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev211210.Link1Builder;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev211210.TerminationPoint1Builder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev230526.Link1Builder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev230526.TerminationPoint1Builder;
 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.state.types.rev191129.State;
 import org.opendaylight.yang.gen.v1.http.org.openroadm.equipment.states.types.rev191129.AdminStates;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev211210.span.attributes.LinkConcatenation1.FiberType;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev211210.span.attributes.LinkConcatenation1Builder;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev211210.networks.network.link.OMSAttributesBuilder;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev211210.networks.network.link.oms.attributes.SpanBuilder;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev211210.OpenroadmLinkType;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev211210.OpenroadmNodeType;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev211210.OpenroadmTpType;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev211210.link.concatenation.LinkConcatenation;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev211210.link.concatenation.LinkConcatenationBuilder;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev211210.link.concatenation.LinkConcatenationKey;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev230526.span.attributes.LinkConcatenation1.FiberType;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.link.rev230526.span.attributes.LinkConcatenation1Builder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev230526.networks.network.link.OMSAttributesBuilder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev230526.networks.network.link.oms.attributes.SpanBuilder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev230526.OpenroadmLinkType;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev230526.OpenroadmNodeType;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev230526.OpenroadmTpType;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev230526.link.concatenation.LinkConcatenation;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev230526.link.concatenation.LinkConcatenationBuilder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.network.types.rev230526.link.concatenation.LinkConcatenationKey;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NodeId;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.Node;
@@ -71,250 +74,207 @@ public class PceLinkTest extends AbstractTest {
     @Mock
     private PortMapping portMapping;
 
-    @Before
-    public void setup() {
+    @BeforeEach
+    void setup() {
         MockitoAnnotations.openMocks(this);
     }
 
     @Test
-    public void testBuildPceLinkRoadmToRoadm() {
-        Link link = createRoadmToRoadm("srcNode",
-            "destNode",
-            "srcTp", "destTp").build();
-        NodeBuilder node1Builder = getNodeBuilder(geSupportingNodes());
-        Node node = node1Builder.setNodeId(new NodeId("test")).build();
-        PceOpticalNode pceOpticalNode1 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        PceOpticalNode pceOpticalNode2 = new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        pceLink = new PceLink(link, pceOpticalNode1, pceOpticalNode2);
+    void testBuildPceLinkRoadmToRoadm() {
+        Node node = getNodeBuilder(geSupportingNodes()).setNodeId(new NodeId("test")).build();
+        pceLink = new PceLink(
+            createRoadmToRoadm("srcNode", "destNode", "srcTp", "destTp").build(),
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50),
+            new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50));
     }
 
     @Test
-    public void testBuildPceLinkRoadmToRoadmWithoutPMD() {
-        Link link = createRoadmToRoadmWithoutPMD("srcNode",
-            "destNode",
-            "srcTp", "destTp").build();
-        NodeBuilder node1Builder = getNodeBuilder(geSupportingNodes());
-        Node node = node1Builder.setNodeId(new NodeId("test")).build();
-        PceOpticalNode pceOpticalNode1 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        PceOpticalNode pceOpticalNode2 = new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        pceLink = new PceLink(link, pceOpticalNode1, pceOpticalNode2);
-        Assert.assertNotNull(MapUtils.getOmsAttributesSpan(link));
-        Assert.assertEquals(1, pceLink.getsrlgList().size());
-        assertEquals("Checking length loss", 20.0, pceLink.getspanLoss(), 0.005);
-        assertEquals("Checking length loss", 825.0, pceLink.getcd(), 0.005);
-        assertEquals("Checking PMDvalue of link", 4.0, pceLink.getpmd2(), 0.005);
+    void testBuildPceLinkRoadmToRoadmWithoutPMD() {
+        Link link = createRoadmToRoadmWithoutPMD("srcNode", "destNode", "srcTp", "destTp").build();
+        Node node = getNodeBuilder(geSupportingNodes()).setNodeId(new NodeId("test")).build();
+        pceLink = new PceLink(
+            link,
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50),
+            new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50));
+        assertNotNull(MapUtils.getOmsAttributesSpan(link));
+        assertEquals(1, pceLink.getsrlgList().size());
+        assertEquals(20.0, pceLink.getspanLoss(), 0.005, "Checking length loss");
+        assertEquals(825.0, pceLink.getcd(), 0.005, "Checking length loss");
+        assertEquals(4.0, pceLink.getpmd2(), 0.005, "Checking PMDvalue of link");
     }
 
     @Test
-    public void testBuildPceLinkRoadmToRoadmWithoutLinkLatency() {
-        Link link = createRoadmToRoadmWithoutLinkLatency("srcNode",
-            "destNode",
-            "srcTp", "destTp").build();
-
-        NodeBuilder node1Builder = getNodeBuilder(geSupportingNodes());
-        Node node = node1Builder.setNodeId(new NodeId("test")).build();
-        PceOpticalNode pceOpticalNode1 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        PceOpticalNode pceOpticalNode2 = new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        pceLink = new PceLink(link, pceOpticalNode1, pceOpticalNode2);
+    void testBuildPceLinkRoadmToRoadmWithoutLinkLatency() {
+        Node node = getNodeBuilder(geSupportingNodes()).setNodeId(new NodeId("test")).build();
+        pceLink = new PceLink(
+            createRoadmToRoadmWithoutLinkLatency("srcNode", "destNode", "srcTp", "destTp").build(),
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50),
+            new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50));
     }
 
     @Test
-    public void testBuildPceLinkOTN() {
-        // TODO: Modify with OTN node not PceO=ticalNode
-        Link link = createOTNLink("srcNode",
-            "destNode",
-            "srcTp", "destTp").build();
-
-        NodeBuilder node1Builder = getNodeBuilder(geSupportingNodes());
-        Node node = node1Builder.setNodeId(new NodeId("test")).build();
-        PceOpticalNode pceOpticalNode1 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.SWITCH, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        PceOpticalNode pceOpticalNode2 = new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
-            OpenroadmNodeType.SWITCH, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        pceLink = new PceLink(link, pceOpticalNode1, pceOpticalNode2);
+    void testBuildPceLinkOTN() {
+        // TODO: Modify with OTN node not PceOpticalNode
+        Node node = getNodeBuilder(geSupportingNodes()).setNodeId(new NodeId("test")).build();
+        pceLink = new PceLink(
+            createOTNLink("srcNode", "destNode", "srcTp", "destTp").build(),
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.SWITCH, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50),
+            new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
+                OpenroadmNodeType.SWITCH, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50));
     }
 
     @Test
-    public void testBuildPceLinkExponder() {
-        Link link = createXponderLink("srcNode",
-            "destNode",
-            "srcTp", "destTp").build();
-
-        NodeBuilder node1Builder = getNodeBuilder(geSupportingNodes());
-        Node node = node1Builder.setNodeId(new NodeId("test")).build();
-        PceOpticalNode pceOpticalNode1 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.XPONDER, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        PceOpticalNode pceOpticalNode2 = new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
-            OpenroadmNodeType.SRG, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        pceLink = new PceLink(link, pceOpticalNode1, pceOpticalNode2);
+    void testBuildPceLinkExponder() {
+        Node node = getNodeBuilder(geSupportingNodes()).setNodeId(new NodeId("test")).build();
+        pceLink = new PceLink(
+            createXponderLink("srcNode", "destNode", "srcTp", "destTp").build(),
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.XPONDER, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50),
+            new PceOpticalNode(deviceNodeId2, serviceType, portMapping, node,
+                OpenroadmNodeType.SRG, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50));
     }
 
     @Test
-    public void testCalcSpanOSNR() {
-        Link link = createRoadmToRoadm("srcNode",
-            "destNode",
-            "srcTp", "destTp").build();
-
-        NodeBuilder node1Builder = getNodeBuilder(geSupportingNodes());
-        Node node = node1Builder.setNodeId(new NodeId("test")).build();
-        PceOpticalNode pceOpticalNode1 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        PceOpticalNode pceOpticalNode2 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        pceLink = new PceLink(link, pceOpticalNode1, pceOpticalNode2);
-        Assert.assertNotNull(MapUtils.getOmsAttributesSpan(link));
-        // Assert.assertNotNull(pceLink.getosnr());
-        Assert.assertEquals(1, pceLink.getsrlgList().size());
-        assertEquals("Checking PMDvalue of link", 0.25, pceLink.getpmd2(), 0.005);
-        assertEquals("Checking CDvalue of link", 825, pceLink.getcd(), 0.005);
-        // Assert.assertTrue(7.857119000000001 == pceLink.getosnr());
-        Assert.assertNull(pceLink.getOppositeLink());
-        Assert.assertNull(pceLink.getOppositeLink());
-        Assert.assertNotNull(pceLink.getDestTP());
-        Assert.assertNotNull(pceLink.getlinkType());
-        Assert.assertNotNull(pceLink.getLinkId());
-        Assert.assertNotNull(pceLink.getSourceId());
-        Assert.assertNotNull(pceLink.getDestId());
-        pceLink.setClient("specific_client");
-        Assert.assertTrue(pceLink.getClient().equals("specific_client"));
-        Assert.assertNotNull(pceLink.getClient());
-        Assert.assertNotNull(pceLink.getLatency());
-        Assert.assertNotNull(pceLink.getAvailableBandwidth());
-        Assert.assertNotNull(pceLink.getUsedBandwidth());
-        Assert.assertNotNull(pceLink.getsourceNetworkSupNodeId());
-        Assert.assertNotNull(pceLink.getdestNetworkSupNodeId());
-        Assert.assertNotNull(pceLink.getSourceTP());
-        Assert.assertNotNull(pceLink.getsourceCLLI());
-        Assert.assertNotNull(pceLink.getdestCLLI());
-        Assert.assertTrue(pceLink.toString().equals("PceLink type=" + pceLink.getlinkType()
+    void testCalcSpanOSNR() {
+        Link link = createRoadmToRoadm("srcNode", "destNode", "srcTp", "destTp").build();
+        Node node = getNodeBuilder(geSupportingNodes()).setNodeId(new NodeId("test")).build();
+        pceLink = new PceLink(
+            link,
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50),
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50));
+        assertNotNull(MapUtils.getOmsAttributesSpan(link));
+        // assertNotNull(pceLink.getosnr());
+        assertEquals(1, pceLink.getsrlgList().size());
+        assertEquals(0.25, pceLink.getpmd2(), 0.005, "Checking PMDvalue of link");
+        assertEquals(825, pceLink.getcd(), 0.005, "Checking CDvalue of link");
+        // assertTrue(7.857119000000001 == pceLink.getosnr());
+        assertNull(pceLink.getOppositeLink());
+        assertNull(pceLink.getOppositeLink());
+        assertNotNull(pceLink.getDestTP());
+        assertNotNull(pceLink.getlinkType());
+        assertNotNull(pceLink.getLinkId());
+        assertNotNull(pceLink.getSourceId());
+        assertNotNull(pceLink.getDestId());
+        pceLink.setClientA("specific_client");
+        assertTrue(pceLink.getClientA().equals("specific_client"));
+        assertNotNull(pceLink.getClientA());
+        assertNotNull(pceLink.getLatency());
+        assertNotNull(pceLink.getAvailableBandwidth());
+        assertNotNull(pceLink.getUsedBandwidth());
+        assertNotNull(pceLink.getsourceNetworkSupNodeId());
+        assertNotNull(pceLink.getdestNetworkSupNodeId());
+        assertNotNull(pceLink.getSourceTP());
+        assertNotNull(pceLink.getsourceCLLI());
+        assertNotNull(pceLink.getdestCLLI());
+        assertTrue(pceLink.toString().equals("PceLink type=" + pceLink.getlinkType()
             + " ID=" + pceLink.getLinkId().getValue() + " latency=" + pceLink.getLatency().intValue()));
     }
 
     @Test
-    public void testWrongSpanLoss() {
-        Link link = createInvalidRoadmToRoadm("srcNode",
-            "destNode",
-            "srcTp", "destTp").build();
-
-        NodeBuilder node1Builder = getNodeBuilder(geSupportingNodes());
-        Node node = node1Builder.setNodeId(new NodeId("test")).build();
-        PceOpticalNode pceOpticalNode1 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        PceOpticalNode pceOpticalNode2 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        pceLink = new PceLink(link, pceOpticalNode1, pceOpticalNode2);
-        Assert.assertNull(MapUtils.getOmsAttributesSpan(link));
-        Assert.assertNull(pceLink.getpmd2());
-        Assert.assertNull(pceLink.getpowerCorrection());
-        Assert.assertNull(pceLink.getcd());
+    void testWrongSpanLoss() {
+        Link link = createInvalidRoadmToRoadm("srcNode", "destNode", "srcTp", "destTp").build();
+        Node node = getNodeBuilder(geSupportingNodes()).setNodeId(new NodeId("test")).build();
+        pceLink = new PceLink(
+            link,
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50),
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50));
+        assertNull(MapUtils.getOmsAttributesSpan(link));
+        assertNull(pceLink.getpmd2());
+        assertNull(pceLink.getpowerCorrection());
+        assertNull(pceLink.getcd());
     }
 
     @Test
-    public void testExtrapolatedPMD() {
-        Link link = createRoadmToRoadmWithoutPMD("srcNode",
-            "destNode",
-            "srcTp", "destTp").build();
-
-        NodeBuilder node1Builder = getNodeBuilder(geSupportingNodes());
-        Node node = node1Builder.setNodeId(new NodeId("test")).build();
-        PceOpticalNode pceOpticalNode1 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        PceOpticalNode pceOpticalNode2 = new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
-            OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
-            GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50);
-        pceLink = new PceLink(link, pceOpticalNode1, pceOpticalNode2);
-        Assert.assertNotNull(MapUtils.getOmsAttributesSpan(link));
-        Assert.assertEquals(1, pceLink.getsrlgList().size());
-        assertEquals("Checking PMDvalue of link", 4.0, pceLink.getpmd2(), 0.005);
+    void testExtrapolatedPMD() {
+        Link link = createRoadmToRoadmWithoutPMD("srcNode", "destNode", "srcTp", "destTp").build();
+        Node node = getNodeBuilder(geSupportingNodes()).setNodeId(new NodeId("test")).build();
+        pceLink = new PceLink(
+            link,
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50),
+            new PceOpticalNode(deviceNodeId, serviceType, portMapping, node,
+                OpenroadmNodeType.DEGREE, StringConstants.OPENROADM_DEVICE_VERSION_2_2_1,
+                GridConstant.SLOT_WIDTH_50, GridConstant.SLOT_WIDTH_50));
+        assertNotNull(MapUtils.getOmsAttributesSpan(link));
+        assertEquals(1, pceLink.getsrlgList().size());
+        assertEquals(4.0, pceLink.getpmd2(), 0.005, "Checking PMDvalue of link");
     }
 
     private static LinkBuilder createOTNLink(String srcNode, String destNode, String srcTp, String destTp) {
-        Link1Builder link1Builder = new Link1Builder()
-            .setLinkType(OpenroadmLinkType.OTNLINK)
-            .setOperationalState(State.InService)
-            .setAdministrativeState(AdminStates.InService);
-        // create source link
-        return createLinkBuilder(false, false, 10000.0, srcNode, destNode, srcTp, destTp, link1Builder);
+        return createLinkBuilder(false, false, 10000.0, srcNode, destNode, srcTp, destTp,
+            new Link1Builder()
+                .setLinkType(OpenroadmLinkType.OTNLINK)
+                .setOperationalState(State.InService)
+                .setAdministrativeState(AdminStates.InService));
     }
 
     private static LinkBuilder createXponderLink(String srcNode, String destNode, String srcTp, String destTp) {
-        Link1Builder link1Builder = new Link1Builder()
-            .setLinkType(OpenroadmLinkType.XPONDERINPUT)
-            .setAdministrativeState(AdminStates.InService)
-            .setOperationalState(State.InService);
         // create source link
-        return createLinkBuilder(false, false, 10.0, srcNode, destNode, srcTp, destTp, link1Builder);
+        return createLinkBuilder(false, false, 10.0, srcNode, destNode, srcTp, destTp,
+            new Link1Builder()
+                .setLinkType(OpenroadmLinkType.XPONDERINPUT)
+                .setAdministrativeState(AdminStates.InService)
+                .setOperationalState(State.InService));
     }
 
     private static LinkBuilder createLinkBuilder(boolean pmdpresent, boolean omspresent, double length,
             String srcNode, String destNode, String srcTp, String destTp, Link1Builder link1Builder) {
-        SourceBuilder ietfSrcLinkBldr = new SourceBuilder().setSourceNode(new NodeId(srcNode))
-            .setSourceTp(new TpId(srcTp));
-        // create destination link
-        DestinationBuilder ietfDestLinkBldr = new DestinationBuilder().setDestNode(new NodeId(destNode))
-            .setDestTp(new TpId(destTp));
         LinkId linkId = new LinkId(String.format(LINK_ID_FORMAT, srcNode, srcTp, destNode, destTp));
-
-        // For setting up attributes for openRoadm augment
-        LinkConcatenation linkConcatenation = new LinkConcatenationBuilder()
-            .withKey(new LinkConcatenationKey(Uint32.valueOf(1)))
-            .setSRLGLength(Decimal64.valueOf(length, RoundingMode.FLOOR))
-            .addAugmentation(new LinkConcatenation1Builder()
-                .setFiberType(FiberType.Smf)
-                .build())
-            .build();
-        if (pmdpresent) {
-            linkConcatenation = new LinkConcatenationBuilder()
-                .withKey(new LinkConcatenationKey(Uint32.valueOf(1)))
-                .setSRLGLength(Decimal64.valueOf(length, RoundingMode.FLOOR))
-                .addAugmentation(new LinkConcatenation1Builder()
-                    .setFiberType(FiberType.Smf)
-                    .setPmd(FiberPmd.getDefaultInstance("0.500"))
-                    .build())
-                .build();
-        }
-        OMSAttributesBuilder omsAttributesBuilder = new OMSAttributesBuilder()
-            .setSpan(new SpanBuilder()
-                // .setSpanlossCurrent(new RatioDB(Decimal64.valueOf("55")))
-                .setSpanlossCurrent(new RatioDB(Decimal64.valueOf("20")))
-                .setLinkConcatenation(Map.of(linkConcatenation.key(), linkConcatenation))
-                .build());
-
         LinkBuilder linkBuilder = new LinkBuilder()
-            .setSource(ietfSrcLinkBldr.build())
-            .setDestination(ietfDestLinkBldr.build())
+            .setSource(
+                new SourceBuilder().setSourceNode(new NodeId(srcNode)).setSourceTp(new TpId(srcTp)).build())
+            .setDestination(
+                new DestinationBuilder().setDestNode(new NodeId(destNode)).setDestTp(new TpId(destTp)).build())
             .setLinkId(linkId)
-            .withKey(new LinkKey(linkId));
-
-        linkBuilder.addAugmentation(link1Builder.build());
+            .withKey(new LinkKey(linkId))
+            .addAugmentation(link1Builder.build());
         if (omspresent) {
-            var linkBuilderNetworkLink = new org.opendaylight.yang.gen.v1.http
-                .org.openroadm.network.topology.rev211210.Link1Builder()
-                    .setOMSAttributes(omsAttributesBuilder
+            LinkConcatenation linkConcatenation = new LinkConcatenationBuilder()
+                .withKey(new LinkConcatenationKey(Uint32.valueOf(1)))
+                .setSRLGLength(Decimal64.valueOf(length, RoundingMode.FLOOR))
+                .addAugmentation(
+                    pmdpresent
+                        ? new LinkConcatenation1Builder().setFiberType(FiberType.Smf)
+                                .setPmd(FiberPmd.getDefaultInstance("0.500")).build()
+                        : new LinkConcatenation1Builder().setFiberType(FiberType.Smf).build())
+                .build();
+            linkBuilder.addAugmentation(
+                new org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev230526.Link1Builder()
+                    .setOMSAttributes(
+                        new OMSAttributesBuilder()
+                            .setSpan(new SpanBuilder()
+                            // .setSpanlossCurrent(new RatioDB(Decimal64.valueOf("55")))
+                            .setSpanlossCurrent(new RatioDB(Decimal64.valueOf("20")))
+                            .setLinkConcatenation(Map.of(linkConcatenation.key(), linkConcatenation))
+                            .build())
+                        .build())
                     .build());
-
-            linkBuilder.addAugmentation(linkBuilderNetworkLink.build());
         }
         return linkBuilder;
     }
@@ -354,45 +314,37 @@ public class PceLinkTest extends AbstractTest {
     }
 
     private Map<SupportingNodeKey, SupportingNode> geSupportingNodes() {
-        Map<SupportingNodeKey, SupportingNode> supportingNodes1 = new HashMap<>();
         SupportingNode supportingNode1 = new SupportingNodeBuilder()
             .setNodeRef(new NodeId("node 1"))
             .setNetworkRef(new NetworkId(NetworkUtils.CLLI_NETWORK_ID))
             .build();
-        supportingNodes1
-            .put(supportingNode1.key(), supportingNode1);
-
         SupportingNode supportingNode2 = new SupportingNodeBuilder()
             .setNodeRef(new NodeId("node 2"))
             .setNetworkRef(new NetworkId(NetworkUtils.UNDERLAY_NETWORK_ID))
             .build();
-        supportingNodes1
-            .put(supportingNode2.key(), supportingNode2);
-        return supportingNodes1;
+        return new HashMap<>(Map.of(
+                supportingNode1.key(), supportingNode1,
+                supportingNode2.key(), supportingNode2));
     }
 
     private NodeBuilder getNodeBuilder(Map<SupportingNodeKey, SupportingNode> supportingNodes1) {
         // update tp of nodes
-        TerminationPointBuilder xpdrTpBldr = new TerminationPointBuilder()
-            .withKey(new TerminationPointKey(new TpId("xpdr")));
-        TerminationPoint1Builder tp1Bldr = new TerminationPoint1Builder();
-
-        tp1Bldr.setTpType(OpenroadmTpType.XPONDERNETWORK).setAdministrativeState(AdminStates.InService)
-            .setOperationalState(State.InService);
-        xpdrTpBldr.addAugmentation(tp1Bldr.build());
-        TerminationPoint xpdr = xpdrTpBldr.build();
-        org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226
-            .Node1 node1 = new Node1Builder()
-                .setTerminationPoint(Map.of(xpdr.key(), xpdr)).build();
-        org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev211210.Node1 node11 =
-            new org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev211210.Node1Builder()
-                .setOperationalState(State.InService).setAdministrativeState(AdminStates.InService).build();
+        TerminationPoint xpdr = new TerminationPointBuilder()
+            .withKey(new TerminationPointKey(new TpId("xpdr")))
+            .addAugmentation(
+                new TerminationPoint1Builder()
+                    .setTpType(OpenroadmTpType.XPONDERNETWORK)
+                    .setAdministrativeState(AdminStates.InService)
+                    .setOperationalState(State.InService)
+                    .build())
+            .build();
         return new NodeBuilder()
             .setNodeId(new NodeId("node 1"))
             .withKey(new NodeKey(new NodeId("node 1")))
-            .addAugmentation(node1)
-            .addAugmentation(node11)
+            .addAugmentation(new Node1Builder().setTerminationPoint(Map.of(xpdr.key(), xpdr)).build())
+            .addAugmentation(
+                new org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev230526.Node1Builder()
+                    .setOperationalState(State.InService).setAdministrativeState(AdminStates.InService).build())
             .setSupportingNode(supportingNodes1);
     }
-
-}
+}
\ No newline at end of file