bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/config/OpenConfigMappingUtilTest.java

   1 /*
   2  * Copyright (c) 2016 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
   9 package org.opendaylight.protocol.bgp.rib.impl.config;
  10
  11 import static junit.framework.TestCase.assertFalse;
  12 import static org.junit.Assert.assertArrayEquals;
  13 import static org.junit.Assert.assertEquals;
  14 import static org.junit.Assert.assertNull;
  15 import static org.junit.Assert.assertTrue;
  16 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.AFI_SAFI;
  17 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.AFI_SAFI_IPV4;
  18 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.MD5_PASSWORD;
  19 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.NEIGHBOR_ADDRESS;
  20 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.PORT;
  21 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.SHORT;
  22 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createAfiSafi;
  23 import static org.opendaylight.protocol.bgp.rib.impl.config.BgpPeerTest.createNeighborExpected;
  24 import static org.opendaylight.protocol.bgp.rib.impl.config.OpenConfigMappingUtil.HOLDTIMER;
  25 import static org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IetfInetUtil.INSTANCE;
  26
  27 import com.google.common.collect.ImmutableList;
  28 import com.google.common.primitives.Shorts;
  29 import java.math.BigDecimal;
  30 import java.nio.charset.StandardCharsets;
  31 import java.util.ArrayList;
  32 import java.util.HashMap;
  33 import java.util.List;
  34 import java.util.Map;
  35 import org.junit.Assert;
  36 import org.junit.Before;
  37 import org.junit.Test;
  38 import org.mockito.Mock;
  39 import org.mockito.Mockito;
  40 import org.mockito.MockitoAnnotations;
  41 import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
  42 import org.opendaylight.protocol.bgp.mode.impl.add.all.paths.AllPathSelection;
  43 import org.opendaylight.protocol.bgp.mode.impl.add.n.paths.AddPathBestNPathSelection;
  44 import org.opendaylight.protocol.bgp.openconfig.spi.BGPTableTypeRegistryConsumer;
  45 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
  46 import org.opendaylight.protocol.bgp.rib.impl.BGPPeerTrackerImpl;
  47 import org.opendaylight.protocol.bgp.rib.impl.spi.RIB;
  48 import org.opendaylight.protocol.bgp.rib.spi.BGPPeerTracker;
  49 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafi;
  50 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.multiprotocol.rev151009.bgp.common.afi.safi.list.AfiSafiBuilder;
  51 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafis;
  52 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.AfiSafisBuilder;
  53 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.ConfigBuilder;
  54 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.RouteReflectorBuilder;
  55 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TimersBuilder;
  56 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.Transport;
  57 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.TransportBuilder;
  58 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor.group.transport.Config;
  59 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.Neighbor;
  60 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.NeighborBuilder;
  61 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbors.NeighborKey;
  62 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.peer.group.PeerGroupBuilder;
  63 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.Bgp;
  64 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.top.bgp.Neighbors;
  65 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV4UNICAST;
  66 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV6LABELLEDUNICAST;
  67 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.IPV6UNICAST;
  68 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.PeerType;
  69 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.RrClusterIdType;
  70 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.NetworkInstances;
  71 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstance;
  72 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.NetworkInstanceKey;
  73 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.Protocols;
  74 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.Protocol;
  75 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.network.instance.rev151018.network.instance.top.network.instances.network.instance.protocols.ProtocolKey;
  76 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.BGP;
  77 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
  78 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  79 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
  80 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.BgpTableType;
  81 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.SendReceive;
  82 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.mp.capabilities.add.path.capability.AddressFamilies;
  83 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
  84 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.GlobalAddPathsConfig;
  85 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.GlobalAddPathsConfigBuilder;
  86 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.GlobalConfigAugmentation;
  87 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.GlobalConfigAugmentationBuilder;
  88 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.NeighborAddPathsConfig;
  89 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.NeighborAddPathsConfigBuilder;
  90 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.NeighborPeerGroupConfig;
  91 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.NeighborPeerGroupConfigBuilder;
  92 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.NeighborTransportConfig;
  93 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.NeighborTransportConfigBuilder;
  94 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.NetworkInstanceProtocol;
  95 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.PeerGroupTransportConfig;
  96 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.openconfig.extensions.rev180321.PeerGroupTransportConfigBuilder;
  97 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.PeerRole;
  98 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.TablesKey;
  99 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
 100 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
 101 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv6AddressFamily;
 102 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.MplsLabeledVpnSubsequentAddressFamily;
 103 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
 104 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 105
 106 public class OpenConfigMappingUtilTest {
 107     private static final Neighbor NEIGHBOR = createNeighborExpected(NEIGHBOR_ADDRESS);
 108     private static final String KEY = "bgp";
 109     private static final InstanceIdentifier<Bgp> BGP_II = InstanceIdentifier.create(NetworkInstances.class)
 110         .child(NetworkInstance.class, new NetworkInstanceKey("identifier-test")).child(Protocols.class)
 111         .child(Protocol.class, new ProtocolKey(BGP.class, KEY)).augmentation(NetworkInstanceProtocol.class)
 112             .child(Bgp.class);
 113     private static final NeighborKey NEIGHBOR_KEY = new NeighborKey(NEIGHBOR_ADDRESS);
 114     private static final Ipv4Address ROUTER_ID = new Ipv4Address("1.2.3.4");
 115     private static final Ipv4Address CLUSTER_ID = new Ipv4Address("4.3.2.1");
 116
 117     private static final Long ALL_PATHS = 0L;
 118     private static final Long N_PATHS = 2L;
 119     private static final BGPPeerTracker PEER_TRACKER = new BGPPeerTrackerImpl();
 120     private static final PathSelectionMode ADD_PATH_BEST_N_PATH_SELECTION = new AddPathBestNPathSelection(N_PATHS,
 121             PEER_TRACKER);
 122     private static final PathSelectionMode ADD_PATH_BEST_ALL_PATH_SELECTION = new AllPathSelection(PEER_TRACKER);
 123     private static final BgpTableType BGP_TABLE_TYPE_IPV4 = new BgpTableTypeImpl(Ipv4AddressFamily.class,
 124             UnicastSubsequentAddressFamily.class);
 125     private static final BgpTableType BGP_TABLE_TYPE_IPV6
 126             = new BgpTableTypeImpl(Ipv6AddressFamily.class, UnicastSubsequentAddressFamily.class);
 127     private static final AfiSafi AFISAFI_IPV4 = new AfiSafiBuilder().setAfiSafiName(IPV4UNICAST.class).build();
 128     private static final TablesKey K4 = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
 129
 130     private static final AsNumber AS = new AsNumber(72L);
 131     private static final BgpId BGP_ID = new BgpId(NEIGHBOR_ADDRESS.getIpv4Address());
 132     private static final List<AddressFamilies> FAMILIES;
 133     private static final List<BgpTableType> TABLE_TYPES;
 134     private static final List<AfiSafi> AFISAFIS = new ArrayList<>();
 135     private static final BigDecimal DEFAULT_TIMERS = BigDecimal.valueOf(30);
 136
 137     static {
 138         FAMILIES = new ArrayList<>();
 139         FAMILIES.add(new AddressFamiliesBuilder()
 140                 .setAfi(Ipv4AddressFamily.class).setSafi(UnicastSubsequentAddressFamily.class)
 141                 .setSendReceive(SendReceive.Both).build());
 142         FAMILIES.add(new AddressFamiliesBuilder()
 143                 .setAfi(Ipv6AddressFamily.class).setSafi(UnicastSubsequentAddressFamily.class)
 144                 .setSendReceive(SendReceive.Send).build());
 145         FAMILIES.add(new AddressFamiliesBuilder()
 146                 .setAfi(Ipv6AddressFamily.class).setSafi(MplsLabeledVpnSubsequentAddressFamily.class)
 147                 .setSendReceive(SendReceive.Receive).build());
 148         TABLE_TYPES = new ArrayList<>();
 149         TABLE_TYPES.add(new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class));
 150         TABLE_TYPES.add(new BgpTableTypeImpl(Ipv6AddressFamily.class, UnicastSubsequentAddressFamily.class));
 151         AFISAFIS.add(new AfiSafiBuilder().setAfiSafiName(IPV4UNICAST.class)
 152                 .addAugmentation(GlobalAddPathsConfig.class, new GlobalAddPathsConfigBuilder().setReceive(true)
 153                         .setSendMax(Shorts.checkedCast(N_PATHS)).build()).build());
 154         AFISAFIS.add(new AfiSafiBuilder().setAfiSafiName(IPV6UNICAST.class)
 155                 .addAugmentation(GlobalAddPathsConfig.class, new GlobalAddPathsConfigBuilder().setReceive(true)
 156                         .setSendMax(Shorts.checkedCast(ALL_PATHS)).build()).build());
 157     }
 158
 159     @Mock
 160     private BGPTableTypeRegistryConsumer tableTypeRegistry;
 161
 162     @Mock
 163     private RIB rib;
 164
 165     @Before
 166     public void setUp() {
 167         MockitoAnnotations.initMocks(this);
 168         Mockito.doReturn(java.util.Optional.of(BGP_TABLE_TYPE_IPV4))
 169             .when(this.tableTypeRegistry).getTableType(IPV4UNICAST.class);
 170         Mockito.doReturn(java.util.Optional.of(BGP_TABLE_TYPE_IPV6))
 171             .when(this.tableTypeRegistry).getTableType(IPV6UNICAST.class);
 172         Mockito.doReturn(java.util.Optional.of(new BgpTableTypeImpl(Ipv6AddressFamily.class,
 173                 MplsLabeledVpnSubsequentAddressFamily.class)))
 174             .when(this.tableTypeRegistry).getTableType(IPV6LABELLEDUNICAST.class);
 175         Mockito.doReturn(java.util.Optional.of(IPV4UNICAST.class))
 176             .when(this.tableTypeRegistry).getAfiSafiType(BGP_TABLE_TYPE_IPV4);
 177         Mockito.doReturn(java.util.Optional.of(IPV6UNICAST.class))
 178             .when(this.tableTypeRegistry).getAfiSafiType(BGP_TABLE_TYPE_IPV6);
 179         Mockito.doReturn(AS).when(this.rib).getLocalAs();
 180     }
 181
 182     @Test
 183     public void testGetRibInstanceName() {
 184         assertEquals(KEY, OpenConfigMappingUtil.getRibInstanceName(BGP_II));
 185     }
 186
 187     @Test
 188     public void testGetHoldTimer() {
 189         TimersBuilder builder = new TimersBuilder().setConfig(new org.opendaylight.yang.gen.v1.http.openconfig.net
 190                 .yang.bgp.rev151009.bgp.neighbor.group.timers.ConfigBuilder().setHoldTime(BigDecimal.TEN).build());
 191
 192         assertEquals(DEFAULT_TIMERS.toBigInteger().intValue(),
 193                 OpenConfigMappingUtil.getHoldTimer(NEIGHBOR, null));
 194         assertEquals(HOLDTIMER,
 195                 OpenConfigMappingUtil.getHoldTimer(new NeighborBuilder().build(), null));
 196
 197         assertEquals(DEFAULT_TIMERS.toBigInteger().intValue(),
 198                 OpenConfigMappingUtil.getHoldTimer(NEIGHBOR, new PeerGroupBuilder().build()));
 199         assertEquals(BigDecimal.TEN.intValue(), OpenConfigMappingUtil.getHoldTimer(NEIGHBOR, new PeerGroupBuilder()
 200                 .setTimers(builder.build()).build()));
 201     }
 202
 203     @Test
 204     public void testGetPeerAs() {
 205         assertEquals(AS, OpenConfigMappingUtil.getPeerAs(NEIGHBOR, null, null));
 206         assertEquals(AS, OpenConfigMappingUtil.getPeerAs(new NeighborBuilder().build(), null, this.rib.getLocalAs()));
 207
 208         assertEquals(AS, OpenConfigMappingUtil.getPeerAs(NEIGHBOR, new PeerGroupBuilder().build(), null));
 209         assertEquals(AS, OpenConfigMappingUtil.getPeerAs(new NeighborBuilder().build(), new PeerGroupBuilder().build(),
 210                 this.rib.getLocalAs()));
 211
 212         assertEquals(AS, OpenConfigMappingUtil.getPeerAs(null, new PeerGroupBuilder()
 213                         .setConfig(new ConfigBuilder().setPeerAs(AS).build()).build(), null));
 214     }
 215
 216     @Test
 217     public void testIsActive() {
 218         final TransportBuilder builder = new TransportBuilder();
 219         assertTrue(OpenConfigMappingUtil.isActive(new NeighborBuilder().build(), null));
 220         assertTrue(OpenConfigMappingUtil.isActive(new NeighborBuilder()
 221                 .setTransport(builder.build()).build(), null));
 222
 223         final Transport activeFalse = builder.setConfig(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp
 224                 .rev151009.bgp.neighbor.group.transport.ConfigBuilder().setPassiveMode(true).build()).build();
 225         assertFalse(OpenConfigMappingUtil.isActive(new NeighborBuilder().setTransport(activeFalse).build(),
 226                 null));
 227
 228         assertTrue(OpenConfigMappingUtil.isActive(new NeighborBuilder().build(),
 229                 new PeerGroupBuilder().build()));
 230         assertFalse(OpenConfigMappingUtil.isActive(new NeighborBuilder().build(),
 231                 new PeerGroupBuilder().setTransport(activeFalse).build()));
 232     }
 233
 234     @Test
 235     public void testGetRetryTimer() {
 236         assertEquals(DEFAULT_TIMERS.toBigInteger().intValue(),
 237                 OpenConfigMappingUtil.getRetryTimer(NEIGHBOR, null));
 238         assertEquals(DEFAULT_TIMERS.toBigInteger().intValue(),
 239                 OpenConfigMappingUtil.getRetryTimer(new NeighborBuilder().build(), null));
 240         TimersBuilder builder = new TimersBuilder().setConfig(new org.opendaylight.yang.gen.v1.http.openconfig.net
 241                 .yang.bgp.rev151009.bgp.neighbor.group.timers.ConfigBuilder().setConnectRetry(BigDecimal.TEN).build());
 242         assertEquals(BigDecimal.TEN.intValue(), OpenConfigMappingUtil.getRetryTimer(new NeighborBuilder()
 243                 .setTimers(builder.build()).build(), null));
 244
 245         assertEquals(DEFAULT_TIMERS.toBigInteger().intValue(),
 246                 OpenConfigMappingUtil.getRetryTimer(NEIGHBOR, new PeerGroupBuilder().build()));
 247         assertEquals(BigDecimal.TEN.intValue(), OpenConfigMappingUtil.getRetryTimer(NEIGHBOR,
 248                 new PeerGroupBuilder().setTimers(builder.build()).build()));
 249     }
 250
 251     @Test
 252     public void testGetNeighborKey() {
 253         assertArrayEquals(MD5_PASSWORD.getBytes(StandardCharsets.US_ASCII),
 254             OpenConfigMappingUtil.getNeighborKey(NEIGHBOR).get(INSTANCE.inetAddressFor(NEIGHBOR_ADDRESS)));
 255         assertNull(OpenConfigMappingUtil.getNeighborKey(new NeighborBuilder().build()));
 256         assertNull(OpenConfigMappingUtil.getNeighborKey(new NeighborBuilder().setConfig(new ConfigBuilder()
 257                 .build()).build()));
 258     }
 259
 260     @Test
 261     public void testGetNeighborInstanceIdentifier() {
 262         assertEquals(BGP_II.child(Neighbors.class).child(Neighbor.class, NEIGHBOR_KEY),
 263             OpenConfigMappingUtil.getNeighborInstanceIdentifier(BGP_II, NEIGHBOR_KEY));
 264
 265     }
 266
 267     @Test
 268     public void testGetNeighborInstanceName() {
 269         assertEquals(NEIGHBOR_ADDRESS.getIpv4Address().getValue(), OpenConfigMappingUtil
 270                 .getNeighborInstanceName(BGP_II.child(Neighbors.class).child(Neighbor.class, NEIGHBOR_KEY)));
 271     }
 272
 273     @Test
 274     public void testGetPort() {
 275         final TransportBuilder transport = new TransportBuilder();
 276         assertEquals(PORT, OpenConfigMappingUtil.getPort(NEIGHBOR, null));
 277         assertEquals(PORT, OpenConfigMappingUtil.getPort(new NeighborBuilder()
 278                 .setTransport(transport.build()).build(), null));
 279         assertEquals(PORT, OpenConfigMappingUtil.getPort(new NeighborBuilder().setTransport(
 280                 transport.setConfig(new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor
 281                         .group.transport.ConfigBuilder().build()).build()).build(), null));
 282         final PortNumber newPort = new PortNumber(111);
 283         final Config portConfig = new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.neighbor
 284                 .group.transport.ConfigBuilder().addAugmentation(NeighborTransportConfig.class,
 285                 new NeighborTransportConfigBuilder().setRemotePort(newPort).build()).build();
 286         assertEquals(newPort, OpenConfigMappingUtil.getPort(new NeighborBuilder().setTransport(
 287                 transport.setConfig(portConfig).build()).build(), null));
 288
 289         assertEquals(newPort, OpenConfigMappingUtil.getPort(new NeighborBuilder()
 290                 .setTransport(transport.setConfig(portConfig).build()).build(), new PeerGroupBuilder().build()));
 291
 292         final Config portConfigGroup = new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp
 293                 .neighbor.group.transport.ConfigBuilder().addAugmentation(PeerGroupTransportConfig.class,
 294                 new PeerGroupTransportConfigBuilder().setRemotePort(newPort).build()).build();
 295         assertEquals(newPort, OpenConfigMappingUtil.getPort(new NeighborBuilder().build(), new PeerGroupBuilder()
 296                 .setTransport(transport.setConfig(portConfigGroup).build()).build()));
 297     }
 298
 299     @Test
 300     public void testGetAfiSafiWithDefault() {
 301         final ImmutableList<AfiSafi> defaultValue
 302                 = ImmutableList.of(new AfiSafiBuilder().setAfiSafiName(IPV4UNICAST.class).build());
 303         assertEquals(defaultValue, OpenConfigMappingUtil.getAfiSafiWithDefault(null, true));
 304         final AfiSafis afiSafi = new AfiSafisBuilder().build();
 305         assertEquals(defaultValue, OpenConfigMappingUtil.getAfiSafiWithDefault(afiSafi, true));
 306
 307         final AfiSafi afiSafiIpv6 = new AfiSafiBuilder().setAfiSafiName(IPV6UNICAST.class)
 308                 .addAugmentation(NeighborAddPathsConfig.class,
 309             new NeighborAddPathsConfigBuilder().setReceive(true).setSendMax(SHORT).build()).build();
 310         final List<AfiSafi> afiSafiIpv6List = new ArrayList<>();
 311         afiSafiIpv6List.add(afiSafiIpv6);
 312
 313         final List<AfiSafi> expected = new ArrayList<>(afiSafiIpv6List);
 314         expected.add(AFI_SAFI_IPV4);
 315         assertEquals(afiSafiIpv6, OpenConfigMappingUtil.getAfiSafiWithDefault(new AfiSafisBuilder()
 316                 .setAfiSafi(afiSafiIpv6List).build(), true).get(0));
 317         assertEquals(new AfiSafiBuilder().setAfiSafiName(IPV4UNICAST.class).build(),
 318                 OpenConfigMappingUtil.getAfiSafiWithDefault(new AfiSafisBuilder().setAfiSafi(afiSafiIpv6List).build(),
 319                         true).get(1));
 320         assertEquals(AFI_SAFI, OpenConfigMappingUtil.getAfiSafiWithDefault(createAfiSafi(), true));
 321
 322         assertTrue(OpenConfigMappingUtil.getAfiSafiWithDefault(null, false).isEmpty());
 323         assertTrue(OpenConfigMappingUtil.getAfiSafiWithDefault(afiSafi, false).isEmpty());
 324         assertEquals(afiSafiIpv6, OpenConfigMappingUtil.getAfiSafiWithDefault(new AfiSafisBuilder()
 325                 .setAfiSafi(afiSafiIpv6List).build(), false).get(0));
 326         assertEquals(new AfiSafiBuilder().setAfiSafiName(IPV4UNICAST.class).build(),
 327             OpenConfigMappingUtil.getAfiSafiWithDefault(new AfiSafisBuilder().setAfiSafi(afiSafiIpv6List).build(),
 328                     false).get(1));
 329         assertEquals(AFI_SAFI, OpenConfigMappingUtil.getAfiSafiWithDefault(createAfiSafi(), false));
 330     }
 331
 332     @Test
 333     public void testGetClusterIdentifier() {
 334         final org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp.global.base
 335                 .ConfigBuilder configBuilder = new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009
 336                 .bgp.global.base.ConfigBuilder();
 337         configBuilder.setRouterId(ROUTER_ID);
 338         assertEquals(ROUTER_ID.getValue(),
 339                 OpenConfigMappingUtil.getClusterIdentifier(configBuilder.build()).getValue());
 340
 341         configBuilder.addAugmentation(GlobalConfigAugmentation.class, new GlobalConfigAugmentationBuilder()
 342                 .setRouteReflectorClusterId(new RrClusterIdType(CLUSTER_ID)).build()).build();
 343         assertEquals(CLUSTER_ID.getValue(),
 344                 OpenConfigMappingUtil.getClusterIdentifier(configBuilder.build()).getValue());
 345     }
 346
 347     @Test
 348     public void isAppNeighbor() {
 349         assertFalse(OpenConfigMappingUtil.isApplicationPeer(new NeighborBuilder()
 350                 .setConfig(new ConfigBuilder().build()).build()));
 351         final Neighbor neighbor = new NeighborBuilder().setConfig(new ConfigBuilder()
 352                 .addAugmentation(NeighborPeerGroupConfig.class, new NeighborPeerGroupConfigBuilder()
 353                         .setPeerGroup(OpenConfigMappingUtil.APPLICATION_PEER_GROUP_NAME).build()).build()).build();
 354         assertTrue(OpenConfigMappingUtil.isApplicationPeer(neighbor));
 355     }
 356
 357     @Test
 358     public void toPathSelectionMode() {
 359         final List<AfiSafi> families = new ArrayList<>();
 360         families.add(new AfiSafiBuilder().setAfiSafiName(IPV4UNICAST.class)
 361             .addAugmentation(GlobalAddPathsConfig.class, new GlobalAddPathsConfigBuilder()
 362                     .setSendMax(Shorts.checkedCast(N_PATHS)).build()).build());
 363         families.add(new AfiSafiBuilder().setAfiSafiName(IPV6UNICAST.class)
 364             .addAugmentation(GlobalAddPathsConfig.class, new GlobalAddPathsConfigBuilder()
 365                     .setSendMax(Shorts.checkedCast(ALL_PATHS)).build()).build());
 366         final Map<BgpTableType, PathSelectionMode> result = OpenConfigMappingUtil
 367                 .toPathSelectionMode(families, this.tableTypeRegistry, PEER_TRACKER);
 368         final Map<BgpTableType, PathSelectionMode> expected = new HashMap<>();
 369         expected.put(new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class),
 370                 ADD_PATH_BEST_N_PATH_SELECTION);
 371         expected.put(new BgpTableTypeImpl(Ipv6AddressFamily.class, UnicastSubsequentAddressFamily.class),
 372                 ADD_PATH_BEST_ALL_PATH_SELECTION);
 373         assertEquals(expected.get(0), result.get(0));
 374         assertEquals(expected.get(1), result.get(1));
 375     }
 376
 377     @Test
 378     public void toPeerRole() {
 379         Neighbor neighbor = new NeighborBuilder().setConfig(new ConfigBuilder()
 380                 .setPeerType(PeerType.EXTERNAL).build()).build();
 381         PeerRole peerRoleResult = OpenConfigMappingUtil.toPeerRole(neighbor);
 382         Assert.assertEquals(PeerRole.Ebgp, peerRoleResult);
 383
 384         neighbor = new NeighborBuilder().setConfig(new ConfigBuilder().setPeerType(PeerType.INTERNAL).build()).build();
 385         peerRoleResult = OpenConfigMappingUtil.toPeerRole(neighbor);
 386         Assert.assertEquals(PeerRole.Ibgp, peerRoleResult);
 387
 388         neighbor = new NeighborBuilder()
 389             .setRouteReflector(new RouteReflectorBuilder().setConfig(
 390                     new org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.rev151009.bgp
 391                 .neighbor.group.route.reflector.ConfigBuilder().setRouteReflectorClient(true).build()).build()).build();
 392         peerRoleResult = OpenConfigMappingUtil.toPeerRole(neighbor);
 393         Assert.assertEquals(PeerRole.RrClient, peerRoleResult);
 394     }
 395
 396     @Test
 397     public void toAddPathCapability() {
 398         final List<AfiSafi> families = new ArrayList<>();
 399         families.add(new AfiSafiBuilder().setAfiSafiName(IPV4UNICAST.class)
 400             .addAugmentation(NeighborAddPathsConfig.class,
 401                 new NeighborAddPathsConfigBuilder()
 402                     .setReceive(true).setSendMax(Shorts.checkedCast(ALL_PATHS)).build()).build());
 403         families.add(new AfiSafiBuilder().setAfiSafiName(IPV6UNICAST.class)
 404             .addAugmentation(NeighborAddPathsConfig.class,
 405                 new NeighborAddPathsConfigBuilder()
 406                     .setReceive(false).setSendMax(Shorts.checkedCast(N_PATHS)).build()).build());
 407         families.add(new AfiSafiBuilder().setAfiSafiName(IPV6LABELLEDUNICAST.class)
 408             .addAugmentation(NeighborAddPathsConfig.class,
 409                 new NeighborAddPathsConfigBuilder()
 410                     .setReceive(false).build()).build());
 411         final List<AddressFamilies> result = OpenConfigMappingUtil
 412                 .toAddPathCapability(families, this.tableTypeRegistry);
 413         assertEquals(FAMILIES, result);
 414     }
 415 }