openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/device/TransactionChainManagerTest.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8
   9 package org.opendaylight.openflowplugin.impl.device;
  10
  11 import com.google.common.util.concurrent.Futures;
  12 import io.netty.util.HashedWheelTimer;
  13 import org.junit.After;
  14 import org.junit.Before;
  15 import org.junit.Test;
  16 import org.junit.runner.RunWith;
  17 import org.mockito.Matchers;
  18 import org.mockito.Mock;
  19 import org.mockito.Mockito;
  20 import org.mockito.runners.MockitoJUnitRunner;
  21 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
  22 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  23 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  24 import org.opendaylight.controller.md.sal.common.api.data.AsyncTransaction;
  25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  26 import org.opendaylight.controller.md.sal.common.api.data.TransactionChain;
  27 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
  28 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  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.NodeBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  34 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  35
  36 /**
  37  * Created by mirehak on 4/5/15.
  38  */
  39 @RunWith(MockitoJUnitRunner.class)
  40 public class TransactionChainManagerTest {
  41
  42     @Mock
  43     private DataBroker dataBroker;
  44     @Mock
  45     private BindingTransactionChain txChain;
  46     @Mock
  47     private WriteTransaction writeTx;
  48     @Mock
  49     private TransactionChain<?, ?> transactionChain;
  50     @Mock
  51     HashedWheelTimer timer;
  52
  53     private TransactionChainManager txChainManager;
  54     private InstanceIdentifier<Node> path;
  55     private NodeId nodeId;
  56
  57     @Before
  58     public void setUp() throws Exception {
  59         Mockito.when(dataBroker.createTransactionChain(Matchers.any(TransactionChainListener.class)))
  60                 .thenReturn(txChain);
  61         txChainManager = new TransactionChainManager(dataBroker, timer, 2, 2);
  62         Mockito.when(txChain.newWriteOnlyTransaction()).thenReturn(writeTx);
  63
  64         nodeId = new NodeId("h2g2:42");
  65         path = InstanceIdentifier.create(Nodes.class).child(Node.class, new NodeKey(nodeId));
  66
  67         Mockito.when(writeTx.submit()).thenReturn(Futures.<Void, TransactionCommitFailedException>immediateCheckedFuture(null));
  68     }
  69
  70     @After
  71     public void tearDown() throws Exception {
  72         Mockito.verifyNoMoreInteractions(txChain, writeTx);
  73     }
  74
  75     @Test
  76     public void testWriteToTransaction() throws Exception {
  77         final Node data = new NodeBuilder().setId(nodeId).build();
  78         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
  79
  80         Mockito.verify(txChain).newWriteOnlyTransaction();
  81         Mockito.verify(writeTx).put(LogicalDatastoreType.CONFIGURATION, path, data);
  82     }
  83
  84     @Test
  85     public void testSubmitTransaction() throws Exception {
  86         final Node data = new NodeBuilder().setId(nodeId).build();
  87         txChainManager.enableSubmit();
  88         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
  89         txChainManager.submitTransaction();
  90
  91         Mockito.verify(txChain).newWriteOnlyTransaction();
  92         Mockito.verify(writeTx).put(LogicalDatastoreType.CONFIGURATION, path, data);
  93         Mockito.verify(writeTx).submit();
  94         Mockito.verify(writeTx).getIdentifier();
  95     }
  96
  97     /**
  98      * test of {@link TransactionChainManager#enableSubmit()}: no submit - counter is not active
  99      * @throws Exception
 100      */
 101     @Test
 102     public void testEnableCounter1() throws Exception {
 103         final Node data = new NodeBuilder().setId(nodeId).build();
 104         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 105         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 106
 107         Mockito.verify(txChain).newWriteOnlyTransaction();
 108         Mockito.verify(writeTx, Mockito.times(2)).put(LogicalDatastoreType.CONFIGURATION, path, data);
 109     }
 110
 111     /**
 112      * test of {@link TransactionChainManager#enableSubmit()}: submit - after counter activated
 113      * @throws Exception
 114      */
 115     @Test
 116     public void testEnableCounter2() throws Exception {
 117         txChainManager.enableSubmit();
 118
 119         final Node data = new NodeBuilder().setId(nodeId).build();
 120         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 121         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 122
 123         Mockito.verify(txChain).newWriteOnlyTransaction();
 124         Mockito.verify(writeTx, Mockito.times(2)).put(LogicalDatastoreType.CONFIGURATION, path, data);
 125         Mockito.verify(writeTx).submit();
 126
 127         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 128         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 129
 130         Mockito.verify(txChain, Mockito.times(2)).newWriteOnlyTransaction();
 131         Mockito.verify(writeTx, Mockito.times(4)).put(LogicalDatastoreType.CONFIGURATION, path, data);
 132         Mockito.verify(writeTx, Mockito.times(2)).submit();
 133         Mockito.verify(writeTx, Mockito.times(2)).getIdentifier();
 134     }
 135
 136     @Test
 137     public void testOnTransactionChainFailed() throws Exception {
 138         txChainManager.onTransactionChainFailed(transactionChain, Mockito.mock(AsyncTransaction.class), Mockito.mock(Throwable.class));
 139
 140         Mockito.verify(txChain).close();
 141         Mockito.verify(dataBroker, Mockito.times(2)).createTransactionChain(txChainManager);
 142     }
 143
 144     @Test
 145     public void testOnTransactionChainSuccessful() throws Exception {
 146         txChainManager.onTransactionChainSuccessful(transactionChain);
 147         // NOOP
 148         Mockito.verifyZeroInteractions(transactionChain);
 149     }
 150
 151     @Test
 152     public void testAddDeleteOperationTotTxChain() throws Exception {
 153         txChainManager.addDeleteOperationTotTxChain(LogicalDatastoreType.CONFIGURATION, path);
 154
 155         Mockito.verify(txChain).newWriteOnlyTransaction();
 156         Mockito.verify(writeTx).delete(LogicalDatastoreType.CONFIGURATION, path);
 157     }
 158 }