2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
\r
4 * This program and the accompanying materials are made available under the
\r
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
\r
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
\r
9 package org.opendaylight.groupbasedpolicy.renderer.ofoverlay.statistics;
\r
11 import static org.mockito.Mockito.mock;
\r
12 import static org.mockito.Mockito.spy;
\r
13 import static org.mockito.Mockito.verify;
\r
14 import static org.mockito.Mockito.when;
\r
16 import java.util.Set;
\r
18 import com.google.common.collect.ImmutableSet;
\r
19 import org.junit.Before;
\r
20 import org.junit.Test;
\r
21 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
\r
22 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
\r
23 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
\r
24 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
\r
25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
\r
26 import org.opendaylight.groupbasedpolicy.api.sf.EtherTypeClassifierDefinition;
\r
27 import org.opendaylight.groupbasedpolicy.api.sf.IpProtoClassifierDefinition;
\r
28 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.test.ParameterValueList;
\r
29 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.test.TestUtils;
\r
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ClassifierName;
\r
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ContractId;
\r
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.EndpointGroupId;
\r
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.RuleName;
\r
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.SubjectName;
\r
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
\r
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.ResolvedPolicies;
\r
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.has.classifiers.Classifier;
\r
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.has.classifiers.ClassifierBuilder;
\r
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.resolved.policies.ResolvedPolicy;
\r
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.resolved.policies.ResolvedPolicyKey;
\r
41 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
\r
43 public class ResolvedPolicyClassifierListenerTest {
\r
45 private ResolvedPolicyClassifierListener classifierListener;
\r
46 private DataObjectModification<ResolvedPolicy> rootNode;
\r
47 private Set<DataTreeModification<ResolvedPolicy>> changes;
\r
49 private InstanceIdentifier<ResolvedPolicy> rootIdentifier;
\r
51 private final EndpointGroupId consumerEpgId = new EndpointGroupId("consumerEpg1");
\r
52 private final EndpointGroupId providerEpgId = new EndpointGroupId("providerEpg1");
\r
53 private final ContractId contractId = new ContractId("contract1");
\r
54 private final TenantId tenantId = new TenantId("tenant1");
\r
55 private final ClassifierName classifierName = new ClassifierName("classifier1");
\r
56 private final SubjectName subjectName = new SubjectName("subject1");
\r
57 private final RuleName ruleName = new RuleName("rule1");
\r
59 @SuppressWarnings("unchecked")
\r
61 public void init() {
\r
62 DataBroker dataProvider = mock(DataBroker.class);
\r
63 OFStatisticsManager ofStatisticsManager = mock(OFStatisticsManager.class);
\r
65 classifierListener = spy(new ResolvedPolicyClassifierListener(dataProvider, ofStatisticsManager));
\r
67 ResolvedPolicyKey key = mock(ResolvedPolicyKey.class);
\r
68 rootNode = mock(DataObjectModification.class);
\r
69 rootIdentifier = InstanceIdentifier.builder(ResolvedPolicies.class).child(ResolvedPolicy.class, key).build();
\r
70 DataTreeIdentifier<ResolvedPolicy> rootPath =
\r
71 new DataTreeIdentifier<>(LogicalDatastoreType.CONFIGURATION, rootIdentifier);
\r
73 DataTreeModification<ResolvedPolicy> change = mock(DataTreeModification.class);
\r
75 when(change.getRootNode()).thenReturn(rootNode);
\r
76 when(change.getRootPath()).thenReturn(rootPath);
\r
78 changes = ImmutableSet.of(change);
\r
80 ParameterValueList parameterValues = new ParameterValueList();
\r
81 parameterValues.addEthertype(EtherTypeClassifierDefinition.IPv4_VALUE)
\r
82 .addProto(IpProtoClassifierDefinition.TCP_VALUE);
\r
84 Classifier classifier = new ClassifierBuilder().setName(classifierName)
\r
85 .setClassifierDefinitionId(IpProtoClassifierDefinition.ID)
\r
86 .setParameterValue(parameterValues)
\r
88 ResolvedPolicy rp = TestUtils.newResolvedPolicy(tenantId, contractId, subjectName, ruleName, consumerEpgId,
\r
89 providerEpgId, classifier);
\r
90 when(rootNode.getDataBefore()).thenReturn(rp);
\r
91 when(rootNode.getDataAfter()).thenReturn(rp);
\r
95 public void testOnWrite() {
\r
96 when(rootNode.getModificationType()).thenReturn(DataObjectModification.ModificationType.WRITE);
\r
98 classifierListener.onDataTreeChanged(changes);
\r
100 verify(classifierListener).onWrite(rootNode, rootIdentifier);
\r
104 public void testOnDelete() {
\r
105 when(rootNode.getModificationType()).thenReturn(DataObjectModification.ModificationType.DELETE);
\r
107 classifierListener.onDataTreeChanged(changes);
\r
109 verify(classifierListener).onDelete(rootNode, rootIdentifier);
\r
113 public void testOnSubtreeModified() {
\r
114 when(rootNode.getModificationType()).thenReturn(DataObjectModification.ModificationType.SUBTREE_MODIFIED);
\r
116 classifierListener.onDataTreeChanged(changes);
\r
118 verify(classifierListener).onSubtreeModified(rootNode, rootIdentifier);
\r