2 * Copyright (c) 2017 Pantheon Technologies s.r.o. 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
9 package org.opendaylight.openflowplugin.impl.services.sal;
11 import static org.mockito.ArgumentMatchers.any;
13 import com.google.common.collect.Lists;
14 import java.util.Iterator;
15 import java.util.List;
16 import java.util.concurrent.Future;
17 import org.junit.After;
18 import org.junit.Assert;
19 import org.junit.Before;
20 import org.junit.Test;
21 import org.junit.runner.RunWith;
22 import org.mockito.ArgumentCaptor;
23 import org.mockito.ArgumentMatchers;
24 import org.mockito.Captor;
25 import org.mockito.InOrder;
26 import org.mockito.Mock;
27 import org.mockito.Mockito;
28 import org.mockito.junit.MockitoJUnitRunner;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.FlowCapableTransactionService;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.SendBarrierOutput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterInput;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterOutput;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterOutputBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterInput;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterOutput;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterOutputBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.SalMeterService;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterInput;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterOutput;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterOutputBuilder;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchInput;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchInputBuilder;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchOutput;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.RemoveMetersBatchInput;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.RemoveMetersBatchInputBuilder;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.RemoveMetersBatchOutput;
53 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.UpdateMetersBatchInput;
54 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.UpdateMetersBatchInputBuilder;
55 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.UpdateMetersBatchOutput;
56 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.add.meters.batch.input.BatchAddMeters;
57 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.add.meters.batch.input.BatchAddMetersBuilder;
58 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.input.update.grouping.OriginalBatchedMeterBuilder;
59 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.input.update.grouping.UpdatedBatchedMeterBuilder;
60 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.output.list.grouping.BatchFailedMetersOutput;
61 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.remove.meters.batch.input.BatchRemoveMeters;
62 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.remove.meters.batch.input.BatchRemoveMetersBuilder;
63 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.update.meters.batch.input.BatchUpdateMeters;
64 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.update.meters.batch.input.BatchUpdateMetersBuilder;
65 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
66 import org.opendaylight.yangtools.yang.common.RpcError;
67 import org.opendaylight.yangtools.yang.common.RpcResult;
68 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
69 import org.opendaylight.yangtools.yang.common.Uint32;
72 * Test for {@link org.opendaylight.openflowplugin.impl.services.sal.SalMetersBatchServiceImpl}.
74 @RunWith(MockitoJUnitRunner.class)
75 public class SalMetersBatchServiceImplTest {
77 public static final NodeId NODE_ID = new NodeId("ut-dummy-node");
78 public static final NodeKey NODE_KEY = new NodeKey(NODE_ID);
79 public static final NodeRef NODE_REF =
80 new NodeRef(InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY));
83 private SalMeterService salMeterService;
85 private FlowCapableTransactionService transactionService;
87 private ArgumentCaptor<RemoveMeterInput> removeMeterInputCpt;
89 private ArgumentCaptor<UpdateMeterInput> updateMeterInputCpt;
91 private ArgumentCaptor<AddMeterInput> addMeterInputCpt;
93 private SalMetersBatchServiceImpl salMetersBatchService;
97 salMetersBatchService = new SalMetersBatchServiceImpl(salMeterService, transactionService);
99 Mockito.when(transactionService.sendBarrier(any()))
100 .thenReturn(RpcResultBuilder.<SendBarrierOutput>success().buildFuture());
104 public void tearDown() {
105 Mockito.verifyNoMoreInteractions(salMeterService, transactionService);
109 public void testUpdateMetersBatch_success() throws Exception {
110 Mockito.when(salMeterService.updateMeter(Mockito.any()))
111 .thenReturn(RpcResultBuilder.success(new UpdateMeterOutputBuilder().build()).buildFuture());
113 final UpdateMetersBatchInput input = new UpdateMetersBatchInputBuilder()
115 .setBarrierAfter(true)
116 .setBatchUpdateMeters(Lists.newArrayList(
117 createEmptyBatchUpdateMeter(42L),
118 createEmptyBatchUpdateMeter(44L)))
121 final Future<RpcResult<UpdateMetersBatchOutput>> resultFuture = salMetersBatchService.updateMetersBatch(input);
123 Assert.assertTrue(resultFuture.isDone());
124 Assert.assertTrue(resultFuture.get().isSuccessful());
126 final InOrder inOrder = Mockito.inOrder(salMeterService, transactionService);
127 inOrder.verify(salMeterService, Mockito.times(2)).updateMeter(updateMeterInputCpt.capture());
128 final List<UpdateMeterInput> allValues = updateMeterInputCpt.getAllValues();
129 Assert.assertEquals(2, allValues.size());
130 Assert.assertEquals(42, allValues.get(0).getOriginalMeter().getMeterId().getValue().longValue());
131 Assert.assertEquals(43, allValues.get(0).getUpdatedMeter().getMeterId().getValue().longValue());
132 Assert.assertEquals(44, allValues.get(1).getOriginalMeter().getMeterId().getValue().longValue());
133 Assert.assertEquals(45, allValues.get(1).getUpdatedMeter().getMeterId().getValue().longValue());
135 inOrder.verify(transactionService).sendBarrier(ArgumentMatchers.any());
139 public void testUpdateMetersBatch_failure() throws Exception {
140 Mockito.when(salMeterService.updateMeter(Mockito.any()))
141 .thenReturn(RpcResultBuilder.<UpdateMeterOutput>failed()
142 .withError(RpcError.ErrorType.APPLICATION, "ur-groupUpdateError")
145 final UpdateMetersBatchInput input = new UpdateMetersBatchInputBuilder()
147 .setBarrierAfter(true)
148 .setBatchUpdateMeters(Lists.newArrayList(
149 createEmptyBatchUpdateMeter(42L),
150 createEmptyBatchUpdateMeter(44L)))
153 final Future<RpcResult<UpdateMetersBatchOutput>> resultFuture = salMetersBatchService.updateMetersBatch(input);
154 Iterator<BatchFailedMetersOutput> iterator = resultFuture.get().getResult().nonnullBatchFailedMetersOutput()
155 .values().iterator();
157 Assert.assertTrue(resultFuture.isDone());
158 Assert.assertFalse(resultFuture.get().isSuccessful());
159 Assert.assertEquals(2, resultFuture.get().getResult().nonnullBatchFailedMetersOutput().size());
160 Assert.assertEquals(43L, iterator.next().getMeterId().getValue().longValue());
161 Assert.assertEquals(45L,iterator.next().getMeterId().getValue().longValue());
162 Assert.assertEquals(2, resultFuture.get().getErrors().size());
165 final InOrder inOrder = Mockito.inOrder(salMeterService, transactionService);
166 inOrder.verify(salMeterService, Mockito.times(2)).updateMeter(updateMeterInputCpt.capture());
167 final List<UpdateMeterInput> allValues = updateMeterInputCpt.getAllValues();
168 Assert.assertEquals(2, allValues.size());
169 Assert.assertEquals(42, allValues.get(0).getOriginalMeter().getMeterId().getValue().longValue());
170 Assert.assertEquals(43, allValues.get(0).getUpdatedMeter().getMeterId().getValue().longValue());
171 Assert.assertEquals(44, allValues.get(1).getOriginalMeter().getMeterId().getValue().longValue());
172 Assert.assertEquals(45, allValues.get(1).getUpdatedMeter().getMeterId().getValue().longValue());
174 inOrder.verify(transactionService).sendBarrier(ArgumentMatchers.any());
179 public void testAddMetersBatch_success() throws Exception {
180 Mockito.when(salMeterService.addMeter(Mockito.any()))
181 .thenReturn(RpcResultBuilder.success(new AddMeterOutputBuilder().build()).buildFuture());
183 final AddMetersBatchInput input = new AddMetersBatchInputBuilder()
185 .setBarrierAfter(true)
186 .setBatchAddMeters(Lists.newArrayList(
187 createEmptyBatchAddMeter(42L),
188 createEmptyBatchAddMeter(43L)))
191 final Future<RpcResult<AddMetersBatchOutput>> resultFuture = salMetersBatchService.addMetersBatch(input);
193 Assert.assertTrue(resultFuture.isDone());
194 Assert.assertTrue(resultFuture.get().isSuccessful());
196 final InOrder inOrder = Mockito.inOrder(salMeterService, transactionService);
197 inOrder.verify(salMeterService, Mockito.times(2)).addMeter(addMeterInputCpt.capture());
198 final List<AddMeterInput> allValues = addMeterInputCpt.getAllValues();
199 Assert.assertEquals(2, allValues.size());
200 Assert.assertEquals(42L, allValues.get(0).getMeterId().getValue().longValue());
201 Assert.assertEquals(43L, allValues.get(1).getMeterId().getValue().longValue());
203 inOrder.verify(transactionService).sendBarrier(ArgumentMatchers.any());
207 public void testAddMetersBatch_failure() throws Exception {
208 Mockito.when(salMeterService.addMeter(Mockito.any()))
209 .thenReturn(RpcResultBuilder.<AddMeterOutput>failed()
210 .withError(RpcError.ErrorType.APPLICATION, "ut-groupAddError")
213 final AddMetersBatchInput input = new AddMetersBatchInputBuilder()
215 .setBarrierAfter(true)
216 .setBatchAddMeters(Lists.newArrayList(
217 createEmptyBatchAddMeter(42L),
218 createEmptyBatchAddMeter(43L)))
221 final Future<RpcResult<AddMetersBatchOutput>> resultFuture = salMetersBatchService.addMetersBatch(input);
222 Iterator<BatchFailedMetersOutput> iterator = resultFuture.get().getResult().nonnullBatchFailedMetersOutput()
223 .values().iterator();
225 Assert.assertTrue(resultFuture.isDone());
226 Assert.assertFalse(resultFuture.get().isSuccessful());
227 Assert.assertEquals(2, resultFuture.get().getResult().nonnullBatchFailedMetersOutput().size());
228 Assert.assertEquals(42L, iterator.next().getMeterId().getValue().longValue());
229 Assert.assertEquals(43L,iterator.next().getMeterId().getValue().longValue());
230 Assert.assertEquals(2, resultFuture.get().getErrors().size());
233 final InOrder inOrder = Mockito.inOrder(salMeterService, transactionService);
234 inOrder.verify(salMeterService, Mockito.times(2)).addMeter(addMeterInputCpt.capture());
235 final List<AddMeterInput> allValues = addMeterInputCpt.getAllValues();
236 Assert.assertEquals(2, allValues.size());
237 Assert.assertEquals(42L, allValues.get(0).getMeterId().getValue().longValue());
238 Assert.assertEquals(43L, allValues.get(1).getMeterId().getValue().longValue());
240 inOrder.verify(transactionService).sendBarrier(ArgumentMatchers.any());
244 public void testRemoveMetersBatch_success() throws Exception {
245 Mockito.when(salMeterService.removeMeter(Mockito.any()))
246 .thenReturn(RpcResultBuilder.success(new RemoveMeterOutputBuilder().build()).buildFuture());
248 final RemoveMetersBatchInput input = new RemoveMetersBatchInputBuilder()
250 .setBarrierAfter(true)
251 .setBatchRemoveMeters(Lists.newArrayList(
252 createEmptyBatchRemoveMeter(42L),
253 createEmptyBatchRemoveMeter(43L)))
256 final Future<RpcResult<RemoveMetersBatchOutput>> resultFuture = salMetersBatchService.removeMetersBatch(input);
258 Assert.assertTrue(resultFuture.isDone());
259 Assert.assertTrue(resultFuture.get().isSuccessful());
261 final InOrder inOrder = Mockito.inOrder(salMeterService, transactionService);
263 inOrder.verify(salMeterService, Mockito.times(2)).removeMeter(removeMeterInputCpt.capture());
264 final List<RemoveMeterInput> allValues = removeMeterInputCpt.getAllValues();
265 Assert.assertEquals(2, allValues.size());
266 Assert.assertEquals(42L, allValues.get(0).getMeterId().getValue().longValue());
267 Assert.assertEquals(43L, allValues.get(1).getMeterId().getValue().longValue());
269 inOrder.verify(transactionService).sendBarrier(ArgumentMatchers.any());
273 public void testRemoveMetersBatch_failure() throws Exception {
274 Mockito.when(salMeterService.removeMeter(Mockito.any()))
275 .thenReturn(RpcResultBuilder.<RemoveMeterOutput>failed()
276 .withError(RpcError.ErrorType.APPLICATION, "ut-groupRemoveError")
279 final RemoveMetersBatchInput input = new RemoveMetersBatchInputBuilder()
281 .setBarrierAfter(true)
282 .setBatchRemoveMeters(Lists.newArrayList(
283 createEmptyBatchRemoveMeter(42L),
284 createEmptyBatchRemoveMeter(43L)))
287 final Future<RpcResult<RemoveMetersBatchOutput>> resultFuture = salMetersBatchService.removeMetersBatch(input);
288 Iterator<BatchFailedMetersOutput> iterator = resultFuture.get().getResult().nonnullBatchFailedMetersOutput()
289 .values().iterator();
291 Assert.assertTrue(resultFuture.isDone());
292 Assert.assertFalse(resultFuture.get().isSuccessful());
293 Assert.assertEquals(2, resultFuture.get().getResult().nonnullBatchFailedMetersOutput().size());
294 Assert.assertEquals(42L, iterator.next().getMeterId().getValue().longValue());
295 Assert.assertEquals(43L, iterator.next().getMeterId().getValue().longValue());
296 Assert.assertEquals(2, resultFuture.get().getErrors().size());
298 final InOrder inOrder = Mockito.inOrder(salMeterService, transactionService);
300 inOrder.verify(salMeterService, Mockito.times(2)).removeMeter(removeMeterInputCpt.capture());
301 final List<RemoveMeterInput> allValues = removeMeterInputCpt.getAllValues();
302 Assert.assertEquals(2, allValues.size());
303 Assert.assertEquals(42L, allValues.get(0).getMeterId().getValue().longValue());
304 Assert.assertEquals(43L, allValues.get(1).getMeterId().getValue().longValue());
306 inOrder.verify(transactionService).sendBarrier(ArgumentMatchers.any());
309 private static BatchAddMeters createEmptyBatchAddMeter(final long groupIdValue) {
310 return new BatchAddMetersBuilder()
311 .setMeterId(new MeterId(Uint32.valueOf(groupIdValue)))
315 private static BatchRemoveMeters createEmptyBatchRemoveMeter(final long groupIdValue) {
316 return new BatchRemoveMetersBuilder()
317 .setMeterId(new MeterId(Uint32.valueOf(groupIdValue)))
321 private static BatchUpdateMeters createEmptyBatchUpdateMeter(final long groupIdValue) {
322 return new BatchUpdateMetersBuilder()
323 .setOriginalBatchedMeter(
324 new OriginalBatchedMeterBuilder(createEmptyBatchAddMeter(groupIdValue)).build())
325 .setUpdatedBatchedMeter(
326 new UpdatedBatchedMeterBuilder(createEmptyBatchAddMeter(groupIdValue + 1)).build())