2 * Copyright (c) 2014 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
8 package org.opendaylight.openflowplugin.applications.frm.impl;
10 import static java.util.concurrent.TimeUnit.SECONDS;
11 import static org.awaitility.Awaitility.await;
12 import static org.junit.Assert.assertEquals;
14 import java.util.List;
15 import org.junit.After;
16 import org.junit.Before;
17 import org.junit.Test;
18 import org.junit.runner.RunWith;
19 import org.mockito.junit.MockitoJUnitRunner;
20 import org.opendaylight.mdsal.binding.api.WriteTransaction;
21 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.Meter;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.MeterKey;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.StaleMeter;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.StaleMeterBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.meters.StaleMeterKey;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterInput;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterInput;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterInput;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId;
37 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
38 import org.opendaylight.yangtools.yang.common.Uint32;
40 @RunWith(MockitoJUnitRunner.class)
41 public class MeterListenerTest extends AbstractFRMTest {
42 private static final NodeId NODE_ID = new NodeId("testnode:1");
43 private static final NodeKey NODE_KEY = new NodeKey(NODE_ID);
47 setUpForwardingRulesManager();
48 setDeviceMastership(NODE_ID);
52 public void addTwoMetersTest() {
53 addFlowCapableNode(NODE_KEY);
55 MeterKey meterKey = new MeterKey(new MeterId(Uint32.valueOf(2000)));
56 InstanceIdentifier<Meter> meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
57 .augmentation(FlowCapableNode.class).child(Meter.class, meterKey);
58 Meter meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_one").build();
60 WriteTransaction writeTx = getDataBroker().newWriteOnlyTransaction();
61 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
62 assertCommit(writeTx.commit());
63 SalMeterServiceMock salMeterService = (SalMeterServiceMock) getForwardingRulesManager().getSalMeterService();
64 await().atMost(10, SECONDS).until(() -> salMeterService.getAddMeterCalls().size() == 1);
65 List<AddMeterInput> addMeterCalls = salMeterService.getAddMeterCalls();
66 assertEquals(1, addMeterCalls.size());
67 assertEquals("DOM-0", addMeterCalls.get(0).getTransactionUri().getValue());
69 meterKey = new MeterKey(new MeterId(Uint32.valueOf(2001)));
70 meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
71 .augmentation(FlowCapableNode.class).child(Meter.class, meterKey);
72 meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_two").setBarrier(true).build();
73 writeTx = getDataBroker().newWriteOnlyTransaction();
74 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
75 assertCommit(writeTx.commit());
76 await().atMost(10, SECONDS).until(() -> salMeterService.getAddMeterCalls().size() == 2);
77 addMeterCalls = salMeterService.getAddMeterCalls();
78 assertEquals(2, addMeterCalls.size());
79 assertEquals("DOM-1", addMeterCalls.get(1).getTransactionUri().getValue());
80 assertEquals(meterII, addMeterCalls.get(1).getMeterRef().getValue());
84 public void updateMeterTest() {
85 addFlowCapableNode(NODE_KEY);
87 MeterKey meterKey = new MeterKey(new MeterId(Uint32.valueOf(2000)));
88 InstanceIdentifier<Meter> meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
89 .augmentation(FlowCapableNode.class).child(Meter.class, meterKey);
90 Meter meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_one").setBarrier(false).build();
92 WriteTransaction writeTx = getDataBroker().newWriteOnlyTransaction();
93 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
94 assertCommit(writeTx.commit());
95 SalMeterServiceMock salMeterService = (SalMeterServiceMock) getForwardingRulesManager().getSalMeterService();
96 await().atMost(10, SECONDS).until(() -> salMeterService.getAddMeterCalls().size() == 1);
97 List<AddMeterInput> addMeterCalls = salMeterService.getAddMeterCalls();
98 assertEquals(1, addMeterCalls.size());
99 assertEquals("DOM-0", addMeterCalls.get(0).getTransactionUri().getValue());
101 meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_two").setBarrier(true).build();
102 writeTx = getDataBroker().newWriteOnlyTransaction();
103 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
104 assertCommit(writeTx.commit());
105 await().atMost(10, SECONDS).until(() -> salMeterService.getUpdateMeterCalls().size() == 1);
106 List<UpdateMeterInput> updateMeterCalls = salMeterService.getUpdateMeterCalls();
107 assertEquals(1, updateMeterCalls.size());
108 assertEquals("DOM-1", updateMeterCalls.get(0).getTransactionUri().getValue());
109 assertEquals(meterII, updateMeterCalls.get(0).getMeterRef().getValue());
113 public void removeMeterTest() {
114 addFlowCapableNode(NODE_KEY);
116 MeterKey meterKey = new MeterKey(new MeterId(Uint32.valueOf(2000)));
117 InstanceIdentifier<Meter> meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
118 .augmentation(FlowCapableNode.class).child(Meter.class, meterKey);
119 Meter meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_one").build();
121 WriteTransaction writeTx = getDataBroker().newWriteOnlyTransaction();
122 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
123 assertCommit(writeTx.commit());
124 SalMeterServiceMock salMeterService = (SalMeterServiceMock) getForwardingRulesManager().getSalMeterService();
125 await().atMost(10, SECONDS).until(() -> salMeterService.getAddMeterCalls().size() == 1);
126 List<AddMeterInput> addMeterCalls = salMeterService.getAddMeterCalls();
127 assertEquals(1, addMeterCalls.size());
128 assertEquals("DOM-0", addMeterCalls.get(0).getTransactionUri().getValue());
130 writeTx = getDataBroker().newWriteOnlyTransaction();
131 writeTx.delete(LogicalDatastoreType.CONFIGURATION, meterII);
132 assertCommit(writeTx.commit());
133 await().atMost(10, SECONDS).until(() -> salMeterService.getRemoveMeterCalls().size() == 1);
134 List<RemoveMeterInput> removeMeterCalls = salMeterService.getRemoveMeterCalls();
135 assertEquals(1, removeMeterCalls.size());
136 assertEquals("DOM-1", removeMeterCalls.get(0).getTransactionUri().getValue());
137 assertEquals(meterII, removeMeterCalls.get(0).getMeterRef().getValue());
141 public void staleMeterCreationTest() {
142 addFlowCapableNode(NODE_KEY);
144 StaleMeterKey meterKey = new StaleMeterKey(new MeterId(Uint32.valueOf(2000)));
145 InstanceIdentifier<StaleMeter> meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
146 .augmentation(FlowCapableNode.class).child(StaleMeter.class, meterKey);
147 StaleMeter meter = new StaleMeterBuilder().withKey(meterKey).setMeterName("stale_meter_one").build();
149 WriteTransaction writeTx = getDataBroker().newWriteOnlyTransaction();
150 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
151 assertCommit(writeTx.commit());
155 public void tearDown() throws Exception {
156 getForwardingRulesManager().close();