add unit test and address comments

This commit is contained in:
Jingfang Liu
2018-06-11 14:52:31 -07:00
parent 37720765fc
commit 934e37d781
7 changed files with 72 additions and 12 deletions

View File

@@ -222,13 +222,13 @@ func MergeWithOverride(maps ...ResMap) (ResMap, error) {
glog.V(4).Infof("Replace %v with %v", result[id].Object, r.Object)
r.Replace(result[id])
result[id] = r
result[id].ChangeBehavior(resource.BehaviorCreate)
result[id].SetBehavior(resource.BehaviorCreate)
case resource.BehaviorMerge:
glog.V(4).Infof("Merging %v with %v", result[id].Object, r.Object)
r.Merge(result[id])
result[id] = r
glog.V(4).Infof("Merged object is %v", result[id].Object)
result[id].ChangeBehavior(resource.BehaviorCreate)
result[id].SetBehavior(resource.BehaviorCreate)
default:
return nil, fmt.Errorf("Id %#v exists; must merge or replace.", id)
}

View File

@@ -151,7 +151,7 @@ metadata:
}
}
func TestMerge(t *testing.T) {
func TestMergeWithoutOverride(t *testing.T) {
input1 := ResMap{
resource.NewResId(deploy, "deploy1"): resource.NewResourceFromMap(
map[string]interface{}{
@@ -199,3 +199,63 @@ func TestMerge(t *testing.T) {
t.Fatalf("%#v doesn't equal expected %#v", merged, expected)
}
}
func TestMergeWithOverride(t *testing.T) {
input1 := ResMap{
resource.NewResId(cmap, "cmap"): resource.NewResourceFromMap(
map[string]interface{}{
"apiVersion": "apps/v1",
"kind": "ConfigMap",
"metadata": map[string]interface{}{
"name": "cmap",
},
"data": map[string]interface{}{
"a": "x",
"b": "y",
},
}),
}
input2 := ResMap{
resource.NewResId(cmap, "cmap"): resource.NewResourceFromMap(
map[string]interface{}{
"apiVersion": "apps/v1",
"kind": "ConfigMap",
"metadata": map[string]interface{}{
"name": "cmap",
},
"data": map[string]interface{}{
"a": "u",
"b": "v",
"c": "w",
},
}),
}
input1[resource.NewResId(cmap, "cmap")].SetBehavior(resource.BehaviorCreate)
input2[resource.NewResId(cmap, "cmap")].SetBehavior(resource.BehaviorMerge)
input := []ResMap{input1, input2}
expected := ResMap{
resource.NewResId(cmap, "cmap"): resource.NewResourceFromMap(
map[string]interface{}{
"apiVersion": "apps/v1",
"kind": "ConfigMap",
"metadata": map[string]interface{}{
"annotations": map[string]interface{}{},
"labels": map[string]interface{}{},
"name": "cmap",
},
"data": map[string]interface{}{
"a": "u",
"b": "v",
"c": "w",
},
}),
}
expected[resource.NewResId(cmap, "cmap")].SetBehavior(resource.BehaviorCreate)
merged, err := MergeWithOverride(input...)
if err != nil {
t.Fatalf("unexpected error: %v", err)
}
if !reflect.DeepEqual(merged, expected) {
t.Fatalf("%#v doesn't equal expected %#v", merged, expected)
}
}