mirror of
https://github.com/kubernetes-sigs/kustomize.git
synced 2026-07-18 18:18:36 +00:00
Adds the PruneOptions and implements the methods for this struct
This commit is contained in:
Sean Sullivan
244
cmd/kubectl/kubectlcobra/prune_test.go
Normal file
244
cmd/kubectl/kubectlcobra/prune_test.go
Normal file
@@ -0,0 +1,244 @@
|
||||
// Copyright 2019 The Kubernetes Authors.
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
// package kubectlcobra contains cobra commands from kubectl
|
||||
package kubectlcobra
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||
"k8s.io/cli-runtime/pkg/resource"
|
||||
)
|
||||
|
||||
var pod1Inv = &Inventory{
|
||||
Namespace: testNamespace,
|
||||
Name: pod1Name,
|
||||
GroupKind: schema.GroupKind{
|
||||
Group: "",
|
||||
Kind: "Pod",
|
||||
},
|
||||
}
|
||||
|
||||
var pod2Inv = &Inventory{
|
||||
Namespace: testNamespace,
|
||||
Name: pod2Name,
|
||||
GroupKind: schema.GroupKind{
|
||||
Group: "",
|
||||
Kind: "Pod",
|
||||
},
|
||||
}
|
||||
|
||||
var pod3Inv = &Inventory{
|
||||
Namespace: testNamespace,
|
||||
Name: pod3Name,
|
||||
GroupKind: schema.GroupKind{
|
||||
Group: "",
|
||||
Kind: "Pod",
|
||||
},
|
||||
}
|
||||
|
||||
var groupingInv = &Inventory{
|
||||
Namespace: testNamespace,
|
||||
Name: groupingObjName,
|
||||
GroupKind: schema.GroupKind{
|
||||
Group: "",
|
||||
Kind: "ConfigMap",
|
||||
},
|
||||
}
|
||||
|
||||
func TestInfoToInventory(t *testing.T) {
|
||||
tests := map[string]struct {
|
||||
info *resource.Info
|
||||
expected *Inventory
|
||||
isError bool
|
||||
}{
|
||||
"Nil info is an error": {
|
||||
info: nil,
|
||||
expected: nil,
|
||||
isError: true,
|
||||
},
|
||||
"Nil info object is an error": {
|
||||
info: nilInfo,
|
||||
expected: nil,
|
||||
isError: true,
|
||||
},
|
||||
"Pod 1 object becomes Pod 1 inventory": {
|
||||
info: pod1Info,
|
||||
expected: pod1Inv,
|
||||
isError: false,
|
||||
},
|
||||
"Grouping object becomes grouping inventory": {
|
||||
info: copyGroupingInfo(),
|
||||
expected: groupingInv,
|
||||
isError: false,
|
||||
},
|
||||
}
|
||||
|
||||
for name, test := range tests {
|
||||
t.Run(name, func(t *testing.T) {
|
||||
actual, err := infoToInventory(test.info)
|
||||
if test.isError && err == nil {
|
||||
t.Errorf("Did not receive expected error.\n")
|
||||
}
|
||||
if !test.isError {
|
||||
if err != nil {
|
||||
t.Errorf("Receieved unexpected error: %s\n", err)
|
||||
}
|
||||
if !test.expected.Equals(actual) {
|
||||
t.Errorf("Expected inventory (%s), got (%s)\n", test.expected, actual)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// Returns a grouping object with the inventory set from
|
||||
// the passed "children".
|
||||
func createGroupingInfo(name string, children ...(*resource.Info)) *resource.Info {
|
||||
groupingObjCopy := groupingObj.DeepCopy()
|
||||
var groupingInfo = &resource.Info{
|
||||
Namespace: testNamespace,
|
||||
Name: groupingObjName,
|
||||
Object: groupingObjCopy,
|
||||
}
|
||||
infos := []*resource.Info{groupingInfo}
|
||||
infos = append(infos, children...)
|
||||
_ = addInventoryToGroupingObj(infos)
|
||||
return groupingInfo
|
||||
}
|
||||
|
||||
func TestUnionPastInventory(t *testing.T) {
|
||||
tests := map[string]struct {
|
||||
groupingInfos []*resource.Info
|
||||
expected []*Inventory
|
||||
}{
|
||||
"Empty grouping objects = empty inventory set": {
|
||||
groupingInfos: []*resource.Info{},
|
||||
expected: []*Inventory{},
|
||||
},
|
||||
"No children in grouping object, equals no inventory": {
|
||||
groupingInfos: []*resource.Info{createGroupingInfo("test-1")},
|
||||
expected: []*Inventory{},
|
||||
},
|
||||
"Grouping object with Pod1 returns inventory with Pod1": {
|
||||
groupingInfos: []*resource.Info{createGroupingInfo("test-1", pod1Info)},
|
||||
expected: []*Inventory{pod1Inv},
|
||||
},
|
||||
"Grouping object with three pods returns inventory with three pods": {
|
||||
groupingInfos: []*resource.Info{
|
||||
createGroupingInfo("test-1", pod1Info, pod2Info, pod3Info),
|
||||
},
|
||||
expected: []*Inventory{pod1Inv, pod2Inv, pod3Inv},
|
||||
},
|
||||
"Two grouping objects with different pods returns inventory with both pods": {
|
||||
groupingInfos: []*resource.Info{
|
||||
createGroupingInfo("test-1", pod1Info),
|
||||
createGroupingInfo("test-2", pod2Info),
|
||||
},
|
||||
expected: []*Inventory{pod1Inv, pod2Inv},
|
||||
},
|
||||
"Two grouping objects with overlapping pods returns set of pods": {
|
||||
groupingInfos: []*resource.Info{
|
||||
createGroupingInfo("test-1", pod1Info, pod2Info),
|
||||
createGroupingInfo("test-2", pod2Info, pod3Info),
|
||||
},
|
||||
expected: []*Inventory{pod1Inv, pod2Inv, pod3Inv},
|
||||
},
|
||||
}
|
||||
|
||||
for name, test := range tests {
|
||||
t.Run(name, func(t *testing.T) {
|
||||
actual, err := unionPastInventory(test.groupingInfos)
|
||||
expected := NewInventorySet(test.expected)
|
||||
if err != nil {
|
||||
t.Errorf("Unexpected error received: %s\n", err)
|
||||
}
|
||||
if !expected.Equals(actual) {
|
||||
t.Errorf("Expected inventory (%s), got (%s)\n", expected, actual)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestCalcPruneSet(t *testing.T) {
|
||||
tests := map[string]struct {
|
||||
past []*resource.Info
|
||||
current *resource.Info
|
||||
expected []*Inventory
|
||||
isError bool
|
||||
}{
|
||||
"Object not unstructured--error": {
|
||||
past: []*resource.Info{nonUnstructuredGroupingInfo},
|
||||
current: &resource.Info{},
|
||||
expected: []*Inventory{},
|
||||
isError: true,
|
||||
},
|
||||
"No past group objects--no prune set": {
|
||||
|
||||
past: []*resource.Info{},
|
||||
current: createGroupingInfo("test-1"),
|
||||
expected: []*Inventory{},
|
||||
isError: false,
|
||||
},
|
||||
"Empty past grouping object--no prune set": {
|
||||
past: []*resource.Info{createGroupingInfo("test-1")},
|
||||
current: createGroupingInfo("test-1"),
|
||||
expected: []*Inventory{},
|
||||
isError: false,
|
||||
},
|
||||
"Pod1 - Pod1 = empty set": {
|
||||
past: []*resource.Info{
|
||||
createGroupingInfo("test-1", pod1Info),
|
||||
},
|
||||
current: createGroupingInfo("test-1", pod1Info),
|
||||
expected: []*Inventory{},
|
||||
isError: false,
|
||||
},
|
||||
"(Pod1, Pod2) - Pod1 = Pod2": {
|
||||
past: []*resource.Info{
|
||||
createGroupingInfo("test-1", pod1Info, pod2Info),
|
||||
},
|
||||
current: createGroupingInfo("test-1", pod1Info),
|
||||
expected: []*Inventory{pod2Inv},
|
||||
isError: false,
|
||||
},
|
||||
"(Pod1, Pod2) - Pod2 = Pod1": {
|
||||
past: []*resource.Info{
|
||||
createGroupingInfo("test-1", pod1Info, pod2Info),
|
||||
},
|
||||
current: createGroupingInfo("test-1", pod2Info),
|
||||
expected: []*Inventory{pod1Inv},
|
||||
isError: false,
|
||||
},
|
||||
"(Pod1, Pod2, Pod3) - Pod2 = Pod1, Pod3": {
|
||||
past: []*resource.Info{
|
||||
createGroupingInfo("test-1", pod1Info, pod2Info),
|
||||
createGroupingInfo("test-1", pod2Info, pod3Info),
|
||||
},
|
||||
current: createGroupingInfo("test-1", pod2Info),
|
||||
expected: []*Inventory{pod1Inv, pod3Inv},
|
||||
isError: false,
|
||||
},
|
||||
}
|
||||
|
||||
for name, test := range tests {
|
||||
t.Run(name, func(t *testing.T) {
|
||||
po := &PruneOptions{}
|
||||
po.currentGroupingObject = test.current
|
||||
actual, err := po.calcPruneSet(test.past)
|
||||
expected := NewInventorySet(test.expected)
|
||||
if test.isError && err == nil {
|
||||
t.Errorf("Did not receive expected error.\n")
|
||||
}
|
||||
if !test.isError {
|
||||
if err != nil {
|
||||
t.Errorf("Unexpected error received: %s\n", err)
|
||||
}
|
||||
if !expected.Equals(actual) {
|
||||
t.Errorf("Expected prune set (%s), got (%s)\n", expected, actual)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user