a52504499f0e8c4c04e1facd8ac5b0dc42a54862
[bgpcep.git] / bgp / l3vpn / src / test / java / org / opendaylight / protocol / bgp / l3vpn / unicast / ipv4 / VpnIpv4RIBSupportTest.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.l3vpn.unicast.ipv4;
10
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertNull;
13 import static org.junit.Assert.assertTrue;
14 import static org.opendaylight.protocol.bgp.l3vpn.unicast.ipv4.VpnIpv4NlriParserTest.DISTINGUISHER;
15 import static org.opendaylight.protocol.bgp.l3vpn.unicast.ipv4.VpnIpv4NlriParserTest.IPV4_PREFIX;
16 import static org.opendaylight.protocol.bgp.l3vpn.unicast.ipv4.VpnIpv4NlriParserTest.IPV4_VPN;
17 import static org.opendaylight.protocol.bgp.l3vpn.unicast.ipv4.VpnIpv4NlriParserTest.LABEL_STACK;
18 import static org.opendaylight.protocol.bgp.parser.spi.PathIdUtil.NON_PATH_ID;
19
20 import com.google.common.collect.ImmutableSet;
21 import java.util.Collection;
22 import java.util.Collections;
23 import org.junit.Assert;
24 import org.junit.Test;
25 import org.opendaylight.protocol.bgp.rib.spi.AbstractRIBSupportTest;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.PathId;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.Update;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.Attributes;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes1;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes2;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Routes;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.ipv4.rev180329.bgp.rib.rib.loc.rib.tables.routes.VpnIpv4RoutesCase;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.ipv4.rev180329.bgp.rib.rib.loc.rib.tables.routes.VpnIpv4RoutesCaseBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.ipv4.rev180329.l3vpn.ipv4.destination.VpnIpv4DestinationBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.ipv4.rev180329.l3vpn.ipv4.routes.VpnIpv4Routes;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.ipv4.rev180329.l3vpn.ipv4.routes.VpnIpv4RoutesBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.ipv4.rev180329.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationVpnIpv4Case;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.ipv4.rev180329.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationVpnIpv4CaseBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.rev180329.l3vpn.ip.route.VpnRoute;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.rev180329.l3vpn.ip.route.VpnRouteBuilder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.rev180329.l3vpn.ip.route.VpnRouteKey;
42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
43 import org.opendaylight.yangtools.yang.common.QName;
44 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
45 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
46 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidateNode;
47 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidates;
48
49 public class VpnIpv4RIBSupportTest extends AbstractRIBSupportTest<VpnIpv4RoutesCase, VpnIpv4Routes, VpnRoute,
50         VpnRouteKey> {
51     private VpnIpv4RIBSupport ribSupport;
52
53     private static final DestinationVpnIpv4Case REACH_NLRI = new DestinationVpnIpv4CaseBuilder().setVpnIpv4Destination(
54         new VpnIpv4DestinationBuilder().setVpnDestination(Collections.singletonList(IPV4_VPN)).build()).build();
55
56     private static final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.ipv4.rev180329.update
57             .attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationVpnIpv4Case UNREACH_NLRI =
58             new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.vpn.ipv4.rev180329.update
59                     .attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationVpnIpv4CaseBuilder()
60                     .setVpnIpv4Destination(new VpnIpv4DestinationBuilder()
61                             .setVpnDestination(Collections.singletonList(IPV4_VPN)).build()).build();
62     private static final VpnRouteKey ROUTE_KEY = new VpnRouteKey(new PathId(0L), "WAABAQIDBAECIgEW");
63
64     private static final VpnRoute ROUTE = new VpnRouteBuilder().setPathId(NON_PATH_ID)
65             .setAttributes(ATTRIBUTES).setPrefix(IPV4_PREFIX)
66         .setLabelStack(LABEL_STACK).setRouteDistinguisher(DISTINGUISHER).withKey(ROUTE_KEY).build();
67     private static final VpnIpv4Routes ROUTES = new VpnIpv4RoutesBuilder()
68             .setVpnRoute(Collections.singletonList(ROUTE)).build();
69
70     @Override
71     public void setUp() throws Exception {
72         super.setUp();
73         this.ribSupport = VpnIpv4RIBSupport.getInstance(this.mappingService);
74         setUpTestCustomizer(this.ribSupport);
75     }
76
77     @Test
78     public void testDeleteRoutes() {
79         this.ribSupport.deleteRoutes(this.tx, getTablePath(), createNlriWithDrawnRoute(UNREACH_NLRI));
80         final InstanceIdentifier<VpnRoute> instanceIdentifier = this.deletedRoutes.get(0);
81         assertEquals(ROUTE_KEY, instanceIdentifier.firstKeyOf(VpnRoute.class));
82     }
83
84     @Test
85     public void testPutRoutes() {
86         this.ribSupport.putRoutes(this.tx, getTablePath(), createNlriAdvertiseRoute(REACH_NLRI), createAttributes());
87         final VpnRoute route = (VpnRoute) this.insertedRoutes.get(0).getValue();
88         assertEquals(ROUTE, route);
89     }
90
91
92     @Test
93     public void testEmptyRoute() {
94         final Routes empty = new VpnIpv4RoutesCaseBuilder().setVpnIpv4Routes(new VpnIpv4RoutesBuilder()
95                 .setVpnRoute(Collections.emptyList()).build()).build();
96         assertEquals(createEmptyTable(empty), this.ribSupport.emptyTable());
97     }
98
99     @Test
100     public void testBuildMpUnreachNlriUpdate() {
101         final Update update = this.ribSupport.buildUpdate(Collections.emptyList(), createRoutes(ROUTES), ATTRIBUTES);
102         assertEquals(UNREACH_NLRI, update.getAttributes().augmentation(Attributes2.class)
103             .getMpUnreachNlri().getWithdrawnRoutes().getDestinationType());
104         assertNull(update.getAttributes().augmentation(Attributes1.class));
105     }
106
107     @Test
108     public void testBuildMpReachNlriUpdate() {
109         final Update update = this.ribSupport.buildUpdate(createRoutes(ROUTES), Collections.emptyList(), ATTRIBUTES);
110         assertEquals(REACH_NLRI, update.getAttributes().augmentation(Attributes1.class).getMpReachNlri()
111                 .getAdvertizedRoutes().getDestinationType());
112         assertNull(update.getAttributes().augmentation(Attributes2.class));
113     }
114
115     @Test
116     public void testCacheableNlriObjects() {
117         assertEquals(ImmutableSet.of(), this.ribSupport.cacheableNlriObjects());
118     }
119
120     @Test
121     public void testCacheableAttributeObjects() {
122         assertEquals(ImmutableSet.of(), this.ribSupport.cacheableAttributeObjects());
123     }
124
125     @Test
126     public void testRouteIdAddPath() {
127         Assert.assertEquals(ROUTE_KEY, this.ribSupport.createRouteListKey(0L, ROUTE_KEY.getRouteKey()));
128     }
129
130     @Test
131     public void testRoutePath() {
132         final NodeIdentifierWithPredicates prefixNii = createRouteNIWP(ROUTES);
133         assertEquals(getRoutePath().node(prefixNii), ribSupport
134                 .routePath(getTablePath().node(Routes.QNAME), prefixNii));
135     }
136
137     @Test
138     public void testRouteAttributesIdentifier() {
139         assertEquals(new NodeIdentifier(QName.create(VpnIpv4Routes.QNAME,
140                 Attributes.QNAME.getLocalName().intern())), this.ribSupport.routeAttributesIdentifier());
141     }
142
143     @Test
144     public void testRoutesCaseClass() {
145         assertEquals(VpnIpv4RoutesCase.class, this.ribSupport.routesCaseClass());
146     }
147
148     @Test
149     public void testRoutesContainerClass() {
150         assertEquals(VpnIpv4Routes.class, this.ribSupport.routesContainerClass());
151     }
152
153     @Test
154     public void testRoutesListClass() {
155         assertEquals(VpnRoute.class, this.ribSupport.routesListClass());
156     }
157
158     @Test
159     public void testChangedRoutes() {
160         final Routes emptyCase = new VpnIpv4RoutesCaseBuilder().build();
161         DataTreeCandidateNode tree = DataTreeCandidates.fromNormalizedNode(getRoutePath(),
162                 createRoutes(emptyCase)).getRootNode();
163         assertTrue(this.ribSupport.changedRoutes(tree).isEmpty());
164
165         final Routes emptyRoutes = new VpnIpv4RoutesCaseBuilder().setVpnIpv4Routes(
166                 new VpnIpv4RoutesBuilder().build()).build();
167         tree = DataTreeCandidates.fromNormalizedNode(getRoutePath(), createRoutes(emptyRoutes)).getRootNode();
168         assertTrue(this.ribSupport.changedRoutes(tree).isEmpty());
169
170         final Routes routes = new VpnIpv4RoutesCaseBuilder().setVpnIpv4Routes(ROUTES).build();
171         tree = DataTreeCandidates.fromNormalizedNode(getRoutePath(), createRoutes(routes)).getRootNode();
172         final Collection<DataTreeCandidateNode> result = this.ribSupport.changedRoutes(tree);
173         Assert.assertFalse(result.isEmpty());
174     }
175 }