/*
* RHQ Management Platform
* Copyright (C) 2005-2008 Red Hat, Inc.
* All rights reserved.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License, version 2, as
* published by the Free Software Foundation, and/or the GNU Lesser
* General Public License, version 2.1, also as published by the Free
* Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License and the GNU Lesser General Public License
* for more details.
*
* You should have received a copy of the GNU General Public License
* and the GNU Lesser General Public License along with this program;
* if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
package org.rhq.core.util.exec;
import java.io.ByteArrayOutputStream;
import java.io.File;
import org.testng.annotations.Test;
/**
* Tests executing processes.
*
* @author John Mazzitelli
*/
@Test
public class ProcessExecTest {
public void testProcessExecOutputStream() {
// run multiple times to ensure race condition fixed
for (int i = 0; i < 100; i++) {
ProcessToStart start = new ProcessToStart();
setupProgram(start);
ByteArrayOutputStream baos = new ByteArrayOutputStream();
start.setOutputStream(baos);
start.setCaptureOutput(Boolean.TRUE);
start.setWaitForExit(Long.valueOf(30000));
ProcessExecutor exec = new ProcessExecutor();
ProcessExecutorResults results = exec.execute(start);
assert results.getError() == null : "Should not have failed: " + results;
assert results.getExitCode() != null : "Should have had exit code: " + results;
byte[] output = baos.toByteArray();
if (output.length == 0) {
assert false : "Should have had some output: " + results;
}
}
}
public void testProcessExecNoWait() {
ProcessToStart start = new ProcessToStart();
setupProgram(start);
ProcessExecutor exec = new ProcessExecutor();
ProcessExecutorResults results = exec.execute(start);
assert results.getError() == null : "Should not have failed: " + results;
assert results.getExitCode() == null : "Didn't wait, should not have had exit code: " + results;
start.setWaitForExit(Long.valueOf(0)); // explicitly tell it not to wait (the same as if leaving it null)
results = exec.execute(start);
assert results.getError() == null : "Should not have failed: " + results;
assert results.getExitCode() == null : "Didn't wait, should not have had exit code: " + results;
}
public void testProcessExec() {
ProcessToStart start = new ProcessToStart();
setupProgram(start);
start.setWaitForExit(Long.valueOf(30000));
ProcessExecutor exec = new ProcessExecutor();
ProcessExecutorResults results = exec.execute(start);
assert results.getError() == null : "Should not have failed - check the error, this test might not be bullet proof: "
+ results;
assert results.getExitCode() != null : "Missing exit code - check the error, this test might not be bullet proof: "
+ results;
}
/**
* Picks a program that can be executed on the test platform. This is not fool proof, might need to tweek this in
* case, for example, some platforms do not have "/bin/ls".
*
* @param start
*/
private void setupProgram(ProcessToStart start) {
// just pick some short-lived executable
if (File.separatorChar == '\\') {
start.setProgramDirectory("C:\\WINDOWS\\system32");
start.setProgramExecutable("find.exe");
start.setArguments(new String[] { "/C", "\"yaddayadda\"", "C:\\WINDOWS\\WIN.INI" });
} else {
start.setProgramDirectory("/bin");
start.setProgramExecutable("ls");
start.setArguments(new String[] { "/bin" });
}
}
}