-
Notifications
You must be signed in to change notification settings - Fork 1
/
service_port_test.go
151 lines (144 loc) · 5 KB
/
service_port_test.go
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
package fluentbitconfig
import (
"testing"
"github.com/alecthomas/assert/v2"
"github.com/calyptia/go-fluentbit-config/v2/networking"
)
func TestConfig_ServicePorts(t *testing.T) {
t.Run("defaults", func(t *testing.T) {
config, err := ParseAs(`
[SERVICE]
http_server on
[INPUT]
name collectd
[INPUT]
name elasticsearch
[INPUT]
name forward
[INPUT]
name http
[INPUT]
name mqtt
[INPUT]
name opentelemetry
[INPUT]
name statsd
[INPUT]
name syslog
mode udp
[INPUT]
name tcp
[INPUT]
name udp
[OUTPUT]
name prometheus_exporter
`, FormatClassic)
assert.NoError(t, err)
assert.Equal(t, ServicePorts{
{Port: 2020, Protocol: networking.ProtocolTCP, Kind: SectionKindService},
{Port: 25826, Protocol: networking.ProtocolUDP, Kind: SectionKindInput, Plugin: &Plugin{ID: "collectd.0", Name: "collectd"}},
{Port: 9200, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "elasticsearch.1", Name: "elasticsearch"}},
{Port: 24224, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "forward.2", Name: "forward"}},
{Port: 9880, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "http.3", Name: "http"}},
{Port: 1883, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "mqtt.4", Name: "mqtt"}},
{Port: 4318, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "opentelemetry.5", Name: "opentelemetry"}},
{Port: 8125, Protocol: networking.ProtocolUDP, Kind: SectionKindInput, Plugin: &Plugin{ID: "statsd.6", Name: "statsd"}},
{Port: 5140, Protocol: networking.ProtocolUDP, Kind: SectionKindInput, Plugin: &Plugin{ID: "syslog.7", Name: "syslog"}},
{Port: 5170, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "tcp.8", Name: "tcp"}},
{Port: 5170, Protocol: networking.ProtocolUDP, Kind: SectionKindInput, Plugin: &Plugin{ID: "udp.9", Name: "udp"}},
{Port: 2021, Protocol: networking.ProtocolTCP, Kind: SectionKindOutput, Plugin: &Plugin{ID: "prometheus_exporter.0", Name: "prometheus_exporter"}},
}, config.ServicePorts())
})
t.Run("explicit", func(t *testing.T) {
config, err := ParseAs(`
[SERVICE]
http_server on
http_port 1
[INPUT]
name collectd
port 2
[INPUT]
name elasticsearch
port 3
[INPUT]
name forward
port 4
[INPUT]
name http
port 5
[INPUT]
name mqtt
port 6
[INPUT]
name opentelemetry
port 7
[INPUT]
name statsd
port 8
[INPUT]
name syslog
mode udp
port 9
[INPUT]
name tcp
port 10
[INPUT]
name udp
port 11
[OUTPUT]
name prometheus_exporter
port 12
`, FormatClassic)
assert.NoError(t, err)
assert.Equal(t, ServicePorts{
{Port: 1, Protocol: networking.ProtocolTCP, Kind: SectionKindService},
{Port: 2, Protocol: networking.ProtocolUDP, Kind: SectionKindInput, Plugin: &Plugin{ID: "collectd.0", Name: "collectd"}},
{Port: 3, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "elasticsearch.1", Name: "elasticsearch"}},
{Port: 4, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "forward.2", Name: "forward"}},
{Port: 5, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "http.3", Name: "http"}},
{Port: 6, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "mqtt.4", Name: "mqtt"}},
{Port: 7, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "opentelemetry.5", Name: "opentelemetry"}},
{Port: 8, Protocol: networking.ProtocolUDP, Kind: SectionKindInput, Plugin: &Plugin{ID: "statsd.6", Name: "statsd"}},
{Port: 9, Protocol: networking.ProtocolUDP, Kind: SectionKindInput, Plugin: &Plugin{ID: "syslog.7", Name: "syslog"}},
{Port: 10, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "tcp.8", Name: "tcp"}},
{Port: 11, Protocol: networking.ProtocolUDP, Kind: SectionKindInput, Plugin: &Plugin{ID: "udp.9", Name: "udp"}},
{Port: 12, Protocol: networking.ProtocolTCP, Kind: SectionKindOutput, Plugin: &Plugin{ID: "prometheus_exporter.0", Name: "prometheus_exporter"}},
}, config.ServicePorts())
})
t.Run("skips", func(t *testing.T) {
config, err := ParseAs(`
[INPUT]
name forward
unix_path /tmp/forward.sock
port 1
[INPUT]
name syslog
port 2
[INPUT]
name syslog
mode unix_udp
port 3
[INPUT]
name syslog
mode unix_tcp
port 4
[INPUT]
name does_not_exists
port 5
`, FormatClassic)
assert.NoError(t, err)
assert.Equal(t, nil, config.ServicePorts())
})
t.Run("with_mode", func(t *testing.T) {
config, err := ParseAs(`
[INPUT]
name syslog
mode tcp
port 1
`, FormatClassic)
assert.NoError(t, err)
assert.Equal(t, ServicePorts{
{Port: 1, Protocol: networking.ProtocolTCP, Kind: SectionKindInput, Plugin: &Plugin{ID: "syslog.0", Name: "syslog"}},
}, config.ServicePorts())
})
}