first import

2021-12-29 16:10:50 -05:00
parent f84ee3a259
commit b0085938f1
66 changed files with 5255 additions and 0 deletions
@@ -0,0 +1,107 @@
+package common
+
+import (
+    "testing"
+
+    "github.com/k8s-at-home/library-charts/test/helmunit"
+    "github.com/stretchr/testify/suite"
+)
+
+type HorizontalPodAutoscalerTestSuite struct {
+    suite.Suite
+    Chart helmunit.HelmChart
+}
+
+func (suite *HorizontalPodAutoscalerTestSuite) SetupSuite() {
+    suite.Chart = helmunit.New("common-test", "../../../../helper-charts/common-test")
+    suite.Chart.UpdateDependencies()
+}
+
+// We need this function to kick off the test suite, otherwise
+// "go test" won't know about our tests
+func TestHorizontalPodAutoscaler(t *testing.T) {
+    suite.Run(t, new(HorizontalPodAutoscalerTestSuite))
+}
+
+func (suite *HorizontalPodAutoscalerTestSuite) TestValues() {
+    baseValues := []string{"autoscaling.enabled=true"}
+    tests := map[string]struct {
+        values                          []string
+        expectedHorizontalPodAutoscaler bool
+        expectedTarget                  string
+        expectedMinReplicas             int
+        expectedMaxReplicas             int
+    }{
+        "Default": {
+            values:                          nil,
+            expectedHorizontalPodAutoscaler: false,
+        },
+        "Enabled": {
+            values:                          baseValues,
+            expectedHorizontalPodAutoscaler: true, expectedTarget: "common-test", expectedMinReplicas: 1, expectedMaxReplicas: 3,
+        },
+        "CustomSettings": {
+            values:                          append(baseValues, "autoscaling.target=common-custom", "autoscaling.minReplicas=4", "autoscaling.maxReplicas=8"),
+            expectedHorizontalPodAutoscaler: true, expectedTarget: "common-custom", expectedMinReplicas: 4, expectedMaxReplicas: 8,
+        },
+    }
+    for name, tc := range tests {
+        suite.Suite.Run(name, func() {
+            err := suite.Chart.Render(nil, tc.values, nil)
+            if err != nil {
+                suite.FailNow(err.Error())
+            }
+
+            manifest := suite.Chart.Manifests.Get("HorizontalPodAutoscaler", "common-test")
+            if tc.expectedHorizontalPodAutoscaler {
+                suite.Assertions.NotEmpty(manifest)
+                suite.Assertions.EqualValues(tc.expectedTarget, manifest.Path("spec.scaleTargetRef.name").Data())
+                suite.Assertions.EqualValues(tc.expectedMinReplicas, manifest.Path("spec.minReplicas").Data())
+                suite.Assertions.EqualValues(tc.expectedMaxReplicas, manifest.Path("spec.maxReplicas").Data())
+            } else {
+                suite.Assertions.Empty(manifest)
+            }
+        })
+    }
+}
+
+func (suite *HorizontalPodAutoscalerTestSuite) TestMetrics() {
+    baseValues := []string{"autoscaling.enabled=true"}
+    tests := map[string]struct {
+        values            []string
+        expectedResources map[string]int
+    }{
+        "targetCPUUtilizationPercentage": {
+            values:            append(baseValues, "autoscaling.targetCPUUtilizationPercentage=60"),
+            expectedResources: map[string]int{"cpu": 60},
+        },
+        "targetMemoryUtilizationPercentage": {
+            values:            append(baseValues, "autoscaling.targetMemoryUtilizationPercentage=70"),
+            expectedResources: map[string]int{"memory": 70},
+        },
+        "Combined": {
+            values:            append(baseValues, "autoscaling.targetCPUUtilizationPercentage=60", "autoscaling.targetMemoryUtilizationPercentage=70"),
+            expectedResources: map[string]int{"cpu": 60, "memory": 70},
+        },
+    }
+    for name, tc := range tests {
+        suite.Suite.Run(name, func() {
+            err := suite.Chart.Render(nil, tc.values, nil)
+            if err != nil {
+                suite.FailNow(err.Error())
+            }
+
+            manifest := suite.Chart.Manifests.Get("HorizontalPodAutoscaler", "common-test")
+            suite.Assertions.NotEmpty(manifest)
+
+            manifestMetrics := manifest.Path("spec.metrics").Children()
+
+            metricsMap := make(map[string]int)
+            for _, manifestMetric := range manifestMetrics {
+                metricsMap[manifestMetric.Path("resource.name").Data().(string)] = int(manifestMetric.Path("resource.targetAverageUtilization").Data().(float64))
+            }
+
+            suite.Assertions.EqualValues(tc.expectedResources, metricsMap)
+        })
+    }
+}