Adding upstream version 1.34.4.

Signed-off-by: Daniel Baumann <daniel@debian.org>

plugins/inputs/kube_inventory/deployment_test.go

@@ -0,0 +1,295 @@
+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"
+	"k8s.io/apimachinery/pkg/util/intstr"
+
+	"github.com/influxdata/telegraf"
+	"github.com/influxdata/telegraf/testutil"
+)
+
+func TestDeployment(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 deployments",
+			handler: &mockHandler{
+				responseMap: map[string]interface{}{
+					"/deployments/": &v1.DeploymentList{},
+				},
+			},
+			hasError: false,
+		},
+		{
+			name: "collect deployments",
+			handler: &mockHandler{
+				responseMap: map[string]interface{}{
+					"/deployments/": &v1.DeploymentList{
+						Items: []v1.Deployment{
+							{
+								Status: v1.DeploymentStatus{
+									Replicas:            3,
+									AvailableReplicas:   1,
+									UnavailableReplicas: 4,
+									UpdatedReplicas:     2,
+									ObservedGeneration:  9121,
+								},
+								Spec: v1.DeploymentSpec{
+									Strategy: v1.DeploymentStrategy{
+										RollingUpdate: &v1.RollingUpdateDeployment{
+											MaxUnavailable: &intstr.IntOrString{
+												IntVal: 30,
+											},
+											MaxSurge: &intstr.IntOrString{
+												IntVal: 20,
+											},
+										},
+									},
+									Replicas: toPtr(int32(4)),
+									Selector: &metav1.LabelSelector{
+										MatchLabels: map[string]string{
+											"select1": "s1",
+											"select2": "s2",
+										},
+									},
+								},
+								ObjectMeta: metav1.ObjectMeta{
+									Generation: 11221,
+									Namespace:  "ns1",
+									Name:       "deploy1",
+									Labels: map[string]string{
+										"lab1": "v1",
+										"lab2": "v2",
+									},
+									CreationTimestamp: metav1.Time{Time: now},
+								},
+							},
+						},
+					},
+				},
+			},
+			output: []telegraf.Metric{
+				testutil.MustMetric(
+					"kubernetes_deployment",
+					map[string]string{
+						"namespace":        "ns1",
+						"deployment_name":  "deploy1",
+						"selector_select1": "s1",
+						"selector_select2": "s2",
+					},
+					map[string]interface{}{
+						"replicas_available":   int32(1),
+						"replicas_unavailable": int32(4),
+						"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["/deployments/"]).(*v1.DeploymentList)).Items
+		for i := range items {
+			ks.gatherDeployment(&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 TestDeploymentSelectorFilter(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{}{
+		"/deployments/": &v1.DeploymentList{
+			Items: []v1.Deployment{
+				{
+					Status: v1.DeploymentStatus{
+						Replicas:            3,
+						AvailableReplicas:   1,
+						UnavailableReplicas: 4,
+						UpdatedReplicas:     2,
+						ObservedGeneration:  9121,
+					},
+					Spec: v1.DeploymentSpec{
+						Strategy: v1.DeploymentStrategy{
+							RollingUpdate: &v1.RollingUpdateDeployment{
+								MaxUnavailable: &intstr.IntOrString{
+									IntVal: 30,
+								},
+								MaxSurge: &intstr.IntOrString{
+									IntVal: 20,
+								},
+							},
+						},
+						Replicas: toPtr(int32(4)),
+						Selector: &metav1.LabelSelector{
+							MatchLabels: map[string]string{
+								"select1": "s1",
+								"select2": "s2",
+							},
+						},
+					},
+					ObjectMeta: metav1.ObjectMeta{
+						Generation: 11221,
+						Namespace:  "ns1",
+						Name:       "deploy1",
+						Labels: map[string]string{
+							"lab1": "v1",
+							"lab2": "v2",
+						},
+						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["/deployments/"]).(*v1.DeploymentList)).Items
+		for i := range items {
+			ks.gatherDeployment(&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)
+	}
+}