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
9 package org.opendaylight.netconf.sal.connect.netconf.sal;
11 import static org.mockito.Matchers.any;
12 import static org.mockito.Matchers.eq;
13 import static org.mockito.Mockito.doNothing;
14 import static org.mockito.Mockito.doReturn;
15 import static org.mockito.Mockito.times;
16 import static org.mockito.Mockito.verify;
17 import static org.mockito.Mockito.when;
19 import com.google.common.util.concurrent.Futures;
20 import java.net.InetSocketAddress;
21 import org.junit.Before;
22 import org.junit.Test;
23 import org.mockito.ArgumentCaptor;
24 import org.mockito.Mock;
25 import org.mockito.MockitoAnnotations;
26 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
27 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
28 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
29 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
30 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
31 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
32 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
33 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
35 public class NetconfDeviceSalProviderTest {
38 private DOMMountPointService mountpointService;
40 private BindingAwareBroker.ProviderContext context;
42 private WriteTransaction tx;
44 private DataBroker dataBroker;
46 private BindingTransactionChain chain;
48 private DOMMountPointService mountPointService;
50 private WriteTransaction writeTx;
51 private NetconfDeviceSalProvider provider;
54 public void setUp() throws Exception {
55 MockitoAnnotations.initMocks(this);
56 doReturn(chain).when(dataBroker).createTransactionChain(any(TransactionChainListener.class));
57 doReturn(writeTx).when(chain).newWriteOnlyTransaction();
58 doNothing().when(writeTx).merge(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
59 doReturn("Some object").when(writeTx).getIdentifier();
60 doReturn(Futures.immediateCheckedFuture(null)).when(writeTx).submit();
61 provider = new NetconfDeviceSalProvider(new RemoteDeviceId("device1",
62 InetSocketAddress.createUnresolved("localhost", 17830)), mountPointService, dataBroker);
63 when(context.getSALService(DataBroker.class)).thenReturn(dataBroker);
64 when(chain.newWriteOnlyTransaction()).thenReturn(tx);
65 when(tx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
66 when(tx.getIdentifier()).thenReturn(tx);
70 public void replaceChainIfFailed() throws Exception {
71 final ArgumentCaptor<TransactionChainListener> captor = ArgumentCaptor.forClass(TransactionChainListener.class);
72 verify(dataBroker).createTransactionChain(captor.capture());
74 captor.getValue().onTransactionChainFailed(chain, tx, new Exception("chain failed"));
75 } catch (final IllegalStateException e) {
78 verify(dataBroker, times(2)).createTransactionChain(any());
82 public void close() throws Exception {
84 verify(chain).close();
88 public void closeWithoutNPE() throws Exception {
91 verify(chain, times(2)).close();