5d0b3f3199d8cbd3417a6bb5da830428e3402155
[mdsal.git] / 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 org.junit.Assert.assertNotNull;
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.Mockito.doNothing;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.verify;
16
17 import com.google.common.collect.ImmutableClassToInstanceMap;
18 import org.junit.Before;
19 import org.junit.Test;
20 import org.junit.runner.RunWith;
21 import org.mockito.Mock;
22 import org.mockito.junit.MockitoJUnitRunner;
23 import org.opendaylight.mdsal.binding.api.DataBroker;
24 import org.opendaylight.mdsal.binding.api.TransactionChainListener;
25 import org.opendaylight.mdsal.binding.dom.codec.spi.BindingDOMCodecServices;
26 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
27 import org.opendaylight.mdsal.dom.api.DOMDataTreeReadTransaction;
28 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
29 import org.opendaylight.mdsal.dom.api.DOMService;
30 import org.opendaylight.mdsal.dom.api.DOMTransactionChain;
31
32 @RunWith(MockitoJUnitRunner.StrictStubs.class)
33 public class BindingDOMTransactionChainAdapterTest {
34
35     @Mock
36     private DOMDataBroker domService;
37
38     @Mock
39     private DOMTransactionChain transactionChain;
40
41     @Mock
42     private TransactionChainListener transactionChainListener;
43
44     @Mock
45     private BindingDOMCodecServices mockCodecRegistry;
46
47     private BindingDOMTransactionChainAdapter bindingDOMTransactionChainAdapter;
48
49     @Before
50     public void setUp() {
51         doReturn(transactionChain).when(domService).createTransactionChain(any());
52         doReturn(ImmutableClassToInstanceMap.of()).when(domService).getExtensions();
53
54         BindingDOMAdapterLoader bindingDOMAdapterLoader = new BindingDOMAdapterLoader(
55                 new ConstantAdapterContext(mockCodecRegistry)) {
56             @Override
57             protected DOMService getDelegate(final Class<? extends DOMService> reqDeleg) {
58                 return domService;
59             }
60         };
61
62         BindingDOMDataBrokerAdapter bindingDOMDataBrokerAdapter =
63                 (BindingDOMDataBrokerAdapter) bindingDOMAdapterLoader.load(DataBroker.class).get();
64         bindingDOMTransactionChainAdapter = (BindingDOMTransactionChainAdapter) bindingDOMDataBrokerAdapter
65                 .createTransactionChain(transactionChainListener);
66         assertNotNull(bindingDOMTransactionChainAdapter.getDelegate());
67         doNothing().when(transactionChain).close();
68         bindingDOMTransactionChainAdapter.close();
69         verify(transactionChain).close();
70     }
71
72     @Test
73     public void readTransactionTest() throws Exception {
74         doReturn(mock(DOMDataTreeReadTransaction.class)).when(transactionChain).newReadOnlyTransaction();
75         assertNotNull(bindingDOMTransactionChainAdapter.newReadOnlyTransaction());
76     }
77
78     @Test
79     public void writeTransactionTest() throws Exception {
80         doReturn(mock(DOMDataTreeWriteTransaction.class)).when(transactionChain).newWriteOnlyTransaction();
81         assertNotNull(bindingDOMTransactionChainAdapter.newWriteOnlyTransaction());
82     }
83 }