Merging upstream version 1.45.3+dfsg.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

3 files changed, 209 insertions, 0 deletions

diff --git a/src/go/collectors/go.d.plugin/agent/discovery/dummy/config.go b/src/go/collectors/go.d.plugin/agent/discovery/dummy/config.go
new file mode 100644
index 000000000..4da80a8dc
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/agent/discovery/dummy/config.go
@@ -0,0 +1,24 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package dummy
+
+import (
+	"errors"
+
+	"github.com/netdata/netdata/go/go.d.plugin/agent/confgroup"
+)
+
+type Config struct {
+	Registry confgroup.Registry
+	Names    []string
+}
+
+func validateConfig(cfg Config) error {
+	if len(cfg.Registry) == 0 {
+		return errors.New("empty config registry")
+	}
+	if len(cfg.Names) == 0 {
+		return errors.New("names not set")
+	}
+	return nil
+}
diff --git a/src/go/collectors/go.d.plugin/agent/discovery/dummy/discovery.go b/src/go/collectors/go.d.plugin/agent/discovery/dummy/discovery.go
new file mode 100644
index 000000000..fed257b2f
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/agent/discovery/dummy/discovery.go
@@ -0,0 +1,76 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package dummy
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+
+	"github.com/netdata/netdata/go/go.d.plugin/agent/confgroup"
+	"github.com/netdata/netdata/go/go.d.plugin/logger"
+)
+
+func NewDiscovery(cfg Config) (*Discovery, error) {
+	if err := validateConfig(cfg); err != nil {
+		return nil, fmt.Errorf("config validation: %v", err)
+	}
+	d := &Discovery{
+		Logger: logger.New().With(
+			slog.String("component", "discovery"),
+			slog.String("discoverer", "dummy"),
+		),
+		reg:   cfg.Registry,
+		names: cfg.Names,
+	}
+	return d, nil
+}
+
+type Discovery struct {
+	*logger.Logger
+
+	reg   confgroup.Registry
+	names []string
+}
+
+func (d *Discovery) String() string {
+	return d.Name()
+}
+
+func (d *Discovery) Name() string {
+	return "dummy discovery"
+}
+
+func (d *Discovery) Run(ctx context.Context, in chan<- []*confgroup.Group) {
+	d.Info("instance is started")
+	defer func() { d.Info("instance is stopped") }()
+
+	select {
+	case <-ctx.Done():
+	case in <- d.groups():
+	}
+
+	close(in)
+}
+
+func (d *Discovery) groups() []*confgroup.Group {
+	group := &confgroup.Group{Source: "internal"}
+
+	for _, name := range d.names {
+		def, ok := d.reg.Lookup(name)
+		if !ok {
+			continue
+		}
+		src := "internal"
+		cfg := confgroup.Config{}
+		cfg.SetModule(name)
+		cfg.SetProvider("dummy")
+		cfg.SetSourceType(confgroup.TypeStock)
+		cfg.SetSource(src)
+		cfg.ApplyDefaults(def)
+
+		group.Configs = append(group.Configs, cfg)
+	}
+
+	return []*confgroup.Group{group}
+}
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)
+}