aclservice/impl/src/test/java/org/opendaylight/netvirt/aclservice/tests/IdentifiedInterfaceWithAclBuilder.java

   1 /*
   2  * Copyright (c) 2016 Red Hat, 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 package org.opendaylight.netvirt.aclservice.tests;
   9
  10 import static org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType.CONFIGURATION;
  11
  12 import java.util.ArrayList;
  13 import java.util.List;
  14 import javax.annotation.concurrent.NotThreadSafe;
  15 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  16 import org.opendaylight.netvirt.aclservice.tests.infra.DataTreeIdentifierDataObjectPairBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.iana._if.type.rev170119.L2vlan;
  18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.Interfaces;
  19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
  20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceKey;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.rev160608.InterfaceAcl;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.rev160608.InterfaceAclBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.rev160608.interfaces._interface.AllowedAddressPairs;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.rev160608.interfaces._interface.SubnetInfo;
  27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  28
  29 @NotThreadSafe
  30 public class IdentifiedInterfaceWithAclBuilder implements DataTreeIdentifierDataObjectPairBuilder<Interface> {
  31
  32     private String interfaceName;
  33     private Boolean portSecurity;
  34     private final List<Uuid> newSecurityGroups = new ArrayList<>();
  35     private final List<AllowedAddressPairs> ifAllowedAddressPairs = new ArrayList<>();
  36     private final List<SubnetInfo> ifSubnetInfo = new ArrayList<>();
  37
  38     @Override
  39     public LogicalDatastoreType type() {
  40         return CONFIGURATION;
  41     }
  42
  43     @Override
  44     public InstanceIdentifier<Interface> identifier() {
  45         return InstanceIdentifier.builder(Interfaces.class)
  46                     .child(Interface.class, new InterfaceKey(interfaceName)).build();
  47     }
  48
  49     @Override
  50     public Interface dataObject() {
  51         return new InterfaceBuilder()
  52             .addAugmentation(InterfaceAcl.class, new InterfaceAclBuilder()
  53                 .setPortSecurityEnabled(portSecurity)
  54                 .setSecurityGroups(newSecurityGroups)
  55                 .setAllowedAddressPairs(ifAllowedAddressPairs)
  56                 .setSubnetInfo(ifSubnetInfo)
  57                 .build())
  58             .setName(interfaceName)
  59             .setType(L2vlan.class)
  60         .build();
  61     }
  62
  63     // see IdentifiedAceBuilder (@Builder)
  64
  65     public IdentifiedInterfaceWithAclBuilder interfaceName(String newInterfaceName) {
  66         this.interfaceName = newInterfaceName;
  67         return this;
  68     }
  69
  70     public IdentifiedInterfaceWithAclBuilder portSecurity(Boolean newPortSecurity) {
  71         this.portSecurity = newPortSecurity;
  72         return this;
  73     }
  74
  75     public IdentifiedInterfaceWithAclBuilder addAllNewSecurityGroups(List<Uuid> addToNewSecurityGroups) {
  76         this.newSecurityGroups.addAll(addToNewSecurityGroups);
  77         return this;
  78     }
  79
  80     public IdentifiedInterfaceWithAclBuilder addAllIfAllowedAddressPairs(
  81             List<AllowedAddressPairs> addToIfAllowedAddressPairs) {
  82         this.ifAllowedAddressPairs.addAll(addToIfAllowedAddressPairs);
  83         return this;
  84     }
  85
  86     public IdentifiedInterfaceWithAclBuilder addAllIfSubnetInfo(List<SubnetInfo> addToIfSubnetInfo) {
  87         this.ifSubnetInfo.addAll(addToIfSubnetInfo);
  88         return this;
  89     }
  90 }