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 java.util.Collections;
14 import com.google.common.base.Optional;
15 import com.google.common.util.concurrent.Futures;
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.ios.rev160308._native.ClassMap;
42 import org.opendaylight.yang.gen.v1.urn.ios.rev160308._native._class.map.Match;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ParameterName;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.subject.feature.instance.ParameterValue;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.subject.feature.instance.ParameterValueBuilder;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.Sgt;
48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
49 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
52 * Test for {@link PolicyManagerUtil}.
54 @RunWith(MockitoJUnitRunner.class)
55 public class PolicyManagerUtilTest {
58 private DataBroker dataBroker;
60 private ReadOnlyTransaction roTx;
62 private ArgumentCaptor<InstanceIdentifier<RenderedServicePath>> rendererServicePathIICaptor;
64 private ServiceFunctionPath serviceFunctionPath;
67 public void setUp() throws Exception {
68 serviceFunctionPath = new ServiceFunctionPathBuilder()
69 .setServiceChainName(new SfcName("sfc-chain-name-42"))
70 .setName(new SfpName("sfp-name-01"))
73 SfcDataStoreAPI.setDataProviderAux(dataBroker);
77 public void testCreateClassMap() {
78 ClassMap cm = PolicyManagerUtil.createClassMap("testName", null);
83 public void testGetServicePath() throws Exception {
84 final ParameterValue paramValueSfc = new ParameterValueBuilder()
85 .setName(new ParameterName(ChainActionDefinition.SFC_CHAIN_NAME))
86 .setStringValue("sfc-chain-name-42")
89 final ServiceFunctionPaths sfPaths = new ServiceFunctionPathsBuilder()
90 .setServiceFunctionPath(Collections.singletonList(serviceFunctionPath))
93 Mockito.when(dataBroker.newReadOnlyTransaction()).thenReturn(roTx);
94 Mockito.when(roTx.read(LogicalDatastoreType.CONFIGURATION, InstanceIdentifier.create(ServiceFunctionPaths.class)))
95 .thenReturn(Futures.immediateCheckedFuture(Optional.of(sfPaths)));
97 final ServiceFunctionPath servicePath = ServiceChainingUtil.findServicePathFromParameterValues(Collections.singletonList(paramValueSfc));
98 Assert.assertEquals(serviceFunctionPath, servicePath);
102 public void testCreateRenderedPath() throws Exception {
103 final String POLICY_MAP = "policy-map";
104 final String INTERFACE = "interface";
105 final NodeId nodeId = new NodeId("node-id");
106 final String IP_ADDRESS = "ip-address";
107 final TenantId tenantId = new TenantId("tenant-id-01");
108 final RenderedServicePath renderedSP = new RenderedServicePathBuilder().build();
109 final PolicyManagerImpl.PolicyMapLocation location = new PolicyManagerImpl.PolicyMapLocation(POLICY_MAP, INTERFACE,
110 nodeId, IP_ADDRESS, dataBroker);
111 final PolicyConfigurationContext context = new PolicyConfigurationContext();
112 context.setPolicyMapLocation(location);
114 Mockito.when(dataBroker.newReadOnlyTransaction()).thenReturn(roTx);
115 Mockito.when(roTx.read(Matchers.eq(LogicalDatastoreType.OPERATIONAL), rendererServicePathIICaptor.capture()))
116 .thenReturn(Futures.immediateCheckedFuture(Optional.of(renderedSP)));
118 final RenderedServicePath renderedPath = ServiceChainingUtil.resolveRenderedServicePath(serviceFunctionPath, tenantId,
119 dataBroker, new Sgt(1), new Sgt(2), context);
120 Assert.assertEquals(renderedSP, renderedPath);
121 final InstanceIdentifier<RenderedServicePath> ii = rendererServicePathIICaptor.getValue();
122 Assert.assertEquals("sfp-name-01-tenant-id-01-gbp-rsp", ii.firstKeyOf(RenderedServicePath.class).getName().getValue());
126 public void testCreateSymmetricRenderedPath() throws Exception {
127 final String POLICY_MAP = "policy-map";
128 final String INTERFACE = "interface";
129 final NodeId nodeId = new NodeId("node-id");
130 final String IP_ADDRESS = "ip-address";
131 final PolicyManagerImpl.PolicyMapLocation location = new PolicyManagerImpl.PolicyMapLocation(POLICY_MAP, INTERFACE,
132 nodeId, IP_ADDRESS, dataBroker);
133 final PolicyConfigurationContext context = new PolicyConfigurationContext();
134 context.setPolicyMapLocation(location);
135 final RenderedServicePath renderedServicePath = new RenderedServicePathBuilder().build();
136 final TenantId tenantId = new TenantId("tenant-id-02");
138 Mockito.when(dataBroker.newReadOnlyTransaction()).thenReturn(roTx);
139 Mockito.when(roTx.read(Matchers.eq(LogicalDatastoreType.OPERATIONAL), rendererServicePathIICaptor.capture()))
140 .thenReturn(Futures.immediateCheckedFuture(Optional.of(renderedServicePath)));
142 final RenderedServicePath symmetricRenderedPath = ServiceChainingUtil.resolveReversedRenderedServicePath(
143 serviceFunctionPath, tenantId, dataBroker, new Sgt(1), new Sgt(2), context);
144 Assert.assertEquals(renderedServicePath, symmetricRenderedPath);
145 final InstanceIdentifier<RenderedServicePath> ii = rendererServicePathIICaptor.getValue();
146 Assert.assertEquals("sfp-name-01-tenant-id-02-gbp-rsp-Reverse", ii.firstKeyOf(RenderedServicePath.class).getName().getValue());
150 public void testMatch() {
151 Match result = PolicyManagerUtil.createSecurityGroupMatch(10, 20);
152 assertNotNull(result);