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
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;
39 import test.mock.util.AbstractFRMTest;
40 import test.mock.util.SalMeterServiceMock;
42 @RunWith(MockitoJUnitRunner.class)
43 public class MeterListenerTest extends AbstractFRMTest {
44 private static final NodeId NODE_ID = new NodeId("testnode:1");
45 private static final NodeKey NODE_KEY = new NodeKey(NODE_ID);
49 setUpForwardingRulesManager();
50 setDeviceMastership(NODE_ID);
54 public void addTwoMetersTest() {
55 addFlowCapableNode(NODE_KEY);
57 MeterKey meterKey = new MeterKey(new MeterId(Uint32.valueOf(2000)));
58 InstanceIdentifier<Meter> meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
59 .augmentation(FlowCapableNode.class).child(Meter.class, meterKey);
60 Meter meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_one").build();
62 WriteTransaction writeTx = getDataBroker().newWriteOnlyTransaction();
63 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
64 assertCommit(writeTx.commit());
65 SalMeterServiceMock salMeterService = (SalMeterServiceMock) getForwardingRulesManager().getSalMeterService();
66 await().atMost(10, SECONDS).until(() -> salMeterService.getAddMeterCalls().size() == 1);
67 List<AddMeterInput> addMeterCalls = salMeterService.getAddMeterCalls();
68 assertEquals(1, addMeterCalls.size());
69 assertEquals("DOM-0", addMeterCalls.get(0).getTransactionUri().getValue());
71 meterKey = new MeterKey(new MeterId(Uint32.valueOf(2001)));
72 meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
73 .augmentation(FlowCapableNode.class).child(Meter.class, meterKey);
74 meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_two").setBarrier(true).build();
75 writeTx = getDataBroker().newWriteOnlyTransaction();
76 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
77 assertCommit(writeTx.commit());
78 await().atMost(10, SECONDS).until(() -> salMeterService.getAddMeterCalls().size() == 2);
79 addMeterCalls = salMeterService.getAddMeterCalls();
80 assertEquals(2, addMeterCalls.size());
81 assertEquals("DOM-1", addMeterCalls.get(1).getTransactionUri().getValue());
82 assertEquals(meterII, addMeterCalls.get(1).getMeterRef().getValue());
86 public void updateMeterTest() {
87 addFlowCapableNode(NODE_KEY);
89 MeterKey meterKey = new MeterKey(new MeterId(Uint32.valueOf(2000)));
90 InstanceIdentifier<Meter> meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
91 .augmentation(FlowCapableNode.class).child(Meter.class, meterKey);
92 Meter meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_one").setBarrier(false).build();
94 WriteTransaction writeTx = getDataBroker().newWriteOnlyTransaction();
95 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
96 assertCommit(writeTx.commit());
97 SalMeterServiceMock salMeterService = (SalMeterServiceMock) getForwardingRulesManager().getSalMeterService();
98 await().atMost(10, SECONDS).until(() -> salMeterService.getAddMeterCalls().size() == 1);
99 List<AddMeterInput> addMeterCalls = salMeterService.getAddMeterCalls();
100 assertEquals(1, addMeterCalls.size());
101 assertEquals("DOM-0", addMeterCalls.get(0).getTransactionUri().getValue());
103 meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_two").setBarrier(true).build();
104 writeTx = getDataBroker().newWriteOnlyTransaction();
105 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
106 assertCommit(writeTx.commit());
107 await().atMost(10, SECONDS).until(() -> salMeterService.getUpdateMeterCalls().size() == 1);
108 List<UpdateMeterInput> updateMeterCalls = salMeterService.getUpdateMeterCalls();
109 assertEquals(1, updateMeterCalls.size());
110 assertEquals("DOM-1", updateMeterCalls.get(0).getTransactionUri().getValue());
111 assertEquals(meterII, updateMeterCalls.get(0).getMeterRef().getValue());
115 public void removeMeterTest() {
116 addFlowCapableNode(NODE_KEY);
118 MeterKey meterKey = new MeterKey(new MeterId(Uint32.valueOf(2000)));
119 InstanceIdentifier<Meter> meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
120 .augmentation(FlowCapableNode.class).child(Meter.class, meterKey);
121 Meter meter = new MeterBuilder().withKey(meterKey).setMeterName("meter_one").build();
123 WriteTransaction writeTx = getDataBroker().newWriteOnlyTransaction();
124 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
125 assertCommit(writeTx.commit());
126 SalMeterServiceMock salMeterService = (SalMeterServiceMock) getForwardingRulesManager().getSalMeterService();
127 await().atMost(10, SECONDS).until(() -> salMeterService.getAddMeterCalls().size() == 1);
128 List<AddMeterInput> addMeterCalls = salMeterService.getAddMeterCalls();
129 assertEquals(1, addMeterCalls.size());
130 assertEquals("DOM-0", addMeterCalls.get(0).getTransactionUri().getValue());
132 writeTx = getDataBroker().newWriteOnlyTransaction();
133 writeTx.delete(LogicalDatastoreType.CONFIGURATION, meterII);
134 assertCommit(writeTx.commit());
135 await().atMost(10, SECONDS).until(() -> salMeterService.getRemoveMeterCalls().size() == 1);
136 List<RemoveMeterInput> removeMeterCalls = salMeterService.getRemoveMeterCalls();
137 assertEquals(1, removeMeterCalls.size());
138 assertEquals("DOM-1", removeMeterCalls.get(0).getTransactionUri().getValue());
139 assertEquals(meterII, removeMeterCalls.get(0).getMeterRef().getValue());
143 public void staleMeterCreationTest() {
144 addFlowCapableNode(NODE_KEY);
146 StaleMeterKey meterKey = new StaleMeterKey(new MeterId(Uint32.valueOf(2000)));
147 InstanceIdentifier<StaleMeter> meterII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY)
148 .augmentation(FlowCapableNode.class).child(StaleMeter.class, meterKey);
149 StaleMeter meter = new StaleMeterBuilder().withKey(meterKey).setMeterName("stale_meter_one").build();
151 WriteTransaction writeTx = getDataBroker().newWriteOnlyTransaction();
152 writeTx.put(LogicalDatastoreType.CONFIGURATION, meterII, meter);
153 assertCommit(writeTx.commit());
157 public void tearDown() throws Exception {
158 getForwardingRulesManager().close();