binding/mdsal-binding-dom-adapter/src/test/java/org/opendaylight/mdsal/binding/dom/adapter/BindingDOMTransactionChainAdapterTest.java

   1 /*
   2  * Copyright (c) 2016 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 package org.opendaylight.mdsal.binding.dom.adapter;
   9
  10 import static java.util.Objects.requireNonNull;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.mockito.ArgumentMatchers.any;
  13 import static org.mockito.Mockito.doCallRealMethod;
  14 import static org.mockito.Mockito.doNothing;
  15 import static org.mockito.Mockito.doReturn;
  16 import static org.mockito.Mockito.mock;
  17 import static org.mockito.Mockito.verify;
  18
  19 import java.util.Collection;
  20 import java.util.List;
  21 import java.util.function.BiFunction;
  22 import org.junit.Before;
  23 import org.junit.Rule;
  24 import org.junit.Test;
  25 import org.junit.runner.RunWith;
  26 import org.junit.runners.Parameterized;
  27 import org.junit.runners.Parameterized.Parameters;
  28 import org.mockito.Mock;
  29 import org.mockito.junit.MockitoJUnit;
  30 import org.mockito.junit.MockitoRule;
  31 import org.mockito.quality.Strictness;
  32 import org.opendaylight.mdsal.binding.api.DataBroker;
  33 import org.opendaylight.mdsal.binding.api.TransactionChain;
  34 import org.opendaylight.mdsal.binding.api.TransactionChainListener;
  35 import org.opendaylight.mdsal.binding.dom.codec.spi.BindingDOMCodecServices;
  36 import org.opendaylight.mdsal.dom.api.DOMDataTreeReadTransaction;
  37 import org.opendaylight.mdsal.dom.api.DOMDataTreeReadWriteTransaction;
  38 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
  39 import org.opendaylight.mdsal.dom.api.DOMService;
  40 import org.opendaylight.mdsal.dom.api.DOMTransactionChain;
  41 import org.opendaylight.mdsal.dom.spi.PingPongMergingDOMDataBroker;
  42
  43 @RunWith(Parameterized.class)
  44 public class BindingDOMTransactionChainAdapterTest {
  45     enum TransactionChainType implements BiFunction<DataBroker, TransactionChainListener, TransactionChain> {
  46         NORMAL {
  47             @Override
  48             public TransactionChain apply(final DataBroker broker, final TransactionChainListener listener) {
  49                 return broker.createTransactionChain(listener);
  50             }
  51
  52             @Override
  53             void mockRead(final DOMTransactionChain chain) {
  54                 doReturn(mock(DOMDataTreeReadTransaction.class)).when(chain).newReadOnlyTransaction();
  55             }
  56
  57             @Override
  58             void mockWrite(final DOMTransactionChain chain) {
  59                 doReturn(mock(DOMDataTreeWriteTransaction.class)).when(chain).newWriteOnlyTransaction();
  60             }
  61
  62         },
  63         MERGING {
  64             @Override
  65             public TransactionChain apply(final DataBroker broker, final TransactionChainListener listener) {
  66                 return broker.createMergingTransactionChain(listener);
  67             }
  68
  69             @Override
  70             void mockRead(final DOMTransactionChain chain) {
  71                 mockReadWrite(chain);
  72             }
  73
  74             @Override
  75             void mockWrite(final DOMTransactionChain chain) {
  76                 mockReadWrite(chain);
  77             }
  78         };
  79
  80         abstract void mockRead(DOMTransactionChain chain);
  81
  82         abstract void mockWrite(DOMTransactionChain chain);
  83     }
  84
  85     @Rule
  86     public MockitoRule rule = MockitoJUnit.rule().strictness(Strictness.STRICT_STUBS);
  87
  88     @Parameters
  89     public static Collection<Object[]> parameters() {
  90         return List.of(new Object[] { TransactionChainType.NORMAL }, new Object[] { TransactionChainType.MERGING });
  91     }
  92
  93     private final TransactionChainType type;
  94
  95     @Mock
  96     private PingPongMergingDOMDataBroker domService;
  97     @Mock
  98     private DOMTransactionChain transactionChain;
  99     @Mock
 100     private TransactionChainListener transactionChainListener;
 101     @Mock
 102     private BindingDOMCodecServices mockCodecRegistry;
 103
 104     private BindingDOMTransactionChainAdapter bindingDOMTransactionChainAdapter;
 105
 106     public BindingDOMTransactionChainAdapterTest(final TransactionChainType type) {
 107         this.type = requireNonNull(type);
 108     }
 109
 110     @Before
 111     public void setUp() {
 112         doCallRealMethod().when(domService).getExtensions();
 113         doReturn(transactionChain).when(domService).createTransactionChain(any());
 114         if (type == TransactionChainType.MERGING) {
 115             doCallRealMethod().when(domService).createMergingTransactionChain(any());
 116         }
 117
 118
 119         BindingDOMAdapterLoader bindingDOMAdapterLoader = new BindingDOMAdapterLoader(
 120                 new ConstantAdapterContext(mockCodecRegistry)) {
 121             @Override
 122             protected DOMService getDelegate(final Class<? extends DOMService> reqDeleg) {
 123                 return domService;
 124             }
 125         };
 126
 127         BindingDOMDataBrokerAdapter bindingDOMDataBrokerAdapter =
 128                 (BindingDOMDataBrokerAdapter) bindingDOMAdapterLoader.load(DataBroker.class).get();
 129         bindingDOMTransactionChainAdapter =
 130             (BindingDOMTransactionChainAdapter) type.apply(bindingDOMDataBrokerAdapter, transactionChainListener);
 131         assertNotNull(bindingDOMTransactionChainAdapter.getDelegate());
 132
 133     }
 134
 135     static void mockReadWrite(final DOMTransactionChain chain) {
 136         doReturn(mock(DOMDataTreeReadWriteTransaction.class)).when(chain).newReadWriteTransaction();
 137     }
 138
 139     @Test
 140     public void closeTest() {
 141         doNothing().when(transactionChain).close();
 142         bindingDOMTransactionChainAdapter.close();
 143         verify(transactionChain).close();
 144     }
 145
 146     @Test
 147     public void readTransactionTest() throws Exception {
 148         type.mockRead(transactionChain);
 149         assertNotNull(bindingDOMTransactionChainAdapter.newReadOnlyTransaction());
 150     }
 151
 152     @Test
 153     public void writeTransactionTest() throws Exception {
 154         type.mockWrite(transactionChain);
 155         assertNotNull(bindingDOMTransactionChainAdapter.newWriteOnlyTransaction());
 156     }
 157
 158     @Test
 159     public void readWriteTransactionTest() throws Exception {
 160         mockReadWrite(transactionChain);
 161         assertNotNull(bindingDOMTransactionChainAdapter.newReadWriteTransaction());
 162     }
 163
 164 }