Merge pull request #130039 from p0lyn0mial/upstream-fake-client-improve-unit-test

fake/clientset: improve TestNewSimpleClientset

Kubernetes-commit: d7db75ce91f74dc560115c9f485032701e913c84
This commit is contained in:
Kubernetes Publisher 2025-02-12 15:16:27 -08:00
commit bf05f9c594

View File

@ -18,9 +18,11 @@ package fake
import (
"context"
"github.com/stretchr/testify/assert"
"testing"
"github.com/google/go-cmp/cmp"
"github.com/stretchr/testify/require"
v1 "k8s.io/api/core/v1"
policy "k8s.io/api/policy/v1"
meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@ -29,35 +31,36 @@ import (
func TestNewSimpleClientset(t *testing.T) {
client := NewSimpleClientset()
client.CoreV1().Pods("default").Create(context.Background(), &v1.Pod{
expectedPods := []*v1.Pod{}
pod, err := client.CoreV1().Pods("default").Create(context.Background(), &v1.Pod{
ObjectMeta: meta_v1.ObjectMeta{
Name: "pod-1",
Namespace: "default",
},
}, meta_v1.CreateOptions{})
client.CoreV1().Pods("default").Create(context.Background(), &v1.Pod{
require.NoError(t, err)
expectedPods = append(expectedPods, pod)
pod, err = client.CoreV1().Pods("default").Create(context.Background(), &v1.Pod{
ObjectMeta: meta_v1.ObjectMeta{
Name: "pod-2",
Namespace: "default",
},
}, meta_v1.CreateOptions{})
err := client.CoreV1().Pods("default").EvictV1(context.Background(), &policy.Eviction{
require.NoError(t, err)
expectedPods = append(expectedPods, pod)
err = client.CoreV1().Pods("default").EvictV1(context.Background(), &policy.Eviction{
ObjectMeta: meta_v1.ObjectMeta{
Name: "pod-2",
},
})
if err != nil {
t.Errorf("TestNewSimpleClientset() res = %v", err.Error())
}
require.NoError(t, err)
pods, err := client.CoreV1().Pods("default").List(context.Background(), meta_v1.ListOptions{})
// err: item[0]: can't assign or convert v1beta1.Eviction into v1.Pod
if err != nil {
t.Errorf("TestNewSimpleClientset() res = %v", err.Error())
} else {
t.Logf("TestNewSimpleClientset() res = %v", pods)
}
require.NoError(t, err)
cmp.Equal(expectedPods, pods.Items)
}
func TestManagedFieldClientset(t *testing.T) {
@ -69,58 +72,47 @@ func TestManagedFieldClientset(t *testing.T) {
ObjectMeta: meta_v1.ObjectMeta{Name: name, Namespace: namespace},
Data: map[string]string{"k0": "v0"},
}, meta_v1.CreateOptions{FieldManager: "test-manager-0"})
if err != nil {
t.Errorf("Failed to create pod: %v", err)
}
assert.Equal(t, map[string]string{"k0": "v0"}, cm.Data)
require.NoError(t, err)
require.Equal(t, map[string]string{"k0": "v0"}, cm.Data)
// Apply with test-manager-1
// Expect data to be shared with initial create
cm, err = client.CoreV1().ConfigMaps("default").Apply(context.Background(),
v1ac.ConfigMap(name, namespace).WithData(map[string]string{"k1": "v1"}),
meta_v1.ApplyOptions{FieldManager: "test-manager-1"})
if err != nil {
t.Errorf("Failed to create pod: %v", err)
}
assert.Equal(t, map[string]string{"k0": "v0", "k1": "v1"}, cm.Data)
require.NoError(t, err)
require.Equal(t, map[string]string{"k0": "v0", "k1": "v1"}, cm.Data)
// Apply conflicting with test-manager-2, expect apply to fail
_, err = client.CoreV1().ConfigMaps("default").Apply(context.Background(),
v1ac.ConfigMap(name, namespace).WithData(map[string]string{"k1": "xyz"}),
meta_v1.ApplyOptions{FieldManager: "test-manager-2"})
if assert.Error(t, err) {
assert.Equal(t, "Apply failed with 1 conflict: conflict with \"test-manager-1\": .data.k1", err.Error())
}
require.Error(t, err)
require.Equal(t, "Apply failed with 1 conflict: conflict with \"test-manager-1\": .data.k1", err.Error())
// Apply with test-manager-2
// Expect data to be shared with initial create and test-manager-1
cm, err = client.CoreV1().ConfigMaps("default").Apply(context.Background(),
v1ac.ConfigMap(name, namespace).WithData(map[string]string{"k2": "v2"}),
meta_v1.ApplyOptions{FieldManager: "test-manager-2"})
if err != nil {
t.Errorf("Failed to create pod: %v", err)
}
assert.Equal(t, map[string]string{"k0": "v0", "k1": "v1", "k2": "v2"}, cm.Data)
require.NoError(t, err)
require.Equal(t, map[string]string{"k0": "v0", "k1": "v1", "k2": "v2"}, cm.Data)
// Apply with test-manager-1
// Expect owned data to be updated
cm, err = client.CoreV1().ConfigMaps("default").Apply(context.Background(),
v1ac.ConfigMap(name, namespace).WithData(map[string]string{"k1": "v101"}),
meta_v1.ApplyOptions{FieldManager: "test-manager-1"})
if err != nil {
t.Errorf("Failed to create pod: %v", err)
}
assert.Equal(t, map[string]string{"k0": "v0", "k1": "v101", "k2": "v2"}, cm.Data)
require.NoError(t, err)
require.Equal(t, map[string]string{"k0": "v0", "k1": "v101", "k2": "v2"}, cm.Data)
// Force apply with test-manager-2
// Expect data owned by test-manager-1 to be updated, expect data already owned but not in apply configuration to be removed
cm, err = client.CoreV1().ConfigMaps("default").Apply(context.Background(),
v1ac.ConfigMap(name, namespace).WithData(map[string]string{"k1": "v202"}),
meta_v1.ApplyOptions{FieldManager: "test-manager-2", Force: true})
if err != nil {
t.Errorf("Failed to create pod: %v", err)
}
assert.Equal(t, map[string]string{"k0": "v0", "k1": "v202"}, cm.Data)
require.NoError(t, err)
require.Equal(t, map[string]string{"k0": "v0", "k1": "v202"}, cm.Data)
// Update with test-manager-1 to perform a force update of the entire resource
cm, err = client.CoreV1().ConfigMaps("default").Update(context.Background(),
@ -137,8 +129,6 @@ func TestManagedFieldClientset(t *testing.T) {
"k99": "v99",
},
}, meta_v1.UpdateOptions{FieldManager: "test-manager-0"})
if err != nil {
t.Errorf("Failed to update pod: %v", err)
}
assert.Equal(t, map[string]string{"k99": "v99"}, cm.Data)
require.NoError(t, err)
require.Equal(t, map[string]string{"k99": "v99"}, cm.Data)
}