You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
1.7 KiB
Go
89 lines
1.7 KiB
Go
4 years ago
|
package main
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"fmt"
|
||
|
"log"
|
||
|
"os"
|
||
|
"os/exec"
|
||
|
"strings"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func TestMain(m *testing.M) {
|
||
|
switch os.Getenv("TEST_MAIN") {
|
||
|
case "":
|
||
|
os.Exit(m.Run())
|
||
|
case "echo":
|
||
|
fmt.Println(strings.Join(os.Args[1:], " "))
|
||
|
case "exit":
|
||
|
os.Exit(42)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestExec(t *testing.T) {
|
||
|
logger := log.New(bytes.NewBuffer([]byte{}), "", 0)
|
||
|
commander := DefaultCommander{logger}
|
||
|
|
||
|
cmd := exec.Command(os.Args[0], "42")
|
||
|
cmd.Env = append(os.Environ(), "TEST_MAIN=echo")
|
||
|
|
||
|
output, err := commander.Exec(cmd)
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error %v", err)
|
||
|
}
|
||
|
|
||
|
if output != "42" {
|
||
|
t.Errorf("expected 42, got %q", output)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestExecError(t *testing.T) {
|
||
|
logger := log.New(bytes.NewBuffer([]byte{}), "", 0)
|
||
|
commander := DefaultCommander{logger}
|
||
|
|
||
|
cmd := exec.Command(os.Args[0], "42")
|
||
|
cmd.Env = append(os.Environ(), "TEST_MAIN=exit")
|
||
|
|
||
|
_, err := commander.Exec(cmd)
|
||
|
if err == nil {
|
||
|
t.Errorf("expected error")
|
||
|
}
|
||
|
|
||
|
got := cmd.ProcessState.ExitCode()
|
||
|
if got != 42 {
|
||
|
t.Errorf("expected %d, got %d", 42, got)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestExecSilently(t *testing.T) {
|
||
|
logger := log.New(bytes.NewBuffer([]byte{}), "", 0)
|
||
|
commander := DefaultCommander{logger}
|
||
|
|
||
|
cmd := exec.Command(os.Args[0], "42")
|
||
|
cmd.Env = append(os.Environ(), "TEST_MAIN=echo")
|
||
|
|
||
|
err := commander.ExecSilently(cmd)
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error %v", err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestExecSilentlyError(t *testing.T) {
|
||
|
logger := log.New(bytes.NewBuffer([]byte{}), "", 0)
|
||
|
commander := DefaultCommander{logger}
|
||
|
|
||
|
cmd := exec.Command(os.Args[0], "42")
|
||
|
cmd.Env = append(os.Environ(), "TEST_MAIN=exit")
|
||
|
|
||
|
err := commander.ExecSilently(cmd)
|
||
|
if err == nil {
|
||
|
t.Errorf("expected error")
|
||
|
}
|
||
|
|
||
|
got := cmd.ProcessState.ExitCode()
|
||
|
if got != 42 {
|
||
|
t.Errorf("expected %d, got %d", 42, got)
|
||
|
}
|
||
|
}
|