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.Mockito.times;
13 import static org.mockito.Mockito.verify;
14 import static org.mockito.Mockito.when;
16 import com.google.common.util.concurrent.Futures;
17 import java.net.InetSocketAddress;
18 import org.junit.Assert;
19 import org.junit.Before;
20 import org.junit.Test;
21 import org.mockito.ArgumentCaptor;
22 import org.mockito.Mock;
23 import org.mockito.MockitoAnnotations;
24 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
25 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
26 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
27 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
28 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
29 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
30 import org.opendaylight.controller.sal.core.api.Broker;
31 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
33 public class NetconfDeviceSalProviderTest {
36 private Broker.ProviderSession session;
38 private DOMMountPointService mountpointService;
40 private BindingAwareBroker.ProviderContext context;
42 private WriteTransaction tx;
44 private DataBroker dataBroker;
46 private BindingTransactionChain chain;
47 private NetconfDeviceSalProvider provider;
50 public void setUp() throws Exception {
51 MockitoAnnotations.initMocks(this);
52 provider = new NetconfDeviceSalProvider(new RemoteDeviceId("device1", InetSocketAddress.createUnresolved("localhost", 17830)));
53 when(session.getService(DOMMountPointService.class)).thenReturn(mountpointService);
54 when(context.getSALService(DataBroker.class)).thenReturn(dataBroker);
55 when(dataBroker.createTransactionChain(any())).thenReturn(chain);
56 when(chain.newWriteOnlyTransaction()).thenReturn(tx);
57 when(tx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
58 when(tx.getIdentifier()).thenReturn(tx);
62 public void onSessionInitiated() throws Exception {
63 provider.onSessionInitiated(session);
64 provider.onSessionInitiated(context);
65 Assert.assertNotNull(provider.getMountInstance());
66 Assert.assertNotNull(provider.getTopologyDatastoreAdapter());
70 public void getProviderFunctionality() throws Exception {
71 Assert.assertTrue(provider.getProviderFunctionality().isEmpty());
75 public void replaceChainIfFailed() throws Exception {
76 provider.onSessionInitiated(session);
77 provider.onSessionInitiated(context);
78 Assert.assertNotNull(provider.getMountInstance());
79 final ArgumentCaptor<TransactionChainListener> captor = ArgumentCaptor.forClass(TransactionChainListener.class);
80 verify(dataBroker).createTransactionChain(captor.capture());
82 captor.getValue().onTransactionChainFailed(chain, tx, new Exception("chain failed"));
83 } catch (final IllegalStateException e) {
86 verify(dataBroker, times(2)).createTransactionChain(any());
90 public void close() throws Exception {
91 provider.onSessionInitiated(session);
92 provider.onSessionInitiated(context);
94 verify(chain).close();