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.controller.sal.core.api.Broker;
34 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
36 public class NetconfDeviceSalProviderTest {
39 private Broker.ProviderSession session;
41 private DOMMountPointService mountpointService;
43 private BindingAwareBroker.ProviderContext context;
45 private WriteTransaction tx;
47 private DataBroker dataBroker;
49 private BindingTransactionChain chain;
51 private DOMMountPointService mountPointService;
53 private WriteTransaction writeTx;
54 private NetconfDeviceSalProvider provider;
57 public void setUp() throws Exception {
58 MockitoAnnotations.initMocks(this);
59 doReturn(chain).when(dataBroker).createTransactionChain(any(TransactionChainListener.class));
60 doReturn(writeTx).when(chain).newWriteOnlyTransaction();
61 doNothing().when(writeTx).merge(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
62 doReturn("Some object").when(writeTx).getIdentifier();
63 doReturn(Futures.immediateCheckedFuture(null)).when(writeTx).submit();
64 provider = new NetconfDeviceSalProvider(new RemoteDeviceId("device1",
65 InetSocketAddress.createUnresolved("localhost", 17830)), mountPointService, dataBroker);
66 when(session.getService(DOMMountPointService.class)).thenReturn(mountpointService);
67 when(context.getSALService(DataBroker.class)).thenReturn(dataBroker);
68 when(chain.newWriteOnlyTransaction()).thenReturn(tx);
69 when(tx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
70 when(tx.getIdentifier()).thenReturn(tx);
74 public void replaceChainIfFailed() throws Exception {
75 final ArgumentCaptor<TransactionChainListener> captor = ArgumentCaptor.forClass(TransactionChainListener.class);
76 verify(dataBroker).createTransactionChain(captor.capture());
78 captor.getValue().onTransactionChainFailed(chain, tx, new Exception("chain failed"));
79 } catch (final IllegalStateException e) {
82 verify(dataBroker, times(2)).createTransactionChain(any());
86 public void close() throws Exception {
88 verify(chain).close();
92 public void closeWithoutNPE() throws Exception {
95 verify(chain, times(2)).close();