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.
119 lines
2.4 KiB
119 lines
2.4 KiB
/*
|
|
Copyright 2016 The Kubernetes Authors All rights reserved.
|
|
|
|
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 main
|
|
|
|
import (
|
|
"sort"
|
|
"testing"
|
|
)
|
|
|
|
func newKindSorter(m []manifest, s SortOrder) *kindSorter {
|
|
o := make(map[string]int, len(s))
|
|
for v, k := range s {
|
|
o[k] = v
|
|
}
|
|
|
|
return &kindSorter{
|
|
manifests: m,
|
|
ordering: o,
|
|
}
|
|
}
|
|
|
|
// sortByKind does an in-place sort of manifests by Kind.
|
|
//
|
|
// Results are sorted by 'ordering'
|
|
func sortByKind(manifests []manifest, ordering SortOrder) []manifest {
|
|
ks := newKindSorter(manifests, ordering)
|
|
sort.Sort(ks)
|
|
return ks.manifests
|
|
}
|
|
|
|
type kindSorter struct {
|
|
ordering map[string]int
|
|
manifests []manifest
|
|
}
|
|
|
|
func (k *kindSorter) Len() int { return len(k.manifests) }
|
|
|
|
func (k *kindSorter) Swap(i, j int) { k.manifests[i], k.manifests[j] = k.manifests[j], k.manifests[i] }
|
|
|
|
func (k *kindSorter) Less(i, j int) bool {
|
|
a := k.manifests[i]
|
|
b := k.manifests[j]
|
|
first, ok := k.ordering[a.head.Kind]
|
|
if !ok {
|
|
// Unknown is always last
|
|
return false
|
|
}
|
|
second, ok := k.ordering[b.head.Kind]
|
|
if !ok {
|
|
return true
|
|
}
|
|
return first < second
|
|
}
|
|
|
|
func TestKindSorter(t *testing.T) {
|
|
manifests := []manifest{
|
|
{
|
|
name: "m",
|
|
content: "",
|
|
head: &simpleHead{Kind: "Deployment"},
|
|
},
|
|
{
|
|
name: "l",
|
|
content: "",
|
|
head: &simpleHead{Kind: "Service"},
|
|
},
|
|
{
|
|
name: "!",
|
|
content: "",
|
|
head: &simpleHead{Kind: "HonkyTonkSet"},
|
|
},
|
|
{
|
|
name: "h",
|
|
content: "",
|
|
head: &simpleHead{Kind: "Namespace"},
|
|
},
|
|
{
|
|
name: "e",
|
|
content: "",
|
|
head: &simpleHead{Kind: "ConfigMap"},
|
|
},
|
|
}
|
|
|
|
res := sortByKind(manifests, InstallOrder)
|
|
got := ""
|
|
expect := "helm!"
|
|
for _, r := range res {
|
|
got += r.name
|
|
}
|
|
if got != expect {
|
|
t.Errorf("Expected %q, got %q", expect, got)
|
|
}
|
|
|
|
expect = "lmeh!"
|
|
got = ""
|
|
res = sortByKind(manifests, UninstallOrder)
|
|
for _, r := range res {
|
|
got += r.name
|
|
}
|
|
if got != expect {
|
|
t.Errorf("Expected %q, got %q", expect, got)
|
|
}
|
|
|
|
}
|