bgp/mvpn/src/test/java/org/opendaylight/protocol/bgp/mvpn/impl/MvpnIpv4RIBSupportTest.java

   1 /*
   2  * Copyright (c) 2018 AT&T Intellectual Property. 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.mvpn.impl;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.assertFalse;
  13 import static org.junit.Assert.assertNull;
  14 import static org.junit.Assert.assertTrue;
  15
  16 import com.google.common.collect.ImmutableSet;
  17 import java.util.ArrayList;
  18 import java.util.Collection;
  19 import java.util.Collections;
  20 import org.junit.Test;
  21 import org.opendaylight.protocol.bgp.rib.spi.AbstractRIBSupportTest;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.PathId;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.Update;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes1;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes2;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.bgp.rib.rib.loc.rib.tables.routes.MvpnRoutesIpv4Case;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.bgp.rib.rib.loc.rib.tables.routes.MvpnRoutesIpv4CaseBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.mvpn.destination.MvpnDestination;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.mvpn.destination.MvpnDestinationBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.mvpn.routes.ipv4.MvpnRoutesIpv4;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.mvpn.routes.ipv4.MvpnRoutesIpv4Builder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationMvpnIpv4AdvertizedCase;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationMvpnIpv4AdvertizedCaseBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination.mvpn.ipv4.advertized._case.DestinationMvpnBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationMvpnIpv4WithdrawnCase;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4.rev180417.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationMvpnIpv4WithdrawnCaseBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.inter.as.i.pmsi.a.d.grouping.InterAsIPmsiADBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.mvpn.MvpnChoice;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.mvpn.mvpn.choice.InterAsIPmsiADCaseBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.mvpn.routes.MvpnRoute;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.mvpn.routes.MvpnRouteBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.mvpn.routes.MvpnRouteKey;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Attributes;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Routes;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.RdIpv4;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.RouteDistinguisher;
  48 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  49 import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
  50 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  51 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  52 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  53 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidateNode;
  54 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidates;
  55
  56 public class MvpnIpv4RIBSupportTest extends AbstractRIBSupportTest<MvpnRoutesIpv4Case, MvpnRoutesIpv4, MvpnRoute,
  57         MvpnRouteKey> {
  58     private static final PathId PATH_ID = new PathId(0L);
  59     private static final MvpnChoice MVPN = new InterAsIPmsiADCaseBuilder().setInterAsIPmsiAD(
  60             new InterAsIPmsiADBuilder()
  61                     .setSourceAs(new AsNumber(1L))
  62                     .setRouteDistinguisher(new RouteDistinguisher(new RdIpv4("1.2.3.4:258")))
  63                     .build()).build();
  64     private static final MvpnRouteKey ROUTE_KEY = new MvpnRouteKey(PATH_ID, "AgwAAQECAwQBAgAAAAE=");
  65     private static final MvpnRoute ROUTE = new MvpnRouteBuilder()
  66             .setRouteKey(ROUTE_KEY.getRouteKey())
  67             .setPathId(ROUTE_KEY.getPathId())
  68             .setAttributes(ATTRIBUTES)
  69             .setMvpnChoice(MVPN)
  70             .build();
  71     private static final MvpnRoutesIpv4 MVPN_ROUTES
  72             = new MvpnRoutesIpv4Builder().setMvpnRoute(Collections.singletonList(ROUTE)).build();
  73
  74     private static final MvpnDestination MVPN_DESTINATION = new MvpnDestinationBuilder()
  75             .setMvpnChoice(MVPN)
  76             .setPathId(PATH_ID)
  77             .build();
  78     private static final DestinationMvpnIpv4AdvertizedCase REACH_NLRI = new DestinationMvpnIpv4AdvertizedCaseBuilder()
  79             .setDestinationMvpn(new DestinationMvpnBuilder()
  80                     .setMvpnDestination(Collections.singletonList(MVPN_DESTINATION)).build()).build();
  81     private static final DestinationMvpnIpv4WithdrawnCase UNREACH_NLRI = new DestinationMvpnIpv4WithdrawnCaseBuilder()
  82             .setDestinationMvpn(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.ipv4
  83                     .rev180417.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.destination.mvpn
  84                     .ipv4.withdrawn._case.DestinationMvpnBuilder()
  85                     .setMvpnDestination(Collections.singletonList(MVPN_DESTINATION)).build()).build();
  86
  87     private MvpnIpv4RIBSupport ribSupport;
  88
  89     @Override
  90     public void setUp() throws Exception {
  91         super.setUp();
  92         this.ribSupport = MvpnIpv4RIBSupport.getInstance(this.mappingService);
  93         setUpTestCustomizer(this.ribSupport);
  94         NlriActivator.registerNlriParsers(new ArrayList<>());
  95     }
  96
  97     @Test
  98     public void testDeleteRoutes() {
  99         final ContainerNode withdraw = createNlriWithDrawnRoute(UNREACH_NLRI);
 100         this.ribSupport.deleteRoutes(this.tx, getTablePath(), withdraw);
 101         final InstanceIdentifier<MvpnRoute> instanceIdentifier = this.deletedRoutes.get(0);
 102         assertEquals(ROUTE_KEY, instanceIdentifier.firstKeyOf(MvpnRoute.class));
 103     }
 104
 105     @Test
 106     public void testPutRoutes() {
 107         this.ribSupport.putRoutes(this.tx, getTablePath(), createNlriAdvertiseRoute(REACH_NLRI), createAttributes());
 108         final MvpnRoute route = (MvpnRoute) this.insertedRoutes.get(0).getValue();
 109         assertEquals(ROUTE, route);
 110     }
 111
 112
 113     @Test
 114     public void testEmptyRoute() {
 115         final Routes empty = new MvpnRoutesIpv4CaseBuilder().setMvpnRoutesIpv4(new MvpnRoutesIpv4Builder()
 116                 .setMvpnRoute(Collections.emptyList()).build()).build();
 117         assertEquals(createEmptyTable(empty), this.ribSupport.emptyTable());
 118     }
 119
 120     @Test
 121     public void testBuildMpUnreachNlriUpdate() {
 122         final Collection<MapEntryNode> routes = createRoutes(MVPN_ROUTES);
 123         final Update update = this.ribSupport.buildUpdate(Collections.emptyList(), routes, ATTRIBUTES);
 124         assertEquals(UNREACH_NLRI, update.getAttributes().augmentation(Attributes2.class).getMpUnreachNlri()
 125                 .getWithdrawnRoutes().getDestinationType());
 126         assertNull(update.getAttributes().augmentation(Attributes1.class));
 127     }
 128
 129     @Test
 130     public void testBuildMpReachNlriUpdate() {
 131         final Collection<MapEntryNode> routes = createRoutes(MVPN_ROUTES);
 132         final Update update = this.ribSupport.buildUpdate(routes, Collections.emptyList(), ATTRIBUTES);
 133         assertEquals(REACH_NLRI, update.getAttributes().augmentation(Attributes1.class).getMpReachNlri()
 134                 .getAdvertizedRoutes().getDestinationType());
 135         assertNull(update.getAttributes().augmentation(Attributes2.class));
 136     }
 137
 138     @Test
 139     public void testCacheableNlriObjects() {
 140         assertEquals(ImmutableSet.of(MvpnRoutesIpv4Case.class), this.ribSupport.cacheableNlriObjects());
 141     }
 142
 143     @Test
 144     public void testCacheableAttributeObjects() {
 145         assertEquals(ImmutableSet.of(), this.ribSupport.cacheableAttributeObjects());
 146     }
 147
 148     @Test
 149     public void testRouteIdAddPath() {
 150         assertEquals(ROUTE_KEY, this.ribSupport.createRouteListKey(ROUTE_KEY.getPathId().getValue(),
 151                 ROUTE_KEY.getRouteKey()));
 152     }
 153
 154     @Test
 155     public void testRoutePath() {
 156         final YangInstanceIdentifier.NodeIdentifierWithPredicates prefixNii = createRouteNIWP(MVPN_ROUTES);
 157         final YangInstanceIdentifier expected = getRoutePath().node(prefixNii);
 158         final YangInstanceIdentifier actual = this.ribSupport.routePath(getTablePath().node(Routes.QNAME), prefixNii);
 159         assertEquals(expected, actual);
 160     }
 161
 162     @Test
 163     public void testRouteAttributesIdentifier() {
 164         assertEquals(new YangInstanceIdentifier.NodeIdentifier(
 165                         Attributes.QNAME.withModule(BindingReflections.getQNameModule(MvpnRoutesIpv4Case.class))),
 166                 this.ribSupport.routeAttributesIdentifier());
 167     }
 168
 169     @Test
 170     public void testRoutesCaseClass() {
 171         assertEquals(MvpnRoutesIpv4Case.class, this.ribSupport.routesCaseClass());
 172     }
 173
 174     @Test
 175     public void testRoutesContainerClass() {
 176         assertEquals(MvpnRoutesIpv4.class, this.ribSupport.routesContainerClass());
 177     }
 178
 179     @Test
 180     public void testRoutesListClass() {
 181         assertEquals(MvpnRoute.class, this.ribSupport.routesListClass());
 182     }
 183
 184     @Test
 185     public void testChangedRoutes() {
 186         final Routes emptyCase = new MvpnRoutesIpv4CaseBuilder().build();
 187         DataTreeCandidateNode tree = DataTreeCandidates.fromNormalizedNode(getRoutePath(),
 188                 createRoutes(emptyCase)).getRootNode();
 189         assertTrue(this.ribSupport.changedRoutes(tree).isEmpty());
 190
 191         final Routes emptyRoutes
 192                 = new MvpnRoutesIpv4CaseBuilder().setMvpnRoutesIpv4(new MvpnRoutesIpv4Builder().build()).build();
 193         tree = DataTreeCandidates.fromNormalizedNode(getRoutePath(), createRoutes(emptyRoutes)).getRootNode();
 194         assertTrue(this.ribSupport.changedRoutes(tree).isEmpty());
 195
 196         final Routes routes = new MvpnRoutesIpv4CaseBuilder().setMvpnRoutesIpv4(MVPN_ROUTES).build();
 197         tree = DataTreeCandidates.fromNormalizedNode(getRoutePath(), createRoutes(routes)).getRootNode();
 198         final Collection<DataTreeCandidateNode> result = this.ribSupport.changedRoutes(tree);
 199         assertFalse(result.isEmpty());
 200     }
 201 }