karmada-io
/
karmada
mirror of https://github.com/karmada-io/karmada.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
karmada/pkg/util/overridemanager/commandargsoverride_test.go

499 lines
14 KiB
Go
Raw Blame History

package overridemanager
import (
"reflect"
"testing"
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
policyv1alpha1 "github.com/karmada-io/karmada/pkg/apis/policy/v1alpha1"
)
func generateTestCommandDeploymentYaml() *unstructured.Unstructured {
return &unstructured.Unstructured{
Object: map[string]interface{}{
"apiVersion": "apps/v1",
"kind": "Deployment",
"metadata": map[string]interface{}{
"name": "nginx",
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"replicas": 1,
"selector": map[string]interface{}{
"matchLabels": map[string]interface{}{
"app": "nginx",
},
},
"template": map[string]interface{}{
"metadata": map[string]interface{}{
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"containers": []interface{}{
map[string]interface{}{
"image": "nginx",
"name": "nginx",
"command": []interface{}{"nginx", "-v", "-t"},
}}}}}}}
}
func generateTestArgsDeploymentYaml() *unstructured.Unstructured {
return &unstructured.Unstructured{
Object: map[string]interface{}{
"apiVersion": "apps/v1",
"kind": "Deployment",
"metadata": map[string]interface{}{
"name": "nginx",
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"replicas": 1,
"selector": map[string]interface{}{
"matchLabels": map[string]interface{}{
"app": "nginx",
},
},
"template": map[string]interface{}{
"metadata": map[string]interface{}{
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"containers": []interface{}{
map[string]interface{}{
"image": "nginx",
"name": "nginx",
"args": []interface{}{"nginx", "-v", "-t"},
}}}}}}}
}
func generateTestCommandPodYaml() *unstructured.Unstructured {
return &unstructured.Unstructured{
Object: map[string]interface{}{
"apiVersion": "v1",
"kind": "Pod",
"metadata": map[string]interface{}{
"name": "nginx",
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"containers": []interface{}{
map[string]interface{}{
"image": "fictional.registry.example/imagename:v1.0.0",
"name": "nginx",
"command": []interface{}{"nginx", "-v", "-t"},
}}}}}
}
func generateTestCommandStatefulSetYaml() *unstructured.Unstructured {
return &unstructured.Unstructured{
Object: map[string]interface{}{
"apiVersion": "apps/v1",
"kind": "StatefulSet",
"metadata": map[string]interface{}{
"name": "web",
},
"spec": map[string]interface{}{
"replicas": 2,
"selector": map[string]interface{}{
"matchLabels": map[string]interface{}{
"app": "nginx",
},
},
"template": map[string]interface{}{
"metadata": map[string]interface{}{
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"containers": []interface{}{
map[string]interface{}{
"image": "fictional.registry.example/imagename:v1.0.0",
"name": "nginx",
"command": []interface{}{"nginx", "-v", "-t"},
}}}}}}}
}
func generateTestCommandReplicaSetYaml() *unstructured.Unstructured {
return &unstructured.Unstructured{
Object: map[string]interface{}{
"apiVersion": "apps/v1",
"kind": "ReplicaSet",
"metadata": map[string]interface{}{
"name": "nginx",
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"replicas": 1,
"selector": map[string]interface{}{
"matchLabels": map[string]interface{}{
"app": "nginx",
},
},
"template": map[string]interface{}{
"metadata": map[string]interface{}{
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"containers": []interface{}{
map[string]interface{}{
"image": "fictional.registry.example/imagename:v1.0.0",
"name": "nginx",
"command": []interface{}{"nginx", "-v", "-t"},
}}}}}}}
}
func generateTestCommandDaemonSetYaml() *unstructured.Unstructured {
return &unstructured.Unstructured{
Object: map[string]interface{}{
"apiVersion": "apps/v1",
"kind": "ReplicaSet",
"metadata": map[string]interface{}{
"name": "nginx",
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"replicas": 1,
"selector": map[string]interface{}{
"matchLabels": map[string]interface{}{
"app": "nginx",
},
},
"template": map[string]interface{}{
"metadata": map[string]interface{}{
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"containers": []interface{}{
map[string]interface{}{
"image": "fictional.registry.example/imagename:v1.0.0",
"name": "nginx",
"command": []interface{}{"nginx", "-v", "-t"},
}}}}}}}
}
func generateTestCommandDeploymentYamlWithTwoContainer() *unstructured.Unstructured {
return &unstructured.Unstructured{
Object: map[string]interface{}{
"apiVersion": "apps/v1",
"kind": "Deployment",
"metadata": map[string]interface{}{
"name": "nginx",
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"replicas": 1,
"selector": map[string]interface{}{
"matchLabels": map[string]interface{}{
"app": "nginx",
},
},
"template": map[string]interface{}{
"metadata": map[string]interface{}{
"labels": map[string]interface{}{
"app": "nginx",
},
},
"spec": map[string]interface{}{
"containers": []interface{}{
map[string]interface{}{
"name": "nginx",
"command": []interface{}{"nginx", "-v", "-t"},
},
map[string]interface{}{
"name": "nginx1",
"command": []interface{}{"nginx", "-v", "-t"},
}}}}}}}
}
func TestParseJSONPatchesByCommandOverrider(t *testing.T) {
type args struct {
rawObj *unstructured.Unstructured
CommandArgsOverrider *policyv1alpha1.CommandArgsOverrider
}
tests := []struct {
name string
args args
want []overrideOption
wantErr bool
}{
{
name: "CommandArgsOverrider, resource kind: Deployment, operator: add",
args: args{
rawObj: generateTestCommandDeploymentYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "add",
Value: []string{"&& echo 'hello karmada'"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/command",
Value: []string{"nginx", "-v", "-t", "&& echo 'hello karmada'"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider, resource kind: Deployment, operator: remove",
args: args{
rawObj: generateTestCommandDeploymentYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "remove",
Value: []string{"-t"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/command",
Value: []string{"nginx", "-v"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider, remove value is empty, resource kind: Deployment, operator: remove",
args: args{
rawObj: generateTestCommandDeploymentYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "remove",
Value: []string{},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/command",
Value: []string{"nginx", "-v", "-t"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider, resource has more than one container",
args: args{
rawObj: generateTestCommandDeploymentYamlWithTwoContainer(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "add",
Value: []string{"echo 'hello karmada'"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/command",
Value: []string{"nginx", "-v", "-t", "echo 'hello karmada'"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider, resource has more than one container",
args: args{
rawObj: generateTestCommandDeploymentYamlWithTwoContainer(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "remove",
Value: []string{"-t"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/command",
Value: []string{"nginx", "-v"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider resource kind: Pod, operator: add",
args: args{
rawObj: generateTestCommandPodYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "add",
Value: []string{"echo 'hello karmada'"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/containers/0/command",
Value: []string{"nginx", "-v", "-t", "echo 'hello karmada'"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider, resource kind: StatefulSet, operator: add",
args: args{
rawObj: generateTestCommandStatefulSetYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "add",
Value: []string{"echo 'hello karmada'"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/command",
Value: []string{"nginx", "-v", "-t", "echo 'hello karmada'"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider, resource kind: ReplicaSet, operator: remove",
args: args{
rawObj: generateTestCommandReplicaSetYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "remove",
Value: []string{"-t"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/command",
Value: []string{"nginx", "-v"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider, resource kind: DaemonSet, operator: remove",
args: args{
rawObj: generateTestCommandDaemonSetYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "remove",
Value: []string{"-t"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/command",
Value: []string{"nginx", "-v"},
},
},
wantErr: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := buildCommandArgsPatches(CommandString, tt.args.rawObj, tt.args.CommandArgsOverrider)
if (err != nil) != tt.wantErr {
t.Errorf("buildCommandArgsPatches() error = %v, wantErr %v", err, tt.wantErr)
return
}
if !reflect.DeepEqual(got, tt.want) {
t.Errorf("buildCommandArgsPatches() = %v, want %v", got, tt.want)
}
})
}
}
func TestParseJSONPatchesByArgsOverrider(t *testing.T) {
type args struct {
rawObj *unstructured.Unstructured
CommandArgsOverrider *policyv1alpha1.CommandArgsOverrider
}
tests := []struct {
name string
args args
want []overrideOption
wantErr bool
}{
{
name: "CommandArgsOverrider, resource kind: Deployment, operator: replace",
args: args{
rawObj: generateTestArgsDeploymentYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "add",
Value: []string{"&& echo 'hello karmada'"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/args",
Value: []string{"nginx", "-v", "-t", "&& echo 'hello karmada'"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider, resource kind: Deployment, operator: replace",
args: args{
rawObj: generateTestArgsDeploymentYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "remove",
Value: []string{"-t"},
},
},
want: []overrideOption{
{
Op: "replace",
Path: "/spec/template/spec/containers/0/args",
Value: []string{"nginx", "-v"},
},
},
wantErr: false,
}, {
name: "CommandArgsOverrider, resource kind: Deployment, operator: add",
args: args{
rawObj: generateTestCommandDeploymentYaml(),
CommandArgsOverrider: &policyv1alpha1.CommandArgsOverrider{
ContainerName: "nginx",
Operator: "add",
Value: []string{"-t"},
},
},
want: []overrideOption{
{
Op: "add",
Path: "/spec/template/spec/containers/0/args",
Value: []string{"-t"},
},
},
wantErr: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := buildCommandArgsPatches(ArgsString, tt.args.rawObj, tt.args.CommandArgsOverrider)
if (err != nil) != tt.wantErr {
t.Errorf("buildCommandPatches() error = %v, wantErr %v", err, tt.wantErr)
return
}
if !reflect.DeepEqual(got, tt.want) {
t.Errorf("buildCommandPatches() = %v, want %v", got, tt.want)
}
})
}
}
Reference in New Issue View Git Blame Copy Permalink