1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
// Copyright Earl Warren <contact@earl-warren.org>
// Copyright Loïc Dachary <loic@dachary.org>
// SPDX-License-Identifier: MIT
package util
import (
"context"
"path/filepath"
"strings"
"testing"
"github.com/stretchr/testify/assert"
)
func TestCommand(t *testing.T) {
something := "SOMETHING"
assert.EqualValues(t, something, Command(context.Background(), nil, "echo", "-n", something))
assert.Panics(t, func() { Command(context.Background(), nil, "false") })
}
func TestCommandWithOptions(t *testing.T) {
assert.EqualValues(t, "", CommandWithOptions(context.Background(), CommandOptions{
ExitCodes: []int{0, 1},
}, "false"))
}
func TestCommandWithErr(t *testing.T) {
assert.NoError(t, CommandWithErr(context.Background(), CommandOptions{}, "true"))
assert.Error(t, CommandWithErr(context.Background(), CommandOptions{}, "false"))
}
func TestCommandTimeout(t *testing.T) {
t.Skip("Requires network access")
tmp := t.TempDir()
ctx, cancel := context.WithCancel(context.Background())
go func() {
// blocks forever because of the firewall at 4.4.4.4 and
// the git clone process forks a git-remote-https child process
assert.PanicsWithValue(t, "killed", func() {
_ = Command(ctx, nil, "git", "clone", "https://4.4.4.4", filepath.Join(tmp, "something"))
})
}()
pattern := "git-remote-https origin https://4.4.4.4"
ps := []string{"-x", "-o", "pid,ppid,pgid,args"}
Retry(func() {
out := Command(context.Background(), nil, "ps", ps...)
if !strings.Contains(out, pattern) {
panic(out + " does not contain " + pattern)
}
}, 5)
cancel()
<-ctx.Done()
Retry(func() {
out := Command(context.Background(), nil, "ps", ps...)
if strings.Contains(out, pattern) {
panic(out + " contains " + pattern)
}
}, 5)
}
|
