renderers/ofoverlay/src/test/java/org/opendaylight/groupbasedpolicy/renderer/ofoverlay/sf/ChainActionTest.java

   1 /*
   2  * Copyright (c) 2015 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.ofoverlay.sf;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.mockito.Matchers.any;
  13 import static org.mockito.Matchers.eq;
  14 import static org.mockito.Mockito.mock;
  15 import static org.mockito.Mockito.when;
  16
  17 import java.util.Collections;
  18 import java.util.HashMap;
  19 import java.util.HashSet;
  20 import java.util.List;
  21 import java.util.Map;
  22 import java.util.Set;
  23
  24 import com.google.common.base.Optional;
  25 import com.google.common.util.concurrent.CheckedFuture;
  26 import org.junit.Assert;
  27 import org.junit.Before;
  28 import org.junit.Test;
  29 import org.junit.runner.RunWith;
  30 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  31 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  32 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  33 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  34 import org.opendaylight.groupbasedpolicy.api.sf.ChainActionDefinition;
  35 import org.opendaylight.groupbasedpolicy.dto.EgKey;
  36 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.OfContext;
  37 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.OfWriter;
  38 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.endpoint.EndpointManager;
  39 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.flow.OrdinalFactory.EndpointFwdCtxOrdinals;
  40 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.mapper.policyenforcer.NetworkElements;
  41 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.common.rev151017.RspName;
  42 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.common.rev151017.SfcName;
  43 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.common.rev151017.SfpName;
  44 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.rsp.rev140701.rendered.service.paths.RenderedServicePath;
  45 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.rsp.rev140701.rendered.service.paths.RenderedServicePathBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sfp.rev140701.service.function.paths.ServiceFunctionPath;
  47 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sfp.rev140701.service.function.paths.ServiceFunctionPathBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.ActionBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.L2BridgeDomainId;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.Endpoint;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.HasDirection.Direction;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.tenant.policy.subject.feature.instances.ActionInstance;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  57 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  58 import org.powermock.api.mockito.PowerMockito;
  59 import org.powermock.core.classloader.annotations.PrepareForTest;
  60 import org.powermock.modules.junit4.PowerMockRunner;
  61
  62 @RunWith(PowerMockRunner.class)
  63 @PrepareForTest(ChainAction.class)
  64 public class ChainActionTest {
  65
  66     private ChainAction chainAction;
  67
  68     private ServiceFunctionPath sfcPath;
  69     private OfContext ctx;
  70     private DataBroker dataBroker;
  71     private ReadOnlyTransaction rTx;
  72     private CheckedFuture<Optional<RenderedServicePath>, ReadFailedException> checkedFuture;
  73     private Optional<RenderedServicePath> optRsp;
  74     private RenderedServicePath rsp;
  75     private NetworkElements netElements;
  76     private NodeId nodeId;
  77     private EndpointFwdCtxOrdinals endpointFwdCtxOrdinals;
  78     private Endpoint endpoint;
  79     private EndpointManager endpointManager;
  80     private EgKey egKey;
  81     private TenantId tenantId = new TenantId("e09a2308-6ffa-40af-92a2-69f54b2cf3e4");
  82
  83     @SuppressWarnings("unchecked")
  84     @Before
  85     public void init() throws Exception {
  86         chainAction = new ChainAction();
  87
  88         sfcPath = new ServiceFunctionPathBuilder().setName(new SfpName("sfcPathName")).setSymmetric(true).build();
  89         rsp = new RenderedServicePathBuilder().setName(new RspName("rspName")).build();
  90         endpoint = new EndpointBuilder().setL2Context(new L2BridgeDomainId("L2context"))
  91             .setMacAddress(new MacAddress("01:23:45:67:89:ab"))
  92             .setTenant(tenantId)
  93             .build();
  94
  95         ctx = mock(OfContext.class);
  96         dataBroker = mock(DataBroker.class);
  97         when(ctx.getDataBroker()).thenReturn(dataBroker);
  98         rTx = mock(ReadOnlyTransaction.class);
  99         when(dataBroker.newReadOnlyTransaction()).thenReturn(rTx);
 100         checkedFuture = mock(CheckedFuture.class);
 101         when(rTx.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class))).thenReturn(checkedFuture);
 102         optRsp = mock(Optional.class);
 103         when(checkedFuture.checkedGet()).thenReturn(optRsp);
 104         when(optRsp.isPresent()).thenReturn(true).thenReturn(false);
 105         when(optRsp.get()).thenReturn(rsp);
 106
 107         netElements = mock(NetworkElements.class);
 108         nodeId = mock(NodeId.class);
 109         when(netElements.getDstNodeId()).thenReturn(nodeId);
 110         endpointFwdCtxOrdinals = mock(EndpointFwdCtxOrdinals.class);
 111         when(netElements.getSrcEpOrdinals()).thenReturn(endpointFwdCtxOrdinals);
 112         when(netElements.getDstEpOrdinals()).thenReturn(endpointFwdCtxOrdinals);
 113         when(netElements.getSrcEp()).thenReturn(endpoint);
 114
 115         endpointManager = mock(EndpointManager.class);
 116         when(ctx.getEndpointManager()).thenReturn(endpointManager);
 117         egKey = mock(EgKey.class);
 118         Set<EgKey> keysForEndpoint = new HashSet<>();
 119         keysForEndpoint.add(egKey);
 120         when(endpointManager.getEgKeysForEndpoint(any(Endpoint.class))).thenReturn(keysForEndpoint);
 121     }
 122
 123     @Test
 124     public void testGetters() {
 125         assertEquals(ChainActionDefinition.ID, chainAction.getId());
 126         assertEquals(ChainActionDefinition.DEFINITION, chainAction.getActionDef());
 127     }
 128
 129     @Test
 130     public void testUpdateAction_DirectionOut_OpendaylightSfcNull() {
 131         ActionBuilder actionBuilder = mock(ActionBuilder.class);
 132         List<ActionBuilder> actions = Collections.singletonList(actionBuilder);
 133         Map<String, Object> params = new HashMap<>();
 134         String chainName = "chainName";
 135         params.put(ChainActionDefinition.SFC_CHAIN_NAME, chainName);
 136         Integer order = 0;
 137         OfWriter ofWriter = mock(OfWriter.class);
 138
 139         PowerMockito.mockStatic(ChainAction.class);
 140         SfcName sfcName = new SfcName(chainName);
 141         when(ChainAction.getSfcPath(eq(sfcName))).thenReturn(sfcPath);
 142         chainAction.setResolvedSymmetricChains(Collections.singletonList(chainName));
 143
 144         List<ActionBuilder> result =
 145                 chainAction.updateAction(actions, params, order, netElements, ofWriter, ctx, Direction.Out);
 146         Assert.assertNull(result);
 147     }
 148
 149     @Test
 150     public void testUpdateAction_ParamsNull() {
 151         ActionBuilder actionBuilder = mock(ActionBuilder.class);
 152         List<ActionBuilder> actions = Collections.singletonList(actionBuilder);
 153         Integer order = 0;
 154         OfWriter ofWriter = mock(OfWriter.class);
 155
 156         List<ActionBuilder> result =
 157                 chainAction.updateAction(actions, null, order, netElements, ofWriter, ctx, Direction.In);
 158         Assert.assertNull(result);
 159     }
 160
 161     @Test
 162     public void testUpdateAction_ChainNameNull() {
 163         ActionBuilder actionBuilder = mock(ActionBuilder.class);
 164         List<ActionBuilder> actions = Collections.singletonList(actionBuilder);
 165         Map<String, Object> params = new HashMap<>();
 166         params.put(ChainActionDefinition.SFC_CHAIN_NAME, null);
 167         Integer order = 0;
 168         NetworkElements netElements = mock(NetworkElements.class);
 169         OfWriter ofWriter = mock(OfWriter.class);
 170
 171         List<ActionBuilder> result =
 172                 chainAction.updateAction(actions, params, order, netElements, ofWriter, ctx, Direction.In);
 173
 174         Assert.assertNull(result);
 175     }
 176
 177     @Test
 178     public void testUpdateAction() {
 179         ActionBuilder actionBuilder = mock(ActionBuilder.class);
 180         List<ActionBuilder> actions = Collections.singletonList(actionBuilder);
 181         Map<String, Object> params = new HashMap<>();
 182         String chainName = "chainName";
 183         params.put(ChainActionDefinition.SFC_CHAIN_NAME, chainName);
 184         Integer order = 0;
 185         OfWriter ofWriter = mock(OfWriter.class);
 186
 187         ServiceFunctionPath sfcPathNameNull = new ServiceFunctionPathBuilder().setName(null).setSymmetric(true).build();
 188
 189         PowerMockito.mockStatic(ChainAction.class);
 190         when(ChainAction.getSfcPath(new SfcName(chainName))).thenReturn(sfcPathNameNull);
 191
 192         List<ActionBuilder> result =
 193                 chainAction.updateAction(actions, params, order, netElements, ofWriter, ctx, Direction.Out);
 194         Assert.assertNull(result);
 195     }
 196
 197     @Test
 198     public void testIsValid_ParameterValueNull() {
 199         ActionInstance actionInstance = mock(ActionInstance.class);
 200         Assert.assertFalse(chainAction.validate(actionInstance).isValid());
 201     }
 202 }