X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;ds=sidebyside;f=pcep%2Ftopology%2Ftopology-provider%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fbgpcep%2Fpcep%2Ftopology%2Fprovider%2FPCEPStatefulPeerProposalTest.java;h=5e2e47c10447c19b452abff613fed85c96dae4d7;hb=61e4a03e5554241bcb21dce49abe3e089e6e0237;hp=8697e9cf9a33f958452fb75affc36d7b9fdab9e0;hpb=d60888a10b4c49cca3e3cc277ced79d05d0b3460;p=bgpcep.git

diff --git a/pcep/topology/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/PCEPStatefulPeerProposalTest.java b/pcep/topology/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/PCEPStatefulPeerProposalTest.java
index 8697e9cf9a..5e2e47c104 100644
--- a/pcep/topology/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/PCEPStatefulPeerProposalTest.java
+++ b/pcep/topology/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/PCEPStatefulPeerProposalTest.java
@@ -23,6 +23,7 @@ import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.ArgumentCaptor;
+import org.mockito.Captor;
 import org.mockito.Mock;
 import org.mockito.junit.MockitoJUnitRunner;
 import org.opendaylight.bgpcep.pcep.topology.provider.PCEPStatefulPeerProposal.LspDbVersionListener;
@@ -49,7 +50,7 @@ import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.
 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
-import org.opendaylight.yangtools.concepts.ListenerRegistration;
+import org.opendaylight.yangtools.concepts.Registration;
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 import org.opendaylight.yangtools.yang.common.Uint64;
 
@@ -66,11 +67,12 @@ public class PCEPStatefulPeerProposalTest {
     @Mock
     private DataBroker dataBroker;
     @Mock
-    private ListenerRegistration<?> listenerReg;
+    private Registration listenerReg;
     @Mock
     private FluentFuture<Optional<LspDbVersion>> listenableFutureMock;
-
+    @Captor
     private ArgumentCaptor<DataTreeChangeListener<?>> captor;
+
     private TlvsBuilder tlvsBuilder;
 
     @Before
@@ -78,8 +80,7 @@ public class PCEPStatefulPeerProposalTest {
         tlvsBuilder = new TlvsBuilder().addAugmentation(new Tlvs1Builder()
             .setStateful(new StatefulBuilder().addAugmentation(new Stateful1Builder().build()).build())
             .build());
-        captor = ArgumentCaptor.forClass(DataTreeChangeListener.class);
-        doReturn(listenerReg).when(dataBroker).registerDataTreeChangeListener(any(), captor.capture());
+        doReturn(listenerReg).when(dataBroker).registerTreeChangeListener(any(), captor.capture());
         doNothing().when(listenerReg).close();
     }
 
@@ -94,12 +95,12 @@ public class PCEPStatefulPeerProposalTest {
                 new NodeKey(ServerSessionManager.createNodeId(ADDRESS.getAddress())));
 
             final var dbverRoot = mock(DataObjectModification.class);
-            doReturn(LSP_DB_VERSION).when(dbverRoot).getDataAfter();
+            doReturn(LSP_DB_VERSION).when(dbverRoot).dataAfter();
             final var dbverMod = mock(DataTreeModification.class);
-            doReturn(DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL, modPath)).when(dbverMod).getRootPath();
+            doReturn(DataTreeIdentifier.of(LogicalDatastoreType.OPERATIONAL, modPath)).when(dbverMod).getRootPath();
             doReturn(dbverRoot).when(dbverMod).getRootNode();
 
-            for (DataTreeChangeListener<?> listener : listeners) {
+            for (var listener : listeners) {
                 if (listener instanceof LspDbVersionListener) {
                     listener.onDataTreeChanged(List.of(dbverMod));
                 }
@@ -121,20 +122,20 @@ public class PCEPStatefulPeerProposalTest {
                 new NodeKey(ServerSessionManager.createNodeId(ADDRESS.getAddress())));
 
             final var dbverRoot = mock(DataObjectModification.class);
-            doReturn(LSP_DB_VERSION).when(dbverRoot).getDataAfter();
+            doReturn(LSP_DB_VERSION).when(dbverRoot).dataAfter();
             final var dbverMod = mock(DataTreeModification.class);
-            doReturn(DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL, modPath)).when(dbverMod).getRootPath();
+            doReturn(DataTreeIdentifier.of(LogicalDatastoreType.OPERATIONAL, modPath)).when(dbverMod).getRootPath();
             doReturn(dbverRoot).when(dbverMod).getRootNode();
 
             final var speakerRoot = mock(DataObjectModification.class);
             doReturn(new PcepNodeSyncConfigBuilder().setSpeakerEntityIdValue(SPEAKER_ID).build()).when(speakerRoot)
-                .getDataAfter();
+                .dataAfter();
             final var speakerMod = mock(DataTreeModification.class);
-            doReturn(DataTreeIdentifier.create(LogicalDatastoreType.CONFIGURATION, modPath)).when(speakerMod)
+            doReturn(DataTreeIdentifier.of(LogicalDatastoreType.CONFIGURATION, modPath)).when(speakerMod)
                 .getRootPath();
             doReturn(speakerRoot).when(speakerMod).getRootNode();
 
-            for (DataTreeChangeListener<?> listener : listeners) {
+            for (var listener : listeners) {
                 if (listener instanceof SpeakerIdListener) {
                     listener.onDataTreeChanged(List.of(speakerMod));
                 } else if (listener instanceof LspDbVersionListener) {