2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.netconf.sal.connect.netconf.sal;
10 import static org.mockito.ArgumentMatchers.any;
11 import static org.mockito.ArgumentMatchers.eq;
12 import static org.mockito.Mockito.doNothing;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.times;
15 import static org.mockito.Mockito.verify;
16 import static org.mockito.Mockito.when;
17 import static org.opendaylight.mdsal.common.api.CommitInfo.emptyFluentFuture;
19 import java.net.InetSocketAddress;
20 import org.junit.Before;
21 import org.junit.Test;
22 import org.junit.runner.RunWith;
23 import org.mockito.ArgumentCaptor;
24 import org.mockito.Mock;
25 import org.mockito.junit.MockitoJUnitRunner;
26 import org.opendaylight.mdsal.binding.api.DataBroker;
27 import org.opendaylight.mdsal.binding.api.TransactionChain;
28 import org.opendaylight.mdsal.binding.api.TransactionChainListener;
29 import org.opendaylight.mdsal.binding.api.WriteTransaction;
30 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
31 import org.opendaylight.mdsal.dom.api.DOMMountPointService;
32 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
34 @RunWith(MockitoJUnitRunner.StrictStubs.class)
35 public class NetconfDeviceSalProviderTest {
38 private WriteTransaction tx;
40 private DataBroker dataBroker;
42 private TransactionChain chain;
44 private DOMMountPointService mountPointService;
46 private WriteTransaction writeTx;
48 private NetconfDeviceSalProvider provider;
51 public void setUp() throws Exception {
52 doReturn(chain).when(dataBroker).createTransactionChain(any(TransactionChainListener.class));
53 doReturn(writeTx).when(chain).newWriteOnlyTransaction();
54 doNothing().when(writeTx).merge(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
55 doReturn("Some object").when(writeTx).getIdentifier();
56 doReturn(emptyFluentFuture()).when(writeTx).commit();
57 provider = new NetconfDeviceSalProvider(new RemoteDeviceId("device1",
58 InetSocketAddress.createUnresolved("localhost", 17830)), mountPointService, dataBroker);
59 when(chain.newWriteOnlyTransaction()).thenReturn(tx);
60 doReturn(emptyFluentFuture()).when(tx).commit();
61 when(tx.getIdentifier()).thenReturn(tx);
65 public void replaceChainIfFailed() throws Exception {
66 final ArgumentCaptor<TransactionChainListener> captor = ArgumentCaptor.forClass(TransactionChainListener.class);
67 verify(dataBroker).createTransactionChain(captor.capture());
69 captor.getValue().onTransactionChainFailed(chain, tx, new Exception("chain failed"));
70 } catch (final IllegalStateException e) {
73 verify(dataBroker, times(2)).createTransactionChain(any());
77 public void close() throws Exception {
79 verify(chain).close();
83 public void closeWithoutNPE() throws Exception {
86 verify(chain, times(2)).close();