Daniel Nephin
e13afe9c59
Signed-off-by: Daniel Nephin <dnephin@docker.com> Upstream-commit: 9db68f4dea6379b7e85ee5c6f2be080a6fcbe4ca Component: engine
56 lines
1.7 KiB
Go
56 lines
1.7 KiB
Go
package testutil
|
|
|
|
import (
|
|
"os"
|
|
"os/exec"
|
|
"runtime"
|
|
"testing"
|
|
)
|
|
|
|
func TestRunCommandPipelineWithOutputWithNotEnoughCmds(t *testing.T) {
|
|
_, _, err := RunCommandPipelineWithOutput(exec.Command("ls"))
|
|
expectedError := "pipeline does not have multiple cmds"
|
|
if err == nil || err.Error() != expectedError {
|
|
t.Fatalf("Expected an error with %s, got err:%s", expectedError, err)
|
|
}
|
|
}
|
|
|
|
func TestRunCommandPipelineWithOutputErrors(t *testing.T) {
|
|
p := "$PATH"
|
|
if runtime.GOOS == "windows" {
|
|
p = "%PATH%"
|
|
}
|
|
cmd1 := exec.Command("ls")
|
|
cmd1.Stdout = os.Stdout
|
|
cmd2 := exec.Command("anything really")
|
|
_, _, err := RunCommandPipelineWithOutput(cmd1, cmd2)
|
|
if err == nil || err.Error() != "cannot set stdout pipe for anything really: exec: Stdout already set" {
|
|
t.Fatalf("Expected an error, got %v", err)
|
|
}
|
|
|
|
cmdWithError := exec.Command("doesnotexists")
|
|
cmdCat := exec.Command("cat")
|
|
_, _, err = RunCommandPipelineWithOutput(cmdWithError, cmdCat)
|
|
if err == nil || err.Error() != `starting doesnotexists failed with error: exec: "doesnotexists": executable file not found in `+p {
|
|
t.Fatalf("Expected an error, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestRunCommandPipelineWithOutput(t *testing.T) {
|
|
//TODO: Should run on Solaris
|
|
if runtime.GOOS == "solaris" {
|
|
t.Skip()
|
|
}
|
|
cmds := []*exec.Cmd{
|
|
// Print 2 characters
|
|
exec.Command("echo", "-n", "11"),
|
|
// Count the number or char from stdin (previous command)
|
|
exec.Command("wc", "-m"),
|
|
}
|
|
out, exitCode, err := RunCommandPipelineWithOutput(cmds...)
|
|
expectedOutput := "2\n"
|
|
if out != expectedOutput || exitCode != 0 || err != nil {
|
|
t.Fatalf("Expected %s for commands %v, got out:%s, exitCode:%d, err:%v", expectedOutput, cmds, out, exitCode, err)
|
|
}
|
|
}
|