mirror of https://github.com/k3s-io/k3s
91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
|
package kubelet
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"sync"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"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/apimachinery/pkg/util/wait"
|
||
|
|
"k8s.io/apimachinery/pkg/watch"
|
||
|
|
"k8s.io/client-go/kubernetes"
|
||
|
|
kubeapiservertesting "k8s.io/kubernetes/cmd/kube-apiserver/app/testing"
|
||
|
|
"k8s.io/kubernetes/pkg/kubelet/util/manager"
|
||
|
|
"k8s.io/kubernetes/test/integration/framework"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWatchBasedManager(t *testing.T) {
|
||
|
|
server := kubeapiservertesting.StartTestServerOrDie(t, nil, nil, framework.SharedEtcd())
|
||
|
|
defer server.TearDownFn()
|
||
|
|
|
||
|
|
server.ClientConfig.QPS = 10000
|
||
|
|
client, err := kubernetes.NewForConfig(server.ClientConfig)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("unexpected error: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
listObj := func(namespace string, options metav1.ListOptions) (runtime.Object, error) {
|
||
|
|
return client.CoreV1().Secrets(namespace).List(options)
|
||
|
|
}
|
||
|
|
watchObj := func(namespace string, options metav1.ListOptions) (watch.Interface, error) {
|
||
|
|
return client.CoreV1().Secrets(namespace).Watch(options)
|
||
|
|
}
|
||
|
|
newObj := func() runtime.Object { return &v1.Secret{} }
|
||
|
|
store := manager.NewObjectCache(listObj, watchObj, newObj, schema.GroupResource{Group: "v1", Resource: "secrets"})
|
||
|
|
|
||
|
|
// create 1000 secrets in parallel
|
||
|
|
t.Log(time.Now(), "creating 1000 secrets")
|
||
|
|
wg := sync.WaitGroup{}
|
||
|
|
for i := 0; i < 10; i++ {
|
||
|
|
wg.Add(1)
|
||
|
|
go func(i int) {
|
||
|
|
defer wg.Done()
|
||
|
|
for j := 0; j < 100; j++ {
|
||
|
|
name := fmt.Sprintf("s%d", i*100+j)
|
||
|
|
if _, err := client.CoreV1().Secrets("default").Create(&v1.Secret{ObjectMeta: metav1.ObjectMeta{Name: name}}); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
fmt.Print(".")
|
||
|
|
}(i)
|
||
|
|
}
|
||
|
|
wg.Wait()
|
||
|
|
t.Log(time.Now(), "finished creating 1000 secrets")
|
||
|
|
|
||
|
|
// fetch all secrets
|
||
|
|
wg = sync.WaitGroup{}
|
||
|
|
for i := 0; i < 10; i++ {
|
||
|
|
wg.Add(1)
|
||
|
|
go func(i int) {
|
||
|
|
defer wg.Done()
|
||
|
|
for j := 0; j < 100; j++ {
|
||
|
|
name := fmt.Sprintf("s%d", i*100+j)
|
||
|
|
start := time.Now()
|
||
|
|
store.AddReference("default", name)
|
||
|
|
err := wait.PollImmediate(10*time.Millisecond, 10*time.Second, func() (bool, error) {
|
||
|
|
obj, err := store.Get("default", name)
|
||
|
|
if err != nil {
|
||
|
|
t.Logf("failed on %s, retrying: %v", name, err)
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
if obj.(*v1.Secret).Name != name {
|
||
|
|
return false, fmt.Errorf("wrong object: %v", obj.(*v1.Secret).Name)
|
||
|
|
}
|
||
|
|
return true, nil
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed on %s: %v", name, err)
|
||
|
|
}
|
||
|
|
if d := time.Now().Sub(start); d > time.Second {
|
||
|
|
t.Logf("%s took %v", name, d)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}(i)
|
||
|
|
}
|
||
|
|
wg.Wait()
|
||
|
|
}
|