renderers/ofoverlay/src/test/java/org/opendaylight/groupbasedpolicy/renderer/ofoverlay/statistics/ResolvedPolicyClassifierListenerTest.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.ofoverlay.statistics;
  10
  11 import static org.mockito.Mockito.mock;
  12 import static org.mockito.Mockito.spy;
  13 import static org.mockito.Mockito.verify;
  14 import static org.mockito.Mockito.when;
  15
  16 import java.util.Set;
  17
  18 import com.google.common.collect.ImmutableSet;
  19 import org.junit.Before;
  20 import org.junit.Test;
  21 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  22 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
  23 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
  24 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
  25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  26 import org.opendaylight.groupbasedpolicy.api.sf.EtherTypeClassifierDefinition;
  27 import org.opendaylight.groupbasedpolicy.api.sf.IpProtoClassifierDefinition;
  28 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.test.ParameterValueList;
  29 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.test.TestUtils;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ClassifierName;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ContractId;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.EndpointGroupId;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.RuleName;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.SubjectName;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.ResolvedPolicies;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.has.classifiers.Classifier;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.has.classifiers.ClassifierBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.resolved.policies.ResolvedPolicy;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.resolved.policies.ResolvedPolicyKey;
  41 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  42
  43 public class ResolvedPolicyClassifierListenerTest {
  44
  45     private ResolvedPolicyClassifierListener classifierListener;
  46     private DataObjectModification<ResolvedPolicy> rootNode;
  47     private Set<DataTreeModification<ResolvedPolicy>> changes;
  48
  49     private InstanceIdentifier<ResolvedPolicy> rootIdentifier;
  50
  51     private final EndpointGroupId consumerEpgId = new EndpointGroupId("consumerEpg1");
  52     private final EndpointGroupId providerEpgId = new EndpointGroupId("providerEpg1");
  53     private final ContractId contractId = new ContractId("contract1");
  54     private final TenantId tenantId = new TenantId("tenant1");
  55     private final ClassifierName classifierName = new ClassifierName("classifier1");
  56     private final SubjectName subjectName = new SubjectName("subject1");
  57     private final RuleName ruleName = new RuleName("rule1");
  58
  59     @SuppressWarnings("unchecked")
  60     @Before
  61     public void init() {
  62         DataBroker dataProvider = mock(DataBroker.class);
  63         OFStatisticsManager ofStatisticsManager = mock(OFStatisticsManager.class);
  64
  65         classifierListener = spy(new ResolvedPolicyClassifierListener(dataProvider, ofStatisticsManager));
  66
  67         ResolvedPolicyKey key = mock(ResolvedPolicyKey.class);
  68         rootNode = mock(DataObjectModification.class);
  69         rootIdentifier = InstanceIdentifier.builder(ResolvedPolicies.class).child(ResolvedPolicy.class, key).build();
  70         DataTreeIdentifier<ResolvedPolicy> rootPath =
  71                 new DataTreeIdentifier<>(LogicalDatastoreType.CONFIGURATION, rootIdentifier);
  72
  73         DataTreeModification<ResolvedPolicy> change = mock(DataTreeModification.class);
  74
  75         when(change.getRootNode()).thenReturn(rootNode);
  76         when(change.getRootPath()).thenReturn(rootPath);
  77
  78         changes = ImmutableSet.of(change);
  79
  80         ParameterValueList parameterValues = new ParameterValueList();
  81         parameterValues.addEthertype(EtherTypeClassifierDefinition.IPv4_VALUE)
  82             .addProto(IpProtoClassifierDefinition.TCP_VALUE);
  83
  84         Classifier classifier = new ClassifierBuilder().setName(classifierName)
  85             .setClassifierDefinitionId(IpProtoClassifierDefinition.ID)
  86             .setParameterValue(parameterValues)
  87             .build();
  88         ResolvedPolicy rp = TestUtils.newResolvedPolicy(tenantId, contractId, subjectName, ruleName, consumerEpgId,
  89                 providerEpgId, classifier);
  90         when(rootNode.getDataBefore()).thenReturn(rp);
  91         when(rootNode.getDataAfter()).thenReturn(rp);
  92     }
  93
  94     @Test
  95     public void testOnWrite() {
  96         when(rootNode.getModificationType()).thenReturn(DataObjectModification.ModificationType.WRITE);
  97
  98         classifierListener.onDataTreeChanged(changes);
  99
 100         verify(classifierListener).onWrite(rootNode, rootIdentifier);
 101     }
 102
 103     @Test
 104     public void testOnDelete() {
 105         when(rootNode.getModificationType()).thenReturn(DataObjectModification.ModificationType.DELETE);
 106
 107         classifierListener.onDataTreeChanged(changes);
 108
 109         verify(classifierListener).onDelete(rootNode, rootIdentifier);
 110     }
 111
 112     @Test
 113     public void testOnSubtreeModified() {
 114         when(rootNode.getModificationType()).thenReturn(DataObjectModification.ModificationType.SUBTREE_MODIFIED);
 115
 116         classifierListener.onDataTreeChanged(changes);
 117
 118         verify(classifierListener).onSubtreeModified(rootNode, rootIdentifier);
 119     }
 120
 121 }