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
9 package org.opendaylight.openflowplugin.impl.util;
11 import com.google.common.base.Function;
12 import com.google.common.util.concurrent.ListenableFuture;
13 import org.apache.commons.lang3.tuple.Pair;
14 import org.junit.After;
15 import org.junit.Assert;
16 import org.junit.Before;
17 import org.junit.Test;
18 import org.junit.runner.RunWith;
19 import org.mockito.ArgumentCaptor;
20 import org.mockito.Captor;
21 import org.mockito.Matchers;
22 import org.mockito.Mock;
23 import org.mockito.Mockito;
24 import org.mockito.runners.MockitoJUnitRunner;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.FlowCapableTransactionService;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.SendBarrierInput;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
32 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
33 import org.opendaylight.yangtools.yang.common.RpcResult;
34 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
37 * Test for {@link BarrierUtil}.
39 @RunWith(MockitoJUnitRunner.class)
40 public class BarrierUtilTest {
42 public static final NodeKey NODE_KEY = new NodeKey(new NodeId("ut-dummy-node"));
43 private static final NodeRef NODE_REF = new NodeRef(InstanceIdentifier.create(Nodes.class)
44 .child(Node.class, NODE_KEY));
47 private FlowCapableTransactionService transactionService;
49 private Function<Pair<RpcResult<String>, RpcResult<Void>>, RpcResult<String>> compositeTransform;
51 private ArgumentCaptor<Pair<RpcResult<String>, RpcResult<Void>>> pairCpt;
54 public void setUp() throws Exception {
55 Mockito.when(transactionService.sendBarrier(Matchers.<SendBarrierInput>any()))
56 .thenReturn(RpcResultBuilder.<Void>success().buildFuture());
60 public void tearDown() throws Exception {
61 Mockito.verifyNoMoreInteractions(transactionService, compositeTransform);
65 public void testChainBarrier() throws Exception {
66 final String data = "ut-data1";
67 final ListenableFuture<RpcResult<String>> input = RpcResultBuilder.success(data).buildFuture();
68 final ListenableFuture<RpcResult<String>> chainResult =
69 BarrierUtil.chainBarrier(input, NODE_REF, transactionService, compositeTransform);
71 Mockito.verify(transactionService).sendBarrier(Matchers.<SendBarrierInput>any());
72 Mockito.verify(compositeTransform).apply(pairCpt.capture());
74 final Pair<RpcResult<String>, RpcResult<Void>> value = pairCpt.getValue();
75 Assert.assertTrue(value.getLeft().isSuccessful());
76 Assert.assertEquals(data, value.getLeft().getResult());
77 Assert.assertTrue(value.getRight().isSuccessful());
78 Assert.assertNull(value.getRight().getResult());
83 public void testCreateSendBarrierInput() throws Exception {
84 final SendBarrierInput barrierInput = BarrierUtil.createSendBarrierInput(NODE_REF);
86 Assert.assertEquals(NODE_REF, barrierInput.getNode());
87 Assert.assertEquals(SendBarrierInput.class, barrierInput.getImplementedInterface());