influx/telegraf
1
0
Fork 0
Code Issues Activity

Adding upstream version 1.34.4.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-05-24 07:26:29 +02:00
parent e393c3af3f
commit 4978089aab
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
4963 changed files with 677545 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

279
plugins/inputs/kube_inventory/daemonset_test.go Normal file
View file

@ -0,0 +1,279 @@
package kube_inventory
import (
"strings"
"testing"
"time"
"github.com/stretchr/testify/require"
apps "k8s.io/api/apps/v1"
meta "k8s.io/apimachinery/pkg/apis/meta/v1"
"github.com/influxdata/telegraf"
"github.com/influxdata/telegraf/testutil"
)
func TestDaemonSet(t *testing.T) {
cli := &client{}
now := time.Now()
now = time.Date(now.Year(), now.Month(), now.Day(), now.Hour(), 1, 36, 0, now.Location())
tests := []struct {
name string
handler *mockHandler
output []telegraf.Metric
hasError bool
}{
{
name: "no daemon set",
handler: &mockHandler{
responseMap: map[string]interface{}{
"/daemonsets/": &apps.DaemonSetList{},
},
},
hasError: false,
},
{
name: "collect daemonsets",
handler: &mockHandler{
responseMap: map[string]interface{}{
"/daemonsets/": &apps.DaemonSetList{
Items: []apps.DaemonSet{
{
Status: apps.DaemonSetStatus{
CurrentNumberScheduled: 3,
DesiredNumberScheduled: 5,
NumberAvailable: 2,
NumberMisscheduled: 2,
NumberReady: 1,
NumberUnavailable: 1,
UpdatedNumberScheduled: 2,
},
ObjectMeta: meta.ObjectMeta{
Generation: 11221,
Namespace: "ns1",
Name: "daemon1",
Labels: map[string]string{
"lab1": "v1",
"lab2": "v2",
},
CreationTimestamp: meta.Time{Time: now},
},
Spec: apps.DaemonSetSpec{
Selector: &meta.LabelSelector{
MatchLabels: map[string]string{
"select1": "s1",
"select2": "s2",
},
},
},
},
},
},
},
},
output: []telegraf.Metric{
testutil.MustMetric(
"kubernetes_daemonset",
map[string]string{
"daemonset_name": "daemon1",
"namespace": "ns1",
"selector_select1": "s1",
"selector_select2": "s2",
},
map[string]interface{}{
"generation": int64(11221),
"current_number_scheduled": int32(3),
"desired_number_scheduled": int32(5),
"number_available": int32(2),
"number_misscheduled": int32(2),
"number_ready": int32(1),
"number_unavailable": int32(1),
"updated_number_scheduled": int32(2),
"created": now.UnixNano(),
},
time.Unix(0, 0),
),
},
hasError: false,
},
}
for _, v := range tests {
ks := &KubernetesInventory{
client: cli,
}
require.NoError(t, ks.createSelectorFilters())
acc := new(testutil.Accumulator)
items := ((v.handler.responseMap["/daemonsets/"]).(*apps.DaemonSetList)).Items
for i := range items {
ks.gatherDaemonSet(&items[i], acc)
}
err := acc.FirstError()
if v.hasError {
require.Errorf(t, err, "%s failed, should have error", v.name)
continue
}
// No error case
require.NoErrorf(t, err, "%s failed, err: %v", v.name, err)
require.Len(t, acc.Metrics, len(v.output))
testutil.RequireMetricsEqual(t, acc.GetTelegrafMetrics(), v.output, testutil.IgnoreTime())
}
}
func TestDaemonSetSelectorFilter(t *testing.T) {
cli := &client{}
responseMap := map[string]interface{}{
"/daemonsets/": &apps.DaemonSetList{
Items: []apps.DaemonSet{
{
Status: apps.DaemonSetStatus{
CurrentNumberScheduled: 3,
DesiredNumberScheduled: 5,
NumberAvailable: 2,
NumberMisscheduled: 2,
NumberReady: 1,
NumberUnavailable: 1,
UpdatedNumberScheduled: 2,
},
ObjectMeta: meta.ObjectMeta{
Generation: 11221,
Namespace: "ns1",
Name: "daemon1",
Labels: map[string]string{
"lab1": "v1",
"lab2": "v2",
},
CreationTimestamp: meta.Time{Time: time.Now()},
},
Spec: apps.DaemonSetSpec{
Selector: &meta.LabelSelector{
MatchLabels: map[string]string{
"select1": "s1",
"select2": "s2",
},
},
},
},
},
},
}
tests := []struct {
name string
handler *mockHandler
hasError bool
include []string
exclude []string
expected map[string]string
}{
{
name: "nil filters equals all selectors",
handler: &mockHandler{
responseMap: responseMap,
},
hasError: false,
include: nil,
exclude: nil,
expected: map[string]string{
"selector_select1": "s1",
"selector_select2": "s2",
},
},
{
name: "empty filters equals all selectors",
handler: &mockHandler{
responseMap: responseMap,
},
hasError: false,
expected: map[string]string{
"selector_select1": "s1",
"selector_select2": "s2",
},
},
{
name: "include filter equals only include-matched selectors",
handler: &mockHandler{
responseMap: responseMap,
},
hasError: false,
include: []string{"select1"},
expected: map[string]string{
"selector_select1": "s1",
},
},
{
name: "exclude filter equals only non-excluded selectors (overrides include filter)",
handler: &mockHandler{
responseMap: responseMap,
},
hasError: false,
exclude: []string{"select2"},
expected: map[string]string{
"selector_select1": "s1",
},
},
{
name: "include glob filter equals only include-matched selectors",
handler: &mockHandler{
responseMap: responseMap,
},
hasError: false,
include: []string{"*1"},
expected: map[string]string{
"selector_select1": "s1",
},
},
{
name: "exclude glob filter equals only non-excluded selectors",
handler: &mockHandler{
responseMap: responseMap,
},
hasError: false,
exclude: []string{"*2"},
expected: map[string]string{
"selector_select1": "s1",
},
},
{
name: "exclude glob filter equals only non-excluded selectors",
handler: &mockHandler{
responseMap: responseMap,
},
hasError: false,
exclude: []string{"*2"},
expected: map[string]string{
"selector_select1": "s1",
},
},
}
for _, v := range tests {
ks := &KubernetesInventory{
client: cli,
}
ks.SelectorInclude = v.include
ks.SelectorExclude = v.exclude
require.NoError(t, ks.createSelectorFilters())
acc := new(testutil.Accumulator)
items := ((v.handler.responseMap["/daemonsets/"]).(*apps.DaemonSetList)).Items
for i := range items {
ks.gatherDaemonSet(&items[i], acc)
}
// Grab selector tags
actual := map[string]string{}
for _, metric := range acc.Metrics {
for key, val := range metric.Tags {
if strings.Contains(key, "selector_") {
actual[key] = val
}
}
}
require.Equalf(t, v.expected, actual,
"actual selector tags (%v) do not match expected selector tags (%v)", actual, v.expected)
}
}