102 lines
3.2 KiB
Go
102 lines
3.2 KiB
Go
// Copyright © 2022 Roberto Hidalgo <chinampa@un.rob.mx>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
package exec_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
. "git.rob.mx/nidito/chinampa/pkg/exec"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var logger = logrus.WithContext(context.Background())
|
|
|
|
func TestSubshellExec(t *testing.T) {
|
|
ExecFunc = WithSubshell
|
|
stdout, directive, err := Exec("test-command", []string{"bash", "-c", `echo "stdout"; echo "stderr" >&2;`}, []string{}, 1*time.Second, logger)
|
|
if err != nil {
|
|
t.Fatalf("good subshell errored: %v", err)
|
|
}
|
|
|
|
if len(stdout) != 1 && stdout[0] == "stdout" {
|
|
t.Fatalf("good subshell returned wrong stdout: %v", stdout)
|
|
}
|
|
|
|
if directive != cobra.ShellCompDirectiveDefault {
|
|
t.Fatalf("good subshell returned wrong directive: %v", directive)
|
|
}
|
|
|
|
stdout, directive, err = Exec("test-command", []string{"bash", "-c", `echo "stdout"; echo "stderr" >&2; exit 2`}, []string{}, 1*time.Second, logger)
|
|
if err == nil {
|
|
t.Fatalf("bad subshell did not error; stdout: %v", stdout)
|
|
}
|
|
|
|
if len(stdout) != 0 {
|
|
t.Fatalf("bad subshell returned non-empty stdout: %v", stdout)
|
|
}
|
|
|
|
if directive != cobra.ShellCompDirectiveError {
|
|
t.Fatalf("bad subshell returned wrong directive: %v", directive)
|
|
}
|
|
}
|
|
|
|
func TestExecTimesOut(t *testing.T) {
|
|
ExecFunc = func(ctx context.Context, env []string, executable string, args ...string) (bytes.Buffer, bytes.Buffer, error) {
|
|
time.Sleep(100 * time.Nanosecond)
|
|
return bytes.Buffer{}, bytes.Buffer{}, context.DeadlineExceeded
|
|
}
|
|
_, _, err := Exec("test-command", []string{"bash", "-c", "sleep", "2"}, []string{}, 10*time.Nanosecond, logger)
|
|
if err == nil {
|
|
t.Fatalf("timeout didn't happen after 10ms: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestExecWorksFine(t *testing.T) {
|
|
ExecFunc = func(ctx context.Context, env []string, executable string, args ...string) (bytes.Buffer, bytes.Buffer, error) {
|
|
var out bytes.Buffer
|
|
fmt.Fprint(&out, strings.Join([]string{
|
|
"a",
|
|
"b",
|
|
"c",
|
|
}, "\n"))
|
|
return out, bytes.Buffer{}, nil
|
|
}
|
|
args := []string{"a", "b", "c"}
|
|
res, directive, err := Exec("test-command", append([]string{"bash", "-c", "echo"}, args...), []string{}, 1*time.Second, logger)
|
|
if err != nil {
|
|
t.Fatalf("good command failed: %v", err)
|
|
}
|
|
|
|
if directive != 0 {
|
|
t.Fatalf("good command resulted in wrong directive, expected %d, got %d", 0, directive)
|
|
}
|
|
|
|
if strings.Join(args, "-") != strings.Join(res, "-") {
|
|
t.Fatalf("good command resulted in wrong results, expected %v, got %v", res, args)
|
|
}
|
|
}
|
|
|
|
func TestExecErrors(t *testing.T) {
|
|
ExecFunc = func(ctx context.Context, env []string, executable string, args ...string) (bytes.Buffer, bytes.Buffer, error) {
|
|
return bytes.Buffer{}, bytes.Buffer{}, fmt.Errorf("bad command is bad")
|
|
}
|
|
res, directive, err := Exec("test-command", []string{"bash", "-c", "bad-command"}, []string{}, 1*time.Second, logger)
|
|
if err == fmt.Errorf("bad command is bad") {
|
|
t.Fatalf("bad command didn't fail: %v", res)
|
|
}
|
|
|
|
if directive != cobra.ShellCompDirectiveError {
|
|
t.Fatalf("bad command resulted in wrong directive, expected %d, got %d", cobra.ShellCompDirectiveError, directive)
|
|
}
|
|
|
|
if len(res) > 0 {
|
|
t.Fatalf("bad command returned values, got %v", res)
|
|
}
|
|
}
|