1 files changed, 109 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/agent/discovery/dummy/discovery_test.go b/src/go/collectors/go.d.plugin/agent/discovery/dummy/discovery_test.go
new file mode 100644
index 000000000..e42ee2041
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/agent/discovery/dummy/discovery_test.go
@@ -0,0 +1,109 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package dummy
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/netdata/netdata/go/go.d.plugin/agent/confgroup"
+	"github.com/netdata/netdata/go/go.d.plugin/agent/module"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestNewDiscovery(t *testing.T) {
+	tests := map[string]struct {
+		cfg     Config
+		wantErr bool
+	}{
+		"valid config": {
+			cfg: Config{
+				Registry: confgroup.Registry{"module1": confgroup.Default{}},
+				Names:    []string{"module1", "module2"},
+			},
+		},
+		"invalid config, registry not set": {
+			cfg: Config{
+				Names: []string{"module1", "module2"},
+			},
+			wantErr: true,
+		},
+		"invalid config, names not set": {
+			cfg: Config{
+				Names: []string{"module1", "module2"},
+			},
+			wantErr: true,
+		},
+	}
+
+	for name, test := range tests {
+		t.Run(name, func(t *testing.T) {
+			d, err := NewDiscovery(test.cfg)
+
+			if test.wantErr {
+				assert.Error(t, err)
+			} else {
+				require.NoError(t, err)
+				assert.NotNil(t, d)
+			}
+		})
+	}
+}
+
+func TestDiscovery_Run(t *testing.T) {
+	expected := []*confgroup.Group{
+		{
+			Source: "internal",
+			Configs: []confgroup.Config{
+				{
+					"name":                "module1",
+					"module":              "module1",
+					"update_every":        module.UpdateEvery,
+					"autodetection_retry": module.AutoDetectionRetry,
+					"priority":            module.Priority,
+					"__provider__":        "dummy",
+					"__source_type__":     confgroup.TypeStock,
+					"__source__":          "internal",
+				},
+				{
+					"name":                "module2",
+					"module":              "module2",
+					"update_every":        module.UpdateEvery,
+					"autodetection_retry": module.AutoDetectionRetry,
+					"priority":            module.Priority,
+					"__provider__":        "dummy",
+					"__source_type__":     confgroup.TypeStock,
+					"__source__":          "internal",
+				},
+			},
+		},
+	}
+
+	reg := confgroup.Registry{
+		"module1": {},
+		"module2": {},
+	}
+	cfg := Config{
+		Registry: reg,
+		Names:    []string{"module1", "module2"},
+	}
+
+	discovery, err := NewDiscovery(cfg)
+	require.NoError(t, err)
+
+	in := make(chan []*confgroup.Group)
+	timeout := time.Second * 2
+
+	go discovery.Run(context.Background(), in)
+
+	var actual []*confgroup.Group
+	select {
+	case actual = <-in:
+	case <-time.After(timeout):
+		t.Logf("discovery timed out after %s", timeout)
+	}
+	assert.Equal(t, expected, actual)
+}