2 * Copyright (c) 2016 Cisco Systems, Inc. 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
9 package org.opendaylight.groupbasedpolicy.renderer.ios_xe_provider.impl.util;
11 import static org.junit.Assert.assertNotNull;
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.rev170511.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;
67 * Test for {@link PolicyManagerUtil}.
69 @RunWith(MockitoJUnitRunner.class)
70 public class PolicyManagerUtilTest {
73 private DataBroker dataBroker;
75 private ReadOnlyTransaction roTx;
77 private ArgumentCaptor<InstanceIdentifier<RenderedServicePath>> rendererServicePathIICaptor;
79 private ServiceFunctionPath serviceFunctionPath;
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"))
88 SfcDataStoreAPI.setDataProviderAux(dataBroker);
92 public void testCreateClassMap() {
93 ClassMap cm = PolicyManagerUtil.createClassMap("testName", null);
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")
104 final ServiceFunctionPaths sfPaths = new ServiceFunctionPathsBuilder()
105 .setServiceFunctionPath(Collections.singletonList(serviceFunctionPath))
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)));
112 final ServiceFunctionPath servicePath = ServiceChainingUtil.findServicePathFromParameterValues(Collections.singletonList(paramValueSfc));
113 Assert.assertEquals(serviceFunctionPath, servicePath);
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);
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)));
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());
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");
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)));
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());
165 public void testMatch() {
166 Match result = PolicyManagerUtil.createSecurityGroupMatch(10, 20);
167 assertNotNull(result);
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()
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)
190 final java.util.Optional<Binding> ipSgtBindingItem = PolicyManagerUtil.createIpSgtBindingItem(sgt, addressEndpointWithLocation);
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));