2 * Copyright (c) 2017 Brocade Communication Systems 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
8 package org.opendaylight.netconf.sal.connect.netconf.util;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertTrue;
12 import static org.mockito.Mockito.when;
14 import org.junit.Before;
15 import org.junit.Test;
16 import org.mockito.Mock;
17 import org.mockito.MockitoAnnotations;
18 import org.opendaylight.aaa.encrypt.AAAEncryptionService;
19 import org.opendaylight.mdsal.binding.api.DataBroker;
20 import org.opendaylight.netconf.sal.connect.util.NetconfTopologyRPCProvider;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Host;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.CreateDeviceInput;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.CreateDeviceInputBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.Credentials;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPw;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPwBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPwUnencrypted;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPwUnencryptedBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.login.pw.LoginPasswordBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.login.pw.unencrypted.LoginPasswordUnencryptedBuilder;
35 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
36 import org.opendaylight.yangtools.yang.common.Uint16;
38 public class NetconfTopologyRPCProviderTest {
39 private static final NodeId NODE_ID = new NodeId("testing-node");
40 private static final String TOPOLOGY_ID = "testing-topology";
41 private static final String TEST_PWD = "test";
42 private static final String ENC_PWD = "4o9/Hn3Pi4150YrP12N/1g==";
45 private DataBroker dataBroker;
48 private AAAEncryptionService encryptionService;
50 private NetconfTopologyRPCProvider rpcProvider ;
54 MockitoAnnotations.initMocks(this);
55 when(encryptionService.encrypt(TEST_PWD)).thenReturn(ENC_PWD);
56 rpcProvider = new NetconfTopologyRPCProvider(dataBroker, encryptionService, TOPOLOGY_ID);
60 public void testEncryptPassword() throws Exception {
62 final NetconfNode encryptedPwNode = rpcProvider.encryptPassword(getInput(true));
64 final Credentials credentials = encryptedPwNode.getCredentials();
65 assertTrue(credentials instanceof LoginPw);
66 final LoginPw loginPw = (LoginPw) credentials;
68 assertEquals(ENC_PWD, loginPw.getLoginPassword().getPassword());
72 public void testNoEncryption() throws Exception {
73 final NetconfNode encryptedPwNode = rpcProvider.encryptPassword(getInput(false));
75 final Credentials credentials = encryptedPwNode.getCredentials();
76 assertTrue(credentials instanceof LoginPwUnencrypted);
77 final LoginPwUnencrypted loginPw = (LoginPwUnencrypted) credentials;
79 assertEquals(TEST_PWD, loginPw.getLoginPasswordUnencrypted().getPassword());
82 private static CreateDeviceInput getInput(final boolean encrypt) {
83 CreateDeviceInputBuilder builder = new CreateDeviceInputBuilder();
84 final Credentials credentials;
86 credentials = new LoginPwBuilder().setLoginPassword(
87 new LoginPasswordBuilder().setUsername("test").setPassword(TEST_PWD).build()).build();
89 credentials = new LoginPwUnencryptedBuilder().setLoginPasswordUnencrypted(
90 new LoginPasswordUnencryptedBuilder().setUsername("test").setPassword(TEST_PWD).build()).build();
93 builder.setCredentials(credentials);
94 builder.setHost(new Host(new IpAddress(new Ipv4Address("10.18.16.188"))));
95 builder.setPort(new PortNumber(Uint16.valueOf(830)));
96 builder.setTcpOnly(Boolean.FALSE);
97 builder.setNodeId(NODE_ID.toString());
98 return builder.build();