mirror of
https://github.com/kubernetes/client-go.git
synced 2025-12-25 22:22:28 +00:00
318 lines
9.6 KiB
Go
318 lines
9.6 KiB
Go
/*
|
|
Copyright 2019 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 metadata
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/client-go/rest"
|
|
"k8s.io/client-go/util/watchlist"
|
|
"k8s.io/klog/v2/ktesting"
|
|
)
|
|
|
|
func TestDoesClientSupportWatchListSemantics(t *testing.T) {
|
|
target, err := NewForConfig(&rest.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if watchlist.DoesClientNotSupportWatchListSemantics(target) {
|
|
t.Fatalf("Metadata client should support WatchList semantics")
|
|
}
|
|
}
|
|
|
|
func TestClient(t *testing.T) {
|
|
gvr := schema.GroupVersionResource{Group: "group", Version: "v1", Resource: "resource"}
|
|
statusOK := &metav1.Status{
|
|
Status: metav1.StatusSuccess,
|
|
Code: http.StatusOK,
|
|
}
|
|
|
|
writeJSON := func(t *testing.T, w http.ResponseWriter, obj runtime.Object) {
|
|
data, err := json.Marshal(obj)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
if _, err := w.Write(data); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
testCases := []struct {
|
|
name string
|
|
handler func(t *testing.T, w http.ResponseWriter, req *http.Request)
|
|
want func(ctx context.Context, t *testing.T, client *Client)
|
|
}{
|
|
{
|
|
name: "GET is able to convert a JSON object to PartialObjectMetadata",
|
|
handler: func(t *testing.T, w http.ResponseWriter, req *http.Request) {
|
|
if req.Header.Get("Accept") != "application/vnd.kubernetes.protobuf;as=PartialObjectMetadata;g=meta.k8s.io;v=v1,application/json;as=PartialObjectMetadata;g=meta.k8s.io;v=v1,application/json" {
|
|
t.Fatal(req.Header.Get("Accept"))
|
|
}
|
|
if req.Method != "GET" && req.URL.String() != "/apis/group/v1/namespaces/ns/resource/name" {
|
|
t.Fatal(req.URL.String())
|
|
}
|
|
writeJSON(t, w, &corev1.Pod{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Pod",
|
|
APIVersion: "v1",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "name",
|
|
Namespace: "ns",
|
|
},
|
|
})
|
|
},
|
|
want: func(ctx context.Context, t *testing.T, client *Client) {
|
|
obj, err := client.Resource(gvr).Namespace("ns").Get(ctx, "name", metav1.GetOptions{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
expect := &metav1.PartialObjectMetadata{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "name",
|
|
Namespace: "ns",
|
|
},
|
|
}
|
|
if !reflect.DeepEqual(expect, obj) {
|
|
t.Fatal(cmp.Diff(expect, obj))
|
|
}
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "LIST is able to convert a JSON object to PartialObjectMetadata",
|
|
handler: func(t *testing.T, w http.ResponseWriter, req *http.Request) {
|
|
if req.Header.Get("Accept") != "application/vnd.kubernetes.protobuf;as=PartialObjectMetadataList;g=meta.k8s.io;v=v1,application/json;as=PartialObjectMetadataList;g=meta.k8s.io;v=v1,application/json" {
|
|
t.Fatal(req.Header.Get("Accept"))
|
|
}
|
|
if req.Method != "GET" && req.URL.String() != "/apis/group/v1/namespaces/ns/resource" {
|
|
t.Fatal(req.URL.String())
|
|
}
|
|
writeJSON(t, w, &corev1.PodList{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "PodList",
|
|
APIVersion: "v1",
|
|
},
|
|
ListMeta: metav1.ListMeta{
|
|
ResourceVersion: "253",
|
|
},
|
|
Items: []corev1.Pod{
|
|
{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Pod",
|
|
APIVersion: "v1",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "name",
|
|
Namespace: "ns",
|
|
},
|
|
},
|
|
},
|
|
})
|
|
},
|
|
want: func(ctx context.Context, t *testing.T, client *Client) {
|
|
objs, err := client.Resource(gvr).Namespace("ns").List(ctx, metav1.ListOptions{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if objs.GetResourceVersion() != "253" {
|
|
t.Fatal(objs)
|
|
}
|
|
expect := []metav1.PartialObjectMetadata{
|
|
{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Pod",
|
|
APIVersion: "v1",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "name",
|
|
Namespace: "ns",
|
|
},
|
|
},
|
|
}
|
|
if !reflect.DeepEqual(expect, objs.Items) {
|
|
t.Fatal(cmp.Diff(expect, objs.Items))
|
|
}
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "GET fails if the object is JSON and has no kind",
|
|
handler: func(t *testing.T, w http.ResponseWriter, req *http.Request) {
|
|
if req.Header.Get("Accept") != "application/vnd.kubernetes.protobuf;as=PartialObjectMetadata;g=meta.k8s.io;v=v1,application/json;as=PartialObjectMetadata;g=meta.k8s.io;v=v1,application/json" {
|
|
t.Fatal(req.Header.Get("Accept"))
|
|
}
|
|
if req.Method != "GET" && req.URL.String() != "/apis/group/v1/namespaces/ns/resource/name" {
|
|
t.Fatal(req.URL.String())
|
|
}
|
|
writeJSON(t, w, &corev1.Pod{
|
|
TypeMeta: metav1.TypeMeta{},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
UID: "123",
|
|
},
|
|
})
|
|
},
|
|
want: func(ctx context.Context, t *testing.T, client *Client) {
|
|
obj, err := client.Resource(gvr).Namespace("ns").Get(ctx, "name", metav1.GetOptions{})
|
|
if err == nil || !runtime.IsMissingKind(err) {
|
|
t.Fatal(err)
|
|
}
|
|
if obj != nil {
|
|
t.Fatal(obj)
|
|
}
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "GET fails if the object is JSON and has no apiVersion",
|
|
handler: func(t *testing.T, w http.ResponseWriter, req *http.Request) {
|
|
if req.Header.Get("Accept") != "application/vnd.kubernetes.protobuf;as=PartialObjectMetadata;g=meta.k8s.io;v=v1,application/json;as=PartialObjectMetadata;g=meta.k8s.io;v=v1,application/json" {
|
|
t.Fatal(req.Header.Get("Accept"))
|
|
}
|
|
if req.Method != "GET" && req.URL.String() != "/apis/group/v1/namespaces/ns/resource/name" {
|
|
t.Fatal(req.URL.String())
|
|
}
|
|
writeJSON(t, w, &corev1.Pod{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Pod",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
UID: "123",
|
|
},
|
|
})
|
|
},
|
|
want: func(ctx context.Context, t *testing.T, client *Client) {
|
|
obj, err := client.Resource(gvr).Namespace("ns").Get(ctx, "name", metav1.GetOptions{})
|
|
if err == nil || !runtime.IsMissingVersion(err) {
|
|
t.Fatal(err)
|
|
}
|
|
if obj != nil {
|
|
t.Fatal(obj)
|
|
}
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "GET fails if the object is JSON and not clearly metadata",
|
|
handler: func(t *testing.T, w http.ResponseWriter, req *http.Request) {
|
|
if req.Header.Get("Accept") != "application/vnd.kubernetes.protobuf;as=PartialObjectMetadata;g=meta.k8s.io;v=v1,application/json;as=PartialObjectMetadata;g=meta.k8s.io;v=v1,application/json" {
|
|
t.Fatal(req.Header.Get("Accept"))
|
|
}
|
|
if req.Method != "GET" && req.URL.String() != "/apis/group/v1/namespaces/ns/resource/name" {
|
|
t.Fatal(req.URL.String())
|
|
}
|
|
writeJSON(t, w, &corev1.Pod{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Pod",
|
|
APIVersion: "v1",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{},
|
|
})
|
|
},
|
|
want: func(ctx context.Context, t *testing.T, client *Client) {
|
|
obj, err := client.Resource(gvr).Namespace("ns").Get(ctx, "name", metav1.GetOptions{})
|
|
if err == nil || !strings.Contains(err.Error(), "object does not appear to match the ObjectMeta schema") {
|
|
t.Fatal(err)
|
|
}
|
|
if obj != nil {
|
|
t.Fatal(obj)
|
|
}
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "Delete fails if DeleteOptions cannot be serialized to JSON",
|
|
handler: func(t *testing.T, w http.ResponseWriter, req *http.Request) {
|
|
if req.Header.Get("Content-Type") != runtime.ContentTypeJSON {
|
|
t.Fatal(req.Header.Get("Content-Type"))
|
|
}
|
|
if req.Method != "DELETE" && req.URL.String() != "/apis/group/v1/namespaces/ns/resource/name" {
|
|
t.Fatal(req.URL.String())
|
|
}
|
|
defer req.Body.Close()
|
|
buf, err := io.ReadAll(req.Body)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !json.Valid(buf) {
|
|
t.Fatalf("request body is not a valid JSON: %s", buf)
|
|
}
|
|
writeJSON(t, w, statusOK)
|
|
},
|
|
want: func(ctx context.Context, t *testing.T, client *Client) {
|
|
err := client.Resource(gvr).Namespace("ns").Delete(ctx, "name", metav1.DeleteOptions{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "DeleteCollection fails if DeleteOptions cannot be serialized to JSON",
|
|
handler: func(t *testing.T, w http.ResponseWriter, req *http.Request) {
|
|
if req.Header.Get("Content-Type") != runtime.ContentTypeJSON {
|
|
t.Fatal(req.Header.Get("Content-Type"))
|
|
}
|
|
if req.Method != "DELETE" && req.URL.String() != "/apis/group/v1/namespaces/ns/resource/name" {
|
|
t.Fatal(req.URL.String())
|
|
}
|
|
defer req.Body.Close()
|
|
buf, err := io.ReadAll(req.Body)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !json.Valid(buf) {
|
|
t.Fatalf("request body is not a valid JSON: %s", buf)
|
|
}
|
|
|
|
writeJSON(t, w, statusOK)
|
|
},
|
|
want: func(ctx context.Context, t *testing.T, client *Client) {
|
|
err := client.Resource(gvr).Namespace("ns").DeleteCollection(ctx, metav1.DeleteOptions{}, metav1.ListOptions{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range testCases {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { tt.handler(t, w, req) }))
|
|
defer s.Close()
|
|
|
|
_, ctx := ktesting.NewTestContext(t)
|
|
cfg := ConfigFor(&rest.Config{Host: s.URL})
|
|
client := NewForConfigOrDie(cfg).(*Client)
|
|
tt.want(ctx, t, client)
|
|
})
|
|
}
|
|
}
|