add test

Signed-off-by: Lyndon Shi <shilyndon@outlook.com>

pkg/action/hooks_test.go

@@ -0,0 +1,52 @@
+/*
+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 action
+
+import (
+	"sort"
+	"testing"
+
+	"helm.sh/helm/v3/pkg/release"
+)
+
+func Test_hookByWeight(t *testing.T) {
+	hooks := []*release.Hook{
+		{Weight: 1, Kind: "Pod", Name: "podA"},
+		{Weight: 1, Kind: "ServiceAccount", Name: "ServiceAccountB"},
+		{Weight: 1, Kind: "ServiceAccount", Name: "ServiceAccountA"},
+		{Weight: 0, Kind: "Job", Name: "Job"},
+		{Weight: -1, Kind: "APIService", Name: "APIServiceB"},
+		{Weight: -1, Kind: "APIService", Name: "APIServiceA"},
+	}
+
+	sort.Stable(hookByWeight(hooks))
+
+	expected := []*release.Hook{
+		{Weight: -1, Kind: "APIService", Name: "APIServiceA"},
+		{Weight: -1, Kind: "APIService", Name: "APIServiceB"},
+		{Weight: 0, Kind: "Job", Name: "Job"},
+		{Weight: 1, Kind: "ServiceAccount", Name: "ServiceAccountA"},
+		{Weight: 1, Kind: "ServiceAccount", Name: "ServiceAccountB"},
+		{Weight: 1, Kind: "Pod", Name: "podA"},
+	}
+
+	for i, hook := range hooks {
+		if hook.Name != expected[i].Name {
+			t.Errorf("Expected hook %d to be %s, got %s", i, expected[i].Name, hook.Name)
+		}
+	}
+}