mirror of https://github.com/databricks/cli.git
92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
|
package process
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestBackgroundUnwrapsNotFound(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
_, err := Background(ctx, []string{"/bin/meeecho", "1"})
|
||
|
|
assert.ErrorIs(t, err, os.ErrNotExist)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBackground(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
res, err := Background(ctx, []string{"echo", "1"}, WithDir("/"))
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "1", strings.TrimSpace(res))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBackgroundOnlyStdoutGetsoutOnSuccess(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
res, err := Background(ctx, []string{
|
||
|
|
"python3", "-c", "import sys; sys.stderr.write('1'); sys.stdout.write('2')",
|
||
|
|
})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "2", res)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBackgroundCombinedOutput(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
buf := bytes.Buffer{}
|
||
|
|
res, err := Background(ctx, []string{
|
||
|
|
"python3", "-c", "import sys, time; " +
|
||
|
|
`sys.stderr.write("1\n"); sys.stderr.flush(); ` +
|
||
|
|
"time.sleep(0.001); " +
|
||
|
|
"print('2', flush=True); sys.stdout.flush(); " +
|
||
|
|
"time.sleep(0.001)",
|
||
|
|
}, WithCombinedOutput(&buf))
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "2", strings.TrimSpace(res))
|
||
|
|
assert.Equal(t, "1\n2\n", strings.ReplaceAll(buf.String(), "\r", ""))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBackgroundCombinedOutputFailure(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
buf := bytes.Buffer{}
|
||
|
|
res, err := Background(ctx, []string{
|
||
|
|
"python3", "-c", "import sys, time; " +
|
||
|
|
`sys.stderr.write("1\n"); sys.stderr.flush(); ` +
|
||
|
|
"time.sleep(0.001); " +
|
||
|
|
"print('2', flush=True); sys.stdout.flush(); " +
|
||
|
|
"time.sleep(0.001); " +
|
||
|
|
"sys.exit(42)",
|
||
|
|
}, WithCombinedOutput(&buf))
|
||
|
|
var processErr *ProcessError
|
||
|
|
if assert.ErrorAs(t, err, &processErr) {
|
||
|
|
assert.Equal(t, "1", strings.TrimSpace(processErr.Stderr))
|
||
|
|
assert.Equal(t, "2", strings.TrimSpace(processErr.Stdout))
|
||
|
|
}
|
||
|
|
assert.Equal(t, "2", strings.TrimSpace(res))
|
||
|
|
assert.Equal(t, "1\n2\n", strings.ReplaceAll(buf.String(), "\r", ""))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBackgroundNoStdin(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
res, err := Background(ctx, []string{"cat"})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "", res)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBackgroundFails(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
_, err := Background(ctx, []string{"ls", "/dev/null/x"})
|
||
|
|
assert.NotNil(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBackgroundFailsOnOption(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
_, err := Background(ctx, []string{"ls", "/dev/null/x"}, func(_ context.Context, c *exec.Cmd) error {
|
||
|
|
return fmt.Errorf("nope")
|
||
|
|
})
|
||
|
|
assert.EqualError(t, err, "nope")
|
||
|
|
}
|