summaryrefslogtreecommitdiffstats
path: root/appc-config/appc-config-adaptor
diff options
context:
space:
mode:
Diffstat (limited to 'appc-config/appc-config-adaptor')
-rw-r--r--appc-config/appc-config-adaptor/provider/src/main/java/org/onap/appc/ccadaptor/SshJcraftWrapper.java310
-rw-r--r--appc-config/appc-config-adaptor/provider/src/test/java/org/onap/appc/ccadaptor/SshJcraftWrapperTest.java663
2 files changed, 714 insertions, 259 deletions
diff --git a/appc-config/appc-config-adaptor/provider/src/main/java/org/onap/appc/ccadaptor/SshJcraftWrapper.java b/appc-config/appc-config-adaptor/provider/src/main/java/org/onap/appc/ccadaptor/SshJcraftWrapper.java
index 28f00069a..d4deb1fa5 100644
--- a/appc-config/appc-config-adaptor/provider/src/main/java/org/onap/appc/ccadaptor/SshJcraftWrapper.java
+++ b/appc-config/appc-config-adaptor/provider/src/main/java/org/onap/appc/ccadaptor/SshJcraftWrapper.java
@@ -5,6 +5,8 @@
* Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
* ================================================================================
* Copyright (C) 2017 Amdocs
+ * ================================================================================
+ * Modifications Copyright (C) 2018 Ericsson
* =============================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -23,8 +25,6 @@
package org.onap.appc.ccadaptor;
-import com.att.eelf.configuration.EELFLogger;
-import com.att.eelf.configuration.EELFManager;
import com.jcraft.jsch.Channel;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.ChannelShell;
@@ -54,10 +54,7 @@ import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.StringTokenizer;
-import java.util.concurrent.TimeUnit;
-import javax.annotation.Nonnull;
import org.apache.commons.lang.StringUtils;
-import org.onap.appc.i18n.Msg;
public class SshJcraftWrapper {
@@ -71,7 +68,6 @@ public class SshJcraftWrapper {
private String RouterName = null;
private int BUFFER_SIZE = 512000;
char[] charBuffer = new char[BUFFER_SIZE];
- // private int BUFFER_SIZE = 4000000;
private DataInputStream dis = null;
private BufferedReader reader = null;
private BufferedWriter out = null;
@@ -93,21 +89,20 @@ public class SshJcraftWrapper {
private Runtime runtime = Runtime.getRuntime();
private DebugLog dbLog = new DebugLog();
- public void SshJcraftWrapper() {
+ public SshJcraftWrapper() {
String fn = "SshJcraftWrapper.SshJcraftWrapper";
- debugLog.printRTAriDebug(fn, "SshJcraftWrapper has been instantated");
+ DebugLog.printRTAriDebug(fn, "SshJcraftWrapper has been instantated");
routerLogFileName = "/tmp/" + host;
- this.host = host;
}
public void connect(String hostname, String username, String password, String prompt, int timeOut)
throws IOException {
String fn = "SshJcraftWrapper.connect";
- jsch = new JSch();
- debugLog.printRTAriDebug(fn,
+ jsch = getJSch();
+ DebugLog.printRTAriDebug(fn,
"Attempting to connect to " + hostname + " username=" + username + " prompt='"
+ prompt + "' timeOut=" + timeOut);
- debugLog.printRTAriDebug(fn, "Trace A");
+ DebugLog.printRTAriDebug(fn, "Trace A");
RouterName = hostname;
hostName = hostname;
userName = username;
@@ -126,16 +121,15 @@ public class SshJcraftWrapper {
dis = new DataInputStream(inputStream);
reader = new BufferedReader(new InputStreamReader(dis), BUFFER_SIZE);
channel.connect();
- debugLog.printRTAriDebug(fn, "Successfully connected.");
- debugLog.printRTAriDebug(fn, "Flushing input buffer");
+ DebugLog.printRTAriDebug(fn, "Successfully connected.");
+ DebugLog.printRTAriDebug(fn, "Flushing input buffer");
try {
receiveUntil(prompt, 3000, "No cmd was sent, just waiting");
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception: Nothing to flush out.");
+ DebugLog.printRTAriDebug(fn, "Caught an Exception: Nothing to flush out.");
}
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception. e=" + e);
- // dbLog.storeData("ErrorMsg= Exception trying to connect to "+hostname +" "+e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception. e=" + e);
throw new IOException(e.toString());
}
}
@@ -144,7 +138,7 @@ public class SshJcraftWrapper {
public void connect(String hostname, String username, String password, String prompt, int timeOut, int portNum)
throws IOException {
String fn = "SshJcraftWrapper.connect";
- debugLog.printRTAriDebug(fn,
+ DebugLog.printRTAriDebug(fn,
":Attempting to connect to " + hostname + " username=" + username + " prompt='"
+ prompt + "' timeOut=" + timeOut + " portNum=" + portNum);
RouterName = hostname;
@@ -152,14 +146,14 @@ public class SshJcraftWrapper {
userName = username;
passWord = password;
RouterName = hostname;
- jsch = new JSch();
+ jsch = getJSch();
try {
session = jsch.getSession(username, hostname, portNum);
UserInfo ui = new MyUserInfo();
session.setPassword(password);
session.setUserInfo(ui);
session.setConfig("StrictHostKeyChecking", "no");
- debugLog.printRTAriDebug(fn, ":StrictHostKeyChecking set to 'no'");
+ DebugLog.printRTAriDebug(fn, ":StrictHostKeyChecking set to 'no'");
session.connect(timeOut);
session.setServerAliveCountMax(
@@ -170,8 +164,8 @@ public class SshJcraftWrapper {
dis = new DataInputStream(inputStream);
reader = new BufferedReader(new InputStreamReader(dis), BUFFER_SIZE);
channel.connect();
- debugLog.printRTAriDebug(fn, ":Successfully connected.");
- debugLog.printRTAriDebug(fn, ":Flushing input buffer");
+ DebugLog.printRTAriDebug(fn, ":Successfully connected.");
+ DebugLog.printRTAriDebug(fn, ":Flushing input buffer");
try {
if (prompt.equals("]]>]]>")) {
receiveUntil("]]>]]>", 10000, "No cmd was sent, just waiting");
@@ -179,13 +173,12 @@ public class SshJcraftWrapper {
receiveUntil(":~#", 5000, "No cmd was sent, just waiting");
}
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception::: Nothing to flush out.");
+ DebugLog.printRTAriDebug(fn, "Caught an Exception::: Nothing to flush out.");
}
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, ":Caught an Exception. e=" + e);
+ DebugLog.printRTAriDebug(fn, ":Caught an Exception. e=" + e);
dbLog.outputStackTrace(e);
- // dbLog.storeData("ErrorMsg= Exception trying to connect to "+hostname +" "+e);
throw new IOException(e.toString());
}
}
@@ -198,7 +191,7 @@ public class SshJcraftWrapper {
boolean cliPromptCmd = false;
StringBuffer sb2 = new StringBuffer();
StringBuffer sbReceive = new StringBuffer();
- debugLog.printRTAriDebug(fn,
+ DebugLog.printRTAriDebug(fn,
"delimeters='" + delimeters + "' timeout=" + timeout + " cmdThatWasSent='" + cmdThatWasSent + "'");
appendToFile(debugLogFileName,
fn + " delimeters='" + delimeters + "' timeout=" + timeout + " cmdThatWasSent='" + cmdThatWasSent + "'\n");
@@ -212,20 +205,20 @@ public class SshJcraftWrapper {
session.setTimeout(timeout); // This is the socket timeout value.
while (!match) {
if (new Date().getTime() > deadline) {
- debugLog.printRTAriDebug(fn,
+ DebugLog.printRTAriDebug(fn,
"Throwing a TimedOutException: time in routine has exceed our deadline: RouterName:"
+ RouterName + " CmdThatWasSent=" + CmdThatWasSent);
throw new TimedOutException("Timeout: time in routine has exceed our deadline");
}
try {
- Thread.sleep(500);
+ delay(500);
} catch (java.lang.InterruptedException ee) {
- boolean ignore = true;
+ Thread.currentThread().interrupt();
}
int len = reader.read(charBuffer, 0, BUFFER_SIZE);
appendToFile(debugLogFileName, fn + " After reader.read cmd: len=" + len + "\n");
if (len <= 0) {
- debugLog.printRTAriDebug(fn,
+ DebugLog.printRTAriDebug(fn,
"Reader read " + len + " bytes. Looks like we timed out, router=" + RouterName);
throw new TimedOutException("Received a SocketTimeoutException router=" + RouterName);
}
@@ -235,7 +228,7 @@ public class SshJcraftWrapper {
// This is a IOS XR sw config file. We will write it to the disk.
timeout = timeout * 2;
deadline = new Date().getTime() + timeout;
- debugLog.printRTAriDebug(fn, "IOS XR upload for software config: timeout=" + timeout);
+ DebugLog.printRTAriDebug(fn, "IOS XR upload for software config: timeout=" + timeout);
StringTokenizer st = new StringTokenizer(cmdThatWasSent);
st.nextToken();
routerFileName = st.nextToken();
@@ -243,10 +236,9 @@ public class SshJcraftWrapper {
out = new BufferedWriter(fileWriter);
routerLogFileName = "/tmp/" + RouterName;
_tmpFile = new File(routerLogFileName);
- debugLog.printRTAriDebug(fn,
+ DebugLog.printRTAriDebug(fn,
"Will write the swConfigFile to disk, routerFileName=" + routerFileName);
}
- int c;
out.write(charBuffer, 0, len);
out.flush();
appendToFile(debugLogFileName, fn + " Wrote " + len + " bytes to the disk\n");
@@ -286,31 +278,14 @@ public class SshJcraftWrapper {
break;
}
}
- } else {
- debugLog.printRTAriDebug(fn, "cliPromptCmd, Trace 2");
- sb2.setLength(0);
- for (int i = 0; i < len; i++) {
- sbReceive.append((char) charBuffer[i]);
- sb2.append((char) charBuffer[i]);
- }
- appendToRouterFile("/tmp/" + RouterName, sb2);
- if (listener != null) {
- listener.receivedString(sb2.toString());
- }
- debugLog.printRTAriDebug(fn, "sb2='" + sb2.toString() + "' delimeters='" + delimeters + "'");
- if (sb2.toString().indexOf("\nariPrompt>") != -1) {
- debugLog.printRTAriDebug(fn, "Found our prompt");
- match = true;
- break;
- }
}
}
} catch (JSchException e) {
- debugLog.printRTAriDebug(fn, "Caught an JSchException e=" + e.toString());
+ DebugLog.printRTAriDebug(fn, "Caught an JSchException e=" + e.toString());
dbLog.outputStackTrace(e);
throw new TimedOutException(e.toString());
} catch (IOException ee) {
- debugLog.printRTAriDebug(fn, "Caught an IOException: ee=" + ee.toString());
+ DebugLog.printRTAriDebug(fn, "Caught an IOException: ee=" + ee.toString());
dbLog.outputStackTrace(ee);
throw new TimedOutException(ee.toString());
} finally {
@@ -319,11 +294,11 @@ public class SshJcraftWrapper {
fileWriter.close();
}
} catch(IOException ex) {
- debugLog.printRTAriDebug(fn, "Failed to close fileWriter output stream: ex=" + ex);
+ DebugLog.printRTAriDebug(fn, "Failed to close fileWriter output stream: ex=" + ex);
}
}
String result = stripOffCmdFromRouterResponse(sbReceive.toString());
- debugLog.printRTAriDebug(fn, "Leaving method successfully");
+ DebugLog.printRTAriDebug(fn, "Leaving method successfully");
return result;
}
@@ -372,7 +347,7 @@ public class SshJcraftWrapper {
appendToFile(debugLogFileName, fn + " Looking for an delimeter of:'" + delimeter + "'\n");
appendToFile(debugLogFileName, fn + " receivedString='" + receivedString);
if (aggregatedReceivedString.indexOf(delimeter) != -1) {
- debugLog.printRTAriDebug(fn, "Found our delimeter, which was: '" + delimeter + "'");
+ DebugLog.printRTAriDebug(fn, "Found our delimeter, which was: '" + delimeter + "'");
aggregatedReceivedString = "";
return (true);
}
@@ -400,7 +375,6 @@ public class SshJcraftWrapper {
e.printStackTrace();
}
} else {
- // DebugLog.printAriDebug(fnName, "TRACE 1: ******************************");
// When looking at the end of the charBuffer, don't include any linefeeds or spaces. We only want to make the smallest string possible.
for (x = len - 1; x >= 0; x--) {
c = charBuffer[x];
@@ -447,7 +421,7 @@ public class SshJcraftWrapper {
public void closeConnection() {
String fn = "SshJcraftWrapper.closeConnection";
- debugLog.printRTAriDebug(fn, "Executing the closeConnection....");
+ DebugLog.printRTAriDebug(fn, "Executing the closeConnection....");
inputStream = null;
outputStream = null;
dis = null;
@@ -459,8 +433,7 @@ public class SshJcraftWrapper {
public void send(String cmd) throws IOException {
String fn = "SshJcraftWrapper.send";
OutputStream out = channel.getOutputStream();
- DataOutputStream dos = new DataOutputStream(out);
-
+ DataOutputStream dos = getDataOutputStream(out);
if ((cmd.charAt(cmd.length() - 1) != '\n') && (cmd.charAt(cmd.length() - 1) != '\r')) {
cmd += "\n";
}
@@ -471,8 +444,7 @@ public class SshJcraftWrapper {
int ncharsSent = 0;
appendToFile(debugLogFileName, fn + ": Sending: '" + cmd);
- // debugLog.printRTAriDebug (fn, "cmd = "+cmd);
- debugLog.printRTAriDebug(fn, "Length of cmd is:" + length); // 2,937,706
+ DebugLog.printRTAriDebug(fn, "Length of cmd is:" + length); // 2,937,706
try {
if (length > 600000) {
int timeout = 9000;
@@ -480,30 +452,30 @@ public class SshJcraftWrapper {
String Cmd = cmd.substring(i, Math.min(length, i + nchars));
ncharsSent = Cmd.length();
ncharsTotalSent = ncharsTotalSent + Cmd.length();
- debugLog.printRTAriDebug(fn, "i=" + i + " Sending Cmd: ncharsSent=" + ncharsSent);
+ DebugLog.printRTAriDebug(fn, "i=" + i + " Sending Cmd: ncharsSent=" + ncharsSent);
dos.writeBytes(Cmd);
dos.flush();
try {
- debugLog.printRTAriDebug(fn, ":::i=" + i + " length=" + length);
+ DebugLog.printRTAriDebug(fn, ":::i=" + i + " length=" + length);
if (ncharsSent < length) {
receiveUntilBufferFlush(ncharsSent, timeout, "buffer flush i=" + i);
} else {
- debugLog.printRTAriDebug(fn, "i=" + i + " No Waiting this time....");
+ DebugLog.printRTAriDebug(fn, "i=" + i + " No Waiting this time....");
dos.flush();
}
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception: Nothing to flush out.");
+ DebugLog.printRTAriDebug(fn, "Caught an Exception: Nothing to flush out.");
}
}
} else {
- debugLog.printRTAriDebug(fn, "Before executing the dos.writeBytes");
+ DebugLog.printRTAriDebug(fn, "Before executing the dos.writeBytes");
dos.writeBytes(cmd);
}
dos.flush();
- debugLog.printRTAriDebug(fn, "Leaving method");
+ DebugLog.printRTAriDebug(fn, "Leaving method");
appendToFile(debugLogFileName, fn + ": Leaving method\n");
} catch (IOException e) {
- debugLog.printRTAriDebug(fn, "Caught an IOException. e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an IOException. e=" + e);
dbLog.outputStackTrace(e);
throw new IOException(e.toString());
}
@@ -513,13 +485,13 @@ public class SshJcraftWrapper {
public void sendChar(int v) throws IOException {
String fn = "SshJcraftWrapper.sendChar";
OutputStream out = channel.getOutputStream();
- DataOutputStream dos = new DataOutputStream(out);
+ DataOutputStream dos = getDataOutputStream(out);
try {
- debugLog.printRTAriDebug(fn, "Sending: '" + v + "'");
+ DebugLog.printRTAriDebug(fn, "Sending: '" + v + "'");
dos.writeChar(v);
dos.flush();
} catch (IOException e) {
- debugLog.printRTAriDebug(fn, "Caught an IOException. e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an IOException. e=" + e);
throw new IOException(e.toString());
}
}
@@ -527,12 +499,12 @@ public class SshJcraftWrapper {
public void send(byte[] b, int off, int len) throws IOException {
String fn = "SshJcraftWrapper.send:byte[]";
OutputStream out = channel.getOutputStream();
- DataOutputStream dos = new DataOutputStream(out);
+ DataOutputStream dos = getDataOutputStream(out);
try {
dos.write(b, off, len);
dos.flush();
} catch (IOException e) {
- debugLog.printRTAriDebug(fn, "Caught an IOException. e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an IOException. e=" + e);
throw new IOException(e.toString());
}
}
@@ -584,15 +556,13 @@ public class SshJcraftWrapper {
File tmpFile = new File(fileName);
if (tmpFile.exists()) {
try(BufferedWriter out = new BufferedWriter(new FileWriter(fileName, true))) {
- // out.write(dataToWrite);
- // out.write(getTheDate() +": " +Thread.currentThread().getName() +": "+dataToWrite);
out.write(getTheDate() + ": " + tId + ": " + dataToWrite);
}
}
} catch (IOException e) {
- debugLog.printRTAriDebug(fn, "Caught an IOException: e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an IOException: e=" + e);
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception: e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception: e=" + e);
}
}
@@ -609,9 +579,9 @@ public class SshJcraftWrapper {
}
}
} catch (IOException e) {
- debugLog.printRTAriDebug(fn, "Caught an IOException: e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an IOException: e=" + e);
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception: e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception: e=" + e);
}
}
@@ -619,7 +589,6 @@ public class SshJcraftWrapper {
public String getTheDate() {
Calendar cal = Calendar.getInstance();
java.util.Date today = cal.getTime();
- DateFormat df1 = DateFormat.getDateInstance();
DateFormat df3 = new SimpleDateFormat("MM/dd/yyyy H:mm:ss ");
return (df3.format(today));
}
@@ -627,16 +596,14 @@ public class SshJcraftWrapper {
public void appendToRouterFile(String fileName, StringBuffer dataToWrite) {
String fnName = "SshJcraftWrapper.appendToRouterFile";
- debugLog.printRTAriDebug(fnName, "Entered.... ");
+ DebugLog.printRTAriDebug(fnName, "Entered.... ");
try {
// First check to see if a file 'fileName' exist, if it does
// write to it. If it does not exist, don't write to it.
File tmpFile = new File(fileName);
{
- // if ((tmpFile.exists()) && (tmpFile.setWritable(true, true)))
if (tmpFile.exists()) {
try(BufferedWriter out = new BufferedWriter(new FileWriter(fileName, true))) {
- // out.write("<!-- "+getTheDate() +": " +tId +" -->\n");
out.write(dataToWrite.toString());
}
}
@@ -648,16 +615,12 @@ public class SshJcraftWrapper {
}
public void appendToRouterFile(String fileName, int len) {
- String fnName = "SshJcraftWrapper.appendToFile";
- // debugLog.printRTAriDebug (fnName, "Entered.... len="+len);
try {
// First check to see if a file 'fileName' exist, if it does
// write to it. If it does not exist, don't write to it.
File tmpFile = new File(fileName);
- // if ((tmpFile.exists()) && (tmpFile.setWritable(true, true)))
if (tmpFile.exists()) {
try(BufferedWriter out = new BufferedWriter(new FileWriter(fileName, true))) {
- // out.write("<!-- "+getTheDate() +": " +tId +" -->\n");
out.write(charBuffer, 0, len);
}
}
@@ -683,18 +646,14 @@ public class SshJcraftWrapper {
}
public String stripOffCmdFromRouterResponse(String routerResponse) {
- String fn = "SshJcraftWrapper.stripOffCmdFromRouterResponse";
- // appendToFile(debugLogFileName, fn+": routerResponse='"+routerResponse +"'\n");
-
// The session of SSH will echo the command sent to the router, in the router's response.
// Since all our commands are terminated by a '\n', strip off the first line
- // of the response from the router. This first line contains the orginal command.
+ // of the response from the router. This first line contains the original command.
StringTokenizer rr = new StringTokenizer(routerResponse, "\n");
StringBuffer sb = new StringBuffer();
int numTokens = rr.countTokens();
- // debugLog.printRTAriDebug (fn, "Number of lines in the response from the router is:" +numTokens);
if (numTokens > 1) {
rr.nextToken(); //Skip the first line.
while (rr.hasMoreTokens()) {
@@ -707,7 +666,7 @@ public class SshJcraftWrapper {
public void setRouterCommandType(String type) {
String fn = "SshJcraftWrapper.setRouterCommandType";
this.routerCmdType = type;
- debugLog.printRTAriDebug(fn, "Setting routerCmdType to a value of '" + type + "'");
+ DebugLog.printRTAriDebug(fn, "Setting routerCmdType to a value of '" + type + "'");
}
public String getLastFewLinesOfFile(File file, int linesToRead) throws IOException {
@@ -725,7 +684,6 @@ public class SshJcraftWrapper {
builder.append(c);
if (c == '\n') {
builder = builder.reverse();
- // System.out.println(builder.toString());
tail = builder.toString() + tail;
lines++;
builder.setLength(0);
@@ -736,7 +694,7 @@ public class SshJcraftWrapper {
}
}
if (!jcraftReadSwConfigFileFromDisk()) {
- debugLog.printRTAriDebug(fn, "tail='" + tail + "'");
+ DebugLog.printRTAriDebug(fn, "tail='" + tail + "'");
}
appendToFile(debugLogFileName, "tail='" + tail + "'\n");
return tail;
@@ -767,12 +725,9 @@ public class SshJcraftWrapper {
public void receiveUntilBufferFlush(int ncharsSent, int timeout, String message)
throws TimedOutException, IOException {
String fn = "SshJcraftWrapper.receiveUntilBufferFlush";
- StringBuffer sb2 = new StringBuffer();
- StringBuffer sbReceive = new StringBuffer();
- debugLog.printRTAriDebug(fn, "ncharsSent=" + ncharsSent + " timeout=" + timeout + " " + message);
+ DebugLog.printRTAriDebug(fn, "ncharsSent=" + ncharsSent + " timeout=" + timeout + " " + message);
int ncharsTotalReceived = 0;
int ncharsRead = 0;
- boolean flag = false;
charactersFromBufferFlush.setLength(0);
long deadline = new Date().getTime() + timeout;
@@ -781,10 +736,9 @@ public class SshJcraftWrapper {
session.setTimeout(timeout); // This is the socket timeout value.
while (true) {
if (new Date().getTime() > deadline) {
- debugLog.printRTAriDebug(fn,
+ DebugLog.printRTAriDebug(fn,
"Throwing a TimedOutException: time in routine has exceed our deadline: ncharsSent="
+ ncharsSent + " ncharsTotalReceived=" + ncharsTotalReceived);
- flag = true;
throw new TimedOutException("Timeout: time in routine has exceed our deadline");
}
ncharsRead = reader.read(charBuffer, 0, BUFFER_SIZE);
@@ -793,9 +747,8 @@ public class SshJcraftWrapper {
}
appendToRouterFile("/tmp/" + RouterName, ncharsRead);
ncharsTotalReceived = ncharsTotalReceived + ncharsRead;
- // debugLog.printRTAriDebug (fn, "::ncharsSent="+ncharsSent+" ncharsTotalReceived="+ncharsTotalReceived +" ncharsRead="+ncharsRead);
if (ncharsTotalReceived >= ncharsSent) {
- debugLog.printRTAriDebug(fn,
+ DebugLog.printRTAriDebug(fn,
"Received the correct number of characters, ncharsSent=" + ncharsSent + " ncharsTotalReceived="
+ ncharsTotalReceived);
logMemoryUsage();
@@ -803,8 +756,8 @@ public class SshJcraftWrapper {
}
}
} catch (JSchException e) {
- debugLog.printRTAriDebug(fn, "Caught an JSchException e=" + e);
- debugLog.printRTAriDebug(fn,
+ DebugLog.printRTAriDebug(fn, "Caught an JSchException e=" + e);
+ DebugLog.printRTAriDebug(fn,
"ncharsSent=" + ncharsSent + " ncharsTotalReceived=" + ncharsTotalReceived + " ncharsRead="
+ ncharsRead);
throw new TimedOutException(e.toString());
@@ -824,26 +777,8 @@ public class SshJcraftWrapper {
}
public void sftpPut(String sourcePath, String destDirectory) throws IOException {
- String fn = "SshJcraftWrapper.sftp";
- try {
- Session sftpSession = jsch.getSession(userName, hostName, 22);
- UserInfo ui = new MyUserInfo();
- sftpSession.setPassword(passWord);
- sftpSession.setUserInfo(ui);
- sftpSession.connect(30 * 1000);
- debugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
- ChannelSftp sftp = (ChannelSftp) sftpSession.openChannel("sftp");
- debugLog.printRTAriDebug(fn, "Connecting....");
- sftp.connect();
- debugLog.printRTAriDebug(fn, "Sending " + sourcePath + " --> " + destDirectory);
- sftp.put(sourcePath, destDirectory, ChannelSftp.OVERWRITE);
- debugLog.printRTAriDebug(fn, "Sent successfully");
- sftpSession.disconnect();
- } catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
- // dbLog.storeData("ErrorMsg= sftp threw an Exception. error is:"+e);
- throw new IOException(e.toString());
- }
+ // delegate to duplicate method
+ put(sourcePath, destDirectory);
}
@@ -855,18 +790,17 @@ public class SshJcraftWrapper {
sftpSession.setPassword(passWord);
sftpSession.setUserInfo(ui);
sftpSession.connect(30 * 1000);
- debugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
+ DebugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
ChannelSftp sftp = (ChannelSftp) sftpSession.openChannel("sftp");
- debugLog.printRTAriDebug(fn, "Connecting....");
+ DebugLog.printRTAriDebug(fn, "Connecting....");
sftp.connect();
InputStream is = new ByteArrayInputStream(stringOfData.getBytes());
- debugLog.printRTAriDebug(fn, "Sending stringOfData --> " + fullPathDest);
+ DebugLog.printRTAriDebug(fn, "Sending stringOfData --> " + fullPathDest);
sftp.put(is, fullPathDest, ChannelSftp.OVERWRITE);
- debugLog.printRTAriDebug(fn, "Sent successfully");
+ DebugLog.printRTAriDebug(fn, "Sent successfully");
sftpSession.disconnect();
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
- // dbLog.storeData("ErrorMsg= sftp threw an Exception. error is:"+e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
throw new IOException(e.toString());
}
}
@@ -879,20 +813,18 @@ public class SshJcraftWrapper {
sftpSession.setPassword(passWord);
sftpSession.setUserInfo(ui);
sftpSession.connect(30 * 1000);
- debugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
+ DebugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
ChannelSftp sftp = (ChannelSftp) sftpSession.openChannel("sftp");
- debugLog.printRTAriDebug(fn, "Connecting....");
+ DebugLog.printRTAriDebug(fn, "Connecting....");
sftp.connect();
InputStream in = null;
in = sftp.get(fullFilePathName);
String sftpFileString = readInputStreamAsString(in);
- debugLog.printRTAriDebug(fn, "Retreived successfully");
- // debugLog.printRTAriDebug (fn, "sftpFileString="+sftpFileString);
+ DebugLog.printRTAriDebug(fn, "Retreived successfully");
sftpSession.disconnect();
return (sftpFileString);
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
- // dbLog.storeData("ErrorMsg= sftp threw an Exception. error is:"+e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
throw new IOException(e.toString());
}
}
@@ -935,11 +867,11 @@ public class SshJcraftWrapper {
String subsystem) throws IOException {
String fn = "SshJcraftWrapper.connect";
- debugLog.printRTAriDebug(fn,
+ DebugLog.printRTAriDebug(fn,
":::Attempting to connect to " + hostname + " username=" + username + " prompt='"
+ prompt + "' timeOut=" + timeOut + " portNum=" + portNum + " subsystem=" + subsystem);
RouterName = hostname;
- jsch = new JSch();
+ jsch = getJSch();
try {
session = jsch.getSession(username, hostname, portNum);
UserInfo ui = new MyUserInfo();
@@ -951,32 +883,31 @@ public class SshJcraftWrapper {
0); // If this is not set to '0', then socket timeout on all reads will not work!!!!
channel = session.openChannel("subsystem");
((ChannelSubsystem) channel).setSubsystem(subsystem);
- // ((ChannelSubsystem)channel).setPtyType("vt102");
((ChannelSubsystem) channel).setPty(true);
inputStream = channel.getInputStream();
dis = new DataInputStream(inputStream);
reader = new BufferedReader(new InputStreamReader(dis), BUFFER_SIZE);
channel.connect();
- debugLog.printRTAriDebug(fn, "Successfully connected.");
- debugLog.printRTAriDebug(fn, "Five second sleep....");
+ DebugLog.printRTAriDebug(fn, "Successfully connected.");
+ DebugLog.printRTAriDebug(fn, "Five second sleep....");
try {
- Thread.sleep(5000);
+ delay(5000);
} catch (java.lang.InterruptedException ee) {
- boolean ignore = true;
+ Thread.currentThread().interrupt();
}
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception. e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception. e=" + e);
throw new IOException(e.toString());
}
}
public void connect(String hostName, String username, String password, int portNumber) throws IOException {
String fn = "SshJcraftWrapper.connect";
- jsch = new JSch();
- debugLog.printRTAriDebug(fn,
+ jsch = getJSch();
+ DebugLog.printRTAriDebug(fn,
"::Attempting to connect to " + hostName + " username=" + username + " portNumber=" + portNumber);
- debugLog.printRTAriDebug(fn, "Trace C");
+ DebugLog.printRTAriDebug(fn, "Trace C");
RouterName = hostName;
this.hostName = hostName;
userName = username;
@@ -985,7 +916,6 @@ public class SshJcraftWrapper {
java.util.Properties config = new java.util.Properties();
config.put("StrictHostKeyChecking", "no");
session = jsch.getSession(username, hostName, 22);
- // session = jsch.getSession(username, hostName, portNumber);
UserInfo ui = new MyUserInfo();
session.setConfig(config);
session.setPassword(password);
@@ -999,17 +929,16 @@ public class SshJcraftWrapper {
dis = new DataInputStream(inputStream);
reader = new BufferedReader(new InputStreamReader(dis), BUFFER_SIZE);
channel.connect();
- debugLog.printRTAriDebug(fn, "::Successfully connected.");
- debugLog.printRTAriDebug(fn, "::Flushing input buffer");
+ DebugLog.printRTAriDebug(fn, "::Successfully connected.");
+ DebugLog.printRTAriDebug(fn, "::Flushing input buffer");
try {
receiveUntil(":~#", 9000, "No cmd was sent, just waiting, but we can stop on a '~#'");
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception::: Nothing to flush out.");
+ DebugLog.printRTAriDebug(fn, "Caught an Exception::: Nothing to flush out.");
}
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception. e=" + e);
- // dbLog.storeData("ErrorMsg= Exception trying to connect to "+hostName +" "+e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception. e=" + e);
throw new IOException(e.toString());
}
}
@@ -1023,17 +952,16 @@ public class SshJcraftWrapper {
sftpSession.setPassword(passWord);
sftpSession.setUserInfo(ui);
sftpSession.connect(30 * 1000);
- debugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
+ DebugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
ChannelSftp sftp = (ChannelSftp) sftpSession.openChannel("sftp");
- debugLog.printRTAriDebug(fn, "Connecting....");
+ DebugLog.printRTAriDebug(fn, "Connecting....");
sftp.connect();
- debugLog.printRTAriDebug(fn, "Sending " + sourcePath + " --> " + destDirectory);
+ DebugLog.printRTAriDebug(fn, "Sending " + sourcePath + " --> " + destDirectory);
sftp.put(sourcePath, destDirectory, ChannelSftp.OVERWRITE);
- debugLog.printRTAriDebug(fn, "Sent successfully");
+ DebugLog.printRTAriDebug(fn, "Sent successfully");
sftpSession.disconnect();
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
- // dbLog.storeData("ErrorMsg= sftp threw an Exception. error is:"+e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
throw new IOException(e.toString());
}
}
@@ -1043,8 +971,8 @@ public class SshJcraftWrapper {
String fn = "SshJcraftWrapper.put";
Session sftpSession = null;
try {
- debugLog.printRTAriDebug(fn, "userName=" + userName + " hostName=" + hostName);
- jsch = new JSch();
+ DebugLog.printRTAriDebug(fn, "userName=" + userName + " hostName=" + hostName);
+ jsch = getJSch();
java.util.Properties config = new java.util.Properties();
config.put("StrictHostKeyChecking", "no");
sftpSession = jsch.getSession(userName, hostName, 22);
@@ -1053,29 +981,28 @@ public class SshJcraftWrapper {
sftpSession.setUserInfo(ui);
sftpSession.setConfig(config);
sftpSession.connect(30 * 1000);
- debugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
+ DebugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
ChannelSftp sftp = (ChannelSftp) sftpSession.openChannel("sftp");
- debugLog.printRTAriDebug(fn, "Connecting....");
+ DebugLog.printRTAriDebug(fn, "Connecting....");
sftp.connect();
String oldFiles = fullPathDest + "*";
- debugLog.printRTAriDebug(fn, "Deleting old files --> " + oldFiles);
+ DebugLog.printRTAriDebug(fn, "Deleting old files --> " + oldFiles);
try {
sftp.rm(oldFiles);
- debugLog.printRTAriDebug(fn, "Sending stringOfData --> " + fullPathDest);
+ DebugLog.printRTAriDebug(fn, "Sending stringOfData --> " + fullPathDest);
} catch (SftpException sft) {
String exp = "No such file";
if (sft.getMessage() != null && sft.getMessage().contains(exp)) {
- debugLog.printRTAriDebug(fn, "No files found -- Continue");
+ DebugLog.printRTAriDebug(fn, "No files found -- Continue");
} else {
- debugLog.printRTAriDebug(fn, "Exception while sftp.rm " + sft.getMessage());
- sft.printStackTrace();
+ DebugLog.printRTAriDebug(fn, "Exception while sftp.rm " + sft.getMessage());
throw sft;
}
}
sftp.put(is, fullPathDest, ChannelSftp.OVERWRITE);
- debugLog.printRTAriDebug(fn, "Sent successfully");
+ DebugLog.printRTAriDebug(fn, "Sent successfully");
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
throw new IOException(e.toString());
} finally {
if(sftpSession != null) {
@@ -1089,8 +1016,8 @@ public class SshJcraftWrapper {
String fn = "SshJcraftWrapper.get";
Session sftpSession = null;
try {
- debugLog.printRTAriDebug(fn, "userName=" + userName + " hostName=" + hostName);
- jsch = new JSch();
+ DebugLog.printRTAriDebug(fn, "userName=" + userName + " hostName=" + hostName);
+ jsch = getJSch();
sftpSession = jsch.getSession(userName, hostName, 22);
java.util.Properties config = new java.util.Properties();
config.put("StrictHostKeyChecking", "no");
@@ -1099,16 +1026,16 @@ public class SshJcraftWrapper {
sftpSession.setUserInfo(ui);
sftpSession.setConfig(config);
sftpSession.connect(30 * 1000);
- debugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
+ DebugLog.printRTAriDebug(fn, "Opening up an sftp channel....");
ChannelSftp sftp = (ChannelSftp) sftpSession.openChannel("sftp");
- debugLog.printRTAriDebug(fn, "Connecting....");
+ DebugLog.printRTAriDebug(fn, "Connecting....");
sftp.connect();
InputStream in = sftp.get(fullFilePathName);
String sftpFileString = readInputStreamAsString(in);
- debugLog.printRTAriDebug(fn, "Retreived successfully");
+ DebugLog.printRTAriDebug(fn, "Retreived successfully");
return sftpFileString;
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an Exception, e=" + e);
throw new IOException(e.toString());
} finally {
if(sftpSession != null) {
@@ -1120,7 +1047,7 @@ public class SshJcraftWrapper {
public String send(String cmd, String delimiter) throws IOException {
String fn = "SshJcraftWrapper.send";
OutputStream out = channel.getOutputStream();
- DataOutputStream dos = new DataOutputStream(out);
+ DataOutputStream dos = getDataOutputStream(out);
if ((cmd.charAt(cmd.length() - 1) != '\n') && (cmd.charAt(cmd.length() - 1) != '\r')) {
cmd += "\n";
@@ -1131,8 +1058,7 @@ public class SshJcraftWrapper {
int ncharsTotalSent = 0;
int ncharsSent = 0;
- debugLog.printRTAriDebug(fn, "Length of cmd is:" + length); // 2,937,706
- debugLog.printRTAriDebug(fn, "Length of cmd is:" + length); // 2,937,706
+ DebugLog.printRTAriDebug(fn, "Length of cmd is:" + length); // 2,937,706
try {
if (length > 600000) {
int timeout = 9000;
@@ -1140,35 +1066,45 @@ public class SshJcraftWrapper {
String Cmd = cmd.substring(i, Math.min(length, i + nchars));
ncharsSent = Cmd.length();
ncharsTotalSent = ncharsTotalSent + Cmd.length();
- debugLog.printRTAriDebug(fn, "i=" + i + " Sending Cmd: ncharsSent=" + ncharsSent);
+ DebugLog.printRTAriDebug(fn, "i=" + i + " Sending Cmd: ncharsSent=" + ncharsSent);
dos.writeBytes(Cmd);
dos.flush();
try {
- debugLog.printRTAriDebug(fn, ":::i=" + i + " length=" + length);
+ DebugLog.printRTAriDebug(fn, ":::i=" + i + " length=" + length);
if (ncharsSent < length) {
receiveUntilBufferFlush(ncharsSent, timeout, "buffer flush i=" + i);
} else {
- debugLog.printRTAriDebug(fn, "i=" + i + " No Waiting this time....");
+ DebugLog.printRTAriDebug(fn, "i=" + i + " No Waiting this time....");
dos.flush();
}
} catch (Exception e) {
- debugLog.printRTAriDebug(fn, "Caught an Exception: Nothing to flush out.");
+ DebugLog.printRTAriDebug(fn, "Caught an Exception: Nothing to flush out.");
}
}
} else {
- debugLog.printRTAriDebug(fn, "Before executing the dos.writeBytes");
+ DebugLog.printRTAriDebug(fn, "Before executing the dos.writeBytes");
dos.writeBytes(cmd);
}
dos.flush();
// Now lets get the response.
String response = receiveUntil(delimiter, 300000, cmd);
- debugLog.printRTAriDebug(fn, "Leaving method");
+ DebugLog.printRTAriDebug(fn, "Leaving method");
return (response);
} catch (IOException e) {
- debugLog.printRTAriDebug(fn, "Caught an IOException. e=" + e);
+ DebugLog.printRTAriDebug(fn, "Caught an IOException. e=" + e);
throw new IOException(e.toString());
}
}
+ protected JSch getJSch() {
+ return new JSch();
+ }
+ protected DataOutputStream getDataOutputStream(OutputStream out) {
+ return new DataOutputStream(out);
+ }
+
+ protected void delay(int milliseconds) throws InterruptedException {
+ Thread.sleep(5000);
+ }
}
diff --git a/appc-config/appc-config-adaptor/provider/src/test/java/org/onap/appc/ccadaptor/SshJcraftWrapperTest.java b/appc-config/appc-config-adaptor/provider/src/test/java/org/onap/appc/ccadaptor/SshJcraftWrapperTest.java
index 7bf6b77d3..70de9e63a 100644
--- a/appc-config/appc-config-adaptor/provider/src/test/java/org/onap/appc/ccadaptor/SshJcraftWrapperTest.java
+++ b/appc-config/appc-config-adaptor/provider/src/test/java/org/onap/appc/ccadaptor/SshJcraftWrapperTest.java
@@ -7,6 +7,8 @@
* Copyright (C) 2017 Amdocs
* =============================================================================
* Modifications Copyright (C) 2018 IBM.
+ * ================================================================================
+ * Modifications Copyright (C) 2018 Ericsson
* =============================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -22,28 +24,14 @@
*
* ============LICENSE_END=========================================================
*/
-
+
package org.onap.appc.ccadaptor;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.fail;
-import static org.mockito.BDDMockito.given;
-import static org.mockito.Matchers.any;
-import static org.mockito.Matchers.anyInt;
-import static org.mockito.Matchers.anyString;
-import static org.mockito.Matchers.eq;
-import static org.mockito.Mockito.atLeastOnce;
-import static org.mockito.Mockito.doThrow;
-import static org.mockito.Mockito.inOrder;
-import static org.mockito.Mockito.never;
-import static org.mockito.Mockito.spy;
-import static org.mockito.Mockito.times;
-import static org.mockito.Mockito.verify;
-import static org.mockito.Mockito.verifyNoMoreInteractions;
-
+import com.google.common.base.Charsets;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.ChannelShell;
import com.jcraft.jsch.ChannelSubsystem;
@@ -51,82 +39,619 @@ import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import com.jcraft.jsch.SftpException;
-import com.jcraft.jsch.UserInfo;
+import java.io.BufferedOutputStream;
+import java.io.BufferedReader;
+import java.io.BufferedWriter;
+import java.io.DataOutputStream;
import java.io.File;
+import java.io.FileInputStream;
import java.io.FileNotFoundException;
+import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
-import java.net.URL;
-
-import org.apache.commons.lang.StringUtils;
+import org.apache.commons.io.IOUtils;
import org.junit.Assert;
import org.junit.Before;
-import org.junit.Ignore;
+import org.junit.Rule;
import org.junit.Test;
+import org.junit.rules.ExpectedException;
import org.junit.runner.RunWith;
-import org.mockito.InOrder;
-import org.mockito.Mock;
-import org.mockito.runners.MockitoJUnitRunner;
+import org.mockito.Mockito;
+import org.mockito.internal.util.reflection.Whitebox;
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
import org.onap.appc.ccadaptor.SshJcraftWrapper.MyUserInfo;
-import org.apache.commons.io.IOUtils;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
-@RunWith(MockitoJUnitRunner.class)
+@RunWith(PowerMockRunner.class)
+@PrepareForTest(DataOutputStream.class)
public class SshJcraftWrapperTest {
+
+ private SshJcraftWrapper wrapper;
+ private File debugFile;
+
+ @Rule
+ public ExpectedException expectedEx = ExpectedException.none();
+
+ @Before
+ public void setupForTests() throws IOException, InterruptedException {
+ wrapper = Mockito.spy(new SshJcraftWrapper());
+ Mockito.doNothing().when(wrapper).delay(Mockito.anyInt());
+ debugFile = new File("src/test/resources/sshJcraftWrapperDebug");
+ File debugFile2 = new File("src/test/resources/sshJcraftWrapperDEBUG");
+ File configFile = new File("src/test/resources/jcraftReadSwConfigFileFromDisk");
+ File proxyRouterLogFile = new File("src/test/resources/proxyRouterLogFile");
+ debugFile.getParentFile().mkdirs();
+ debugFile.createNewFile();
+ debugFile2.createNewFile();
+ configFile.createNewFile();
+ proxyRouterLogFile.createNewFile();
+ Whitebox.setInternalState(wrapper, "debugLogFileName",
+ "src/test/resources/sshJcraftWrapperDebug");
+ Whitebox.setInternalState(wrapper, "extraDebugFile", debugFile2);
+ Whitebox.setInternalState(wrapper, "jcraftReadSwConfigFileFromDisk", configFile);
+ }
+
@Test
- public void TestCheckIfReceivedStringMatchesDelimeter(){
- SshJcraftWrapper wrapper = new SshJcraftWrapper();
- wrapper.getTheDate();
- boolean result = wrapper.checkIfReceivedStringMatchesDelimeter("#", "test#", "test#");
- Assert.assertEquals(true, result);
+ public void testConnect() throws IOException, JSchException, InterruptedException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ InputStream stubInputStream = IOUtils.toInputStream("hello\n]]>]]>", Charsets.UTF_8);
+ Mockito.doReturn(stubInputStream).when(mockChannel).getInputStream();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("shell");
+ Mockito.doReturn(mockSession).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ wrapper.connect("testHost", "testUser", "testPswd", "]]>]]>", 1000);
+ Mockito.verify(wrapper, Mockito.times(9)).appendToFile(Mockito.anyString(),
+ Mockito.anyString());
}
@Test
- public void testRemoveWhiteSpaceAndNewLineCharactersAroundString(){
- SshJcraftWrapper wrapper = new SshJcraftWrapper();
- String nameSpace = wrapper.removeWhiteSpaceAndNewLineCharactersAroundString("namespace ");
- Assert.assertEquals("namespace", nameSpace);
+ public void testConnectExceptionFlow() throws IOException, JSchException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Mockito.doThrow(new JSchException()).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("com.jcraft.jsch.JSchException");
+ wrapper.connect("testHost", "testUser", "testPswd", "]]>]]>", 1000);
}
@Test
- public void testStripOffCmdFromRouterResponse(){
- SshJcraftWrapper wrapper = new SshJcraftWrapper();
- String result = wrapper.stripOffCmdFromRouterResponse("test\nsuccess");
- Assert.assertEquals("success\n", result);
+ public void testConnectExceptionFlow2() throws IOException, JSchException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("shell");
+ Mockito.doThrow(new IOException()).when(wrapper).receiveUntil("]]>]]>", 3000,
+ "No cmd was sent, just waiting");
+ Mockito.doReturn(mockSession).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ wrapper.connect("testHost", "testUser", "testPswd", "]]>]]>", 1000);
+ Mockito.verify(wrapper, Mockito.times(1)).receiveUntil("]]>]]>", 3000,
+ "No cmd was sent, just waiting");
}
- //@Test
- public void testGetLastFewLinesOfFile() throws FileNotFoundException, IOException{
- SshJcraftWrapper wrapper = new SshJcraftWrapper();
- URL path = SshJcraftWrapperTest.class.getResource("Test");
- File file = new File(path.getFile());
- String value = wrapper.getLastFewLinesOfFile(file,1);
- Assert.assertEquals("\nTest data 3", value);
+ @Test
+ public void testConnectWithPortNumber() throws IOException, JSchException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ InputStream stubInputStream = IOUtils.toInputStream("hello\n:~#", Charsets.UTF_8);
+ Mockito.doReturn(stubInputStream).when(mockChannel).getInputStream();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("shell");
+ Mockito.doReturn(mockSession).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ wrapper.connect("testHost", "testUser", "testPswd", ":~#", 1000, 22);
+ Mockito.verify(wrapper, Mockito.times(9)).appendToFile(Mockito.anyString(),
+ Mockito.anyString());
}
- @Test(expected=Exception.class)
- public void testSetRouterCommandType() throws IOException{
- SshJcraftWrapper wrapper = new SshJcraftWrapper();
- wrapper.setRouterCommandType("test");
- wrapper.receiveUntil("test", 2, "test");
+ @Test
+ public void testConnectWithPortNumberSuccessFlow2()
+ throws IOException, JSchException, InterruptedException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ InputStream stubInputStream = IOUtils.toInputStream("hello\n]]>]]>", Charsets.UTF_8);
+ Mockito.doReturn(stubInputStream).when(mockChannel).getInputStream();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("shell");
+ Mockito.doReturn(mockSession).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ wrapper.connect("testHost", "testUser", "testPswd", "]]>]]>", 1000, 22);
+ Mockito.verify(wrapper, Mockito.times(9)).appendToFile(Mockito.anyString(),
+ Mockito.anyString());
+ }
+
+ @Test
+ public void testConnectWithPortNumberExceptionFlow() throws IOException, JSchException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Mockito.doThrow(new JSchException()).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("com.jcraft.jsch.JSchException");
+ wrapper.connect("testHost", "testUser", "testPswd", "]]>]]>", 1000, 22);
+ }
+
+ @Test
+ public void testConnectWithPortNumberExceptionFlow2() throws IOException, JSchException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ InputStream stubInputStream = IOUtils.toInputStream("", Charsets.UTF_8);
+ Mockito.doReturn(stubInputStream).when(mockChannel).getInputStream();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("shell");
+ Mockito.doReturn(mockSession).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ wrapper.connect("testHost", "testUser", "testPswd", "]]>]]>", 1000, 22);
+ Mockito.verify(wrapper, Mockito.times(1)).receiveUntil("]]>]]>", 10000,
+ "No cmd was sent, just waiting");
+ }
+
+ @Test
+ public void testReceiveUntilTimeout() throws TimedOutException, IOException, JSchException {
+ Session mockSession = Mockito.mock(Session.class);
+ Whitebox.setInternalState(wrapper, "session", mockSession);
+ // The sleep is required to make sure that the system clock has incremented by
+ // (at least) 1 millisecond between passing in our timeout value and setting a deadline
+ // and checking to see if the deadline has been missed
+ Mockito.doAnswer(new Answer<String>() {
+ @Override
+ public String answer(InvocationOnMock invocation) throws InterruptedException {
+ Thread.sleep(1);
+ return null;
+ }
+ }).when(mockSession).setTimeout(0);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("Timeout: time in routine has exceed our deadline");
+ wrapper.receiveUntil("", 0, "");
+ }
+
+ @Test
+ public void testReceiveUntilReaderTimeout()
+ throws TimedOutException, IOException, JSchException {
+ Session mockSession = Mockito.mock(Session.class);
+ Whitebox.setInternalState(wrapper, "session", mockSession);
+ BufferedReader mockReader = Mockito.mock(BufferedReader.class);
+ Whitebox.setInternalState(wrapper, "reader", mockReader);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("Received a SocketTimeoutException router=");
+ wrapper.receiveUntil("", 3000, "");
}
@Test
- public void testValues() throws IOException{
+ public void testReceiveUntilIosXr() throws TimedOutException, IOException, JSchException {
+ Session mockSession = Mockito.mock(Session.class);
+ Whitebox.setInternalState(wrapper, "session", mockSession);
+ BufferedReader mockReader = Mockito.mock(BufferedReader.class);
+ Mockito.doReturn(3).when(mockReader).read(Mockito.anyObject(), Mockito.anyInt(),
+ Mockito.anyInt());
+ Whitebox.setInternalState(wrapper, "reader", mockReader);
+ Mockito.doReturn(false).when(wrapper).jcraftReadSwConfigFileFromDisk();
+ Mockito.doReturn("\nXML>").when(wrapper).getLastFewLinesOfFile(Mockito.anyObject(),
+ Mockito.anyInt());
+
+ assertNull(wrapper.receiveUntil("]]>]]>", 3000, "IOS_XR_uploadedSwConfigCmd\nOTHER\nXML>"));
+ }
+
+ @Test
+ public void testReceiveStringDelimiters() {
+ assertEquals(false, wrapper.checkIfReceivedStringMatchesDelimeter("#$", "", ""));
+ }
+
+ @Test
+ public void testReceiveStringDelimitersShowConfig() {
+ assertEquals(true, wrapper.checkIfReceivedStringMatchesDelimeter("]]>]]>", "]]>]]>\n #",
+ "show config"));
+ }
+
+ @Test
+ public void testReceiveStringDelimitersTwoArg() throws IOException {
+ SshJcraftWrapper localWrapper = Mockito.spy(new SshJcraftWrapper());
+ Mockito.doReturn(true).when(localWrapper).jcraftReadSwConfigFileFromDisk();
+ Mockito.doThrow(new IOException()).when(localWrapper)
+ .getLastFewLinesOfFile(Mockito.anyObject(), Mockito.anyInt());
+ Whitebox.setInternalState(localWrapper, "routerFileName", "DUMMY_FILE_NAME");
+ assertEquals(false, localWrapper.checkIfReceivedStringMatchesDelimeter(3, "]]>]]>\n #"));
+ }
+
+ @Test
+ public void testCloseConnection() {
+ Session mockSession = Mockito.mock(Session.class);
+ Whitebox.setInternalState(wrapper, "session", mockSession);
+ wrapper.closeConnection();
+ Mockito.verify(mockSession, Mockito.times(1)).disconnect();
+ }
+
+ @Test
+ public void testSend() throws IOException {
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ OutputStream mockOutputStream = Mockito.mock(BufferedOutputStream.class);
+ Mockito.doReturn(mockOutputStream).when(mockChannel).getOutputStream();
+ DataOutputStream mockDos = PowerMockito.spy(new DataOutputStream(mockOutputStream));
+ PowerMockito.doReturn(mockDos).when(wrapper).getDataOutputStream(Mockito.anyObject());
+ Whitebox.setInternalState(wrapper, "channel", mockChannel);
+ wrapper.send("TEST COMMAND\n");
+ Mockito.verify(wrapper, Mockito.times(2)).appendToFile(Mockito.anyString(),
+ Mockito.anyString());
+ }
+
+ @Test
+ public void testSendExceptionFlow() throws IOException {
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ OutputStream mockOutputStream = Mockito.mock(BufferedOutputStream.class);
+ Mockito.doReturn(mockOutputStream).when(mockChannel).getOutputStream();
+ DataOutputStream mockDos = PowerMockito.spy(new DataOutputStream(mockOutputStream));
+ PowerMockito.doThrow(new IOException()).when(mockDos).flush();
+ PowerMockito.doReturn(mockDos).when(wrapper).getDataOutputStream(Mockito.anyObject());
+ Whitebox.setInternalState(wrapper, "channel", mockChannel);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("java.io.IOException");
+ wrapper.send("TEST COMMAND");
+ }
+
+ @Test
+ public void testSendChar() throws IOException {
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ OutputStream mockOutputStream = Mockito.mock(BufferedOutputStream.class);
+ Mockito.doReturn(mockOutputStream).when(mockChannel).getOutputStream();
+ DataOutputStream mockDos = PowerMockito.spy(new DataOutputStream(mockOutputStream));
+ PowerMockito.doReturn(mockDos).when(wrapper).getDataOutputStream(Mockito.anyObject());
+ Whitebox.setInternalState(wrapper, "channel", mockChannel);
+ wrapper.sendChar(74);
+ Mockito.verify(mockDos).flush();
+ }
+
+ @Test
+ public void testSendCharExceptionFlow() throws IOException {
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ OutputStream mockOutputStream = Mockito.mock(BufferedOutputStream.class);
+ Mockito.doReturn(mockOutputStream).when(mockChannel).getOutputStream();
+ DataOutputStream mockDos = PowerMockito.spy(new DataOutputStream(mockOutputStream));
+ PowerMockito.doThrow(new IOException()).when(mockDos).flush();
+ PowerMockito.doReturn(mockDos).when(wrapper).getDataOutputStream(Mockito.anyObject());
+ Whitebox.setInternalState(wrapper, "channel", mockChannel);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("java.io.IOException");
+ wrapper.sendChar(65);
+ }
+
+ @Test
+ public void testSendByteArrayExceptionFlow() throws IOException {
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ OutputStream mockOutputStream = Mockito.mock(BufferedOutputStream.class);
+ Mockito.doReturn(mockOutputStream).when(mockChannel).getOutputStream();
+ DataOutputStream mockDos = PowerMockito.spy(new DataOutputStream(mockOutputStream));
+ PowerMockito.doThrow(new IOException()).when(mockDos).flush();
+ PowerMockito.doReturn(mockDos).when(wrapper).getDataOutputStream(Mockito.anyObject());
+ Whitebox.setInternalState(wrapper, "channel", mockChannel);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("java.io.IOException");
+ byte[] byteArray = new byte[] {65, 74};
+ wrapper.send(byteArray, 0, 2);
+ }
+
+ @Test
+ public void testGetLastFewLinesOfFile() throws FileNotFoundException, IOException {
+ File file = new File("src/test/resources/TEST_FILE.txt");
+ BufferedWriter writer = new BufferedWriter(new FileWriter(file.getPath()));
+ writer.write("line1\nline2");
+ writer.flush();
+ writer.close();
+ assertEquals("\nline2", wrapper.getLastFewLinesOfFile(file, 2));
+ }
+
+ @Test
+ public void testReceiveUntilBufferFlush() throws TimedOutException, IOException {
+ Session mockSession = Mockito.mock(Session.class);
+ Whitebox.setInternalState(wrapper, "session", mockSession);
+ BufferedReader mockReader = Mockito.mock(BufferedReader.class);
+ Mockito.doReturn(12).when(mockReader).read(Mockito.anyObject(), Mockito.anyInt(),
+ Mockito.anyInt());
+ Whitebox.setInternalState(wrapper, "reader", mockReader);
+
+ wrapper.receiveUntilBufferFlush(12, 100, "TEST_MESSAGE");
+ Mockito.verify(wrapper, Mockito.times(2)).logMemoryUsage();
+ }
+
+ @Test
+ public void testReceiveUntilBufferFlushTimeout()
+ throws TimedOutException, IOException, JSchException {
+ Session mockSession = Mockito.mock(Session.class);
+ // The sleep is required to make sure that the system clock has incremented by
+ // (at least) 1 millisecond between passing in our timeout value and setting a deadline
+ // and checking to see if the deadline has been missed
+ Mockito.doAnswer(new Answer<String>() {
+ @Override
+ public String answer(InvocationOnMock invocation) throws InterruptedException {
+ Thread.sleep(1);
+ return null;
+ }
+ }).when(mockSession).setTimeout(0);
+ Whitebox.setInternalState(wrapper, "session", mockSession);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("Timeout: time in routine has exceed our deadline");
+ wrapper.receiveUntilBufferFlush(10, 0, "TEST_MESSAGE");
+ }
+
+ @Test
+ public void testReceiveUntilBufferFlushJSchException()
+ throws TimedOutException, IOException, JSchException {
+ Session mockSession = Mockito.mock(Session.class);
+ Mockito.doThrow(new JSchException()).when(mockSession).setTimeout(0);
+ Whitebox.setInternalState(wrapper, "session", mockSession);
+ expectedEx.expect(TimedOutException.class);
+ expectedEx.expectMessage("com.jcraft.jsch.JSchException");
+ wrapper.receiveUntilBufferFlush(10, 0, "TEST_MESSAGE");
+ }
+
+ @Test
+ public void testSftpPutSourceToDest() throws JSchException, IOException {
+ Whitebox.setInternalState(wrapper, "hostName", "testHost");
+ Whitebox.setInternalState(wrapper, "userName", "testUser");
+ Whitebox.setInternalState(wrapper, "passWord", "testPwd");
+ ChannelSftp mockChannel = Mockito.mock(ChannelSftp.class);
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ Mockito.doReturn(mockSession).when(mockJsch).getSession("testUser", "testHost", 22);
+ Mockito.doNothing().when(mockSession).setPassword(Mockito.anyString());
+ Mockito.doNothing().when(mockSession).connect();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("sftp");
+ Whitebox.setInternalState(wrapper, "jsch", mockJsch);
+ wrapper.sftpPut("DUMMY_SRC_PATH", "DUMMY_DEST_DIRECTORY");
+ Mockito.verify(mockSession).disconnect();
+ }
+
+ @Test
+ public void testSftpPutSourceToDestExceptionFlow() throws JSchException, IOException {
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Mockito.doThrow(new JSchException()).when(mockJsch).getSession(null, null, 22);
+ Whitebox.setInternalState(wrapper, "jsch", mockJsch);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("com.jcraft.jsch.JSchException");
+ wrapper.sftpPut("DUMMY_SRC_PATH", "DUMMY_DEST_DIRECTORY");
+ }
+
+ @Test
+ public void testSftpPutStringToDest() throws JSchException, IOException {
+ Whitebox.setInternalState(wrapper, "hostName", "testHost");
+ Whitebox.setInternalState(wrapper, "userName", "testUser");
+ Whitebox.setInternalState(wrapper, "passWord", "testPwd");
+ ChannelSftp mockChannel = Mockito.mock(ChannelSftp.class);
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ Mockito.doReturn(mockSession).when(mockJsch).getSession("testUser", "testHost", 22);
+ Mockito.doNothing().when(mockSession).setPassword(Mockito.anyString());
+ Mockito.doNothing().when(mockSession).connect();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("sftp");
+ Whitebox.setInternalState(wrapper, "jsch", mockJsch);
+ wrapper.SftpPut("DUMMY_STRING", "DUMMY_DEST_DIRECTORY");
+ Mockito.verify(mockSession).disconnect();
+ }
+
+ @Test
+ public void testSftpPutStringToDestExceptionFlow() throws JSchException, IOException {
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Mockito.doThrow(new JSchException()).when(mockJsch).getSession(null, null, 22);
+ Whitebox.setInternalState(wrapper, "jsch", mockJsch);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("com.jcraft.jsch.JSchException");
+ wrapper.SftpPut("DUMMY_STRING", "DUMMY_DEST_DIRECTORY");
+ }
+
+ @Test
+ public void testSftpGet() throws JSchException, IOException, SftpException {
+ File file = new File("src/test/resources/TEST_FILE.txt");
+ BufferedWriter writer = new BufferedWriter(new FileWriter(file.getPath()));
+ writer.write("line1\nline2");
+ writer.flush();
+ Whitebox.setInternalState(wrapper, "hostName", "testHost");
+ Whitebox.setInternalState(wrapper, "userName", "testUser");
+ Whitebox.setInternalState(wrapper, "passWord", "testPwd");
+ ChannelSftp mockChannel = Mockito.mock(ChannelSftp.class);
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ Mockito.doReturn(mockSession).when(mockJsch).getSession("testUser", "testHost", 22);
+ Mockito.doNothing().when(mockSession).setPassword(Mockito.anyString());
+ Mockito.doNothing().when(mockSession).connect();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("sftp");
+ Mockito.doReturn(new FileInputStream(file)).when(mockChannel)
+ .get("src/test/resources/TEST_FILE.txt");
+ Whitebox.setInternalState(wrapper, "jsch", mockJsch);;
+ assertEquals("line1\nline2", wrapper.sftpGet("src/test/resources/TEST_FILE.txt"));
+ }
+
+ @Test
+ public void testSftpGetExceptionFlow() throws JSchException, IOException {
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Mockito.doThrow(new JSchException()).when(mockJsch).getSession(null, null, 22);
+ Whitebox.setInternalState(wrapper, "jsch", mockJsch);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("com.jcraft.jsch.JSchException");
+ wrapper.sftpGet("DUMMY_FILE_PATH");
+ }
+
+ @Test
+ public void testConnectWithSubsystem() throws IOException, JSchException, InterruptedException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ ChannelSubsystem mockChannel = Mockito.mock(ChannelSubsystem.class);
+ InputStream stubInputStream = IOUtils.toInputStream("hello\n:~#", Charsets.UTF_8);
+ Mockito.doReturn(stubInputStream).when(mockChannel).getInputStream();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("subsystem");
+ Mockito.doReturn(mockSession).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ wrapper.connect("testHost", "testUser", "testPswd", ":~#", 1000, 22, "testSubsystem");
+ Mockito.verify(mockChannel).connect();
+ }
+
+ @Test
+ public void testConnectWithSubsystemExceptionFlow() throws IOException, JSchException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Mockito.doThrow(new JSchException()).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("com.jcraft.jsch.JSchException");
+ wrapper.connect("testHost", "testUser", "testPswd", "]]>]]>", 1000, 22, "testSubsystem");
+ }
+
+ @Test
+ public void testConnectShellFourParameters() throws IOException, JSchException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ InputStream stubInputStream = IOUtils.toInputStream("hello\n]]>]]>", Charsets.UTF_8);
+ Mockito.doReturn(stubInputStream).when(mockChannel).getInputStream();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("shell");
+ Mockito.doReturn(mockSession).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(null).when(wrapper).receiveUntil(":~#", 9000,
+ "No cmd was sent, just waiting, but we can stop on a '~#'");
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ wrapper.connect("testHost", "testUser", "testPswd", 22);
+ Mockito.verify(mockChannel).connect();
+ }
+
+ @Test
+ public void testConnectShellFourParametersExceptionFlow() throws IOException, JSchException {
+ JSch mockJSch = Mockito.mock(JSch.class);
+ Mockito.doThrow(new JSchException()).when(mockJSch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJSch).when(wrapper).getJSch();
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("com.jcraft.jsch.JSchException");
+ wrapper.connect("testHost", "testUser", "testPswd", 22);
+ }
+
+ @Test
+ public void testPutInputStreamToDest() throws JSchException, IOException {
+ ChannelSftp mockChannel = Mockito.mock(ChannelSftp.class);
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ Mockito.doReturn(mockSession).when(mockJsch).getSession("testUser", "testHost", 22);
+ Mockito.doNothing().when(mockSession).setPassword(Mockito.anyString());
+ Mockito.doNothing().when(mockSession).connect(30 * 1000);
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("sftp");
+ Mockito.doReturn(mockJsch).when(wrapper).getJSch();
+ InputStream inputStream = Mockito.mock(InputStream.class);
+ wrapper.put(inputStream, "DUMMY_DEST_PATH/", "testHost", "testUser", "testPswd");
+ Mockito.verify(mockSession).disconnect();
+ }
+
+ @Test
+ public void testPutInputStreamToDestExceptionFlow()
+ throws JSchException, IOException, SftpException {
+ ChannelSftp mockChannel = Mockito.mock(ChannelSftp.class);
+ Mockito.doThrow(new SftpException(0, null)).when(mockChannel).rm("DUMMY_DEST_PATH/*");
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ Mockito.doReturn(mockSession).when(mockJsch).getSession("testUser", "testHost", 22);
+ Mockito.doNothing().when(mockSession).setPassword(Mockito.anyString());
+ Mockito.doNothing().when(mockSession).connect(30 * 1000);
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("sftp");
+ Mockito.doReturn(mockJsch).when(wrapper).getJSch();
+ InputStream inputStream = Mockito.mock(InputStream.class);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("0: null");
+ wrapper.put(inputStream, "DUMMY_DEST_PATH/", "testHost", "testUser", "testPswd");
+ }
+
+ @Test
+ public void testPutInputStreamToDestExceptionFlow2()
+ throws JSchException, IOException, SftpException {
+ ChannelSftp mockChannel = Mockito.mock(ChannelSftp.class);
+ Mockito.doThrow(new SftpException(0, "No such file")).when(mockChannel)
+ .rm("DUMMY_DEST_PATH/*");
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ Mockito.doReturn(mockSession).when(mockJsch).getSession("testUser", "testHost", 22);
+ Mockito.doNothing().when(mockSession).setPassword(Mockito.anyString());
+ Mockito.doNothing().when(mockSession).connect(30 * 1000);
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("sftp");
+ Mockito.doReturn(mockJsch).when(wrapper).getJSch();
+ InputStream inputStream = Mockito.mock(InputStream.class);
+ wrapper.put(inputStream, "DUMMY_DEST_PATH/", "testHost", "testUser", "testPswd");
+ Mockito.verify(mockSession).disconnect();
+ }
+
+ @Test
+ public void testGet() throws JSchException, IOException, SftpException {
+ File file = new File("src/test/resources/TEST_FILE.txt");
+ BufferedWriter writer = new BufferedWriter(new FileWriter(file.getPath()));
+ writer.write("line1\nline2");
+ writer.flush();
+ Whitebox.setInternalState(wrapper, "hostName", "testHost");
+ Whitebox.setInternalState(wrapper, "userName", "testUser");
+ Whitebox.setInternalState(wrapper, "passWord", "testPwd");
+ ChannelSftp mockChannel = Mockito.mock(ChannelSftp.class);
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Session mockSession = Mockito.mock(Session.class);
+ Mockito.doReturn(mockSession).when(mockJsch).getSession("testUser", "testHost", 22);
+ Mockito.doNothing().when(mockSession).setPassword(Mockito.anyString());
+ Mockito.doNothing().when(mockSession).connect();
+ Mockito.doReturn(mockChannel).when(mockSession).openChannel("sftp");
+ Mockito.doReturn(new FileInputStream(file)).when(mockChannel)
+ .get("src/test/resources/TEST_FILE.txt");
+ Mockito.doReturn(mockJsch).when(wrapper).getJSch();
+ assertEquals("line1\nline2",
+ wrapper.get("src/test/resources/TEST_FILE.txt", "testHost", "testUser", "testPwd"));
+ }
+
+ @Test
+ public void testGetExceptionFlow() throws JSchException, IOException {
+ JSch mockJsch = Mockito.mock(JSch.class);
+ Mockito.doThrow(new JSchException()).when(mockJsch).getSession("testUser", "testHost", 22);
+ Mockito.doReturn(mockJsch).when(wrapper).getJSch();
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("com.jcraft.jsch.JSchException");
+ wrapper.get("src/test/resources/TEST_FILE.txt", "testHost", "testUser", "testPwd");
+ }
+
+ @Test
+ public void testSendWithDelimiter() throws IOException {
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ OutputStream mockOutputStream = Mockito.mock(BufferedOutputStream.class);
+ Mockito.doReturn(mockOutputStream).when(mockChannel).getOutputStream();
+ DataOutputStream mockDos = PowerMockito.spy(new DataOutputStream(mockOutputStream));
+ PowerMockito.doReturn("TEST RESPONSE").when(wrapper).receiveUntil("#$", 300000,
+ "TEST COMMAND\n");
+ PowerMockito.doReturn(mockDos).when(wrapper).getDataOutputStream(Mockito.anyObject());
+ Whitebox.setInternalState(wrapper, "channel", mockChannel);
+ assertEquals("TEST RESPONSE", wrapper.send("TEST COMMAND\n", "#$"));
+ }
+
+ @Test
+ public void testSendWithDelimiterExceptionFlow() throws IOException {
+ ChannelShell mockChannel = Mockito.mock(ChannelShell.class);
+ OutputStream mockOutputStream = Mockito.mock(BufferedOutputStream.class);
+ Mockito.doReturn(mockOutputStream).when(mockChannel).getOutputStream();
+ DataOutputStream mockDos = PowerMockito.spy(new DataOutputStream(mockOutputStream));
+ PowerMockito.doThrow(new IOException()).when(mockDos).flush();
+ PowerMockito.doReturn(mockDos).when(wrapper).getDataOutputStream(Mockito.anyObject());
+ Whitebox.setInternalState(wrapper, "channel", mockChannel);
+ expectedEx.expect(IOException.class);
+ expectedEx.expectMessage("java.io.IOException");
+ wrapper.send("TEST COMMAND", "]]>]]>");
+ }
+
+ @Test
+ public void testValues() throws IOException {
SshJcraftWrapper wrapper = new SshJcraftWrapper();
wrapper.setEquipNameCode("testcode");
wrapper.setRouterCommandType("testcommand");
- String equipName =wrapper.getEquipNameCode();
- wrapper.getHostName();
- wrapper.getPassWord();
- wrapper.getRouterName();
- wrapper.getUserName();
- wrapper.getTheDate();
+ String equipName = wrapper.getEquipNameCode();
+ assertNull(wrapper.getHostName());
+ assertNull(wrapper.getPassWord());
+ assertNull(wrapper.getRouterName());
+ assertNull(wrapper.getUserName());
+ assertTrue(
+ wrapper.getTheDate().indexOf('/') > -1 && wrapper.getTheDate().indexOf(':') > -1);
Assert.assertEquals("testcode", equipName);
}
- @Test(expected=Exception.class)
- public void testSetRouterCommandType2() throws IOException{
+ @Test(expected = Exception.class)
+ public void testSetRouterCommandType2() throws IOException {
SshJcraftWrapper wrapper = new SshJcraftWrapper();
wrapper.appendToRouterFile("test", 2);
StringBuffer buffer = new StringBuffer();
@@ -135,27 +660,21 @@ public class SshJcraftWrapperTest {
wrapper.receiveUntilBufferFlush(3, 4, "test");
}
- @Test(expected=Exception.class)
- public void testSetRouterCommandType3() throws IOException{
+ @Test(expected = Exception.class)
+ public void testSetRouterCommandType3() throws IOException {
SshJcraftWrapper wrapper = new SshJcraftWrapper();
wrapper.checkIfReceivedStringMatchesDelimeter(3, "test");
}
-
- @Test(expected=IOException.class)
- public void testConnect() throws IOException{
- SshJcraftWrapper wrapper = new SshJcraftWrapper();
- wrapper.connect("testHost", "testUser", "testPswd", "3000", 1000);
- }
-
+
@Test
public void testMyUserInfoGetPassword() {
- MyUserInfo myUserInfo=new MyUserInfo();
+ MyUserInfo myUserInfo = new MyUserInfo();
assertNull(myUserInfo.getPassword());
}
-
+
@Test
public void testMyUserInfoPromptYesNo() {
- MyUserInfo myUserInfo=new MyUserInfo();
+ MyUserInfo myUserInfo = new MyUserInfo();
assertFalse(myUserInfo.promptYesNo(""));
}
}