1 files changed, 113 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/agent/discovery/sd/discoverer/netlisteners/netlisteners_test.go b/src/go/collectors/go.d.plugin/agent/discovery/sd/discoverer/netlisteners/netlisteners_test.go
new file mode 100644
index 0000000000..1a14b48d39
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/agent/discovery/sd/discoverer/netlisteners/netlisteners_test.go
@@ -0,0 +1,113 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package netlisteners
+
+import (
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/netdata/netdata/go/go.d.plugin/agent/discovery/sd/model"
+)
+
+var (
+	localListenersOutputSample = []byte(`
+UDP6|::1|8125|/opt/netdata/usr/sbin/netdata -P /run/netdata/netdata.pid -D
+TCP6|::1|8125|/opt/netdata/usr/sbin/netdata -P /run/netdata/netdata.pid -D
+TCP|127.0.0.1|8125|/opt/netdata/usr/sbin/netdata -P /run/netdata/netdata.pid -D
+UDP|127.0.0.1|53768|/opt/netdata/usr/libexec/netdata/plugins.d/go.d.plugin 1
+`)
+)
+
+func TestDiscoverer_Discover(t *testing.T) {
+	tests := map[string]discoverySim{
+		"valid response": {
+			mock:                 &mockLocalListenersExec{},
+			wantDoneBeforeCancel: false,
+			wantTargetGroups: []model.TargetGroup{&targetGroup{
+				provider: "sd:net_listeners",
+				source:   "discoverer=net_listeners,host=localhost",
+				targets: []model.Target{
+					withHash(&target{
+						Protocol: "UDP6",
+						Address:  "::1",
+						Port:     "8125",
+						Comm:     "netdata",
+						Cmdline:  "/opt/netdata/usr/sbin/netdata -P /run/netdata/netdata.pid -D",
+					}),
+					withHash(&target{
+						Protocol: "TCP6",
+						Address:  "::1",
+						Port:     "8125",
+						Comm:     "netdata",
+						Cmdline:  "/opt/netdata/usr/sbin/netdata -P /run/netdata/netdata.pid -D",
+					}),
+					withHash(&target{
+						Protocol: "TCP",
+						Address:  "127.0.0.1",
+						Port:     "8125",
+						Comm:     "netdata",
+						Cmdline:  "/opt/netdata/usr/sbin/netdata -P /run/netdata/netdata.pid -D",
+					}),
+					withHash(&target{
+						Protocol: "UDP",
+						Address:  "127.0.0.1",
+						Port:     "53768",
+						Comm:     "go.d.plugin",
+						Cmdline:  "/opt/netdata/usr/libexec/netdata/plugins.d/go.d.plugin 1",
+					}),
+				},
+			}},
+		},
+		"empty response": {
+			mock:                 &mockLocalListenersExec{emptyResponse: true},
+			wantDoneBeforeCancel: false,
+			wantTargetGroups: []model.TargetGroup{&targetGroup{
+				provider: "sd:net_listeners",
+				source:   "discoverer=net_listeners,host=localhost",
+			}},
+		},
+		"error on exec": {
+			mock:                 &mockLocalListenersExec{err: true},
+			wantDoneBeforeCancel: true,
+			wantTargetGroups:     nil,
+		},
+		"invalid data": {
+			mock:                 &mockLocalListenersExec{invalidResponse: true},
+			wantDoneBeforeCancel: true,
+			wantTargetGroups:     nil,
+		},
+	}
+
+	for name, sim := range tests {
+		t.Run(name, func(t *testing.T) {
+			sim.run(t)
+		})
+	}
+}
+
+func withHash(l *target) *target {
+	l.hash, _ = calcHash(l)
+	tags, _ := model.ParseTags("hostnetsocket")
+	l.Tags().Merge(tags)
+	return l
+}
+
+type mockLocalListenersExec struct {
+	err             bool
+	emptyResponse   bool
+	invalidResponse bool
+}
+
+func (m *mockLocalListenersExec) discover(context.Context) ([]byte, error) {
+	if m.err {
+		return nil, errors.New("mock discover() error")
+	}
+	if m.emptyResponse {
+		return nil, nil
+	}
+	if m.invalidResponse {
+		return []byte("this is very incorrect data"), nil
+	}
+	return localListenersOutputSample, nil
+}