* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
-
package org.opendaylight.bgpcep.pcep.topology.provider;
import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
-import static org.mockito.Mockito.*;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.doNothing;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.mock;
-import com.google.common.base.Optional;
-import com.google.common.util.concurrent.CheckedFuture;
-import java.math.BigInteger;
-import java.util.concurrent.ExecutionException;
-import java.util.concurrent.Executor;
+import com.google.common.util.concurrent.FluentFuture;
+import java.net.InetSocketAddress;
+import java.util.List;
+import java.util.Optional;
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.MockitoAnnotations;
-import org.opendaylight.controller.md.sal.binding.api.DataBroker;
-import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
-import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
-import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev171025.Stateful1;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev171025.Stateful1Builder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev171025.Tlvs3;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev171025.lsp.db.version.tlv.LspDbVersion;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev171025.lsp.db.version.tlv.LspDbVersionBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev171025.Tlvs1;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev171025.Tlvs1Builder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev171025.stateful.capability.tlv.StatefulBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.opendaylight.bgpcep.pcep.topology.provider.PCEPStatefulPeerProposal.LspDbVersionListener;
+import org.opendaylight.bgpcep.pcep.topology.provider.PCEPStatefulPeerProposal.SpeakerIdListener;
+import org.opendaylight.mdsal.binding.api.DataBroker;
+import org.opendaylight.mdsal.binding.api.DataObjectModification;
+import org.opendaylight.mdsal.binding.api.DataTreeChangeListener;
+import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
+import org.opendaylight.mdsal.binding.api.DataTreeModification;
+import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Stateful1Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Tlvs3;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Tlvs3Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.lsp.db.version.tlv.LspDbVersion;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.lsp.db.version.tlv.LspDbVersionBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.speaker.entity.id.tlv.SpeakerEntityIdBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.Tlvs1Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.stateful.capability.tlv.StatefulBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.open.TlvsBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.sync.optimizations.config.rev181109.PcepNodeSyncConfigBuilder;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
-import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.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.Registration;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.common.Uint64;
+@RunWith(MockitoJUnitRunner.StrictStubs.class)
public class PCEPStatefulPeerProposalTest {
-
private static final InstanceIdentifier<Topology> TOPOLOGY_IID = InstanceIdentifier.create(NetworkTopology.class)
.child(Topology.class, new TopologyKey(new TopologyId("topology")));
- private static final NodeId NODE_ID = new NodeId("node");
- private static final LspDbVersion LSP_DB_VERSION = new LspDbVersionBuilder().setLspDbVersionValue(
- BigInteger.ONE).build();
+ private static final LspDbVersion LSP_DB_VERSION = new LspDbVersionBuilder()
+ .setLspDbVersionValue(Uint64.ONE)
+ .build();
+ private static final byte[] SPEAKER_ID = {0x01, 0x02, 0x03, 0x04};
+ private static final InetSocketAddress ADDRESS = new InetSocketAddress(4321);
@Mock
private DataBroker dataBroker;
@Mock
- private CheckedFuture<Optional<LspDbVersion>, ReadFailedException> listenableFutureMock;
+ private Registration listenerReg;
@Mock
- private ReadOnlyTransaction rTxMock;
+ private FluentFuture<Optional<LspDbVersion>> listenableFutureMock;
+ @Captor
+ private ArgumentCaptor<DataTreeChangeListener<?>> captor;
+
private TlvsBuilder tlvsBuilder;
- @SuppressWarnings("unchecked")
@Before
- public void setUp() throws InterruptedException, ExecutionException {
- MockitoAnnotations.initMocks(this);
- this.tlvsBuilder = new TlvsBuilder().addAugmentation(
- Tlvs1.class,
- new Tlvs1Builder().setStateful(
- new StatefulBuilder().addAugmentation(Stateful1.class, new Stateful1Builder().build()).build())
- .build());
- doReturn(this.rTxMock).when(this.dataBroker).newReadOnlyTransaction();
- doNothing().when(this.rTxMock).close();
- doReturn(this.listenableFutureMock).when(this.rTxMock)
- .read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class));
- doReturn(true).when(this.listenableFutureMock).isDone();
- doAnswer(invocation -> {
- final Runnable runnable = (Runnable) invocation.getArguments()[0];
- runnable.run();
- return null;
- }).when(this.listenableFutureMock).addListener(any(Runnable.class), any(Executor.class));
+ public void setUp() {
+ tlvsBuilder = new TlvsBuilder().addAugmentation(new Tlvs1Builder()
+ .setStateful(new StatefulBuilder().addAugmentation(new Stateful1Builder().build()).build())
+ .build());
+ doReturn(listenerReg).when(dataBroker).registerTreeChangeListener(any(), captor.capture());
+ doNothing().when(listenerReg).close();
}
@Test
- public void testSetPeerProposalSuccess() throws InterruptedException, ExecutionException {
- doReturn(Optional.of(LSP_DB_VERSION)).when(this.listenableFutureMock).get();
- final PCEPStatefulPeerProposal peerProposal = PCEPStatefulPeerProposal.createStatefulPeerProposal(this.dataBroker,
- TOPOLOGY_IID);
- peerProposal.setPeerProposal(NODE_ID, this.tlvsBuilder);
- assertEquals(LSP_DB_VERSION, this.tlvsBuilder.getAugmentation(Tlvs3.class).getLspDbVersion());
+ public void testSetPeerProposalSuccess() throws Exception {
+ updateBuilder(() -> {
+ final var listeners = captor.getAllValues();
+ assertEquals(2, listeners.size());
+
+ // not entirely accurate, but works well enough
+ final var modPath = TOPOLOGY_IID.child(Node.class,
+ new NodeKey(ServerSessionManager.createNodeId(ADDRESS.getAddress())));
+
+ final var dbverRoot = mock(DataObjectModification.class);
+ doReturn(LSP_DB_VERSION).when(dbverRoot).dataAfter();
+ final var dbverMod = mock(DataTreeModification.class);
+ doReturn(DataTreeIdentifier.of(LogicalDatastoreType.OPERATIONAL, modPath)).when(dbverMod).getRootPath();
+ doReturn(dbverRoot).when(dbverMod).getRootNode();
+
+ for (var listener : listeners) {
+ if (listener instanceof LspDbVersionListener) {
+ listener.onDataTreeChanged(List.of(dbverMod));
+ }
+ }
+
+ // Mock lspdb
+ });
+ assertEquals(new Tlvs3Builder().setLspDbVersion(LSP_DB_VERSION).build(), tlvsBuilder.augmentation(Tlvs3.class));
}
@Test
- public void testSetPeerProposalAbsent() throws InterruptedException, ExecutionException {
- doReturn(Optional.absent()).when(this.listenableFutureMock).get();
- final PCEPStatefulPeerProposal peerProposal = PCEPStatefulPeerProposal.createStatefulPeerProposal(this.dataBroker,
- TOPOLOGY_IID);
- peerProposal.setPeerProposal(NODE_ID, this.tlvsBuilder);
- assertNull(this.tlvsBuilder.getAugmentation(Tlvs3.class));
+ public void testSetPeerProposalWithEntityIdSuccess() throws Exception {
+ updateBuilder(() -> {
+ final var listeners = captor.getAllValues();
+ assertEquals(2, listeners.size());
+
+ // not entirely accurate, but works well enough
+ final var modPath = TOPOLOGY_IID.child(Node.class,
+ new NodeKey(ServerSessionManager.createNodeId(ADDRESS.getAddress())));
+
+ final var dbverRoot = mock(DataObjectModification.class);
+ doReturn(LSP_DB_VERSION).when(dbverRoot).dataAfter();
+ final var dbverMod = mock(DataTreeModification.class);
+ 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)
+ .dataAfter();
+ final var speakerMod = mock(DataTreeModification.class);
+ doReturn(DataTreeIdentifier.of(LogicalDatastoreType.CONFIGURATION, modPath)).when(speakerMod)
+ .getRootPath();
+ doReturn(speakerRoot).when(speakerMod).getRootNode();
+
+ for (var listener : listeners) {
+ if (listener instanceof SpeakerIdListener) {
+ listener.onDataTreeChanged(List.of(speakerMod));
+ } else if (listener instanceof LspDbVersionListener) {
+ listener.onDataTreeChanged(List.of(dbverMod));
+ }
+ }
+ });
+ final Tlvs3 aug = tlvsBuilder.augmentation(Tlvs3.class);
+ assertNotNull(aug);
+ assertEquals(LSP_DB_VERSION, aug.getLspDbVersion());
+ assertEquals(new SpeakerEntityIdBuilder().setSpeakerEntityIdValue(SPEAKER_ID).build(),
+ aug.getSpeakerEntityId());
}
@Test
- public void testSetPeerProposalFailure() throws InterruptedException, ExecutionException {
- doThrow(new RuntimeException()).when(this.listenableFutureMock).get();
- final PCEPStatefulPeerProposal peerProposal = PCEPStatefulPeerProposal.createStatefulPeerProposal(this.dataBroker,
- TOPOLOGY_IID);
- peerProposal.setPeerProposal(NODE_ID, this.tlvsBuilder);
- assertNull(this.tlvsBuilder.getAugmentation(Tlvs3.class));
+ public void testSetPeerProposalAbsent() throws Exception {
+ updateBuilder();
+ assertNull(tlvsBuilder.augmentation(Tlvs3.class));
}
+ private void updateBuilder() {
+ updateBuilder(null);
+ }
+
+ private void updateBuilder(final Runnable customizer) {
+ try (var proposal = new PCEPStatefulPeerProposal(dataBroker, TOPOLOGY_IID)) {
+ if (customizer != null) {
+ customizer.run();
+ }
+ proposal.setPeerSpecificProposal(ADDRESS, tlvsBuilder);
+ }
+ }
}