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 org.junit.After;
  12 import org.junit.Before;
  13 import org.junit.Test;
  14 import org.junit.runner.RunWith;
  15 import org.mockito.Matchers;
  16 import org.mockito.Mock;
  17 import org.mockito.Mockito;
  18 import org.mockito.runners.MockitoJUnitRunner;
  19 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
  20 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  21 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  22 import org.opendaylight.controller.md.sal.common.api.data.AsyncTransaction;
  23 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  24 import org.opendaylight.controller.md.sal.common.api.data.TransactionChain;
  25 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  31 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  32
  33 /**
  34  * Created by mirehak on 4/5/15.
  35  */
  36 @RunWith(MockitoJUnitRunner.class)
  37 public class TransactionChainManagerTest {
  38
  39     @Mock
  40     private DataBroker dataBroker;
  41     @Mock
  42     private BindingTransactionChain txChain;
  43     @Mock
  44     private WriteTransaction writeTx;
  45     @Mock
  46     private TransactionChain<?, ?> transactionChain;
  47
  48     private TransactionChainManager txChainManager;
  49     private InstanceIdentifier<Node> path;
  50     private NodeId nodeId;
  51
  52     @Before
  53     public void setUp() throws Exception {
  54         Mockito.when(dataBroker.createTransactionChain(Matchers.any(TransactionChainListener.class)))
  55                 .thenReturn(txChain);
  56         txChainManager = new TransactionChainManager(dataBroker, 2);
  57         Mockito.when(txChain.newWriteOnlyTransaction()).thenReturn(writeTx);
  58
  59         nodeId = new NodeId("h2g2:42");
  60         path = InstanceIdentifier.create(Nodes.class).child(Node.class, new NodeKey(nodeId));
  61     }
  62
  63     @After
  64     public void tearDown() throws Exception {
  65         Mockito.verifyNoMoreInteractions(txChain, writeTx);
  66     }
  67
  68     @Test
  69     public void testWriteToTransaction() throws Exception {
  70         Node data = new NodeBuilder().setId(nodeId).build();
  71         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
  72
  73         Mockito.verify(txChain).newWriteOnlyTransaction();
  74         Mockito.verify(writeTx).put(LogicalDatastoreType.CONFIGURATION, path, data);
  75     }
  76
  77     @Test
  78     public void testSubmitTransaction() throws Exception {
  79         Node data = new NodeBuilder().setId(nodeId).build();
  80         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
  81         txChainManager.submitTransaction();
  82
  83         Mockito.verify(txChain).newWriteOnlyTransaction();
  84         Mockito.verify(writeTx).put(LogicalDatastoreType.CONFIGURATION, path, data);
  85         Mockito.verify(writeTx).submit();
  86     }
  87
  88     /**
  89      * test of {@link TransactionChainManager#enableCounter()}: no submit - counter is not active
  90      * @throws Exception
  91      */
  92     @Test
  93     public void testEnableCounter1() throws Exception {
  94         Node data = new NodeBuilder().setId(nodeId).build();
  95         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
  96         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
  97
  98         Mockito.verify(txChain).newWriteOnlyTransaction();
  99         Mockito.verify(writeTx, Mockito.times(2)).put(LogicalDatastoreType.CONFIGURATION, path, data);
 100     }
 101
 102     /**
 103      * test of {@link TransactionChainManager#enableCounter()}: submit - after counter activated
 104      * @throws Exception
 105      */
 106     @Test
 107     public void testEnableCounter2() throws Exception {
 108         txChainManager.enableCounter();
 109
 110         Node data = new NodeBuilder().setId(nodeId).build();
 111         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 112         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 113
 114         Mockito.verify(txChain).newWriteOnlyTransaction();
 115         Mockito.verify(writeTx, Mockito.times(2)).put(LogicalDatastoreType.CONFIGURATION, path, data);
 116         Mockito.verify(writeTx).submit();
 117
 118         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 119         txChainManager.writeToTransaction(LogicalDatastoreType.CONFIGURATION, path, data);
 120
 121         Mockito.verify(txChain, Mockito.times(2)).newWriteOnlyTransaction();
 122         Mockito.verify(writeTx, Mockito.times(4)).put(LogicalDatastoreType.CONFIGURATION, path, data);
 123         Mockito.verify(writeTx, Mockito.times(2)).submit();
 124     }
 125
 126     @Test
 127     public void testOnTransactionChainFailed() throws Exception {
 128         txChainManager.onTransactionChainSuccessful(transactionChain);
 129         // NOOP
 130     }
 131
 132     @Test
 133     public void testOnTransactionChainSuccessful() throws Exception {
 134         txChainManager.onTransactionChainFailed(transactionChain, Mockito.mock(AsyncTransaction.class), Mockito.mock(Throwable.class));
 135
 136         Mockito.verify(txChain).close();
 137         Mockito.verify(dataBroker, Mockito.times(2)).createTransactionChain(txChainManager);
 138     }
 139 }