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;
11 import static org.mockito.Matchers.any;
12 import static org.mockito.Mockito.doNothing;
13 import static org.mockito.Mockito.mock;
14 import static org.mockito.Mockito.verify;
16 import com.google.common.collect.Lists;
17 import java.net.InetSocketAddress;
18 import java.util.ArrayList;
19 import java.util.Collection;
20 import org.junit.Test;
21 import org.mockito.Mockito;
22 import org.opendaylight.controller.md.sal.dom.api.DOMNotification;
23 import org.opendaylight.controller.md.sal.dom.api.DOMRpcService;
24 import org.opendaylight.netconf.api.NetconfMessage;
25 import org.opendaylight.netconf.api.xml.XmlNetconfConstants;
26 import org.opendaylight.netconf.sal.connect.api.RemoteDeviceHandler;
27 import org.opendaylight.netconf.sal.connect.netconf.listener.NetconfDeviceCommunicator;
28 import org.opendaylight.netconf.sal.connect.netconf.listener.NetconfSessionPreferences;
29 import org.opendaylight.netconf.sal.connect.netconf.sal.NetconfDeviceRpc;
30 import org.opendaylight.netconf.sal.connect.netconf.schema.mapping.SchemalessMessageTransformer;
31 import org.opendaylight.netconf.sal.connect.netconf.util.NetconfMessageTransformUtil;
32 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
33 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
35 public class SchemalessNetconfDeviceTest {
37 private static final String TEST_NAMESPACE = "test:namespace";
38 private static final String TEST_MODULE = "test-module";
39 private static final String TEST_REVISION = "2013-07-22";
42 public void testSessionOnMethods() throws Exception {
43 final RemoteDeviceHandler<NetconfSessionPreferences> facade = getFacade();
44 final NetconfDeviceCommunicator listener = mockCloseableClass(NetconfDeviceCommunicator.class);
45 final SchemalessMessageTransformer messageTransformer = mock(SchemalessMessageTransformer.class);
46 final RemoteDeviceId remoteDeviceId = new RemoteDeviceId("test-D",
47 InetSocketAddress.createUnresolved("localhost", 22));
48 final Throwable throwable = new Throwable();
50 final SchemalessNetconfDevice device = new SchemalessNetconfDevice(remoteDeviceId, facade, messageTransformer);
52 final NetconfSessionPreferences sessionCaps = getSessionCaps(true,
53 Lists.newArrayList(TEST_NAMESPACE + "?module=" + TEST_MODULE + "&revision=" + TEST_REVISION));
55 final NetconfMessage netconfMessage = mock(NetconfMessage.class);
57 device.onRemoteSessionUp(sessionCaps, listener);
58 verify(facade).onDeviceConnected(any(SchemaContext.class), any(NetconfSessionPreferences.class), any(DOMRpcService.class));
60 device.onNotification(netconfMessage);
61 verify(facade).onNotification(any(DOMNotification.class));
63 device.onRemoteSessionDown();
64 verify(facade).onDeviceDisconnected();
66 device.onRemoteSessionFailed(throwable);
67 verify(facade).onDeviceFailed(throwable);
70 @SuppressWarnings("unchecked")
71 private static RemoteDeviceHandler<NetconfSessionPreferences> getFacade() throws Exception {
72 final RemoteDeviceHandler<NetconfSessionPreferences> remoteDeviceHandler = mockCloseableClass(RemoteDeviceHandler.class);
73 doNothing().when(remoteDeviceHandler).onDeviceConnected(any(SchemaContext.class), any(NetconfSessionPreferences.class), any(NetconfDeviceRpc.class));
74 doNothing().when(remoteDeviceHandler).onDeviceDisconnected();
75 doNothing().when(remoteDeviceHandler).onNotification(any(DOMNotification.class));
76 return remoteDeviceHandler;
79 private static <T extends AutoCloseable> T mockCloseableClass(final Class<T> remoteDeviceHandlerClass) throws Exception {
80 final T mock = mockClass(remoteDeviceHandlerClass);
81 doNothing().when(mock).close();
85 private static <T> T mockClass(final Class<T> remoteDeviceHandlerClass) {
86 final T mock = mock(remoteDeviceHandlerClass);
87 Mockito.doReturn(remoteDeviceHandlerClass.getSimpleName()).when(mock).toString();
91 private static NetconfSessionPreferences getSessionCaps(final boolean addMonitor, final Collection<String> additionalCapabilities) {
92 final ArrayList<String> capabilities = Lists.newArrayList(
93 XmlNetconfConstants.URN_IETF_PARAMS_NETCONF_BASE_1_0,
94 XmlNetconfConstants.URN_IETF_PARAMS_NETCONF_BASE_1_1);
97 capabilities.add(NetconfMessageTransformUtil.IETF_NETCONF_MONITORING.getNamespace().toString());
100 capabilities.addAll(additionalCapabilities);
102 return NetconfSessionPreferences.fromStrings(