bgp/extensions/route-target/src/test/java/org/opendaylight/protocol/bgp/route/targetcontrain/impl/RouteTargetConstrainRIBSupportTest.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.route.targetcontrain.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.mdsal.binding.spec.reflect.BindingReflections;
  22 import org.opendaylight.protocol.bgp.rib.spi.AbstractRIBSupportTest;
  23 import org.opendaylight.protocol.bgp.route.targetcontrain.impl.activators.NlriActivator;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.PathId;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.Update;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes1;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes2;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Attributes;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Routes;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.bgp.rib.rib.loc.rib.tables.routes.RouteTargetConstrainRoutesCase;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.bgp.rib.rib.loc.rib.tables.routes.RouteTargetConstrainRoutesCaseBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.RouteTargetConstrainChoice;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.destination.RouteTargetConstrainDestination;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.destination.RouteTargetConstrainDestinationBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.route.target.constrain.choice.RouteTargetConstrainAs4ExtendedCommunityCaseBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.routes.RouteTargetConstrainRoutes;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.routes.RouteTargetConstrainRoutesBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.routes.route.target.constrain.routes.RouteTargetConstrainRoute;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.routes.route.target.constrain.routes.RouteTargetConstrainRouteBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.routes.route.target.constrain.routes.RouteTargetConstrainRouteKey;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationRouteTargetConstrainAdvertizedCase;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationRouteTargetConstrainAdvertizedCaseBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination.route.target.constrain.advertized._case.DestinationRouteTargetConstrainBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationRouteTargetConstrainWithdrawnCase;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationRouteTargetConstrainWithdrawnCaseBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.as._4.route.target.extended.community.grouping.As4RouteTargetExtendedCommunityBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.as._4.spec.common.As4SpecificCommon;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.as._4.spec.common.As4SpecificCommonBuilder;
  50 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  51 import org.opendaylight.yangtools.yang.common.Uint16;
  52 import org.opendaylight.yangtools.yang.common.Uint32;
  53 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  54 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  55 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  56 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  57 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidateNode;
  58 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidates;
  59
  60 public class RouteTargetConstrainRIBSupportTest extends AbstractRIBSupportTest<RouteTargetConstrainRoutesCase,
  61         RouteTargetConstrainRoutes, RouteTargetConstrainRoute, RouteTargetConstrainRouteKey> {
  62     private static final AsNumber ORIGIN_AS = new AsNumber(Uint32.valueOf(72));
  63     private static final PathId PATH_ID = new PathId(Uint32.ZERO);
  64     private static final As4SpecificCommon AS_COMMON = new As4SpecificCommonBuilder()
  65             .setAsNumber(new AsNumber(Uint32.valueOf(20)))
  66             .setLocalAdministrator(Uint16.valueOf(100))
  67             .build();
  68     private static final RouteTargetConstrainChoice RT = new RouteTargetConstrainAs4ExtendedCommunityCaseBuilder()
  69             .setAs4RouteTargetExtendedCommunity(new As4RouteTargetExtendedCommunityBuilder()
  70                     .setAs4SpecificCommon(AS_COMMON).build())
  71             .build();
  72     private static final RouteTargetConstrainRouteKey ROUTE_KEY
  73             = new RouteTargetConstrainRouteKey(PATH_ID, "AgIAAAAUAGQ=");
  74     private static final RouteTargetConstrainRoute ROUTE = new RouteTargetConstrainRouteBuilder()
  75             .setRouteKey(ROUTE_KEY.getRouteKey())
  76             .setPathId(ROUTE_KEY.getPathId())
  77             .setAttributes(ATTRIBUTES)
  78             .setRouteTargetConstrainChoice(RT)
  79             .setOriginAs(ORIGIN_AS)
  80             .build();
  81     private static final RouteTargetConstrainRoutes RT_ROUTES
  82             = new RouteTargetConstrainRoutesBuilder()
  83             .setRouteTargetConstrainRoute(Collections.singletonList(ROUTE)).build();
  84
  85     private static final RouteTargetConstrainDestination RT_DESTINATION = new RouteTargetConstrainDestinationBuilder()
  86             .setRouteTargetConstrainChoice(RT)
  87             .setOriginAs(ORIGIN_AS)
  88             .setPathId(PATH_ID)
  89             .build();
  90     private static final DestinationRouteTargetConstrainAdvertizedCase REACH_NLRI
  91             = new DestinationRouteTargetConstrainAdvertizedCaseBuilder()
  92             .setDestinationRouteTargetConstrain(new DestinationRouteTargetConstrainBuilder()
  93                     .setRouteTargetConstrainDestination(Collections.singletonList(RT_DESTINATION)).build()).build();
  94     private static final DestinationRouteTargetConstrainWithdrawnCase UNREACH_NLRI
  95             = new DestinationRouteTargetConstrainWithdrawnCaseBuilder()
  96             .setDestinationRouteTargetConstrain(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp
  97                     .route.target.constrain.rev180618.update.attributes.mp.unreach.nlri.withdrawn.routes.destination
  98                     .type.destination.route.target.constrain.withdrawn._case.DestinationRouteTargetConstrainBuilder()
  99                     .setRouteTargetConstrainDestination(Collections.singletonList(RT_DESTINATION)).build()).build();
 100
 101     private RouteTargetConstrainRIBSupport ribSupport;
 102
 103     @Override
 104     public void setUp() throws Exception {
 105         super.setUp();
 106         NlriActivator.registerNlriParsers(new ArrayList<>());
 107         this.ribSupport = RouteTargetConstrainRIBSupport.getInstance(this.mappingService);
 108         setUpTestCustomizer(this.ribSupport);
 109     }
 110
 111     @Test
 112     public void testDeleteRoutes() {
 113         final ContainerNode withdraw = createNlriWithDrawnRoute(UNREACH_NLRI);
 114         this.ribSupport.deleteRoutes(this.tx, getTablePath(), withdraw);
 115         final InstanceIdentifier<RouteTargetConstrainRoute> instanceIdentifier = this.deletedRoutes.get(0);
 116         assertEquals(ROUTE_KEY, instanceIdentifier.firstKeyOf(RouteTargetConstrainRoute.class));
 117     }
 118
 119     @Test
 120     public void testPutRoutes() {
 121         this.ribSupport.putRoutes(this.tx, getTablePath(), createNlriAdvertiseRoute(REACH_NLRI), createAttributes());
 122         final RouteTargetConstrainRoute route = (RouteTargetConstrainRoute) this.insertedRoutes.get(0).getValue();
 123         assertEquals(ROUTE, route);
 124     }
 125
 126
 127     @Test
 128     public void testEmptyRoute() {
 129         assertEquals(createEmptyTable(), this.ribSupport.emptyTable());
 130     }
 131
 132     @Test
 133     public void testBuildMpUnreachNlriUpdate() {
 134         final Collection<MapEntryNode> routes = createRoutes(RT_ROUTES);
 135         final Update update = this.ribSupport.buildUpdate(Collections.emptyList(), routes, ATTRIBUTES);
 136         assertEquals(UNREACH_NLRI, update.getAttributes().augmentation(Attributes2.class).getMpUnreachNlri()
 137                 .getWithdrawnRoutes().getDestinationType());
 138         assertNull(update.getAttributes().augmentation(Attributes1.class));
 139     }
 140
 141     @Test
 142     public void testBuildMpReachNlriUpdate() {
 143         final Collection<MapEntryNode> routes = createRoutes(RT_ROUTES);
 144         final Update update = this.ribSupport.buildUpdate(routes, Collections.emptyList(), ATTRIBUTES);
 145         assertEquals(REACH_NLRI, update.getAttributes().augmentation(Attributes1.class).getMpReachNlri()
 146                 .getAdvertizedRoutes().getDestinationType());
 147         assertNull(update.getAttributes().augmentation(Attributes2.class));
 148     }
 149
 150     @Test
 151     public void testCacheableNlriObjects() {
 152         assertEquals(ImmutableSet.of(RouteTargetConstrainRoutesCase.class), this.ribSupport.cacheableNlriObjects());
 153     }
 154
 155     @Test
 156     public void testCacheableAttributeObjects() {
 157         assertEquals(ImmutableSet.of(), this.ribSupport.cacheableAttributeObjects());
 158     }
 159
 160     @Test
 161     public void testRouteIdAddPath() {
 162         assertEquals(ROUTE_KEY, this.ribSupport.createRouteListKey(ROUTE_KEY.getPathId(), ROUTE_KEY.getRouteKey()));
 163     }
 164
 165     @Test
 166     public void testRoutePath() {
 167         final YangInstanceIdentifier.NodeIdentifierWithPredicates prefixNii = createRouteNIWP(RT_ROUTES);
 168         final YangInstanceIdentifier expected = getRoutePath().node(prefixNii);
 169         final YangInstanceIdentifier actual = this.ribSupport.routePath(getTablePath(), prefixNii);
 170         assertEquals(expected, actual);
 171     }
 172
 173     @Test
 174     public void testRouteAttributesIdentifier() {
 175         assertEquals(new NodeIdentifier(Attributes.QNAME.withModule(BindingReflections
 176                         .getQNameModule(RouteTargetConstrainRoutesCase.class))),
 177                 this.ribSupport.routeAttributesIdentifier());
 178     }
 179
 180     @Test
 181     public void testRoutesCaseClass() {
 182         assertEquals(RouteTargetConstrainRoutesCase.class, this.ribSupport.routesCaseClass());
 183     }
 184
 185     @Test
 186     public void testRoutesContainerClass() {
 187         assertEquals(RouteTargetConstrainRoutes.class, this.ribSupport.routesContainerClass());
 188     }
 189
 190     @Test
 191     public void testRoutesListClass() {
 192         assertEquals(RouteTargetConstrainRoute.class, this.ribSupport.routesListClass());
 193     }
 194
 195     @Test
 196     public void testChangedRoutes() {
 197         final Routes emptyCase = new RouteTargetConstrainRoutesCaseBuilder().build();
 198         DataTreeCandidateNode tree = DataTreeCandidates.fromNormalizedNode(getRoutePath(),
 199                 createRoutes(emptyCase)).getRootNode();
 200         assertTrue(this.ribSupport.changedRoutes(tree).isEmpty());
 201
 202         final Routes emptyRoutes = new RouteTargetConstrainRoutesCaseBuilder()
 203                 .setRouteTargetConstrainRoutes(new RouteTargetConstrainRoutesBuilder().build()).build();
 204         tree = DataTreeCandidates.fromNormalizedNode(getRoutePath(), createRoutes(emptyRoutes)).getRootNode();
 205         assertTrue(this.ribSupport.changedRoutes(tree).isEmpty());
 206
 207         final Routes routes = new RouteTargetConstrainRoutesCaseBuilder()
 208                 .setRouteTargetConstrainRoutes(RT_ROUTES).build();
 209         tree = DataTreeCandidates.fromNormalizedNode(getRoutePath(), createRoutes(routes)).getRootNode();
 210         final Collection<DataTreeCandidateNode> result = this.ribSupport.changedRoutes(tree);
 211         assertFalse(result.isEmpty());
 212     }
 213 }