mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-10-31 13:50:01 +00:00
87 lines
2.7 KiB
Go
87 lines
2.7 KiB
Go
/*
|
|
Copyright 2015 The Kubernetes Authors.
|
|
|
|
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 testing
|
|
|
|
import (
|
|
"bytes"
|
|
"math/rand"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
fuzz "github.com/google/gofuzz"
|
|
|
|
"k8s.io/apimachinery/pkg/api/apitesting/fuzzer"
|
|
"k8s.io/apimachinery/pkg/api/apitesting/roundtrip"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/kubernetes/pkg/api/legacyscheme"
|
|
)
|
|
|
|
func TestDeepCopyApiObjects(t *testing.T) {
|
|
for i := 0; i < *roundtrip.FuzzIters; i++ {
|
|
for _, version := range []schema.GroupVersion{{Group: "", Version: runtime.APIVersionInternal}, {Group: "", Version: "v1"}} {
|
|
f := fuzzer.FuzzerFor(FuzzerFuncs, rand.NewSource(rand.Int63()), legacyscheme.Codecs)
|
|
for kind := range legacyscheme.Scheme.KnownTypes(version) {
|
|
doDeepCopyTest(t, version.WithKind(kind), f)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func doDeepCopyTest(t *testing.T, kind schema.GroupVersionKind, f *fuzz.Fuzzer) {
|
|
item, err := legacyscheme.Scheme.New(kind)
|
|
if err != nil {
|
|
t.Fatalf("Could not create a %v: %s", kind, err)
|
|
}
|
|
f.Fuzz(item)
|
|
itemCopy := item.DeepCopyObject()
|
|
if !reflect.DeepEqual(item, itemCopy) {
|
|
t.Errorf("\nexpected: %#v\n\ngot: %#v\n\ndiff: %v", item, itemCopy, cmp.Diff(item, itemCopy))
|
|
}
|
|
|
|
prefuzzData := &bytes.Buffer{}
|
|
if err := legacyscheme.Codecs.LegacyCodec(kind.GroupVersion()).Encode(item, prefuzzData); err != nil {
|
|
t.Errorf("Could not encode a %v: %s", kind, err)
|
|
return
|
|
}
|
|
|
|
// Refuzz the copy, which should have no effect on the original
|
|
f.Fuzz(itemCopy)
|
|
|
|
postfuzzData := &bytes.Buffer{}
|
|
if err := legacyscheme.Codecs.LegacyCodec(kind.GroupVersion()).Encode(item, postfuzzData); err != nil {
|
|
t.Errorf("Could not encode a %v: %s", kind, err)
|
|
return
|
|
}
|
|
|
|
if !bytes.Equal(prefuzzData.Bytes(), postfuzzData.Bytes()) {
|
|
t.Log(cmp.Diff(prefuzzData.String(), postfuzzData.String()))
|
|
t.Errorf("Fuzzing copy modified original of %#v", kind)
|
|
return
|
|
}
|
|
}
|
|
|
|
func TestDeepCopySingleType(t *testing.T) {
|
|
for i := 0; i < *roundtrip.FuzzIters; i++ {
|
|
for _, version := range []schema.GroupVersion{{Group: "", Version: runtime.APIVersionInternal}, {Group: "", Version: "v1"}} {
|
|
f := fuzzer.FuzzerFor(FuzzerFuncs, rand.NewSource(rand.Int63()), legacyscheme.Codecs)
|
|
doDeepCopyTest(t, version.WithKind("Pod"), f)
|
|
}
|
|
}
|
|
}
|