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
|
// SPDX-License-Identifier: GPL-3.0-or-later
package sd
import (
"testing"
"github.com/netdata/netdata/go/go.d.plugin/agent/discovery/sd/pipeline"
"gopkg.in/yaml.v2"
)
func TestServiceDiscovery_Run(t *testing.T) {
tests := map[string]discoverySim{
"add pipeline": {
configs: []confFile{
prepareConfigFile("source", "name"),
},
wantPipelines: []*mockPipeline{
{name: "name", started: true, stopped: false},
},
},
"add disabled pipeline": {
configs: []confFile{
prepareDisabledConfigFile("source", "name"),
},
wantPipelines: nil,
},
"remove pipeline": {
configs: []confFile{
prepareConfigFile("source", "name"),
prepareEmptyConfigFile("source"),
},
wantPipelines: []*mockPipeline{
{name: "name", started: true, stopped: true},
},
},
"re-add pipeline multiple times": {
configs: []confFile{
prepareConfigFile("source", "name"),
prepareConfigFile("source", "name"),
prepareConfigFile("source", "name"),
},
wantPipelines: []*mockPipeline{
{name: "name", started: true, stopped: true},
{name: "name", started: true, stopped: true},
{name: "name", started: true, stopped: false},
},
},
"restart pipeline": {
configs: []confFile{
prepareConfigFile("source", "name1"),
prepareConfigFile("source", "name2"),
},
wantPipelines: []*mockPipeline{
{name: "name1", started: true, stopped: true},
{name: "name2", started: true, stopped: false},
},
},
"invalid pipeline config": {
configs: []confFile{
prepareConfigFile("source", "invalid"),
},
wantPipelines: nil,
},
"invalid config for running pipeline": {
configs: []confFile{
prepareConfigFile("source", "name"),
prepareConfigFile("source", "invalid"),
},
wantPipelines: []*mockPipeline{
{name: "name", started: true, stopped: false},
},
},
}
for name, sim := range tests {
t.Run(name, func(t *testing.T) {
sim.run(t)
})
}
}
func prepareConfigFile(source, name string) confFile {
bs, _ := yaml.Marshal(pipeline.Config{Name: name})
return confFile{
source: source,
content: bs,
}
}
func prepareEmptyConfigFile(source string) confFile {
return confFile{
source: source,
}
}
func prepareDisabledConfigFile(source, name string) confFile {
bs, _ := yaml.Marshal(pipeline.Config{Name: name, Disabled: true})
return confFile{
source: source,
content: bs,
}
}
|
