2 * Copyright (c) 2016 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.impl.util;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertNull;
12 import static org.junit.Assert.assertTrue;
13 import static org.mockito.ArgumentMatchers.any;
14 import static org.mockito.Mockito.verify;
15 import static org.mockito.Mockito.verifyNoMoreInteractions;
16 import static org.mockito.Mockito.when;
18 import com.google.common.base.Function;
19 import com.google.common.util.concurrent.ListenableFuture;
20 import org.apache.commons.lang3.tuple.Pair;
21 import org.junit.After;
22 import org.junit.Before;
23 import org.junit.Test;
24 import org.junit.runner.RunWith;
25 import org.mockito.ArgumentCaptor;
26 import org.mockito.Captor;
27 import org.mockito.Mock;
28 import org.mockito.junit.MockitoJUnitRunner;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.SendBarrier;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.SendBarrierInput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.SendBarrierOutput;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
37 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
38 import org.opendaylight.yangtools.yang.common.RpcResult;
39 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
42 * Test for {@link BarrierUtil}.
44 @RunWith(MockitoJUnitRunner.class)
45 public class BarrierUtilTest {
46 public static final NodeKey NODE_KEY = new NodeKey(new NodeId("ut-dummy-node"));
47 private static final NodeRef NODE_REF = new NodeRef(InstanceIdentifier.create(Nodes.class)
48 .child(Node.class, NODE_KEY));
51 private SendBarrier sendBarrier;
53 private Function<Pair<RpcResult<String>, RpcResult<SendBarrierOutput>>, RpcResult<String>> compositeTransform;
55 private ArgumentCaptor<Pair<RpcResult<String>, RpcResult<SendBarrierOutput>>> pairCpt;
59 when(sendBarrier.invoke(any()))
60 .thenReturn(RpcResultBuilder.<SendBarrierOutput>success().buildFuture());
64 public void tearDown() {
65 verifyNoMoreInteractions(sendBarrier, compositeTransform);
69 public void testChainBarrier() {
70 final String data = "ut-data1";
71 final ListenableFuture<RpcResult<String>> input = RpcResultBuilder.success(data).buildFuture();
72 final ListenableFuture<RpcResult<String>> chainResult =
73 BarrierUtil.chainBarrier(input, NODE_REF, sendBarrier, compositeTransform);
75 verify(sendBarrier).invoke(any());
76 verify(compositeTransform).apply(pairCpt.capture());
78 final Pair<RpcResult<String>, RpcResult<SendBarrierOutput>> value = pairCpt.getValue();
79 assertTrue(value.getLeft().isSuccessful());
80 assertEquals(data, value.getLeft().getResult());
81 assertTrue(value.getRight().isSuccessful());
82 assertNull(value.getRight().getResult());
86 public void testCreateSendBarrierInput() {
87 final SendBarrierInput barrierInput = BarrierUtil.createSendBarrierInput(NODE_REF);
89 assertEquals(NODE_REF, barrierInput.getNode());
90 assertEquals(SendBarrierInput.class, barrierInput.implementedInterface());