From b5f8ee61a7f7e9bd291dd26b0585d03eb686c941 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 5 May 2024 13:19:16 +0200
Subject: Adding upstream version 1.46.3.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 .../go.d.plugin/agent/filestatus/manager_test.go   | 122 +++++++++++++++++++++
 1 file changed, 122 insertions(+)
 create mode 100644 src/go/collectors/go.d.plugin/agent/filestatus/manager_test.go

(limited to 'src/go/collectors/go.d.plugin/agent/filestatus/manager_test.go')

diff --git a/src/go/collectors/go.d.plugin/agent/filestatus/manager_test.go b/src/go/collectors/go.d.plugin/agent/filestatus/manager_test.go
new file mode 100644
index 000000000..7d45c64a2
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/agent/filestatus/manager_test.go
@@ -0,0 +1,122 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package filestatus
+
+import (
+	"context"
+	"os"
+	"path"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/netdata/netdata/go/go.d.plugin/agent/confgroup"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestNewManager(t *testing.T) {
+	mgr := NewManager("")
+	assert.NotNil(t, mgr.store)
+}
+
+func TestManager_Run(t *testing.T) {
+	type testAction struct {
+		name   string
+		cfg    confgroup.Config
+		status string
+	}
+	tests := map[string]struct {
+		actions  []testAction
+		wantFile string
+	}{
+		"save": {
+			actions: []testAction{
+				{
+					name: "save", status: "ok",
+					cfg: prepareConfig("module", "module1", "name", "name1"),
+				},
+				{
+					name: "save", status: "ok",
+					cfg: prepareConfig("module", "module2", "name", "name2"),
+				},
+			},
+			wantFile: `
+{
+ "module1": {
+  "name1:5956328514325012774": "ok"
+ },
+ "module2": {
+  "name2:14684454322123948394": "ok"
+ }
+}
+`,
+		},
+		"remove": {
+			actions: []testAction{
+				{
+					name: "save", status: "ok",
+					cfg: prepareConfig("module", "module1", "name", "name1"),
+				},
+				{
+					name: "save", status: "ok",
+					cfg: prepareConfig("module", "module2", "name", "name2"),
+				},
+				{
+					name: "remove",
+					cfg:  prepareConfig("module", "module2", "name", "name2"),
+				},
+			},
+			wantFile: `
+{
+ "module1": {
+  "name1:5956328514325012774": "ok"
+ }
+}
+`,
+		},
+	}
+
+	for name, test := range tests {
+		t.Run(name, func(t *testing.T) {
+			dir, err := os.MkdirTemp(os.TempDir(), "netdata-go-test-filestatus-run")
+			require.NoError(t, err)
+			defer func() { assert.NoError(t, os.RemoveAll(dir)) }()
+
+			filename := path.Join(dir, "filestatus")
+
+			mgr := NewManager(filename)
+
+			ctx, cancel := context.WithCancel(context.Background())
+			done := make(chan struct{})
+			go func() { defer close(done); mgr.Run(ctx) }()
+
+			for _, v := range test.actions {
+				switch v.name {
+				case "save":
+					mgr.Save(v.cfg, v.status)
+				case "remove":
+					mgr.Remove(v.cfg)
+				}
+			}
+
+			cancel()
+
+			timeout := time.Second * 5
+			tk := time.NewTimer(timeout)
+			defer tk.Stop()
+
+			select {
+			case <-done:
+			case <-tk.C:
+				t.Errorf("timed out after %s", timeout)
+			}
+
+			bs, err := os.ReadFile(filename)
+			require.NoError(t, err)
+
+			assert.Equal(t, strings.TrimSpace(test.wantFile), string(bs))
+		})
+	}
+}
-- 
cgit v1.2.3