renderers/vpp/src/test/java/org/opendaylight/groupbasedpolicy/renderer/vpp/commands/LoopbackCommandTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems. 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.groupbasedpolicy.renderer.vpp.commands;
  10
  11 import com.google.common.base.Optional;
  12 import org.junit.Assert;
  13 import org.junit.Before;
  14 import org.junit.Test;
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  16 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  17 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  18 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  19 import org.opendaylight.groupbasedpolicy.renderer.vpp.VppRendererDataBrokerTest;
  20 import org.opendaylight.groupbasedpolicy.renderer.vpp.util.General;
  21 import org.opendaylight.groupbasedpolicy.renderer.vpp.util.VppIidFactory;
  22 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  23 import org.opendaylight.groupbasedpolicy.util.NetUtils;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4AddressNoZone;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceKey;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ip.rev140616.Interface1;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ip.rev140616.Interface1Builder;
  34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ip.rev140616.interfaces._interface.Ipv4Builder;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ip.rev140616.interfaces._interface.ipv4.AddressBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ip.rev140616.interfaces._interface.ipv4.address.Subnet;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ip.rev140616.interfaces._interface.ipv4.address.subnet.PrefixLength;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.ip.rev140616.interfaces._interface.ipv4.address.subnet.PrefixLengthBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.PhysAddress;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang._interface.nat.rev170816.NatInterfaceAugmentation;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang._interface.nat.rev170816.NatInterfaceAugmentationBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang._interface.nat.rev170816._interface.nat.attributes.NatBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang._interface.nat.rev170816._interface.nat.attributes.nat.InboundBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev170607.Loopback;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev170607.VppInterfaceAugmentation;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev170607.VppInterfaceAugmentationBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev170607.interfaces._interface.L2Builder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev170607.interfaces._interface.LoopbackBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev170607.l2.config.attributes.interconnection.BridgeBasedBuilder;
  50
  51 import java.util.Collections;
  52 import java.util.concurrent.ExecutionException;
  53
  54 public class LoopbackCommandTest extends VppRendererDataBrokerTest {
  55
  56     private final static String DESCRIPTION = "used for testing";
  57     private final static String INTERFACE_NAME = "testInterface";
  58     private final static String BRIDGE_DOMAIN = "testBD";
  59     private final static boolean IS_BVI = true;
  60     private final static PhysAddress MAC_ADDRESS = new PhysAddress("00:11:22:33:44:55");
  61     private final static IpAddress IP_ADDRESS = new IpAddress(new Ipv4Address("10.0.0.1"));
  62     private final static IpPrefix IP_PREFIX = new IpPrefix(new Ipv4Prefix("10.0.0.1/24"));
  63
  64     private final static String UPD_DESCRIPTION = "updated description";
  65     private final static PhysAddress UPD_MAC_ADDRESS = new PhysAddress("55:44:33:22:11:00");
  66     private final static IpAddress UPD_IP_ADDRESS = new IpAddress(new Ipv4Address("20.0.0.1"));
  67     private final static IpPrefix UPD_IP_PREFIX = new IpPrefix(new Ipv4Prefix("20.0.0.1/24"));
  68
  69     private final static VppInterfaceAugmentationBuilder vppAugmentationBuilder = new VppInterfaceAugmentationBuilder()
  70         .setLoopback(new LoopbackBuilder().setMac(MAC_ADDRESS).build());
  71
  72     private final static VppInterfaceAugmentationBuilder vppAugmentationBuilderWithBD =
  73             new VppInterfaceAugmentationBuilder(vppAugmentationBuilder.build())
  74                 .setL2(new L2Builder().setInterconnection(new BridgeBasedBuilder().setBridgeDomain(BRIDGE_DOMAIN)
  75                     .setBridgedVirtualInterface(IS_BVI).setSplitHorizonGroup((short)0)
  76                     .build()).build());
  77
  78     private final static InterfaceBuilder interfaceBuilder =
  79             new InterfaceBuilder().setKey(new InterfaceKey(INTERFACE_NAME))
  80                 .setEnabled(true)
  81                 .setDescription(DESCRIPTION)
  82                 .setType(Loopback.class)
  83                 .setName(INTERFACE_NAME)
  84                 .setLinkUpDownTrapEnable(Interface.LinkUpDownTrapEnable.Enabled);
  85
  86     private final static Interface1Builder
  87         interface1Builder =
  88         new Interface1Builder().setIpv4(new Ipv4Builder().setAddress(Collections.singletonList(
  89             new AddressBuilder()
  90                 .setIp(new Ipv4AddressNoZone(IP_ADDRESS.getIpv4Address()))
  91                 .setSubnet(new PrefixLengthBuilder()
  92                     .setPrefixLength((short) NetUtils.getMaskFromPrefix(IP_PREFIX.getIpv4Prefix().getValue())).build())
  93                 .build()))
  94             .setEnabled(true)
  95             .setForwarding(false)
  96             .build());
  97
  98     private final static Interface BASIC_INTERFACE =
  99             interfaceBuilder.addAugmentation(VppInterfaceAugmentation.class, vppAugmentationBuilder.build())
 100                 .addAugmentation(NatInterfaceAugmentation.class, new NatInterfaceAugmentationBuilder().setNat(
 101                     new NatBuilder().setInbound(
 102                         new InboundBuilder().setNat44Support(true).setPostRouting(false).build()).build())
 103                     .build())
 104                 .addAugmentation(Interface1.class, interface1Builder.build()).build();
 105
 106     private final static Interface BASIC_INTERFACE_WITH_BD = interfaceBuilder
 107         .addAugmentation(VppInterfaceAugmentation.class, vppAugmentationBuilderWithBD.build()).build();
 108
 109     private DataBroker dataBroker;
 110
 111     @Before
 112     public void init() {
 113         dataBroker = getDataBroker();
 114     }
 115
 116     @Test
 117     public void testAddLoopback() throws ExecutionException, InterruptedException {
 118         ReadWriteTransaction rwTx = dataBroker.newReadWriteTransaction();
 119         LoopbackCommand addCommand = LoopbackCommand.builder()
 120             .setOperation(General.Operations.PUT)
 121             .setInterfaceName(INTERFACE_NAME)
 122             .setDescription(DESCRIPTION)
 123             .setBvi(IS_BVI)
 124             .setPhysAddress(MAC_ADDRESS)
 125             .setIpPrefix(IP_PREFIX)
 126             .setIpAddress(IP_ADDRESS)
 127             .setEnabled(true)
 128             .build();
 129
 130         Assert.assertEquals(IS_BVI, addCommand.getBvi());
 131         Assert.assertEquals(MAC_ADDRESS, addCommand.getPhysAddress());
 132         Assert.assertEquals(IP_ADDRESS, addCommand.getIpAddress());
 133         Assert.assertEquals(IP_PREFIX, addCommand.getIpPrefix());
 134
 135         Optional<Interface> optional = executeCommand(rwTx, addCommand);
 136
 137         Assert.assertTrue(optional.isPresent());
 138         Assert.assertEquals(BASIC_INTERFACE, optional.get());
 139
 140     }
 141
 142     @Test
 143     public void testAddLoopback_WithBridgeDomain() throws ExecutionException, InterruptedException {
 144         ReadWriteTransaction rwTx = dataBroker.newReadWriteTransaction();
 145         LoopbackCommand addCommand = LoopbackCommand.builder()
 146             .setOperation(General.Operations.PUT)
 147             .setInterfaceName(INTERFACE_NAME)
 148             .setDescription(DESCRIPTION)
 149             .setBvi(true)
 150             .setPhysAddress(MAC_ADDRESS)
 151             .setIpPrefix(IP_PREFIX)
 152             .setIpAddress(IP_ADDRESS)
 153             .setBridgeDomain(BRIDGE_DOMAIN)
 154             .setEnabled(true)
 155             .build();
 156
 157         Assert.assertEquals(BRIDGE_DOMAIN, addCommand.getBridgeDomain());
 158
 159         Optional<Interface> optional = executeCommand(rwTx, addCommand);
 160
 161         Assert.assertTrue(optional.isPresent());
 162         Assert.assertEquals(BASIC_INTERFACE_WITH_BD, optional.get());
 163     }
 164
 165     private Optional<Interface> executeCommand(ReadWriteTransaction rwTx, LoopbackCommand addCommand)
 166             throws ExecutionException, InterruptedException {
 167         addCommand.execute(rwTx);
 168
 169         rwTx.submit().get();
 170
 171         ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction();
 172
 173         return DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 174                 VppIidFactory.getInterfaceIID(new InterfaceKey(INTERFACE_NAME)), rTx);
 175     }
 176
 177     @Test
 178     public void testDeleteLoopbackPort() throws ExecutionException, InterruptedException {
 179         ReadWriteTransaction rwTx = dataBroker.newReadWriteTransaction();
 180
 181         rwTx.put(LogicalDatastoreType.CONFIGURATION, VppIidFactory.getInterfaceIID(BASIC_INTERFACE.getKey()),
 182                 BASIC_INTERFACE, true);
 183         rwTx.submit().get();
 184
 185         Optional<Interface> optional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 186                 VppIidFactory.getInterfaceIID(BASIC_INTERFACE.getKey()), dataBroker.newReadOnlyTransaction());
 187
 188         Assert.assertTrue(optional.isPresent());
 189
 190         LoopbackCommand deleteCommand = LoopbackCommand.builder()
 191             .setOperation(General.Operations.DELETE)
 192             .setInterfaceName(INTERFACE_NAME)
 193             .setIpPrefix(IP_PREFIX)
 194             .setIpAddress(IP_ADDRESS)
 195             .build();
 196
 197         ReadWriteTransaction rwTxDel = dataBroker.newReadWriteTransaction();
 198         deleteCommand.execute(rwTxDel);
 199         rwTxDel.submit();
 200
 201         Optional<Interface> optionalDeleted = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 202                 VppIidFactory.getInterfaceIID(new InterfaceKey(deleteCommand.getName())),
 203                 dataBroker.newReadOnlyTransaction());
 204
 205         Assert.assertFalse(optionalDeleted.isPresent());
 206     }
 207
 208     @Test
 209     public void testUpdateLoopbackPort() throws ExecutionException, InterruptedException {
 210         ReadWriteTransaction rwTx = dataBroker.newReadWriteTransaction();
 211
 212         rwTx.put(LogicalDatastoreType.CONFIGURATION, VppIidFactory.getInterfaceIID(BASIC_INTERFACE.getKey()),
 213                 BASIC_INTERFACE, true);
 214         rwTx.submit().get();
 215
 216         Optional<Interface> optional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 217                 VppIidFactory.getInterfaceIID(BASIC_INTERFACE.getKey()), dataBroker.newReadOnlyTransaction());
 218
 219         Assert.assertTrue(optional.isPresent());
 220
 221         LoopbackCommand updateCommand = LoopbackCommand.builder()
 222             .setOperation(General.Operations.MERGE)
 223             .setInterfaceName(INTERFACE_NAME)
 224             .setDescription(UPD_DESCRIPTION)
 225             .setPhysAddress(UPD_MAC_ADDRESS)
 226             .setIpPrefix(UPD_IP_PREFIX)
 227             .setIpAddress(UPD_IP_ADDRESS)
 228             .setEnabled(false)
 229             .build();
 230
 231         ReadWriteTransaction rwTxUpd = dataBroker.newReadWriteTransaction();
 232         updateCommand.execute(rwTxUpd);
 233         rwTxUpd.submit().get();
 234
 235         Optional<Interface> optionalUpdated = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 236                 VppIidFactory.getInterfaceIID(new InterfaceKey(updateCommand.getName())),
 237                 dataBroker.newReadOnlyTransaction());
 238
 239         Assert.assertTrue(optionalUpdated.isPresent());
 240         Interface updatedInterface = optionalUpdated.get();
 241
 242         Assert.assertEquals(UPD_DESCRIPTION, updatedInterface.getDescription());
 243         Assert.assertFalse(updatedInterface.isEnabled());
 244         VppInterfaceAugmentation augmentation = updatedInterface.getAugmentation(VppInterfaceAugmentation.class);
 245         Assert.assertEquals(INTERFACE_NAME, updatedInterface.getName());
 246         Assert.assertEquals(UPD_MAC_ADDRESS, augmentation.getLoopback().getMac());
 247         Interface1 interface1 = updatedInterface.getAugmentation(Interface1.class);
 248
 249         String ip = interface1.getIpv4().getAddress().get(0).getIp().getValue();
 250         Subnet subnet = interface1.getIpv4().getAddress().get(0).getSubnet();
 251         String prefix = "";
 252
 253         if ( subnet instanceof PrefixLength){
 254             prefix = ((PrefixLength) subnet).getPrefixLength().toString();
 255         }
 256         IpPrefix ipPrefix = new IpPrefix(new Ipv4Prefix(ip + "/" + prefix));
 257         IpAddress ipAddress = new IpAddress( new Ipv4Address(ip));
 258         Assert.assertEquals(UPD_IP_PREFIX, ipPrefix);
 259         Assert.assertEquals(UPD_IP_ADDRESS, ipAddress);
 260     }
 261 }