validate api version in kubectl --patch and --override.

satnam6502 · Mar 2, 2015 · 25fac68 · 25fac68
1 parent 93c0052
commit 25fac68
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/pkg/kubectl/cmd/util/helpers.go b/pkg/kubectl/cmd/util/helpers.go
@@ -26,7 +26,7 @@ import (
 	"strings"
 	"time"
 
-	"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
+	"github.com/GoogleCloudPlatform/kubernetes/pkg/api/latest"
 	"github.com/GoogleCloudPlatform/kubernetes/pkg/runtime"
 
 	"github.com/evanphx/json-patch"
@@ -161,18 +161,21 @@ func Merge(dst runtime.Object, fragment, kind string) (runtime.Object, error) {
 	if !ok {
 		return nil, fmt.Errorf("apiVersion must be a string")
 	}
+	i, err := latest.InterfacesFor(versionString)
+	if err != nil {
+		return nil, err
+	}
 
-	codec := runtime.CodecFor(api.Scheme, versionString)
 	// encode dst into versioned json and apply fragment directly too it
-	target, err := codec.Encode(dst)
+	target, err := i.Codec.Encode(dst)
 	if err != nil {
 		return nil, err
 	}
 	patched, err := jsonpatch.MergePatch(target, []byte(fragment))
 	if err != nil {
 		return nil, err
 	}
-	out, err := codec.Decode(patched)
+	out, err := i.Codec.Decode(patched)
 	if err != nil {
 		return nil, err
 	}

diff --git a/pkg/kubectl/cmd/util/helpers_test.go b/pkg/kubectl/cmd/util/helpers_test.go
@@ -130,6 +130,12 @@ func TestMerge(t *testing.T) {
 				},
 			},
 		},
+		{
+			kind:      "Service",
+			obj:       &api.Service{},
+			fragment:  `{ "apiVersion": "badVersion" }`,
+			expectErr: true,
+		},
 		{
 			kind: "Service",
 			obj: &api.Service{