1 files changed, 155 insertions, 0 deletions
diff --git a/appc-adapters/appc-ssh-adapter/appc-ssh-adapter-tests/src/main/java/org/openecomp/appc/adapter/ssh/SshConnectionMock.java b/appc-adapters/appc-ssh-adapter/appc-ssh-adapter-tests/src/main/java/org/openecomp/appc/adapter/ssh/SshConnectionMock.java
new file mode 100644
index 000000000..3194edcfb
--- /dev/null
+++ b/appc-adapters/appc-ssh-adapter/appc-ssh-adapter-tests/src/main/java/org/openecomp/appc/adapter/ssh/SshConnectionMock.java
@@ -0,0 +1,155 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * openECOMP : APP-C
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights
+ * 						reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.openecomp.appc.adapter.ssh;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.openecomp.appc.adapter.ssh.SshConnection;
+
+public class SshConnectionMock implements SshConnection {
+
+	private static final int DEF_SUCCESS_STATUS = 0;
+
+	private String host;
+	private int port;
+	private String username;
+	private String password;
+	private long timeout;
+
+	private int returnStatus = DEF_SUCCESS_STATUS;
+	private String returnStdout;
+	private String returnStderr;
+
+	private int connectCallCount = 0;
+	private int disconnectCallCount = 0;
+	private List<String> executedCommands = new ArrayList<>();
+
+	public SshConnectionMock(String host, int port, String username, String password) {
+		this.host = host;
+		this.port = port;
+		this.username = username;
+		this.password = password;
+	}
+
+	@Override
+	public void connect() {
+		connectCallCount++;
+	}
+
+	@Override
+	public void disconnect() {
+		disconnectCallCount++;
+	}
+
+	@Override
+	public int execCommand(String cmd, OutputStream out, OutputStream err) {
+		return execCommand(cmd, out, err, false);
+	}
+
+	@Override
+	public int execCommandWithPty(String cmd, OutputStream out) {
+		return execCommand(cmd, out, out, true);
+	}
+
+	private int execCommand(String cmd, OutputStream out, OutputStream err, boolean usePty) {
+		executedCommands.add(cmd);
+		try {
+			if((out != null) && (returnStdout != null)) {
+				out.write(returnStdout.getBytes());
+			}
+		} catch(IOException e) {
+			throw new RuntimeException("Error writing to stdout output stream", e);
+		}
+		try {
+			if((err != null) && (returnStderr != null)) {
+				err.write(returnStderr.getBytes());
+			}
+		} catch(IOException e) {
+			throw new RuntimeException("Error writing to stderr output stream", e);
+		}
+		return returnStatus;
+	}
+
+	@Override
+	public void setExecTimeout(long timeout) {
+		this.timeout = timeout;
+	}
+
+	public long getExecTimeout() {
+		return timeout;
+	}
+
+	public String getHost() {
+		return host;
+	}
+
+	public int getPort() {
+		return port;
+	}
+
+	public String getUsername() {
+		return username;
+	}
+
+	public String getPassword() {
+		return password;
+	}
+
+	public int getConnectCallCount() {
+		return connectCallCount;
+	}
+
+	public int getDisconnectCallCount() {
+		return disconnectCallCount;
+	}
+
+	public List<String> getExecutedCommands() {
+		return executedCommands;
+	}
+
+	public int getReturnStatus() {
+		return returnStatus;
+	}
+
+	public void setReturnStatus(int returnStatus) {
+		this.returnStatus = returnStatus;
+	}
+
+	public String getReturnStdout() {
+		return returnStdout;
+	}
+
+	public void setReturnStdout(String returnStdout) {
+		this.returnStdout = returnStdout;
+	}
+
+	public String getReturnStderr() {
+		return returnStderr;
+	}
+
+	public void setReturnStderr(String returnStderr) {
+		this.returnStderr = returnStderr;
+	}
+}