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
262
plugins/outputs/sensu/sensu_test.go
Normal file
262
plugins/outputs/sensu/sensu_test.go
Normal file
|
|
@ -0,0 +1,262 @@
|
|||
package sensu
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"math"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"slices"
|
||||
"testing"
|
||||
|
||||
corev2 "github.com/sensu/sensu-go/api/core/v2"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/influxdata/telegraf"
|
||||
"github.com/influxdata/telegraf/testutil"
|
||||
)
|
||||
|
||||
func TestResolveEventEndpointUrl(t *testing.T) {
|
||||
agentAPIURL := "http://127.0.0.1:3031"
|
||||
backendAPIURL := "http://127.0.0.1:8080"
|
||||
entityNamespace := "test-namespace"
|
||||
emptyString := ""
|
||||
tests := []struct {
|
||||
name string
|
||||
plugin *Sensu
|
||||
expectedEndpointURL string
|
||||
}{
|
||||
{
|
||||
name: "agent event endpoint",
|
||||
plugin: &Sensu{
|
||||
AgentAPIURL: &agentAPIURL,
|
||||
Log: testutil.Logger{},
|
||||
},
|
||||
expectedEndpointURL: "http://127.0.0.1:3031/events",
|
||||
},
|
||||
{
|
||||
name: "backend event endpoint with default namespace",
|
||||
plugin: &Sensu{
|
||||
AgentAPIURL: &agentAPIURL,
|
||||
BackendAPIURL: &backendAPIURL,
|
||||
Log: testutil.Logger{},
|
||||
},
|
||||
expectedEndpointURL: "http://127.0.0.1:8080/api/core/v2/namespaces/default/events",
|
||||
},
|
||||
{
|
||||
name: "backend event endpoint with namespace declared",
|
||||
plugin: &Sensu{
|
||||
AgentAPIURL: &agentAPIURL,
|
||||
BackendAPIURL: &backendAPIURL,
|
||||
Entity: &sensuEntity{
|
||||
Namespace: &entityNamespace,
|
||||
},
|
||||
Log: testutil.Logger{},
|
||||
},
|
||||
expectedEndpointURL: "http://127.0.0.1:8080/api/core/v2/namespaces/test-namespace/events",
|
||||
},
|
||||
{
|
||||
name: "agent event endpoint due to empty AgentAPIURL",
|
||||
plugin: &Sensu{
|
||||
AgentAPIURL: &emptyString,
|
||||
Log: testutil.Logger{},
|
||||
},
|
||||
expectedEndpointURL: "http://127.0.0.1:3031/events",
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
err := tt.plugin.setEndpointURL()
|
||||
require.Equal(t, err, error(nil))
|
||||
require.Equal(t, tt.expectedEndpointURL, tt.plugin.EndpointURL)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestConnectAndWrite(t *testing.T) {
|
||||
ts := httptest.NewServer(http.NotFoundHandler())
|
||||
defer ts.Close()
|
||||
|
||||
testURL := "http://" + ts.Listener.Addr().String()
|
||||
testAPIKey := "a0b1c2d3-e4f5-g6h7-i8j9-k0l1m2n3o4p5"
|
||||
testCheck := "telegraf"
|
||||
testEntity := "entity1"
|
||||
testNamespace := "default"
|
||||
testHandler := "influxdb"
|
||||
testTagName := "myTagName"
|
||||
testTagValue := "myTagValue"
|
||||
expectedAuthHeader := "Key " + testAPIKey
|
||||
expectedURL := fmt.Sprintf("/api/core/v2/namespaces/%s/events", testNamespace)
|
||||
expectedPointName := "cpu"
|
||||
expectedPointValue := float64(42)
|
||||
|
||||
plugin := &Sensu{
|
||||
AgentAPIURL: nil,
|
||||
BackendAPIURL: &testURL,
|
||||
APIKey: &testAPIKey,
|
||||
Check: &sensuCheck{
|
||||
Name: &testCheck,
|
||||
},
|
||||
Entity: &sensuEntity{
|
||||
Name: &testEntity,
|
||||
Namespace: &testNamespace,
|
||||
},
|
||||
Metrics: &sensuMetrics{
|
||||
Handlers: []string{testHandler},
|
||||
},
|
||||
Tags: map[string]string{testTagName: testTagValue},
|
||||
Log: testutil.Logger{},
|
||||
}
|
||||
|
||||
t.Run("connect", func(t *testing.T) {
|
||||
err := plugin.Connect()
|
||||
require.NoError(t, err)
|
||||
})
|
||||
|
||||
t.Run("write", func(t *testing.T) {
|
||||
ts.Config.Handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if urlString := r.URL.String(); urlString != expectedURL {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", expectedURL, urlString)
|
||||
return
|
||||
}
|
||||
if authHeader := r.Header.Get("Authorization"); authHeader != expectedAuthHeader {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", expectedAuthHeader, authHeader)
|
||||
return
|
||||
}
|
||||
|
||||
// let's make sure what we received is a valid Sensu event that contains all of the expected data
|
||||
body, err := io.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
receivedEvent := &corev2.Event{}
|
||||
|
||||
if err = json.Unmarshal(body, receivedEvent); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
if receivedEvent.Check.Name != testCheck {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", testCheck, receivedEvent.Check.Name)
|
||||
return
|
||||
}
|
||||
if receivedEvent.Entity.Name != testEntity {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("Not equal, expected: %q, actual: %q", testEntity, receivedEvent.Entity.Name)
|
||||
return
|
||||
}
|
||||
if receivedEvent.Metrics == nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("'receivedEvent.Metrics' should not be nil")
|
||||
return
|
||||
}
|
||||
if !slices.Contains(receivedEvent.Metrics.Handlers, testHandler) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("'receivedEvent.Metrics.Handlers' should contain %q", testHandler)
|
||||
return
|
||||
}
|
||||
if len(receivedEvent.Metrics.Points) == 0 {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("'receivedEvent.Metrics.Points' should not be empty")
|
||||
return
|
||||
}
|
||||
pointFound := false
|
||||
tagFound := false
|
||||
for _, p := range receivedEvent.Metrics.Points {
|
||||
if p.Name == expectedPointName+".value" && p.Value == expectedPointValue {
|
||||
pointFound = true
|
||||
if len(p.Tags) == 0 {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("'p.Tags' should not be empty")
|
||||
return
|
||||
}
|
||||
|
||||
for _, t := range p.Tags {
|
||||
if t.Name == testTagName && t.Value == testTagValue {
|
||||
tagFound = true
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if !pointFound {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("'pointFound' should be true")
|
||||
return
|
||||
}
|
||||
if !tagFound {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Errorf("'tagFound' should be true")
|
||||
return
|
||||
}
|
||||
w.WriteHeader(http.StatusCreated)
|
||||
})
|
||||
err := plugin.Write([]telegraf.Metric{testutil.TestMetric(expectedPointValue, expectedPointName)})
|
||||
require.NoError(t, err)
|
||||
})
|
||||
}
|
||||
|
||||
func TestGetFloat(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
value interface{}
|
||||
expectedReturn float64
|
||||
}{
|
||||
{
|
||||
name: "getfloat with float64",
|
||||
value: float64(42),
|
||||
expectedReturn: 42,
|
||||
},
|
||||
{
|
||||
name: "getfloat with float32",
|
||||
value: float32(42),
|
||||
expectedReturn: 42,
|
||||
},
|
||||
{
|
||||
name: "getfloat with int64",
|
||||
value: int64(42),
|
||||
expectedReturn: 42,
|
||||
},
|
||||
{
|
||||
name: "getfloat with int32",
|
||||
value: int32(42),
|
||||
expectedReturn: 42,
|
||||
},
|
||||
{
|
||||
name: "getfloat with int",
|
||||
value: int(42),
|
||||
expectedReturn: 42,
|
||||
},
|
||||
{
|
||||
name: "getfloat with uint64",
|
||||
value: uint64(42),
|
||||
expectedReturn: 42,
|
||||
},
|
||||
{
|
||||
name: "getfloat with uint32",
|
||||
value: uint32(42),
|
||||
expectedReturn: 42,
|
||||
},
|
||||
{
|
||||
name: "getfloat with uint",
|
||||
value: uint(42),
|
||||
expectedReturn: 42,
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
require.InDelta(t, tt.expectedReturn, getFloat(tt.value), testutil.DefaultDelta)
|
||||
})
|
||||
}
|
||||
// Since math.NaN() == math.NaN() returns false
|
||||
t.Run("getfloat NaN special case", func(t *testing.T) {
|
||||
f := getFloat("42")
|
||||
require.True(t, math.IsNaN(f))
|
||||
})
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue