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
378
plugins/inputs/kube_inventory/statefulset_test.go
Normal file
378
plugins/inputs/kube_inventory/statefulset_test.go
Normal file
|
|
@ -0,0 +1,378 @@
|
|||
package kube_inventory
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"k8s.io/api/apps/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
|
||||
"github.com/influxdata/telegraf"
|
||||
"github.com/influxdata/telegraf/testutil"
|
||||
)
|
||||
|
||||
func TestStatefulSet(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 statefulsets",
|
||||
handler: &mockHandler{
|
||||
responseMap: map[string]interface{}{
|
||||
"/statefulsets/": &v1.StatefulSetList{},
|
||||
},
|
||||
},
|
||||
hasError: false,
|
||||
},
|
||||
{
|
||||
name: "collect statefulsets",
|
||||
handler: &mockHandler{
|
||||
responseMap: map[string]interface{}{
|
||||
"/statefulsets/": &v1.StatefulSetList{
|
||||
Items: []v1.StatefulSet{
|
||||
{
|
||||
Status: v1.StatefulSetStatus{
|
||||
Replicas: 2,
|
||||
CurrentReplicas: 4,
|
||||
ReadyReplicas: 1,
|
||||
UpdatedReplicas: 3,
|
||||
ObservedGeneration: 119,
|
||||
},
|
||||
Spec: v1.StatefulSetSpec{
|
||||
Replicas: toPtr(int32(3)),
|
||||
Selector: &metav1.LabelSelector{
|
||||
MatchLabels: map[string]string{
|
||||
"select1": "s1",
|
||||
"select2": "s2",
|
||||
},
|
||||
},
|
||||
},
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Generation: 332,
|
||||
Namespace: "ns1",
|
||||
Name: "sts1",
|
||||
CreationTimestamp: metav1.Time{Time: now},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
output: []telegraf.Metric{
|
||||
testutil.MustMetric(
|
||||
"kubernetes_statefulset",
|
||||
map[string]string{
|
||||
"namespace": "ns1",
|
||||
"statefulset_name": "sts1",
|
||||
"selector_select1": "s1",
|
||||
"selector_select2": "s2",
|
||||
},
|
||||
map[string]interface{}{
|
||||
"generation": int64(332),
|
||||
"observed_generation": int64(119),
|
||||
"created": now.UnixNano(),
|
||||
"spec_replicas": int32(3),
|
||||
"replicas": int32(2),
|
||||
"replicas_current": int32(4),
|
||||
"replicas_ready": int32(1),
|
||||
"replicas_updated": int32(3),
|
||||
},
|
||||
time.Unix(0, 0),
|
||||
),
|
||||
},
|
||||
hasError: false,
|
||||
},
|
||||
{
|
||||
name: "no label selector",
|
||||
handler: &mockHandler{
|
||||
responseMap: map[string]interface{}{
|
||||
"/statefulsets/": &v1.StatefulSetList{
|
||||
Items: []v1.StatefulSet{
|
||||
{
|
||||
Status: v1.StatefulSetStatus{
|
||||
Replicas: 2,
|
||||
CurrentReplicas: 4,
|
||||
ReadyReplicas: 1,
|
||||
UpdatedReplicas: 3,
|
||||
ObservedGeneration: 119,
|
||||
},
|
||||
Spec: v1.StatefulSetSpec{
|
||||
Replicas: toPtr(int32(3)),
|
||||
Selector: nil,
|
||||
},
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Generation: 332,
|
||||
Namespace: "ns1",
|
||||
Name: "sts1",
|
||||
CreationTimestamp: metav1.Time{Time: now},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
output: []telegraf.Metric{
|
||||
testutil.MustMetric(
|
||||
"kubernetes_statefulset",
|
||||
map[string]string{
|
||||
"namespace": "ns1",
|
||||
"statefulset_name": "sts1",
|
||||
},
|
||||
map[string]interface{}{
|
||||
"generation": int64(332),
|
||||
"observed_generation": int64(119),
|
||||
"created": now.UnixNano(),
|
||||
"spec_replicas": int32(3),
|
||||
"replicas": int32(2),
|
||||
"replicas_current": int32(4),
|
||||
"replicas_ready": int32(1),
|
||||
"replicas_updated": int32(3),
|
||||
},
|
||||
time.Unix(0, 0),
|
||||
),
|
||||
},
|
||||
hasError: false,
|
||||
},
|
||||
{
|
||||
name: "no desired number of replicas",
|
||||
handler: &mockHandler{
|
||||
responseMap: map[string]interface{}{
|
||||
"/statefulsets/": &v1.StatefulSetList{
|
||||
Items: []v1.StatefulSet{
|
||||
{
|
||||
Status: v1.StatefulSetStatus{
|
||||
Replicas: 2,
|
||||
CurrentReplicas: 4,
|
||||
ReadyReplicas: 1,
|
||||
UpdatedReplicas: 3,
|
||||
ObservedGeneration: 119,
|
||||
},
|
||||
Spec: v1.StatefulSetSpec{
|
||||
Replicas: nil,
|
||||
Selector: &metav1.LabelSelector{
|
||||
MatchLabels: map[string]string{
|
||||
"select1": "s1",
|
||||
"select2": "s2",
|
||||
},
|
||||
},
|
||||
},
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Generation: 332,
|
||||
Namespace: "ns1",
|
||||
Name: "sts1",
|
||||
CreationTimestamp: metav1.Time{Time: now},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
output: []telegraf.Metric{
|
||||
testutil.MustMetric(
|
||||
"kubernetes_statefulset",
|
||||
map[string]string{
|
||||
"namespace": "ns1",
|
||||
"statefulset_name": "sts1",
|
||||
"selector_select1": "s1",
|
||||
"selector_select2": "s2",
|
||||
},
|
||||
map[string]interface{}{
|
||||
"generation": int64(332),
|
||||
"observed_generation": int64(119),
|
||||
"created": now.UnixNano(),
|
||||
"replicas": int32(2),
|
||||
"replicas_current": int32(4),
|
||||
"replicas_ready": int32(1),
|
||||
"replicas_updated": int32(3),
|
||||
},
|
||||
time.Unix(0, 0),
|
||||
),
|
||||
},
|
||||
hasError: false,
|
||||
},
|
||||
}
|
||||
|
||||
for _, v := range tests {
|
||||
ks := &KubernetesInventory{
|
||||
client: cli,
|
||||
}
|
||||
require.NoError(t, ks.createSelectorFilters())
|
||||
acc := &testutil.Accumulator{}
|
||||
items := ((v.handler.responseMap["/statefulsets/"]).(*v1.StatefulSetList)).Items
|
||||
for i := range items {
|
||||
ks.gatherStatefulSet(&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 TestStatefulSetSelectorFilter(t *testing.T) {
|
||||
cli := &client{}
|
||||
now := time.Now()
|
||||
now = time.Date(now.Year(), now.Month(), now.Day(), now.Hour(), 1, 36, 0, now.Location())
|
||||
|
||||
responseMap := map[string]interface{}{
|
||||
"/statefulsets/": &v1.StatefulSetList{
|
||||
Items: []v1.StatefulSet{
|
||||
{
|
||||
Status: v1.StatefulSetStatus{
|
||||
Replicas: 2,
|
||||
CurrentReplicas: 4,
|
||||
ReadyReplicas: 1,
|
||||
UpdatedReplicas: 3,
|
||||
ObservedGeneration: 119,
|
||||
},
|
||||
Spec: v1.StatefulSetSpec{
|
||||
Replicas: toPtr(int32(3)),
|
||||
Selector: &metav1.LabelSelector{
|
||||
MatchLabels: map[string]string{
|
||||
"select1": "s1",
|
||||
"select2": "s2",
|
||||
},
|
||||
},
|
||||
},
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Generation: 332,
|
||||
Namespace: "ns1",
|
||||
Name: "sts1",
|
||||
CreationTimestamp: metav1.Time{Time: now},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
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["/statefulsets/"]).(*v1.StatefulSetList)).Items
|
||||
for i := range items {
|
||||
ks.gatherStatefulSet(&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)
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue