Modify spectrum assignment management in PCE

[transportpce.git] / pce / src / test / java / org / opendaylight / transportpce / pce / impl / PceProviderTest.java

diff --git a/pce/src/test/java/org/opendaylight/transportpce/pce/impl/PceProviderTest.java b/pce/src/test/java/org/opendaylight/transportpce/pce/impl/PceProviderTest.java
index b04a75adf70936e6f3b53bba9648542a325e61db..02665c1ebe4554b4228e7f8d27644adfe6750eb6 100644 (file)
--- a/pce/src/test/java/org/opendaylight/transportpce/pce/impl/PceProviderTest.java
+++ b/pce/src/test/java/org/opendaylight/transportpce/pce/impl/PceProviderTest.java
@@ -8,7 +8,6 @@
 
 package org.opendaylight.transportpce.pce.impl;
 
-import static org.mockito.ArgumentMatchers.anyObject;
 import static org.mockito.ArgumentMatchers.eq;
 
 import org.eclipse.jdt.annotation.NonNull;
@@ -23,10 +22,9 @@ import org.opendaylight.transportpce.pce.service.PathComputationService;
 import org.opendaylight.transportpce.pce.service.PathComputationServiceImpl;
 import org.opendaylight.transportpce.pce.utils.NotificationPublishServiceMock;
 import org.opendaylight.transportpce.test.AbstractTest;
-import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.pce.rev200128.TransportpcePceService;
+import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.pce.rev210701.TransportpcePceService;
 import org.opendaylight.yangtools.concepts.ObjectRegistration;
 
-
 public class PceProviderTest extends AbstractTest {
 
     private RpcProviderService rpcService;
@@ -43,7 +41,8 @@ public class PceProviderTest extends AbstractTest {
         notificationPublishService = new NotificationPublishServiceMock();
         requestProcessor = Mockito.mock(RequestProcessor.class);
         networkTransaction = new NetworkTransactionImpl(requestProcessor);
-        pathComputationService = new PathComputationServiceImpl(networkTransaction, notificationPublishService);
+        pathComputationService = new PathComputationServiceImpl(networkTransaction, notificationPublishService,
+                null, null);
         pceProvider = new PceProvider(rpcService, pathComputationService);
     }
 
@@ -61,11 +60,9 @@ public class PceProviderTest extends AbstractTest {
 
             }
         };
-
-
         Mockito
                 .when(rpcService
-                        .registerRpcImplementation(eq(TransportpcePceService.class), anyObject()))
+                        .registerRpcImplementation(eq(TransportpcePceService.class), Mockito.any()))
                 .thenReturn(rpcRegistration);
         pceProvider.init();
         pceProvider.close();