Adding upstream version 1.34.4.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
e393c3af3f
commit
4978089aab
GPG key ID:
FBB4F0E80A80222F
4963 changed files with 677545 additions and 0 deletions
401
plugins/outputs/groundwork/groundwork_test.go
Normal file
401
plugins/outputs/groundwork/groundwork_test.go
Normal file
|
|
@ -0,0 +1,401 @@
|
|||
package groundwork
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/gwos/tcg/sdk/clients"
|
||||
"github.com/gwos/tcg/sdk/transit"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/influxdata/telegraf"
|
||||
"github.com/influxdata/telegraf/config"
|
||||
"github.com/influxdata/telegraf/logger"
|
||||
"github.com/influxdata/telegraf/testutil"
|
||||
)
|
||||
|
||||
const (
|
||||
defaultTestAgentID = "ec1676cc-583d-48ee-b035-7fb5ed0fcf88"
|
||||
defaultHost = "telegraf"
|
||||
defaultAppType = "TELEGRAF"
|
||||
customAppType = "SYSLOG"
|
||||
)
|
||||
|
||||
func TestWriteWithDebug(t *testing.T) {
|
||||
// Generate test metric with default name to test Write logic
|
||||
intMetric := testutil.TestMetric(42, "IntMetric")
|
||||
srvTok := "88fcf0de5bf7-530b-ee84-d385-cc6761ce"
|
||||
|
||||
// Simulate Groundwork server that should receive custom metrics
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
body, err := io.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Decode body to use in assertions below
|
||||
var obj transit.ResourcesWithServicesRequest
|
||||
if err = json.Unmarshal(body, &obj); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Check if server gets proper data
|
||||
if obj.Resources[0].Services[0].Name != "IntMetric" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "IntMetric", obj.Resources[0].Services[0].Name)
|
||||
return
|
||||
}
|
||||
if *obj.Resources[0].Services[0].Metrics[0].Value.IntegerValue != int64(42) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %v, actual: %v", int64(42), *obj.Resources[0].Services[0].Metrics[0].Value.IntegerValue)
|
||||
return
|
||||
}
|
||||
|
||||
// Send back details
|
||||
ans := "Content-type: application/json\n\n" + `{"message":"` + srvTok + `"}`
|
||||
if _, err = fmt.Fprintln(w, ans); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
}))
|
||||
|
||||
i := Groundwork{
|
||||
Server: server.URL,
|
||||
AgentID: defaultTestAgentID,
|
||||
Username: config.NewSecret([]byte(`tu ser`)),
|
||||
Password: config.NewSecret([]byte(`pu ser`)),
|
||||
DefaultAppType: defaultAppType,
|
||||
DefaultHost: defaultHost,
|
||||
DefaultServiceState: string(transit.ServiceOk),
|
||||
ResourceTag: "host",
|
||||
Log: testutil.Logger{},
|
||||
}
|
||||
|
||||
buf := new(bytes.Buffer)
|
||||
require.NoError(t, logger.SetupLogging(&logger.Config{Debug: true}))
|
||||
logger.RedirectLogging(buf)
|
||||
|
||||
require.NoError(t, i.Init())
|
||||
require.NoError(t, i.Write([]telegraf.Metric{intMetric}))
|
||||
|
||||
require.NoError(t, logger.CloseLogging())
|
||||
require.Contains(t, buf.String(), defaultTestAgentID)
|
||||
require.Contains(t, buf.String(), srvTok)
|
||||
|
||||
server.Close()
|
||||
}
|
||||
|
||||
func TestWriteWithDefaults(t *testing.T) {
|
||||
// Generate test metric with default name to test Write logic
|
||||
intMetric := testutil.TestMetric(42, "IntMetric")
|
||||
|
||||
// Simulate Groundwork server that should receive custom metrics
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
body, err := io.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Decode body to use in assertions below
|
||||
var obj transit.ResourcesWithServicesRequest
|
||||
if err = json.Unmarshal(body, &obj); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Check if server gets proper data
|
||||
if obj.Context.AgentID != defaultTestAgentID {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", defaultTestAgentID, obj.Context.AgentID)
|
||||
return
|
||||
}
|
||||
if obj.Context.AppType != customAppType {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", customAppType, obj.Context.AppType)
|
||||
return
|
||||
}
|
||||
if obj.Resources[0].Name != defaultHost {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", defaultHost, obj.Resources[0].Name)
|
||||
return
|
||||
}
|
||||
if obj.Resources[0].Services[0].Status != transit.MonitorStatus("SERVICE_OK") {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", transit.MonitorStatus("SERVICE_OK"), obj.Resources[0].Services[0].Status)
|
||||
return
|
||||
}
|
||||
if obj.Resources[0].Services[0].Name != "IntMetric" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "IntMetric", obj.Resources[0].Services[0].Name)
|
||||
return
|
||||
}
|
||||
if *obj.Resources[0].Services[0].Metrics[0].Value.IntegerValue != int64(42) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %v, actual: %v", int64(42), *obj.Resources[0].Services[0].Metrics[0].Value.IntegerValue)
|
||||
return
|
||||
}
|
||||
if len(obj.Groups) != 0 {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("'obj.Groups' should not be empty")
|
||||
return
|
||||
}
|
||||
|
||||
if _, err = fmt.Fprintln(w, "OK"); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
}))
|
||||
|
||||
i := Groundwork{
|
||||
Log: testutil.Logger{},
|
||||
Server: server.URL,
|
||||
AgentID: defaultTestAgentID,
|
||||
DefaultHost: defaultHost,
|
||||
DefaultAppType: customAppType,
|
||||
client: clients.GWClient{
|
||||
AppName: "telegraf",
|
||||
AppType: customAppType,
|
||||
GWConnection: &clients.GWConnection{
|
||||
HostName: server.URL,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
err := i.Write([]telegraf.Metric{intMetric})
|
||||
require.NoError(t, err)
|
||||
|
||||
server.Close()
|
||||
}
|
||||
|
||||
func TestWriteWithFields(t *testing.T) {
|
||||
// Generate test metric with fields to test Write logic
|
||||
floatMetric := testutil.TestMetric(1.0, "FloatMetric")
|
||||
floatMetric.AddField("value_cr", 3.0)
|
||||
floatMetric.AddField("value_wn", 2.0)
|
||||
floatMetric.AddField("message", "Test Message")
|
||||
floatMetric.AddField("status", "SERVICE_WARNING")
|
||||
|
||||
// Simulate Groundwork server that should receive custom metrics
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
body, err := io.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Decode body to use in assertions below
|
||||
var obj transit.ResourcesWithServicesRequest
|
||||
if err = json.Unmarshal(body, &obj); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Check if server gets proper data
|
||||
if obj.Resources[0].Services[0].LastPluginOutput != "Test Message" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "Test Message", obj.Resources[0].Services[0].LastPluginOutput)
|
||||
return
|
||||
}
|
||||
if obj.Resources[0].Services[0].Status != transit.MonitorStatus("SERVICE_WARNING") {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", transit.MonitorStatus("SERVICE_WARNING"), obj.Resources[0].Services[0].Status)
|
||||
return
|
||||
}
|
||||
if dt := float64(1.0) - *obj.Resources[0].Services[0].Metrics[0].Value.DoubleValue; !testutil.WithinDefaultDelta(dt) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Max difference between %v and %v allowed is %v, but difference was %v",
|
||||
float64(1.0), *obj.Resources[0].Services[0].Metrics[0].Value.DoubleValue, testutil.DefaultDelta, dt)
|
||||
return
|
||||
}
|
||||
if dt := float64(3.0) - *obj.Resources[0].Services[0].Metrics[0].Thresholds[0].Value.DoubleValue; !testutil.WithinDefaultDelta(dt) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Max difference between %v and %v allowed is %v, but difference was %v",
|
||||
float64(3.0), *obj.Resources[0].Services[0].Metrics[0].Thresholds[0].Value.DoubleValue, testutil.DefaultDelta, dt)
|
||||
return
|
||||
}
|
||||
if dt := float64(2.0) - *obj.Resources[0].Services[0].Metrics[0].Thresholds[1].Value.DoubleValue; !testutil.WithinDefaultDelta(dt) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Max difference between %v and %v allowed is %v, but difference was %v",
|
||||
float64(2.0), *obj.Resources[0].Services[0].Metrics[0].Thresholds[1].Value.DoubleValue, testutil.DefaultDelta, dt)
|
||||
return
|
||||
}
|
||||
|
||||
if _, err = fmt.Fprintln(w, "OK"); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
}))
|
||||
|
||||
i := Groundwork{
|
||||
Log: testutil.Logger{},
|
||||
Server: server.URL,
|
||||
AgentID: defaultTestAgentID,
|
||||
DefaultHost: defaultHost,
|
||||
DefaultAppType: defaultAppType,
|
||||
GroupTag: "group",
|
||||
ResourceTag: "host",
|
||||
client: clients.GWClient{
|
||||
AppName: "telegraf",
|
||||
AppType: defaultAppType,
|
||||
GWConnection: &clients.GWConnection{
|
||||
HostName: server.URL,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
err := i.Write([]telegraf.Metric{floatMetric})
|
||||
require.NoError(t, err)
|
||||
|
||||
server.Close()
|
||||
}
|
||||
|
||||
func TestWriteWithTags(t *testing.T) {
|
||||
// Generate test metric with tags to test Write logic
|
||||
floatMetric := testutil.TestMetric(1.0, "FloatMetric")
|
||||
floatMetric.AddField("value_cr", 3.0)
|
||||
floatMetric.AddField("value_wn", 2.0)
|
||||
floatMetric.AddField("message", "Test Message")
|
||||
floatMetric.AddField("status", "SERVICE_WARNING")
|
||||
floatMetric.AddTag("value_cr", "9.0")
|
||||
floatMetric.AddTag("value_wn", "6.0")
|
||||
floatMetric.AddTag("message", "Test Tag")
|
||||
floatMetric.AddTag("status", "SERVICE_PENDING")
|
||||
floatMetric.AddTag("group-tag", "Group01")
|
||||
floatMetric.AddTag("resource-tag", "Host01")
|
||||
floatMetric.AddTag("service", "Service01")
|
||||
floatMetric.AddTag("facility", "FACILITY")
|
||||
floatMetric.AddTag("severity", "SEVERITY")
|
||||
|
||||
// Simulate Groundwork server that should receive custom metrics
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
body, err := io.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Decode body to use in assertions below
|
||||
var obj transit.ResourcesWithServicesRequest
|
||||
if err = json.Unmarshal(body, &obj); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Check if server gets proper data
|
||||
if obj.Context.AgentID != defaultTestAgentID {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", defaultTestAgentID, obj.Context.AgentID)
|
||||
return
|
||||
}
|
||||
if obj.Context.AppType != defaultAppType {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", defaultAppType, obj.Context.AppType)
|
||||
return
|
||||
}
|
||||
if obj.Resources[0].Name != "Host01" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "Host01", obj.Resources[0].Name)
|
||||
return
|
||||
}
|
||||
if obj.Resources[0].Services[0].Name != "Service01" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "Service01", obj.Resources[0].Services[0].Name)
|
||||
return
|
||||
}
|
||||
if *obj.Resources[0].Services[0].Properties["facility"].StringValue != "FACILITY" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "FACILITY", *obj.Resources[0].Services[0].Properties["facility"].StringValue)
|
||||
return
|
||||
}
|
||||
if *obj.Resources[0].Services[0].Properties["severity"].StringValue != "SEVERITY" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "SEVERITY", *obj.Resources[0].Services[0].Properties["severity"].StringValue)
|
||||
return
|
||||
}
|
||||
if obj.Groups[0].GroupName != "Group01" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "Group01", obj.Groups[0].GroupName)
|
||||
return
|
||||
}
|
||||
if obj.Groups[0].Resources[0].Name != "Host01" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "Host01", obj.Groups[0].Resources[0].Name)
|
||||
return
|
||||
}
|
||||
if obj.Resources[0].Services[0].LastPluginOutput != "Test Tag" {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", "Test Tag", obj.Resources[0].Services[0].LastPluginOutput)
|
||||
return
|
||||
}
|
||||
if obj.Resources[0].Services[0].Status != transit.MonitorStatus("SERVICE_PENDING") {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", transit.MonitorStatus("SERVICE_PENDING"), obj.Resources[0].Services[0].Status)
|
||||
return
|
||||
}
|
||||
if dt := float64(1.0) - *obj.Resources[0].Services[0].Metrics[0].Value.DoubleValue; !testutil.WithinDefaultDelta(dt) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Max difference between %v and %v allowed is %v, but difference was %v",
|
||||
float64(1.0), *obj.Resources[0].Services[0].Metrics[0].Value.DoubleValue, testutil.DefaultDelta, dt)
|
||||
return
|
||||
}
|
||||
if dt := float64(9.0) - *obj.Resources[0].Services[0].Metrics[0].Thresholds[0].Value.DoubleValue; !testutil.WithinDefaultDelta(dt) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Max difference between %v and %v allowed is %v, but difference was %v",
|
||||
float64(9.0), *obj.Resources[0].Services[0].Metrics[0].Thresholds[0].Value.DoubleValue, testutil.DefaultDelta, dt)
|
||||
return
|
||||
}
|
||||
if dt := float64(6.0) - *obj.Resources[0].Services[0].Metrics[0].Thresholds[1].Value.DoubleValue; !testutil.WithinDefaultDelta(dt) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Max difference between %v and %v allowed is %v, but difference was %v",
|
||||
float64(6.0), *obj.Resources[0].Services[0].Metrics[0].Thresholds[1].Value.DoubleValue, testutil.DefaultDelta, dt)
|
||||
return
|
||||
}
|
||||
|
||||
if _, err = fmt.Fprintln(w, "OK"); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
}))
|
||||
|
||||
i := Groundwork{
|
||||
Log: testutil.Logger{},
|
||||
Server: server.URL,
|
||||
AgentID: defaultTestAgentID,
|
||||
DefaultHost: defaultHost,
|
||||
DefaultAppType: defaultAppType,
|
||||
GroupTag: "group-tag",
|
||||
ResourceTag: "resource-tag",
|
||||
client: clients.GWClient{
|
||||
AppName: "telegraf",
|
||||
AppType: defaultAppType,
|
||||
GWConnection: &clients.GWConnection{
|
||||
HostName: server.URL,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
err := i.Write([]telegraf.Metric{floatMetric})
|
||||
require.NoError(t, err)
|
||||
|
||||
server.Close()
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue