1 files changed, 55 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/agent/module/mock.go b/src/go/collectors/go.d.plugin/agent/module/mock.go
new file mode 100644
index 0000000000..c4353eb524
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/agent/module/mock.go
@@ -0,0 +1,55 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package module
+
+// MockModule MockModule.
+type MockModule struct {
+	Base
+
+	InitFunc    func() bool
+	CheckFunc   func() bool
+	ChartsFunc  func() *Charts
+	CollectFunc func() map[string]int64
+	CleanupFunc func()
+	CleanupDone bool
+}
+
+// Init invokes InitFunc.
+func (m MockModule) Init() bool {
+	if m.InitFunc == nil {
+		return true
+	}
+	return m.InitFunc()
+}
+
+// Check invokes CheckFunc.
+func (m MockModule) Check() bool {
+	if m.CheckFunc == nil {
+		return true
+	}
+	return m.CheckFunc()
+}
+
+// Charts invokes ChartsFunc.
+func (m MockModule) Charts() *Charts {
+	if m.ChartsFunc == nil {
+		return nil
+	}
+	return m.ChartsFunc()
+}
+
+// Collect invokes CollectDunc.
+func (m MockModule) Collect() map[string]int64 {
+	if m.CollectFunc == nil {
+		return nil
+	}
+	return m.CollectFunc()
+}
+
+// Cleanup sets CleanupDone to true.
+func (m *MockModule) Cleanup() {
+	if m.CleanupFunc != nil {
+		m.CleanupFunc()
+	}
+	m.CleanupDone = true
+}