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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
//go:build !gitaly_test_sha256
package blackbox
import (
"errors"
"net/url"
"testing"
"time"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/v15/internal/helper/duration"
)
func TestConfigParseFailures(t *testing.T) {
testCases := []struct {
desc string
in string
expectedErr error
}{
{
desc: "empty config",
expectedErr: errors.New("missing prometheus_listen_addr"),
},
{
desc: "probe without name",
in: "prometheus_listen_addr = 'foo'\n[[probe]]\n",
expectedErr: errors.New("all probes must have a 'name' attribute"),
},
{
desc: "unsupported probe url",
in: "prometheus_listen_addr = 'foo'\n[[probe]]\nname='foo'\nurl='ssh://not:supported'",
expectedErr: &url.Error{
Op: "parse",
URL: "ssh://not:supported",
Err: errors.New("invalid port \":supported\" after host"),
},
},
{
desc: "missing probe url",
in: "prometheus_listen_addr = 'foo'\n[[probe]]\nname='foo'\n",
expectedErr: errors.New("unsupported probe URL scheme: "),
},
{
desc: "negative sleep",
in: "prometheus_listen_addr = 'foo'\nsleep=-1\n[[probe]]\nname='foo'\nurl='http://foo/bar'",
expectedErr: errors.New("sleep time is less than 0"),
},
{
desc: "no listen addr",
in: "[[probe]]\nname='foo'\nurl='http://foo/bar'",
expectedErr: errors.New("missing prometheus_listen_addr"),
},
{
desc: "invalid probe type",
in: "prometheus_listen_addr = 'foo'\n[[probe]]\nname='foo'\nurl='http://foo/bar'\ntype='foo'",
expectedErr: errors.New("unsupported probe type: \"foo\""),
},
{
desc: "valid configuration",
in: "prometheus_listen_addr = 'foo'\n[[probe]]\nname='foo'\nurl='http://foo/bar'\n",
},
{
desc: "valid configuration with explicit type",
in: "prometheus_listen_addr = 'foo'\n[[probe]]\nname='foo'\nurl='http://foo/bar'\ntype='fetch'",
},
}
for _, tc := range testCases {
t.Run(tc.desc, func(t *testing.T) {
_, err := ParseConfig(tc.in)
require.Equal(t, tc.expectedErr, err)
})
}
}
func TestConfigSleep(t *testing.T) {
testCases := []struct {
desc string
in string
out duration.Duration
}{
{
desc: "default sleep time",
out: duration.Duration(15 * time.Minute),
},
{
desc: "1 second",
in: "sleep = 1\n",
out: duration.Duration(time.Second),
},
}
const validConfig = `
prometheus_listen_addr = ':9687'
[[probe]]
name = 'foo'
url = 'http://foo/bar'
`
for _, tc := range testCases {
t.Run(tc.desc, func(t *testing.T) {
cfg, err := ParseConfig(tc.in + validConfig)
require.NoError(t, err, "parse config")
require.Equal(t, tc.out, cfg.sleepDuration, "parsed sleep time")
})
}
}
|