pcep/topology/topology-provider/src/test/java/org/opendaylight/bgpcep/pcep/topology/provider/PCEPStatefulPeerProposalTest.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.bgpcep.pcep.topology.provider;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.junit.Assert.assertNull;
  13 import static org.mockito.ArgumentMatchers.any;
  14 import static org.mockito.Mockito.doNothing;
  15 import static org.mockito.Mockito.doReturn;
  16 import static org.mockito.Mockito.mock;
  17
  18 import com.google.common.util.concurrent.FluentFuture;
  19 import java.net.InetSocketAddress;
  20 import java.util.List;
  21 import java.util.Optional;
  22 import org.junit.Before;
  23 import org.junit.Test;
  24 import org.junit.runner.RunWith;
  25 import org.mockito.ArgumentCaptor;
  26 import org.mockito.Mock;
  27 import org.mockito.junit.MockitoJUnitRunner;
  28 import org.opendaylight.bgpcep.pcep.topology.provider.PCEPStatefulPeerProposal.LspDbVersionListener;
  29 import org.opendaylight.bgpcep.pcep.topology.provider.PCEPStatefulPeerProposal.SpeakerIdListener;
  30 import org.opendaylight.mdsal.binding.api.DataBroker;
  31 import org.opendaylight.mdsal.binding.api.DataObjectModification;
  32 import org.opendaylight.mdsal.binding.api.DataTreeChangeListener;
  33 import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
  34 import org.opendaylight.mdsal.binding.api.DataTreeModification;
  35 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Stateful1Builder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Tlvs3;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Tlvs3Builder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.lsp.db.version.tlv.LspDbVersion;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.lsp.db.version.tlv.LspDbVersionBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.speaker.entity.id.tlv.SpeakerEntityIdBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.Tlvs1Builder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.stateful.capability.tlv.StatefulBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.open.TlvsBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.sync.optimizations.config.rev181109.PcepNodeSyncConfigBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  49 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  50 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  51 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  52 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  53 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  54 import org.opendaylight.yangtools.yang.common.Uint64;
  55
  56 @RunWith(MockitoJUnitRunner.StrictStubs.class)
  57 public class PCEPStatefulPeerProposalTest {
  58     private static final InstanceIdentifier<Topology> TOPOLOGY_IID = InstanceIdentifier.create(NetworkTopology.class)
  59             .child(Topology.class, new TopologyKey(new TopologyId("topology")));
  60     private static final LspDbVersion LSP_DB_VERSION = new LspDbVersionBuilder()
  61         .setLspDbVersionValue(Uint64.ONE)
  62         .build();
  63     private static final byte[] SPEAKER_ID = {0x01, 0x02, 0x03, 0x04};
  64     private static final InetSocketAddress ADDRESS = new InetSocketAddress(4321);
  65
  66     @Mock
  67     private DataBroker dataBroker;
  68     @Mock
  69     private ListenerRegistration<?> listenerReg;
  70     @Mock
  71     private FluentFuture<Optional<LspDbVersion>> listenableFutureMock;
  72
  73     private ArgumentCaptor<DataTreeChangeListener<?>> captor;
  74     private TlvsBuilder tlvsBuilder;
  75
  76     @Before
  77     public void setUp() {
  78         tlvsBuilder = new TlvsBuilder().addAugmentation(new Tlvs1Builder()
  79             .setStateful(new StatefulBuilder().addAugmentation(new Stateful1Builder().build()).build())
  80             .build());
  81         captor = ArgumentCaptor.forClass(DataTreeChangeListener.class);
  82         doReturn(listenerReg).when(dataBroker).registerDataTreeChangeListener(any(), captor.capture());
  83         doNothing().when(listenerReg).close();
  84     }
  85
  86     @Test
  87     public void testSetPeerProposalSuccess() throws Exception {
  88         updateBuilder(() -> {
  89             final var listeners = captor.getAllValues();
  90             assertEquals(2, listeners.size());
  91
  92             // not entirely accurate, but works well enough
  93             final var modPath = TOPOLOGY_IID.child(Node.class,
  94                 new NodeKey(ServerSessionManager.createNodeId(ADDRESS.getAddress())));
  95
  96             final var dbverRoot = mock(DataObjectModification.class);
  97             doReturn(LSP_DB_VERSION).when(dbverRoot).getDataAfter();
  98             final var dbverMod = mock(DataTreeModification.class);
  99             doReturn(DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL, modPath)).when(dbverMod).getRootPath();
 100             doReturn(dbverRoot).when(dbverMod).getRootNode();
 101
 102             for (DataTreeChangeListener<?> listener : listeners) {
 103                 if (listener instanceof LspDbVersionListener) {
 104                     listener.onDataTreeChanged(List.of(dbverMod));
 105                 }
 106             }
 107
 108             // Mock lspdb
 109         });
 110         assertEquals(new Tlvs3Builder().setLspDbVersion(LSP_DB_VERSION).build(), tlvsBuilder.augmentation(Tlvs3.class));
 111     }
 112
 113     @Test
 114     public void testSetPeerProposalWithEntityIdSuccess() throws Exception {
 115         updateBuilder(() -> {
 116             final var listeners = captor.getAllValues();
 117             assertEquals(2, listeners.size());
 118
 119             // not entirely accurate, but works well enough
 120             final var modPath = TOPOLOGY_IID.child(Node.class,
 121                 new NodeKey(ServerSessionManager.createNodeId(ADDRESS.getAddress())));
 122
 123             final var dbverRoot = mock(DataObjectModification.class);
 124             doReturn(LSP_DB_VERSION).when(dbverRoot).getDataAfter();
 125             final var dbverMod = mock(DataTreeModification.class);
 126             doReturn(DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL, modPath)).when(dbverMod).getRootPath();
 127             doReturn(dbverRoot).when(dbverMod).getRootNode();
 128
 129             final var speakerRoot = mock(DataObjectModification.class);
 130             doReturn(new PcepNodeSyncConfigBuilder().setSpeakerEntityIdValue(SPEAKER_ID).build()).when(speakerRoot)
 131                 .getDataAfter();
 132             final var speakerMod = mock(DataTreeModification.class);
 133             doReturn(DataTreeIdentifier.create(LogicalDatastoreType.CONFIGURATION, modPath)).when(speakerMod)
 134                 .getRootPath();
 135             doReturn(speakerRoot).when(speakerMod).getRootNode();
 136
 137             for (DataTreeChangeListener<?> listener : listeners) {
 138                 if (listener instanceof SpeakerIdListener) {
 139                     listener.onDataTreeChanged(List.of(speakerMod));
 140                 } else if (listener instanceof LspDbVersionListener) {
 141                     listener.onDataTreeChanged(List.of(dbverMod));
 142                 }
 143             }
 144         });
 145         final Tlvs3 aug = tlvsBuilder.augmentation(Tlvs3.class);
 146         assertNotNull(aug);
 147         assertEquals(LSP_DB_VERSION, aug.getLspDbVersion());
 148         assertEquals(new SpeakerEntityIdBuilder().setSpeakerEntityIdValue(SPEAKER_ID).build(),
 149             aug.getSpeakerEntityId());
 150     }
 151
 152     @Test
 153     public void testSetPeerProposalAbsent() throws Exception {
 154         updateBuilder();
 155         assertNull(tlvsBuilder.augmentation(Tlvs3.class));
 156     }
 157
 158     private void updateBuilder() {
 159         updateBuilder(null);
 160     }
 161
 162     private void updateBuilder(final Runnable customizer) {
 163         try (var proposal = new PCEPStatefulPeerProposal(dataBroker, TOPOLOGY_IID)) {
 164             if (customizer != null) {
 165                 customizer.run();
 166             }
 167             proposal.setPeerSpecificProposal(ADDRESS, tlvsBuilder);
 168         }
 169     }
 170 }