Create NetconfDataTreeService with base and additional operations for netconf
[netconf.git] / netconf / sal-netconf-connector / src / test / java / org / opendaylight / netconf / sal / connect / netconf / sal / NetconfDeviceSalFacadeTest.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
9 package org.opendaylight.netconf.sal.connect.netconf.sal;
10
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.ArgumentMatchers.eq;
13 import static org.mockito.ArgumentMatchers.isNull;
14 import static org.mockito.Mockito.doNothing;
15 import static org.mockito.Mockito.doReturn;
16 import static org.mockito.Mockito.mock;
17 import static org.mockito.Mockito.times;
18 import static org.mockito.Mockito.verify;
19 import static org.mockito.MockitoAnnotations.initMocks;
20
21 import java.net.InetSocketAddress;
22 import java.util.Arrays;
23 import java.util.List;
24 import org.junit.Before;
25 import org.junit.Test;
26 import org.mockito.Mock;
27 import org.opendaylight.mdsal.binding.api.DataBroker;
28 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
29 import org.opendaylight.mdsal.dom.api.DOMNotification;
30 import org.opendaylight.mdsal.dom.api.DOMRpcService;
31 import org.opendaylight.netconf.dom.api.NetconfDataTreeService;
32 import org.opendaylight.netconf.sal.connect.netconf.listener.NetconfDeviceCapabilities;
33 import org.opendaylight.netconf.sal.connect.netconf.listener.NetconfSessionPreferences;
34 import org.opendaylight.netconf.sal.connect.netconf.util.NetconfMessageTransformUtil;
35 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
36 import org.opendaylight.yangtools.rcf8528.data.util.EmptyMountPointContext;
37 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
38
39 public class NetconfDeviceSalFacadeTest {
40
41     private NetconfDeviceSalFacade deviceFacade;
42
43     @Mock
44     private NetconfDeviceTopologyAdapter netconfDeviceTopologyAdapter;
45     @Mock
46     private NetconfDeviceSalProvider.MountInstance mountInstance;
47     @Mock
48     private NetconfDeviceSalProvider salProvider;
49     @Mock
50     private DataBroker dataBroker;
51
52     @Before
53     public void setUp() throws Exception {
54         initMocks(this);
55         final InetSocketAddress address = new InetSocketAddress("127.0.0.1", 8000);
56         final RemoteDeviceId remoteDeviceId = new RemoteDeviceId("test", address);
57
58         deviceFacade = new NetconfDeviceSalFacade(remoteDeviceId, salProvider, dataBroker, "mockTopo");
59
60         doReturn(netconfDeviceTopologyAdapter).when(salProvider).getTopologyDatastoreAdapter();
61         doNothing().when(netconfDeviceTopologyAdapter)
62                 .updateDeviceData(any(Boolean.class), any(NetconfDeviceCapabilities.class));
63
64         doReturn(mountInstance).when(salProvider).getMountInstance();
65         doNothing().when(mountInstance).onTopologyDeviceDisconnected();
66     }
67
68     @Test
69     public void testOnDeviceDisconnected() {
70         deviceFacade.onDeviceDisconnected();
71
72         verify(netconfDeviceTopologyAdapter).updateDeviceData(eq(false), any(NetconfDeviceCapabilities.class));
73         verify(mountInstance, times(1)).onTopologyDeviceDisconnected();
74
75     }
76
77     @Test
78     public void testOnDeviceFailed() {
79         final Throwable throwable = new Throwable();
80         deviceFacade.onDeviceFailed(throwable);
81
82         verify(netconfDeviceTopologyAdapter).setDeviceAsFailed(throwable);
83         verify(mountInstance, times(1)).onTopologyDeviceDisconnected();
84     }
85
86     @Test
87     public void testOnDeviceClose() throws Exception {
88         deviceFacade.close();
89         verify(salProvider).close();
90     }
91
92     @Test
93     public void testOnDeviceConnected() {
94         final EffectiveModelContext schemaContext = mock(EffectiveModelContext.class);
95
96         final NetconfSessionPreferences netconfSessionPreferences =
97                 NetconfSessionPreferences.fromStrings(getCapabilities());
98
99         final DOMRpcService deviceRpc = mock(DOMRpcService.class);
100         deviceFacade.onDeviceConnected(new EmptyMountPointContext(schemaContext), netconfSessionPreferences, deviceRpc,
101             null);
102
103         verify(mountInstance, times(1)).onTopologyDeviceConnected(eq(schemaContext),
104                 any(DOMDataBroker.class), any(NetconfDataTreeService.class), eq(deviceRpc),
105                 any(NetconfDeviceNotificationService.class), isNull());
106         verify(netconfDeviceTopologyAdapter,
107                 times(1)).updateDeviceData(true, netconfSessionPreferences.getNetconfDeviceCapabilities());
108     }
109
110     @Test
111     public void testOnDeviceNotification() throws Exception {
112         final DOMNotification domNotification = mock(DOMNotification.class);
113         deviceFacade.onNotification(domNotification);
114         verify(mountInstance).publish(domNotification);
115     }
116
117     private static List<String> getCapabilities() {
118         return Arrays.asList(NetconfMessageTransformUtil.NETCONF_CANDIDATE_URI.toString());
119     }
120 }