mirror of https://github.com/helm/helm
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
316 lines
6.5 KiB
316 lines
6.5 KiB
/*
|
|
Copyright The Helm Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package releaseutil
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"helm.sh/helm/v3/pkg/release"
|
|
)
|
|
|
|
func TestKindSorter(t *testing.T) {
|
|
manifests := []Manifest{
|
|
{
|
|
Name: "i",
|
|
Head: &SimpleHead{Kind: "ClusterRole"},
|
|
},
|
|
{
|
|
Name: "I",
|
|
Head: &SimpleHead{Kind: "ClusterRoleList"},
|
|
},
|
|
{
|
|
Name: "j",
|
|
Head: &SimpleHead{Kind: "ClusterRoleBinding"},
|
|
},
|
|
{
|
|
Name: "J",
|
|
Head: &SimpleHead{Kind: "ClusterRoleBindingList"},
|
|
},
|
|
{
|
|
Name: "f",
|
|
Head: &SimpleHead{Kind: "ConfigMap"},
|
|
},
|
|
{
|
|
Name: "u",
|
|
Head: &SimpleHead{Kind: "CronJob"},
|
|
},
|
|
{
|
|
Name: "2",
|
|
Head: &SimpleHead{Kind: "CustomResourceDefinition"},
|
|
},
|
|
{
|
|
Name: "n",
|
|
Head: &SimpleHead{Kind: "DaemonSet"},
|
|
},
|
|
{
|
|
Name: "r",
|
|
Head: &SimpleHead{Kind: "Deployment"},
|
|
},
|
|
{
|
|
Name: "!",
|
|
Head: &SimpleHead{Kind: "HonkyTonkSet"},
|
|
},
|
|
{
|
|
Name: "v",
|
|
Head: &SimpleHead{Kind: "Ingress"},
|
|
},
|
|
{
|
|
Name: "t",
|
|
Head: &SimpleHead{Kind: "Job"},
|
|
},
|
|
{
|
|
Name: "c",
|
|
Head: &SimpleHead{Kind: "LimitRange"},
|
|
},
|
|
{
|
|
Name: "a",
|
|
Head: &SimpleHead{Kind: "Namespace"},
|
|
},
|
|
{
|
|
Name: "A",
|
|
Head: &SimpleHead{Kind: "NetworkPolicy"},
|
|
},
|
|
{
|
|
Name: "g",
|
|
Head: &SimpleHead{Kind: "PersistentVolume"},
|
|
},
|
|
{
|
|
Name: "h",
|
|
Head: &SimpleHead{Kind: "PersistentVolumeClaim"},
|
|
},
|
|
{
|
|
Name: "o",
|
|
Head: &SimpleHead{Kind: "Pod"},
|
|
},
|
|
{
|
|
Name: "3",
|
|
Head: &SimpleHead{Kind: "PodDisruptionBudget"},
|
|
},
|
|
{
|
|
Name: "C",
|
|
Head: &SimpleHead{Kind: "PodSecurityPolicy"},
|
|
},
|
|
{
|
|
Name: "q",
|
|
Head: &SimpleHead{Kind: "ReplicaSet"},
|
|
},
|
|
{
|
|
Name: "p",
|
|
Head: &SimpleHead{Kind: "ReplicationController"},
|
|
},
|
|
{
|
|
Name: "b",
|
|
Head: &SimpleHead{Kind: "ResourceQuota"},
|
|
},
|
|
{
|
|
Name: "k",
|
|
Head: &SimpleHead{Kind: "Role"},
|
|
},
|
|
{
|
|
Name: "K",
|
|
Head: &SimpleHead{Kind: "RoleList"},
|
|
},
|
|
{
|
|
Name: "l",
|
|
Head: &SimpleHead{Kind: "RoleBinding"},
|
|
},
|
|
{
|
|
Name: "L",
|
|
Head: &SimpleHead{Kind: "RoleBindingList"},
|
|
},
|
|
{
|
|
Name: "e",
|
|
Head: &SimpleHead{Kind: "Secret"},
|
|
},
|
|
{
|
|
Name: "m",
|
|
Head: &SimpleHead{Kind: "Service"},
|
|
},
|
|
{
|
|
Name: "d",
|
|
Head: &SimpleHead{Kind: "ServiceAccount"},
|
|
},
|
|
{
|
|
Name: "s",
|
|
Head: &SimpleHead{Kind: "StatefulSet"},
|
|
},
|
|
{
|
|
Name: "1",
|
|
Head: &SimpleHead{Kind: "StorageClass"},
|
|
},
|
|
{
|
|
Name: "w",
|
|
Head: &SimpleHead{Kind: "APIService"},
|
|
},
|
|
{
|
|
Name: "x",
|
|
Head: &SimpleHead{Kind: "HorizontalPodAutoscaler"},
|
|
},
|
|
}
|
|
|
|
for _, test := range []struct {
|
|
description string
|
|
order KindSortOrder
|
|
expected string
|
|
}{
|
|
{"install", InstallOrder, "aAbcC3def1gh2iIjJkKlLmnopqrxstuvw!"},
|
|
{"uninstall", UninstallOrder, "wvmutsxrqponLlKkJjIi2hg1fed3CcbAa!"},
|
|
} {
|
|
var buf bytes.Buffer
|
|
t.Run(test.description, func(t *testing.T) {
|
|
if got, want := len(test.expected), len(manifests); got != want {
|
|
t.Fatalf("Expected %d names in order, got %d", want, got)
|
|
}
|
|
defer buf.Reset()
|
|
for _, r := range manifestsSortedByKind(manifests, test.order) {
|
|
buf.WriteString(r.Name)
|
|
}
|
|
if got := buf.String(); got != test.expected {
|
|
t.Errorf("Expected %q, got %q", test.expected, got)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// TestKindSorterKeepOriginalOrder verifies manifests of same kind are kept in original order
|
|
func TestKindSorterKeepOriginalOrder(t *testing.T) {
|
|
manifests := []Manifest{
|
|
{
|
|
Name: "a",
|
|
Head: &SimpleHead{Kind: "ClusterRole"},
|
|
},
|
|
{
|
|
Name: "A",
|
|
Head: &SimpleHead{Kind: "ClusterRole"},
|
|
},
|
|
{
|
|
Name: "0",
|
|
Head: &SimpleHead{Kind: "ConfigMap"},
|
|
},
|
|
{
|
|
Name: "1",
|
|
Head: &SimpleHead{Kind: "ConfigMap"},
|
|
},
|
|
{
|
|
Name: "z",
|
|
Head: &SimpleHead{Kind: "ClusterRoleBinding"},
|
|
},
|
|
{
|
|
Name: "!",
|
|
Head: &SimpleHead{Kind: "ClusterRoleBinding"},
|
|
},
|
|
{
|
|
Name: "u2",
|
|
Head: &SimpleHead{Kind: "Unknown"},
|
|
},
|
|
{
|
|
Name: "u1",
|
|
Head: &SimpleHead{Kind: "Unknown"},
|
|
},
|
|
{
|
|
Name: "t3",
|
|
Head: &SimpleHead{Kind: "Unknown2"},
|
|
},
|
|
}
|
|
for _, test := range []struct {
|
|
description string
|
|
order KindSortOrder
|
|
expected string
|
|
}{
|
|
// expectation is sorted by kind (unknown is last) and within each group of same kind, the order is kept
|
|
{"cm,clusterRole,clusterRoleBinding,Unknown,Unknown2", InstallOrder, "01aAz!u2u1t3"},
|
|
} {
|
|
var buf bytes.Buffer
|
|
t.Run(test.description, func(t *testing.T) {
|
|
defer buf.Reset()
|
|
for _, r := range manifestsSortedByKind(manifests, test.order) {
|
|
buf.WriteString(r.Name)
|
|
}
|
|
if got := buf.String(); got != test.expected {
|
|
t.Errorf("Expected %q, got %q", test.expected, got)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestKindSorterNamespaceAgainstUnknown(t *testing.T) {
|
|
unknown := Manifest{
|
|
Name: "a",
|
|
Head: &SimpleHead{Kind: "Unknown"},
|
|
}
|
|
namespace := Manifest{
|
|
Name: "b",
|
|
Head: &SimpleHead{Kind: "Namespace"},
|
|
}
|
|
|
|
manifests := []Manifest{unknown, namespace}
|
|
manifests = manifestsSortedByKind(manifests, InstallOrder)
|
|
|
|
expectedOrder := []Manifest{namespace, unknown}
|
|
for i, manifest := range manifests {
|
|
if expectedOrder[i].Name != manifest.Name {
|
|
t.Errorf("Expected %s, got %s", expectedOrder[i].Name, manifest.Name)
|
|
}
|
|
}
|
|
}
|
|
|
|
// test hook sorting with a small subset of kinds, since it uses the same algorithm as manifestsSortedByKind
|
|
func TestKindSorterForHooks(t *testing.T) {
|
|
hooks := []*release.Hook{
|
|
{
|
|
Name: "i",
|
|
Kind: "ClusterRole",
|
|
},
|
|
{
|
|
Name: "j",
|
|
Kind: "ClusterRoleBinding",
|
|
},
|
|
{
|
|
Name: "c",
|
|
Kind: "LimitRange",
|
|
},
|
|
{
|
|
Name: "a",
|
|
Kind: "Namespace",
|
|
},
|
|
}
|
|
|
|
for _, test := range []struct {
|
|
description string
|
|
order KindSortOrder
|
|
expected string
|
|
}{
|
|
{"install", InstallOrder, "acij"},
|
|
{"uninstall", UninstallOrder, "jica"},
|
|
} {
|
|
var buf bytes.Buffer
|
|
t.Run(test.description, func(t *testing.T) {
|
|
if got, want := len(test.expected), len(hooks); got != want {
|
|
t.Fatalf("Expected %d names in order, got %d", want, got)
|
|
}
|
|
defer buf.Reset()
|
|
for _, r := range hooksSortedByKind(hooks, test.order) {
|
|
buf.WriteString(r.Name)
|
|
}
|
|
if got := buf.String(); got != test.expected {
|
|
t.Errorf("Expected %q, got %q", test.expected, got)
|
|
}
|
|
})
|
|
}
|
|
}
|