2 * Copyright © 2017 Ericsson India Global Services Pvt Ltd. and others. All rights reserved.
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
8 package org.opendaylight.netvirt.elanmanager.tests.utils;
10 import static org.opendaylight.netvirt.elanmanager.tests.ElanServiceTestBase.DCGW_TEPIP;
11 import static org.opendaylight.netvirt.elanmanager.tests.ElanServiceTestBase.ELAN1;
12 import static org.opendaylight.netvirt.elanmanager.tests.ElanServiceTestBase.ELAN1_SEGMENT_ID;
13 import static org.opendaylight.netvirt.elanmanager.tests.ElanServiceTestBase.EVPN1;
14 import static org.opendaylight.netvirt.elanmanager.tests.ElanServiceTestBase.RD;
16 import java.util.ArrayList;
17 import java.util.List;
18 import java.util.stream.Collectors;
19 import javax.inject.Inject;
20 import javax.inject.Singleton;
21 import org.apache.commons.lang3.StringUtils;
22 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
23 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
24 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
25 import org.opendaylight.genius.datastoreutils.SingleTransactionDataBroker;
26 import org.opendaylight.netvirt.elanmanager.api.ElanHelper;
27 import org.opendaylight.netvirt.elanmanager.tests.ExpectedObjects;
28 import org.opendaylight.netvirt.fibmanager.api.FibHelper;
29 import org.opendaylight.netvirt.fibmanager.api.RouteOrigin;
30 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.Bgp;
31 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.Networks;
32 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.bgp.NetworksKey;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.EvpnAugmentation;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.EvpnAugmentationBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstance;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstanceBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.FibEntries;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.fibentries.VrfTables;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.fibentries.VrfTablesKey;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.macvrfentries.MacVrfEntry;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.macvrfentries.MacVrfEntryBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.macvrfentries.MacVrfEntryKey;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.vrfentries.VrfEntry;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.vrfentrybase.RoutePaths;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.EvpnRdToNetworks;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.evpn.rd.to.networks.EvpnRdToNetwork;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.evpn.rd.to.networks.EvpnRdToNetworkBuilder;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.evpn.rd.to.networks.EvpnRdToNetworkKey;
49 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
50 import org.slf4j.Logger;
51 import org.slf4j.LoggerFactory;
56 public class EvpnTestHelper {
57 private static final Logger LOG = LoggerFactory.getLogger(EvpnTestHelper.class);
58 private SingleTransactionDataBroker singleTxdataBroker;
61 public EvpnTestHelper(SingleTransactionDataBroker singleTxdataBroker) {
62 this.singleTxdataBroker = singleTxdataBroker;
65 public void updateRdtoNetworks(ElanInstance actualElanInstances) throws TransactionCommitFailedException {
66 EvpnRdToNetworkBuilder evpnRdToNetworkBuilder = new EvpnRdToNetworkBuilder()
67 .withKey(new EvpnRdToNetworkKey(RD));
68 evpnRdToNetworkBuilder.setRd(RD);
69 evpnRdToNetworkBuilder.setNetworkId(ELAN1);
70 LOG.info("updating Evpn {} with elaninstance {} and rd {}", "evpn1", actualElanInstances, RD);
71 singleTxdataBroker.syncWrite(LogicalDatastoreType.CONFIGURATION, getRdToNetworkIdentifier(RD),
72 evpnRdToNetworkBuilder.build());
75 public InstanceIdentifier<EvpnRdToNetwork> getRdToNetworkIdentifier(String vrfId) {
76 return InstanceIdentifier.builder(EvpnRdToNetworks.class)
77 .child(EvpnRdToNetwork.class, new EvpnRdToNetworkKey(vrfId)).build();
80 public void updateEvpnNameInElan(String elanInstanceName, String evpnName)
81 throws ReadFailedException, TransactionCommitFailedException {
82 InstanceIdentifier<ElanInstance> elanIid = ElanHelper.getElanInstanceConfigurationDataPath(elanInstanceName);
83 ElanInstance elanInstance = singleTxdataBroker.syncRead(LogicalDatastoreType.CONFIGURATION, elanIid);
84 EvpnAugmentationBuilder evpnAugmentationBuilder = new EvpnAugmentationBuilder();
85 ElanInstanceBuilder elanInstanceBuilder = new ElanInstanceBuilder(elanInstance);
86 evpnAugmentationBuilder.setEvpnName(evpnName);
87 LOG.debug("Writing Elan-EvpnAugmentation evpnName {} with key {}", evpnName, elanInstanceName);
88 elanInstanceBuilder.addAugmentation(EvpnAugmentation.class, evpnAugmentationBuilder.build());
89 singleTxdataBroker.syncWrite(LogicalDatastoreType.CONFIGURATION, elanIid, elanInstanceBuilder.build());
92 public void deleteEvpnNameInElan(String elanInstanceName)
93 throws ReadFailedException, TransactionCommitFailedException {
94 InstanceIdentifier<ElanInstance> elanIid = ElanHelper.getElanInstanceConfigurationDataPath(elanInstanceName);
95 ElanInstance elanInstance = singleTxdataBroker.syncRead(LogicalDatastoreType.CONFIGURATION, elanIid);
97 EvpnAugmentationBuilder evpnAugmentationBuilder = new EvpnAugmentationBuilder();
98 ElanInstanceBuilder elanInstanceBuilder = new ElanInstanceBuilder(elanInstance);
99 evpnAugmentationBuilder.setEvpnName(null);
100 LOG.debug("deleting evpn name from Elan-EvpnAugmentation {} ", elanInstanceName);
101 elanInstanceBuilder.addAugmentation(EvpnAugmentation.class, evpnAugmentationBuilder.build());
102 singleTxdataBroker.syncWrite(LogicalDatastoreType.CONFIGURATION, elanIid, elanInstanceBuilder.build());
105 public void deleteRdtoNetworks() throws TransactionCommitFailedException {
106 LOG.info("deleting rd {}", "evpn1", RD);
107 singleTxdataBroker.syncDelete(LogicalDatastoreType.CONFIGURATION, getRdToNetworkIdentifier(RD));
110 public void addMacVrfEntryToDS(String rd, String macAddress, String prefix,
111 List<String> nextHopList, VrfEntry.EncapType encapType,
112 long l2vni, String gatewayMacAddress, RouteOrigin origin)
113 throws TransactionCommitFailedException {
114 MacVrfEntryBuilder macEntryBuilder = new MacVrfEntryBuilder().setOrigin(origin.getValue());
115 buildVpnEncapSpecificInfo(macEntryBuilder, encapType, l2vni, macAddress,
116 gatewayMacAddress, nextHopList);
117 macEntryBuilder.setMac(macAddress);
118 macEntryBuilder.setDestPrefix(prefix);
119 InstanceIdentifier<MacVrfEntry> macEntryId =
120 InstanceIdentifier.builder(FibEntries.class)
121 .child(VrfTables.class, new VrfTablesKey(rd))
122 .child(MacVrfEntry.class, new MacVrfEntryKey(macAddress)).build();
124 singleTxdataBroker.syncWrite(LogicalDatastoreType.CONFIGURATION, macEntryId, macEntryBuilder.build());
127 public void deleteMacVrfEntryToDS(String rd, String macAddress) throws TransactionCommitFailedException {
128 InstanceIdentifier<MacVrfEntry> macEntryId =
129 InstanceIdentifier.builder(FibEntries.class)
130 .child(VrfTables.class, new VrfTablesKey(rd))
131 .child(MacVrfEntry.class, new MacVrfEntryKey(macAddress)).build();
133 singleTxdataBroker.syncDelete(LogicalDatastoreType.CONFIGURATION, macEntryId);
136 private static void buildVpnEncapSpecificInfo(MacVrfEntryBuilder builder,
137 VrfEntry.EncapType encapType, long l2vni, String macAddress,
138 String gatewayMac, List<String> nextHopList) {
139 builder.setEncapType(encapType);
140 builder.setGatewayMacAddress(gatewayMac);
141 builder.setL2vni(l2vni);
142 List<RoutePaths> routePaths = nextHopList.stream()
143 .filter(StringUtils::isNotEmpty)
145 return FibHelper.buildRoutePath(nextHop, null);
146 }).collect(Collectors.toList());
147 builder.setRoutePaths(routePaths);
150 public void attachEvpnToNetwork(ElanInstance elanInstance)
151 throws TransactionCommitFailedException, ReadFailedException {
152 // update RD to networks
153 updateRdtoNetworks(elanInstance);
155 // Attach EVPN to a network
156 updateEvpnNameInElan(ExpectedObjects.ELAN1, EVPN1);
159 public void detachEvpnToNetwork(String elanName) throws TransactionCommitFailedException, ReadFailedException {
160 deleteEvpnNameInElan(elanName);
161 deleteRdtoNetworks();
164 public void handleEvpnRt2Recvd(String macRecvd, String prefixRecvd) throws TransactionCommitFailedException {
165 List<String> nextHopList = new ArrayList<>();
166 nextHopList.add(DCGW_TEPIP);
167 addMacVrfEntryToDS(RD, macRecvd, prefixRecvd, nextHopList, VrfEntry.EncapType.Vxlan,
168 ELAN1_SEGMENT_ID, null, RouteOrigin.BGP);
171 public InstanceIdentifier<MacVrfEntry> buildMacVrfEntryIid(String mac) {
172 return InstanceIdentifier.builder(FibEntries.class)
173 .child(VrfTables.class, new VrfTablesKey(RD))
174 .child(MacVrfEntry.class, new MacVrfEntryKey(mac)).build();
177 public InstanceIdentifier<Networks> buildBgpNetworkIid(String prefix) {
178 return InstanceIdentifier.builder(Bgp.class)
179 .child(Networks.class, new NetworksKey(prefix, RD)).build();