renderers/ios-xe/src/test/java/org/opendaylight/groupbasedpolicy/renderer/ios_xe_provider/impl/util/PolicyManagerUtilTest.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.groupbasedpolicy.renderer.ios_xe_provider.impl.util;
  10
  11 import static org.junit.Assert.assertNotNull;
  12
  13 import com.google.common.base.Optional;
  14 import com.google.common.util.concurrent.Futures;
  15 import java.util.Collections;
  16 import org.junit.Assert;
  17 import org.junit.Before;
  18 import org.junit.Test;
  19 import org.junit.runner.RunWith;
  20 import org.mockito.ArgumentCaptor;
  21 import org.mockito.Captor;
  22 import org.mockito.Matchers;
  23 import org.mockito.Mock;
  24 import org.mockito.Mockito;
  25 import org.mockito.runners.MockitoJUnitRunner;
  26 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  27 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  28 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  29 import org.opendaylight.groupbasedpolicy.api.sf.ChainActionDefinition;
  30 import org.opendaylight.groupbasedpolicy.renderer.ios_xe_provider.impl.manager.PolicyConfigurationContext;
  31 import org.opendaylight.groupbasedpolicy.renderer.ios_xe_provider.impl.manager.PolicyManagerImpl;
  32 import org.opendaylight.sfc.provider.api.SfcDataStoreAPI;
  33 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.common.rev151017.SfcName;
  34 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.common.rev151017.SfpName;
  35 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.rsp.rev140701.rendered.service.paths.RenderedServicePath;
  36 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.rsp.rev140701.rendered.service.paths.RenderedServicePathBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sfp.rev140701.ServiceFunctionPaths;
  38 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sfp.rev140701.ServiceFunctionPathsBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sfp.rev140701.service.function.paths.ServiceFunctionPath;
  40 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sfp.rev140701.service.function.paths.ServiceFunctionPathBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
  42 import org.opendaylight.yang.gen.v1.urn.ios.rev160308._native.ClassMap;
  43 import org.opendaylight.yang.gen.v1.urn.ios.rev160308._native._class.map.Match;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.has.absolute.location.AbsoluteLocationBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.has.absolute.location.absolute.location.location.type.ExternalLocationCaseBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ParameterName;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.forwarding.l2_l3.rev160427.IpPrefixType;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.subject.feature.instance.ParameterValue;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.subject.feature.instance.ParameterValueBuilder;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.renderer.policy.configuration.endpoints.AddressEndpointWithLocation;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.renderer.policy.configuration.endpoints.AddressEndpointWithLocationBuilder;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.ip.sgt.distribution.rev160715.rpc.fields.Binding;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.ip.sgt.distribution.rev160715.rpc.fields.binding.PeerNode;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.ip.sgt.distribution.rev160715.rpc.fields.binding.PeerNodeBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.Sgt;
  57 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  58 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  59 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  60 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  61 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  62 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  63 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  64 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  65
  66 /**
  67  * Test for {@link PolicyManagerUtil}.
  68  */
  69 @RunWith(MockitoJUnitRunner.class)
  70 public class PolicyManagerUtilTest {
  71
  72     @Mock
  73     private DataBroker dataBroker;
  74     @Mock
  75     private ReadOnlyTransaction roTx;
  76     @Captor
  77     private ArgumentCaptor<InstanceIdentifier<RenderedServicePath>> rendererServicePathIICaptor;
  78
  79     private ServiceFunctionPath serviceFunctionPath;
  80
  81     @Before
  82     public void setUp() throws Exception {
  83         serviceFunctionPath = new ServiceFunctionPathBuilder()
  84                 .setServiceChainName(new SfcName("sfc-chain-name-42"))
  85                 .setName(new SfpName("sfp-name-01"))
  86                 .build();
  87
  88         SfcDataStoreAPI.setDataProviderAux(dataBroker);
  89     }
  90
  91     @Test
  92     public void testCreateClassMap() {
  93         ClassMap cm = PolicyManagerUtil.createClassMap("testName", null);
  94         assertNotNull(cm);
  95     }
  96
  97     @Test
  98     public void testGetServicePath() throws Exception {
  99         final ParameterValue paramValueSfc = new ParameterValueBuilder()
 100                 .setName(new ParameterName(ChainActionDefinition.SFC_CHAIN_NAME))
 101                 .setStringValue("sfc-chain-name-42")
 102                 .build();
 103
 104         final ServiceFunctionPaths sfPaths = new ServiceFunctionPathsBuilder()
 105                 .setServiceFunctionPath(Collections.singletonList(serviceFunctionPath))
 106                 .build();
 107
 108         Mockito.when(dataBroker.newReadOnlyTransaction()).thenReturn(roTx);
 109         Mockito.when(roTx.read(LogicalDatastoreType.CONFIGURATION, InstanceIdentifier.create(ServiceFunctionPaths.class)))
 110                 .thenReturn(Futures.immediateCheckedFuture(Optional.of(sfPaths)));
 111
 112         final ServiceFunctionPath servicePath = ServiceChainingUtil.findServicePathFromParameterValues(Collections.singletonList(paramValueSfc));
 113         Assert.assertEquals(serviceFunctionPath, servicePath);
 114     }
 115
 116     @Test
 117     public void testCreateRenderedPath() throws Exception {
 118         final String POLICY_MAP = "policy-map";
 119         final String INTERFACE = "interface";
 120         final NodeId nodeId = new NodeId("node-id");
 121         final String IP_ADDRESS = "ip-address";
 122         final TenantId tenantId = new TenantId("tenant-id-01");
 123         final RenderedServicePath renderedSP = new RenderedServicePathBuilder().build();
 124         final PolicyManagerImpl.PolicyMapLocation location = new PolicyManagerImpl.PolicyMapLocation(POLICY_MAP, INTERFACE,
 125                 nodeId, IP_ADDRESS, dataBroker);
 126         final PolicyConfigurationContext context = new PolicyConfigurationContext();
 127         context.setPolicyMapLocation(location);
 128
 129         Mockito.when(dataBroker.newReadOnlyTransaction()).thenReturn(roTx);
 130         Mockito.when(roTx.read(Matchers.eq(LogicalDatastoreType.OPERATIONAL), rendererServicePathIICaptor.capture()))
 131                 .thenReturn(Futures.immediateCheckedFuture(Optional.of(renderedSP)));
 132
 133         final RenderedServicePath renderedPath = ServiceChainingUtil.resolveRenderedServicePath(serviceFunctionPath, tenantId,
 134                 dataBroker, new Sgt(1), new Sgt(2), context);
 135         Assert.assertEquals(renderedSP, renderedPath);
 136         final InstanceIdentifier<RenderedServicePath> ii = rendererServicePathIICaptor.getValue();
 137         Assert.assertEquals("sfp-name-01-tenant-id-01-gbp-rsp", ii.firstKeyOf(RenderedServicePath.class).getName().getValue());
 138     }
 139
 140     @Test
 141     public void testCreateSymmetricRenderedPath() throws Exception {
 142         final String POLICY_MAP = "policy-map";
 143         final String INTERFACE = "interface";
 144         final NodeId nodeId = new NodeId("node-id");
 145         final String IP_ADDRESS = "ip-address";
 146         final PolicyManagerImpl.PolicyMapLocation location = new PolicyManagerImpl.PolicyMapLocation(POLICY_MAP, INTERFACE,
 147                 nodeId, IP_ADDRESS, dataBroker);
 148         final PolicyConfigurationContext context = new PolicyConfigurationContext();
 149         context.setPolicyMapLocation(location);
 150         final RenderedServicePath renderedServicePath = new RenderedServicePathBuilder().build();
 151         final TenantId tenantId = new TenantId("tenant-id-02");
 152
 153         Mockito.when(dataBroker.newReadOnlyTransaction()).thenReturn(roTx);
 154         Mockito.when(roTx.read(Matchers.eq(LogicalDatastoreType.OPERATIONAL), rendererServicePathIICaptor.capture()))
 155                 .thenReturn(Futures.immediateCheckedFuture(Optional.of(renderedServicePath)));
 156
 157         final RenderedServicePath symmetricRenderedPath = ServiceChainingUtil.resolveReversedRenderedServicePath(
 158                 serviceFunctionPath, tenantId, dataBroker, new Sgt(1), new Sgt(2), context);
 159         Assert.assertEquals(renderedServicePath, symmetricRenderedPath);
 160         final InstanceIdentifier<RenderedServicePath> ii = rendererServicePathIICaptor.getValue();
 161         Assert.assertEquals("sfp-name-01-tenant-id-02-gbp-rsp-Reverse", ii.firstKeyOf(RenderedServicePath.class).getName().getValue());
 162     }
 163
 164     @Test
 165     public void testMatch() {
 166         Match result = PolicyManagerUtil.createSecurityGroupMatch(10, 20);
 167         assertNotNull(result);
 168     }
 169
 170     @Test
 171     public void createIpSgtBindingItem() throws Exception {
 172         final Sgt sgt = new Sgt(42);
 173         final IpPrefix ipPrefix = new IpPrefix("10.11.12.42/32".toCharArray());
 174         final InstanceIdentifier<Node> nodeII = InstanceIdentifier.create(NetworkTopology.class)
 175                 .child(Topology.class, new TopologyKey(new TopologyId("dummy-topology")))
 176                 .child(Node.class, new NodeKey(new NodeId("dummy-node-1")));
 177         final PeerNode peerNode = new PeerNodeBuilder()
 178                 .setNodeIid(nodeII)
 179                 .build();
 180         final AddressEndpointWithLocation addressEndpointWithLocation = new AddressEndpointWithLocationBuilder()
 181                 .setAddressType(IpPrefixType.class)
 182                 .setAddress(ipPrefix.getIpv4Prefix().getValue())
 183                 .setAbsoluteLocation(new AbsoluteLocationBuilder()
 184                         .setLocationType(new ExternalLocationCaseBuilder()
 185                                 .setExternalNodeMountPoint(nodeII)
 186                                 .build())
 187                         .build())
 188                 .build();
 189
 190         final java.util.Optional<Binding> ipSgtBindingItem = PolicyManagerUtil.createIpSgtBindingItem(sgt, addressEndpointWithLocation);
 191
 192         Assert.assertTrue(ipSgtBindingItem.isPresent());
 193         final Binding binding = ipSgtBindingItem.get();
 194         Assert.assertEquals(sgt, binding.getSgt());
 195         Assert.assertEquals(ipPrefix, binding.getIpPrefix());
 196         Assert.assertEquals(1, binding.getPeerNode().size());
 197         Assert.assertEquals(peerNode, binding.getPeerNode().get(0));
 198     }
 199 }