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().setKey(new EvpnRdToNetworkKey(RD));
67 evpnRdToNetworkBuilder.setRd(RD);
68 evpnRdToNetworkBuilder.setNetworkId(ELAN1);
69 LOG.info("updating Evpn {} with elaninstance {} and rd {}", "evpn1", actualElanInstances, RD);
70 singleTxdataBroker.syncWrite(LogicalDatastoreType.CONFIGURATION, getRdToNetworkIdentifier(RD),
71 evpnRdToNetworkBuilder.build());
74 public InstanceIdentifier<EvpnRdToNetwork> getRdToNetworkIdentifier(String vrfId) {
75 return InstanceIdentifier.builder(EvpnRdToNetworks.class)
76 .child(EvpnRdToNetwork.class, new EvpnRdToNetworkKey(vrfId)).build();
79 public void updateEvpnNameInElan(String elanInstanceName, String evpnName)
80 throws ReadFailedException, TransactionCommitFailedException {
81 InstanceIdentifier<ElanInstance> elanIid = ElanHelper.getElanInstanceConfigurationDataPath(elanInstanceName);
82 ElanInstance elanInstance = singleTxdataBroker.syncRead(LogicalDatastoreType.CONFIGURATION, elanIid);
83 EvpnAugmentationBuilder evpnAugmentationBuilder = new EvpnAugmentationBuilder();
84 ElanInstanceBuilder elanInstanceBuilder = new ElanInstanceBuilder(elanInstance);
85 evpnAugmentationBuilder.setEvpnName(evpnName);
86 LOG.debug("Writing Elan-EvpnAugmentation evpnName {} with key {}", evpnName, elanInstanceName);
87 elanInstanceBuilder.addAugmentation(EvpnAugmentation.class, evpnAugmentationBuilder.build());
88 singleTxdataBroker.syncWrite(LogicalDatastoreType.CONFIGURATION, elanIid, elanInstanceBuilder.build());
91 public void deleteEvpnNameInElan(String elanInstanceName)
92 throws ReadFailedException, TransactionCommitFailedException {
93 InstanceIdentifier<ElanInstance> elanIid = ElanHelper.getElanInstanceConfigurationDataPath(elanInstanceName);
94 ElanInstance elanInstance = singleTxdataBroker.syncRead(LogicalDatastoreType.CONFIGURATION, elanIid);
96 EvpnAugmentationBuilder evpnAugmentationBuilder = new EvpnAugmentationBuilder();
97 ElanInstanceBuilder elanInstanceBuilder = new ElanInstanceBuilder(elanInstance);
98 evpnAugmentationBuilder.setEvpnName(null);
99 LOG.debug("deleting evpn name from Elan-EvpnAugmentation {} ", elanInstanceName);
100 elanInstanceBuilder.addAugmentation(EvpnAugmentation.class, evpnAugmentationBuilder.build());
101 singleTxdataBroker.syncWrite(LogicalDatastoreType.CONFIGURATION, elanIid, elanInstanceBuilder.build());
104 public void deleteRdtoNetworks() throws TransactionCommitFailedException {
105 LOG.info("deleting rd {}", "evpn1", RD);
106 singleTxdataBroker.syncDelete(LogicalDatastoreType.CONFIGURATION, getRdToNetworkIdentifier(RD));
109 public void addMacVrfEntryToDS(String rd, String macAddress, String prefix,
110 List<String> nextHopList, VrfEntry.EncapType encapType,
111 long l2vni, String gatewayMacAddress, RouteOrigin origin)
112 throws TransactionCommitFailedException {
113 MacVrfEntryBuilder macEntryBuilder = new MacVrfEntryBuilder().setOrigin(origin.getValue());
114 buildVpnEncapSpecificInfo(macEntryBuilder, encapType, l2vni, macAddress,
115 gatewayMacAddress, nextHopList);
116 macEntryBuilder.setMac(macAddress);
117 macEntryBuilder.setDestPrefix(prefix);
118 InstanceIdentifier<MacVrfEntry> macEntryId =
119 InstanceIdentifier.builder(FibEntries.class)
120 .child(VrfTables.class, new VrfTablesKey(rd))
121 .child(MacVrfEntry.class, new MacVrfEntryKey(macAddress)).build();
123 singleTxdataBroker.syncWrite(LogicalDatastoreType.CONFIGURATION, macEntryId, macEntryBuilder.build());
126 public void deleteMacVrfEntryToDS(String rd, String macAddress) throws TransactionCommitFailedException {
127 InstanceIdentifier<MacVrfEntry> macEntryId =
128 InstanceIdentifier.builder(FibEntries.class)
129 .child(VrfTables.class, new VrfTablesKey(rd))
130 .child(MacVrfEntry.class, new MacVrfEntryKey(macAddress)).build();
132 singleTxdataBroker.syncDelete(LogicalDatastoreType.CONFIGURATION, macEntryId);
135 private static void buildVpnEncapSpecificInfo(MacVrfEntryBuilder builder,
136 VrfEntry.EncapType encapType, long l2vni, String macAddress,
137 String gatewayMac, List<String> nextHopList) {
138 builder.setEncapType(encapType);
139 builder.setGatewayMacAddress(gatewayMac);
140 builder.setL2vni(l2vni);
141 List<RoutePaths> routePaths = nextHopList.stream()
142 .filter(StringUtils::isNotEmpty)
144 return FibHelper.buildRoutePath(nextHop, null);
145 }).collect(Collectors.toList());
146 builder.setRoutePaths(routePaths);
149 public void attachEvpnToNetwork(ElanInstance elanInstance)
150 throws TransactionCommitFailedException, ReadFailedException {
151 // update RD to networks
152 updateRdtoNetworks(elanInstance);
154 // Attach EVPN to a network
155 updateEvpnNameInElan(ExpectedObjects.ELAN1, EVPN1);
158 public void detachEvpnToNetwork(String elanName) throws TransactionCommitFailedException, ReadFailedException {
159 deleteEvpnNameInElan(elanName);
160 deleteRdtoNetworks();
163 public void handleEvpnRt2Recvd(String macRecvd, String prefixRecvd) throws TransactionCommitFailedException {
164 List<String> nextHopList = new ArrayList<>();
165 nextHopList.add(DCGW_TEPIP);
166 addMacVrfEntryToDS(RD, macRecvd, prefixRecvd, nextHopList, VrfEntry.EncapType.Vxlan,
167 ELAN1_SEGMENT_ID, null, RouteOrigin.BGP);
170 public InstanceIdentifier<MacVrfEntry> buildMacVrfEntryIid(String mac) {
171 return InstanceIdentifier.builder(FibEntries.class)
172 .child(VrfTables.class, new VrfTablesKey(RD))
173 .child(MacVrfEntry.class, new MacVrfEntryKey(mac)).build();
176 public InstanceIdentifier<Networks> buildBgpNetworkIid(String prefix) {
177 return InstanceIdentifier.builder(Bgp.class)
178 .child(Networks.class, new NetworksKey(prefix, RD)).build();