Apache
/
jclouds
mirror of https://github.com/apache/jclouds.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

new demo javaoverssh 

git-svn-id: http://jclouds.googlecode.com/svn/trunk@2566 3d8758e0-26b5-11de-8745-db77d3ebf521
This commit is contained in:
adrian.f.cole 2010-01-02 03:29:50 +00:00
parent edc6ccc503
commit 26b2e20640
9 changed files with 900 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
aws/demos/createlamp/src/main/java/org/jclouds/aws/ec2/demos/createlamp/MainApp.java
View File

@ -157,10 +157,10 @@ public class MainApp {
}
static RunningInstance runInstance(EC2Client client, String securityGroupName, String keyPairName) {
String script = new ScriptBuilder() // lamp install script
.addStatement(exec("runurl run.alestic.com/apt/upgrade"))//
.addStatement(exec("runurl run.alestic.com/install/lamp"))//
.addStatement(exec("apt-get -y install openjdk-6-jdk"))// no license agreement!
.build(OsFamily.UNIX);
System.out.printf("%d: running instance%n", System.currentTimeMillis());
@ -183,8 +183,6 @@ public class MainApp {
// create utilities that wait for the instance to finish
RetryablePredicate<RunningInstance> runningTester = new RetryablePredicate<RunningInstance>(
new InstanceStateRunning(client.getInstanceServices()), 180, 5, TimeUnit.SECONDS);
RetryablePredicate<InetSocketAddress> socketTester = new RetryablePredicate<InetSocketAddress>(
new SocketOpen(), 180, 1, TimeUnit.SECONDS);
System.out.printf("%d: %s awaiting instance to run %n", System.currentTimeMillis(), instance
.getId());
@ -193,6 +191,8 @@ public class MainApp {
instance = findInstanceById(client, instance.getId());
RetryablePredicate<InetSocketAddress> socketTester = new RetryablePredicate<InetSocketAddress>(
new SocketOpen(), 300, 1, TimeUnit.SECONDS);
System.out.printf("%d: %s awaiting ssh service to start%n", System.currentTimeMillis(),
instance.getIpAddress());
if (!socketTester.apply(new InetSocketAddress(instance.getIpAddress(), 22)))

87
tools/antcontrib/pom.xml Normal file
View File

@ -0,0 +1,87 @@
<?xml version="1.0" encoding="UTF-8"?>
<!--
Copyright (C) 2009 Cloud Conscious, LLC.
<info@cloudconscious.com>
====================================================================
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.
====================================================================
-->
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<parent>
<groupId>org.jclouds</groupId>
<artifactId>jclouds-tools-project</artifactId>
<version>1.0-SNAPSHOT</version>
</parent>
<artifactId>jclouds-antcontrib</artifactId>
<name>Ant test</name>
<properties>
<jclouds.test.listener></jclouds.test.listener>
</properties>
<dependencies>
<dependency>
<groupId>org.jboss.shrinkwrap</groupId>
<artifactId>shrinkwrap-impl-base</artifactId>
<version>1.0.0-alpha-3</version>
</dependency>
<dependency>
<groupId>${project.groupId}</groupId>
<artifactId>jclouds-scriptbuilder</artifactId>
<version>${project.version}</version>
</dependency>
<dependency>
<groupId>org.apache.ant</groupId>
<artifactId>ant</artifactId>
<version>1.7.1</version>
<scope>provided</scope>
</dependency>
<dependency>
<groupId>org.apache.ant</groupId>
<artifactId>ant-jsch</artifactId>
<version>1.7.1</version>
</dependency>
<dependency>
<groupId>com.jcraft</groupId>
<artifactId>jsch</artifactId>
<version>0.1.42</version>
</dependency>
</dependencies>
<build>
<finalName>${project.artifactId}</finalName>
<plugins>
<plugin>
<artifactId>maven-assembly-plugin</artifactId>
<executions>
<execution>
<phase>package</phase>
<goals>
<goal>single</goal>
</goals>
<configuration>
<descriptorRefs>
<descriptorRef>jar-with-dependencies</descriptorRef>
</descriptorRefs>
</configuration>
</execution>
</executions>
</plugin>
</plugins>
</build>
</project>

25
tools/antcontrib/samples/javaoverssh/README.txt Executable file
View File

@ -0,0 +1,25 @@
====
Copyright (C) 2009 Cloud Conscious, LLC. <info@cloudconscious.com>
====================================================================
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.
====================================================================
====
this is a simple ant script that executes a java command on a remote machine via ssh
1. find or download a copy of jclouds-antcontrib-1.0-SNAPSHOT-jar-with-dependencies.jar
- ex. ~/.m2/repository/org/jclouds/jclouds-antcontrib/1.0-SNAPSHOT/jclouds-antcontrib-1.0-SNAPSHOT-jar-with-dependencies.jar
- ex. curl http://jclouds.rimuhosting.com/maven2/snapshots/org/jclouds/jclouds-antcontrib/1.0-SNAPSHOT/jclouds-antcontrib-1.0-20091215.023231-1-jar-with-dependencies.jar >jclouds-antcontrib-all.jar
2. invoke ant, adding the library above, and passing the properties 'host' 'username' 'keyfile' which corresponds to your remote credentials
- ex. ant -lib ~/.m2/repository/org/jclouds/jclouds-antcontrib/1.0-SNAPSHOT/jclouds-antcontrib-1.0-SNAPSHOT-jar-with-dependencies.jar -Dhost=67.202.42.237 -Dusername=root -Dkeyfile=$HOME/.ssh/id_dsa
- ex. ant -lib jclouds-antcontrib-all.jar -Dhost=67.202.42.237 -Dusername=root -Dkeyfile=$HOME/.ssh/id_dsa

83
tools/antcontrib/samples/javaoverssh/build.xml Normal file
View File

@ -0,0 +1,83 @@
<?xml version="1.0"?>
<!--
Copyright (C) 2009 Cloud Conscious, LLC. <info@cloudconscious.com>
====================================================================
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.
====================================================================
-->
<project name="javassh" default="javassh" basedir=".">
<description>
simple example build file
</description>
<!-- set global properties for this build -->
<property name="src" location="src"/>
<property name="build" location="build"/>
<property name="dist" location="dist"/>
<input
message="Please enter the ip or hostname of the ssh machine"
addproperty="host"
/>
<input
message="Please enter the user you will connect as"
addproperty="username"
/>
<input
message="Please enter the path to the dsa key"
addproperty="keyfile"
/>
<input
message="Please enter the strings you'd like to test"
addproperty="line"
/>
<target name="init">
<!-- Create the time stamp -->
<tstamp/>
<!-- Create the build directory structure used by compile -->
<mkdir dir="${build}"/>
</target>
<target name="compile" depends="init" description="compile the source ">
<!-- Compile the java code from ${src} into ${build} -->
<javac srcdir="${src}" destdir="${build}"/>
</target>
<target name="clean" description="clean up">
<!-- Delete the ${build} and ${dist} directory trees -->
<delete dir="${build}"/>
<delete dir="${dist}"/>
</target>
<taskdef name="javassh" classname="org.jclouds.tools.ant.JavaOverSsh"/>
<target name="javassh" depends="compile" description="remote execute the java command">
<echo message="normal java task"/>
<java classname="TestClass" classpath="${build}" dir="${user.dir}" >
<arg line="${line}"/>
</java>
<echo message="java task over ssh"/>
<javassh classname="TestClass" classpath="${build}" dir="${user.dir}" host="${host}" username="${username}" keyfile="${keyfile}" trust="true" remotedir="/tmp/test" >
<arg line="${line}"/>
</javassh>
</target>
</project>

16
tools/antcontrib/samples/javaoverssh/src/TestClass.java Normal file
View File

@ -0,0 +1,16 @@
import java.io.File;
import java.util.Arrays;
public class TestClass {
public static void main(String... args) {
File cwd = new File(System.getProperty("user.dir"));
System.out.println("children:");
for (File child : cwd.listFiles())
System.out.println(" " + child);
System.out.println("what you wrote:");
System.out.println(Arrays.asList(args));
System.err.println("this is the error stream");
System.out.println("will exit 3:");
System.exit(3);
}
}

560
tools/antcontrib/src/main/java/org/jclouds/tools/ant/JavaOverSsh.java Normal file
View File

@ -0,0 +1,560 @@
/**
*
* Copyright (C) 2009 Cloud Conscious, LLC. <info@cloudconscious.com>
*
* ====================================================================
* 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.
* ====================================================================
*/
package org.jclouds.tools.ant;
import static com.google.common.base.Preconditions.checkNotNull;
import static org.jclouds.scriptbuilder.domain.Statements.exec;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.util.List;
import java.util.zip.ZipFile;
import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Project;
import org.apache.tools.ant.Task;
import org.apache.tools.ant.taskdefs.Java;
import org.apache.tools.ant.taskdefs.optional.ssh.SSHUserInfo;
import org.apache.tools.ant.types.CommandlineJava;
import org.apache.tools.ant.types.Environment;
import org.apache.tools.ant.util.KeepAliveOutputStream;
import org.apache.tools.ant.util.TeeOutputStream;
import org.jboss.shrinkwrap.api.Archives;
import org.jboss.shrinkwrap.api.exporter.ZipExporter;
import org.jboss.shrinkwrap.api.importer.ExplodedImporter;
import org.jboss.shrinkwrap.api.importer.ZipImporter;
import org.jboss.shrinkwrap.api.spec.JavaArchive;
import org.jclouds.scriptbuilder.domain.OsFamily;
import org.jclouds.scriptbuilder.domain.Statement;
import org.jclouds.scriptbuilder.domain.StatementList;
import com.google.common.collect.Lists;
import com.google.common.io.Closeables;
import com.jcraft.jsch.ChannelExec;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
/**
* Ported from Jsch SSHExec task.
*
* @author Adrian Cole
*/
public class JavaOverSsh extends Java {
/** Default listen port for SSH daemon */
private static final int SSH_PORT = 22;
private String jvm = "/usr/bin/java";
private File localDirectory;
private File remoteDirectory;
private Environment env = new Environment();
private String host;
private String knownHosts;
private int port = SSH_PORT;
private SSHUserInfo userInfo = new SSHUserInfo();
private OsFamily osFamily = OsFamily.UNIX;
/** units are milliseconds, default is 0=infinite */
private long maxwait = 0;
/** for waiting for the command to finish */
private Thread watchDog = null;
private File outputFile;
private String outputProperty;
private String resultProperty;
private File errorFile;
private String errorProperty;
private boolean append;
private static final String TIMEOUT_MESSAGE = "Timeout period exceeded, connection dropped.";
public JavaOverSsh() {
super();
setFork(true);
}
public JavaOverSsh(Task owner) {
super(owner);
setFork(true);
}
@Override
public int executeJava() throws BuildException {
String command = convertJavaToScript(getCommandLine());
InputStream classpathJar = (getCommandLine().getClasspath() != null) ? makeClasspathJarOrNull(getCommandLine()
.getClasspath().list())
: null;
InputStream bootClasspathJar = (getCommandLine().getBootclasspath() != null) ? makeClasspathJarOrNull(getCommandLine()
.getBootclasspath().list())
: null;
InputStream currentDirectoryZip = Archives.create("cwd.zip", ZipExporter.class).as(
ExplodedImporter.class).importDirectory(localDirectory).as(ZipExporter.class)
.exportZip();
if (getHost() == null) {
throw new BuildException("Host is required.");
}
if (getUserInfo().getName() == null) {
throw new BuildException("Username is required.");
}
if (getUserInfo().getKeyfile() == null && getUserInfo().getPassword() == null) {
throw new BuildException("Password or Keyfile is required.");
}
Session session = null;
try {
// execute the command
session = openSession();
session.setTimeout((int) maxwait);
ChannelSftp sftp = null;
sftp = (ChannelSftp) session.openChannel("sftp");
sftp.connect();
sftp.put(currentDirectoryZip, remoteDirectory + "/cwd.zip");
Closeables.closeQuietly(currentDirectoryZip);
if (classpathJar != null || bootClasspathJar != null) {
if (classpathJar != null) {
sftp.put(classpathJar, remoteDirectory + "/classpath.jar");
Closeables.closeQuietly(classpathJar);
}
if (bootClasspathJar != null) {
sftp.put(classpathJar, remoteDirectory + "/boot-classpath.jar");
Closeables.closeQuietly(classpathJar);
}
}
final ChannelExec channel = (ChannelExec) session.openChannel("exec");
channel.setCommand(command);
ByteArrayOutputStream out = new ByteArrayOutputStream();
TeeOutputStream teeOut = new TeeOutputStream(out, new KeepAliveOutputStream(System.out));
ByteArrayOutputStream err = new ByteArrayOutputStream();
TeeOutputStream teeErr = new TeeOutputStream(err, new KeepAliveOutputStream(System.err));
channel.setOutputStream(teeOut);
channel.setExtOutputStream(teeOut);
channel.setErrStream(teeErr);
channel.connect();
// wait for it to finish
watchDog = new Thread() {
public void run() {
while (!channel.isEOF()) {
if (watchDog == null) {
return;
}
try {
sleep(500);
} catch (Exception e) {
// ignored
}
}
}
};
watchDog.start();
watchDog.join(maxwait);
if (watchDog.isAlive()) {
// ran out of time
throw new BuildException(TIMEOUT_MESSAGE);
} else {
// completed successfully
return writeOutputAndReturnExitStatus(channel, out, err);
}
} catch (BuildException e) {
throw e;
} catch (JSchException e) {
if (e.getMessage().indexOf("session is down") >= 0) {
throw new BuildException(TIMEOUT_MESSAGE, e);
} else {
throw new BuildException(e);
}
} catch (Exception e) {
throw new BuildException(e);
} finally {
if (session != null && session.isConnected()) {
session.disconnect();
}
}
}
InputStream makeClasspathJarOrNull(String... paths) {
if (paths != null && paths.length > 0) {
JavaArchive classpathArchive = Archives.create("classpath.jar", JavaArchive.class);
for (String path : paths) {
File file = new File(path);
if (file.exists()) {
if (file.isFile()) {
try {
classpathArchive.as(ZipImporter.class).importZip(
new ZipFile(file.getAbsolutePath()));
} catch (IOException e) {
throw new BuildException(e);
}
} else {
classpathArchive.as(ExplodedImporter.class).importDirectory(file);
}
}
}
return classpathArchive.as(ZipExporter.class).exportZip();
}
return null;
}
private int writeOutputAndReturnExitStatus(final ChannelExec channel, ByteArrayOutputStream out,
ByteArrayOutputStream err) throws IOException {
if (outputProperty != null) {
getProject().setProperty(outputProperty, out.toString());
}
if (outputFile != null) {
writeToFile(err.toString(), append, outputFile);
}
if (errorProperty != null) {
getProject().setProperty(errorProperty, err.toString());
}
if (errorFile != null) {
writeToFile(out.toString(), append, errorFile);
}
if (resultProperty != null) {
getProject().setProperty(resultProperty, channel.getExitStatus() + "");
}
return channel.getExitStatus();
}
/**
* Writes a string to a file. If destination file exists, it may be overwritten depending on the
* "append" value.
*
* @param from
* string to write
* @param to
* file to write to
* @param append
* if true, append to existing file, else overwrite
* @exception Exception
* most likely an IOException
*/
private void writeToFile(String from, boolean append, File to) throws IOException {
FileWriter out = null;
try {
out = new FileWriter(to.getAbsolutePath(), append);
StringReader in = new StringReader(from);
char[] buffer = new char[8192];
int bytesRead;
while (true) {
bytesRead = in.read(buffer);
if (bytesRead == -1) {
break;
}
out.write(buffer, 0, bytesRead);
}
out.flush();
} finally {
if (out != null) {
out.close();
}
}
}
String convertJavaToScript(CommandlineJava commandLine) {
checkNotNull(jvm, "jvm must be set");
checkNotNull(localDirectory, "dir must be set");
checkNotNull(remoteDirectory, "remotedir must be set");
List<Statement> statements = Lists.newArrayList();
String[] environment = env.getVariables();
if (environment != null) {
for (int i = 0; i < environment.length; i++) {
log("Setting environment variable: " + environment[i], Project.MSG_VERBOSE);
statements.add(exec("{export} " + environment[i]));
}
}
statements.add(exec("{cd} " + remoteDirectory));
statements.add(exec("jar -xf cwd.zip"));
StringBuilder commandBuilder = new StringBuilder(jvm);
if (commandLine.getBootclasspath() != null
&& commandLine.getBootclasspath().list().length > 0) {
commandBuilder.append(" -Xbootclasspath:boot-classpath.jar");
}
if (commandLine.getVmCommand().getArguments() != null
&& commandLine.getVmCommand().getArguments().length > 0) {
commandBuilder.append(" ");
String[] variables = commandLine.getVmCommand().getArguments();
for (int i = 0; i < variables.length; i++) {
commandBuilder.append(variables[i]);
if (i + 1 < variables.length)
commandBuilder.append(" ");
}
}
if (commandLine.getClasspath() != null && commandLine.getClasspath().list().length > 0) {
commandBuilder.append(" -cp classpath.jar");
}
if (commandLine.getSystemProperties() != null
&& commandLine.getSystemProperties().getVariables() != null
&& commandLine.getSystemProperties().getVariables().length > 0) {
commandBuilder.append(" ");
String[] variables = commandLine.getSystemProperties().getVariables();
for (int i = 0; i < variables.length; i++) {
commandBuilder.append(variables[i]);
if (i + 1 < variables.length)
commandBuilder.append(" ");
}
}
commandBuilder.append(" ").append(commandLine.getClassname());
if (commandLine.getJavaCommand().getArguments() != null
&& commandLine.getJavaCommand().getArguments().length > 0) {
commandBuilder.append(" ");
String[] variables = commandLine.getJavaCommand().getArguments();
for (int i = 0; i < variables.length; i++) {
commandBuilder.append(variables[i]);
if (i + 1 < variables.length)
commandBuilder.append(" ");
}
}
statements.add(exec(commandBuilder.toString()));
String command = new StatementList(statements).render(osFamily);
return command;
}
@Override
public void addEnv(Environment.Variable var) {
env.addVariable(var);
}
@Override
public void setDir(File localDir) {
this.localDirectory = checkNotNull(localDir, "dir");
}
public void setRemotedir(File remotedir) {
this.remoteDirectory = checkNotNull(remotedir, "remotedir");
}
@Override
public void setFork(boolean fork) {
if (!fork)
throw new IllegalArgumentException("this only operates when fork is set");
}
@Override
public void setJvm(String jvm) {
this.jvm = checkNotNull(jvm, "jvm");
}
/**
* Remote host, either DNS name or IP.
*
* @param host
* The new host value
*/
public void setHost(String host) {
this.host = host;
}
/**
* Get the host.
*
* @return the host
*/
public String getHost() {
return host;
}
/**
* Username known to remote host.
*
* @param username
* The new username value
*/
public void setUsername(String username) {
userInfo.setName(username);
}
/**
* Sets the password for the user.
*
* @param password
* The new password value
*/
public void setPassword(String password) {
userInfo.setPassword(password);
}
/**
* Sets the keyfile for the user.
*
* @param keyfile
* The new keyfile value
*/
public void setKeyfile(String keyfile) {
userInfo.setKeyfile(keyfile);
}
/**
* Sets the passphrase for the users key.
*
* @param passphrase
* The new passphrase value
*/
public void setPassphrase(String passphrase) {
userInfo.setPassphrase(passphrase);
}
/**
* Sets the path to the file that has the identities of all known hosts. This is used by SSH
* protocol to validate the identity of the host. The default is
* <i>${user.home}/.ssh/known_hosts</i>.
*
* @param knownHosts
* a path to the known hosts file.
*/
public void setKnownhosts(String knownHosts) {
this.knownHosts = knownHosts;
}
/**
* Setting this to true trusts hosts whose identity is unknown.
*
* @param yesOrNo
* if true trust the identity of unknown hosts.
*/
public void setTrust(boolean yesOrNo) {
userInfo.setTrust(yesOrNo);
}
/**
* Changes the port used to connect to the remote host.
*
* @param port
* port number of remote host.
*/
public void setPort(int port) {
this.port = port;
}
/**
* Get the port attribute.
*
* @return the port
*/
public int getPort() {
return port;
}
/**
* Initialize the task. This initializizs the known hosts and sets the default port.
*
* @throws BuildException
* on error
*/
public void init() throws BuildException {
super.init();
this.knownHosts = System.getProperty("user.home") + "/.ssh/known_hosts";
this.port = SSH_PORT;
}
/**
* Open an ssh seession.
*
* @return the opened session
* @throws JSchException
* on error
*/
protected Session openSession() throws JSchException {
JSch jsch = new JSch();
if (null != userInfo.getKeyfile()) {
jsch.addIdentity(userInfo.getKeyfile());
}
if (!userInfo.getTrust() && knownHosts != null) {
log("Using known hosts: " + knownHosts, Project.MSG_DEBUG);
jsch.setKnownHosts(knownHosts);
}
Session session = jsch.getSession(userInfo.getName(), host, port);
session.setUserInfo(userInfo);
log("Connecting to " + host + ":" + port);
session.connect();
return session;
}
/**
* Get the user information.
*
* @return the user information
*/
protected SSHUserInfo getUserInfo() {
return userInfo;
}
/**
* The connection can be dropped after a specified number of milliseconds. This is sometimes
* useful when a connection may be flaky. Default is 0, which means &quot;wait forever&quot;.
*
* @param timeout
* The new timeout value in seconds
*/
public void setTimeout(long timeout) {
maxwait = timeout;
}
@Override
public void setError(File out) {
this.errorFile = out;
}
@Override
public void setErrorProperty(String errorProp) {
this.errorProperty = errorProp;
}
@Override
public void setOutput(File out) {
this.outputFile = out;
}
@Override
public void setOutputproperty(String outputProp) {
this.outputProperty = outputProp;
}
@Override
public void setResultProperty(String resultProperty) {
this.resultProperty = resultProperty;
}
@Override
public void setAppend(boolean append) {
this.append = append;
}
}

107
tools/antcontrib/src/test/java/org/jclouds/tools/ant/JavaOverSshTest.java Normal file
View File

@ -0,0 +1,107 @@
/**
*
* Copyright (C) 2009 Cloud Conscious, LLC. <info@cloudconscious.com>
*
* ====================================================================
* 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.
* ====================================================================
*/
package org.jclouds.tools.ant;
import static org.testng.Assert.assertEquals;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.net.InetAddress;
import org.apache.tools.ant.Project;
import org.apache.tools.ant.types.Environment;
import org.apache.tools.ant.types.Path;
import org.apache.tools.ant.types.Environment.Variable;
import org.testng.annotations.Test;
/**
* @author Adrian Cole
*/
@Test(groups = "unit", testName = "jclouds.JavaOverSshTest")
public class JavaOverSshTest {
public void testFull() throws SecurityException, NoSuchMethodException {
JavaOverSsh task = createTask();
assertEquals(
String
.format(
"cd /tmp/foo\njar -xf cwd.zip\n%s -Xms256 -cp classpath.jar -Dfooble=baz -Dfoo=bar org.jclouds.tools.ant.TestClass hello world\n",
System.getProperty("java.home") + "/bin/java", System
.getProperty("user.dir")), task.convertJavaToScript(task
.getCommandLine()));
}
private JavaOverSsh createTask() {
Project p = new Project();
JavaOverSsh task = new JavaOverSsh();
task.setProject(p);
task.setClassname(TestClass.class.getName());
task.createClasspath().add(new Path(p, "target/test-classes"));
Variable prop1 = new Environment.Variable();
prop1.setKey("fooble");
prop1.setValue("baz");
task.addSysproperty(prop1);
Variable prop2 = new Environment.Variable();
prop2.setKey("foo");
prop2.setValue("bar");
task.addSysproperty(prop2);
task.createJvmarg().setValue("-Xms256");
task.createArg().setValue("hello");
task.createArg().setValue("world");
task.setDir(new File(System.getProperty("user.dir")));
task.setRemotedir(new File("/tmp/foo"));
task.setFork(true);
task.setJvm(System.getProperty("java.home") + "/bin/java");
return task;
}
@Test(enabled = false, groups = { "live" })
public void testSsh() throws NumberFormatException, FileNotFoundException, IOException {
String sshHost = System.getProperty("jclouds.test.ssh.host");
String sshPort = System.getProperty("jclouds.test.ssh.port");
String sshUser = System.getProperty("jclouds.test.ssh.username");
String sshPass = System.getProperty("jclouds.test.ssh.password");
String sshKeyFile = System.getProperty("jclouds.test.ssh.keyfile");
int port = (sshPort != null) ? Integer.parseInt(sshPort) : 22;
InetAddress host = (sshHost != null) ? InetAddress.getByName(sshHost) : InetAddress
.getLocalHost();
JavaOverSsh task = createTask();
task.setHost(host.getHostAddress());
task.setPort(port);
task.setTrust(true);
task.setUsername(sshUser);
if (sshKeyFile != null && !sshKeyFile.trim().equals("")) {
task.setKeyfile(sshKeyFile);
} else {
task.setPassword(sshPass);
}
task.setOutputproperty("out");
task.setErrorProperty("err");
task.setResultProperty("result");
task.execute();
assertEquals(task.getProject().getProperty("out"), "[hello, world]\n");
assertEquals(task.getProject().getProperty("err"), "err\n");
assertEquals(task.getProject().getProperty("result"), "3");
}
}

18
tools/antcontrib/src/test/java/org/jclouds/tools/ant/TestClass.java Normal file
View File

@ -0,0 +1,18 @@
package org.jclouds.tools.ant;
import java.io.File;
import java.util.Arrays;
public class TestClass {
public static void main(String... args) {
File cwd = new File(System.getProperty("user.dir"));
System.out.println("children:");
for (File child : cwd.listFiles())
System.out.println(" " + child);
System.out.println("what you wrote:");
System.out.println(Arrays.asList(args));
System.err.println("this is the error stream");
System.out.println("will exit 3:");
System.exit(3);
}
}

1
tools/pom.xml
View File

@ -34,6 +34,7 @@
<modules>
<module>ant-plugin</module>
<module>getpath</module>
<module>antcontrib</module>
<module>vfs</module>
</modules>
</project>