Move transaction chain handling
[netconf.git] / netconf / sal-netconf-connector / src / test / java / org / opendaylight / netconf / sal / connect / netconf / sal / NetconfDeviceSalProviderTest.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.netconf.sal.connect.netconf.sal;
9
10 import static org.mockito.Mockito.doReturn;
11 import static org.mockito.Mockito.verify;
12
13 import java.net.InetSocketAddress;
14 import org.junit.Before;
15 import org.junit.Test;
16 import org.junit.runner.RunWith;
17 import org.mockito.ArgumentCaptor;
18 import org.mockito.Captor;
19 import org.mockito.Mock;
20 import org.mockito.junit.MockitoJUnitRunner;
21 import org.opendaylight.mdsal.binding.api.DataBroker;
22 import org.opendaylight.mdsal.binding.api.TransactionChain;
23 import org.opendaylight.mdsal.binding.api.TransactionChainListener;
24 import org.opendaylight.mdsal.binding.api.WriteTransaction;
25 import org.opendaylight.mdsal.common.api.CommitInfo;
26 import org.opendaylight.mdsal.dom.api.DOMMountPointService;
27 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
28
29 @RunWith(MockitoJUnitRunner.StrictStubs.class)
30 public class NetconfDeviceSalProviderTest {
31     @Mock
32     private WriteTransaction tx;
33     @Mock
34     private DataBroker dataBroker;
35     @Mock
36     private TransactionChain chain;
37     @Mock
38     private DOMMountPointService mountPointService;
39     @Mock
40     private WriteTransaction writeTx;
41     @Captor
42     private ArgumentCaptor<TransactionChainListener> listeners;
43
44     private NetconfDeviceSalProvider provider;
45
46     @Before
47     public void setUp() {
48         doReturn(chain).when(dataBroker).createMergingTransactionChain(listeners.capture());
49         doReturn(writeTx).when(chain).newWriteOnlyTransaction();
50         doReturn("Some object").when(writeTx).getIdentifier();
51         doReturn(CommitInfo.emptyFluentFuture()).when(writeTx).commit();
52         provider = new NetconfDeviceSalProvider(new RemoteDeviceId("device1",
53                 InetSocketAddress.createUnresolved("localhost", 17830)), mountPointService, dataBroker);
54         doReturn(tx).when(chain).newWriteOnlyTransaction();
55         doReturn(CommitInfo.emptyFluentFuture()).when(tx).commit();
56         doReturn(tx).when(tx).getIdentifier();
57     }
58
59     @Test
60     public void close() {
61         listeners.getValue().onTransactionChainSuccessful(chain);
62         provider.close();
63         verify(chain).close();
64     }
65
66     @Test
67     public void closeWithoutNPE()  {
68         close();
69
70         // No further interations
71         provider.close();
72     }
73 }