Solve issue for testCalculateSpanlossBase

[transportpce.git] / olm / src / test / java / org / opendaylight / transportpce / olm / util / OlmPowerServiceRpcImplUtil.java

diff --git a/olm/src/test/java/org/opendaylight/transportpce/olm/util/OlmPowerServiceRpcImplUtil.java b/olm/src/test/java/org/opendaylight/transportpce/olm/util/OlmPowerServiceRpcImplUtil.java
index ec9955157bfaecaffd188b68cf23e3e382b679cd..8ba6ce8f0e8794e0b964e4b126d93b1e1a18d8c7 100644 (file)
--- a/olm/src/test/java/org/opendaylight/transportpce/olm/util/OlmPowerServiceRpcImplUtil.java
+++ b/olm/src/test/java/org/opendaylight/transportpce/olm/util/OlmPowerServiceRpcImplUtil.java
@@ -147,11 +147,19 @@ public final class OlmPowerServiceRpcImplUtil {
         return input;
     }
 
-    public static CalculateSpanlossBaseInput getCalculateSpanlossBaseInput() {
+    public static CalculateSpanlossBaseInput getCalculateSpanlossBaseInputLink() {
         CalculateSpanlossBaseInput input = new CalculateSpanlossBaseInputBuilder()
-            .setLinkId(new LinkId("link 1"))
-            .setSrcType(CalculateSpanlossBaseInput.SrcType.Link)
-            .addAugmentation(null, null).build();
+                .setLinkId(new LinkId("ROADM-A1-to-ROADM-C1"))
+                .setSrcType(CalculateSpanlossBaseInput.SrcType.Link)
+                .addAugmentation(null, null).build();
+        return input;
+    }
+
+    public static CalculateSpanlossBaseInput getCalculateSpanlossBaseInputAll() {
+        CalculateSpanlossBaseInput input = new CalculateSpanlossBaseInputBuilder()
+                .setLinkId(new LinkId("ROADM-A1-to-ROADM-C1"))
+                .setSrcType(CalculateSpanlossBaseInput.SrcType.All)
+                .addAugmentation(null, null).build();
         return input;
     }
 
@@ -174,5 +182,4 @@ public final class OlmPowerServiceRpcImplUtil {
             .setServiceName("service 1").build();
         return input;
     }
-
 }