mirror of https://github.com/k3s-io/k3s
Add an e2e test in go. Also adjust validation logic a little to make it more reasonable.
Not integrated into existing e2e yet.pull/6/head
parent
f603785698
commit
8d38f8890e
|
@ -0,0 +1,174 @@
|
||||||
|
/*
|
||||||
|
Copyright 2014 Google Inc. All rights reserved.
|
||||||
|
|
||||||
|
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 main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"runtime"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
|
||||||
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api/latest"
|
||||||
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/client"
|
||||||
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/kubecfg"
|
||||||
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/labels"
|
||||||
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/util"
|
||||||
|
"github.com/golang/glog"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
authConfig = flag.String("auth_config", os.Getenv("HOME")+"/.kubernetes_auth", "Path to the auth info file.")
|
||||||
|
host = flag.String("host", "", "The host to connect to")
|
||||||
|
)
|
||||||
|
|
||||||
|
func waitForPodRunning(c *client.Client, id string) {
|
||||||
|
for {
|
||||||
|
ctx := api.NewContext()
|
||||||
|
time.Sleep(5 * time.Second)
|
||||||
|
pod, err := c.GetPod(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
glog.Warningf("Get pod failed: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if pod.CurrentState.Status == api.PodRunning {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
glog.Infof("Waiting for pod status to be running (%s)", pod.CurrentState.Status)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadObjectOrDie(filePath string) interface{} {
|
||||||
|
data, err := ioutil.ReadFile(filePath)
|
||||||
|
if err != nil {
|
||||||
|
glog.Fatalf("Failed to read pod: %v", err)
|
||||||
|
}
|
||||||
|
obj, err := latest.Codec.Decode(data)
|
||||||
|
if err != nil {
|
||||||
|
glog.Fatalf("Failed to decode pod: %v", err)
|
||||||
|
}
|
||||||
|
return obj
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadPodOrDie(filePath string) *api.Pod {
|
||||||
|
obj := loadObjectOrDie(filePath)
|
||||||
|
pod, ok := obj.(*api.Pod)
|
||||||
|
if !ok {
|
||||||
|
glog.Fatalf("Failed to load pod: %v", obj)
|
||||||
|
}
|
||||||
|
return pod
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadClientOrDie() *client.Client {
|
||||||
|
config := client.Config{
|
||||||
|
Host: *host,
|
||||||
|
}
|
||||||
|
auth, err := kubecfg.LoadAuthInfo(*authConfig, os.Stdin)
|
||||||
|
if err != nil {
|
||||||
|
glog.Fatalf("Error loading auth: %v", err)
|
||||||
|
}
|
||||||
|
config.Username = auth.User
|
||||||
|
config.Password = auth.Password
|
||||||
|
config.CAFile = auth.CAFile
|
||||||
|
config.CertFile = auth.CertFile
|
||||||
|
config.KeyFile = auth.KeyFile
|
||||||
|
if auth.Insecure != nil {
|
||||||
|
config.Insecure = *auth.Insecure
|
||||||
|
}
|
||||||
|
c, err := client.New(&config)
|
||||||
|
if err != nil {
|
||||||
|
glog.Fatalf("Error creating client")
|
||||||
|
}
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPodUpdate(c *client.Client) bool {
|
||||||
|
ctx := api.NewContext()
|
||||||
|
|
||||||
|
pod := loadPodOrDie("./api/examples/pod.json")
|
||||||
|
value := strconv.Itoa(time.Now().Nanosecond())
|
||||||
|
pod.Labels["time"] = value
|
||||||
|
|
||||||
|
_, err := c.CreatePod(ctx, pod)
|
||||||
|
if err != nil {
|
||||||
|
glog.Errorf("Failed to create pod: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
defer c.DeletePod(ctx, pod.ID)
|
||||||
|
waitForPodRunning(c, pod.ID)
|
||||||
|
pods, err := c.ListPods(ctx, labels.SelectorFromSet(labels.Set(map[string]string{"time": value})))
|
||||||
|
if len(pods.Items) != 1 {
|
||||||
|
glog.Errorf("Failed to find the correct pod")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
podOut, err := c.GetPod(ctx, pod.ID)
|
||||||
|
if err != nil {
|
||||||
|
glog.Errorf("Failed to get pod: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
value = "time" + value
|
||||||
|
pod.Labels["time"] = value
|
||||||
|
pod.ResourceVersion = podOut.ResourceVersion
|
||||||
|
pod.DesiredState.Manifest.UUID = podOut.DesiredState.Manifest.UUID
|
||||||
|
pod, err = c.UpdatePod(ctx, pod)
|
||||||
|
if err != nil {
|
||||||
|
glog.Errorf("Failed to update pod: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
waitForPodRunning(c, pod.ID)
|
||||||
|
pods, err = c.ListPods(ctx, labels.SelectorFromSet(labels.Set(map[string]string{"time": value})))
|
||||||
|
if len(pods.Items) != 1 {
|
||||||
|
glog.Errorf("Failed to find the correct pod after update.")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
glog.Infof("pod update OK")
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
flag.Parse()
|
||||||
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
||||||
|
util.ReallyCrash = true
|
||||||
|
util.InitLogs()
|
||||||
|
defer util.FlushLogs()
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
defer util.FlushLogs()
|
||||||
|
time.Sleep(3 * time.Minute)
|
||||||
|
glog.Fatalf("This test has timed out.")
|
||||||
|
}()
|
||||||
|
|
||||||
|
c := loadClientOrDie()
|
||||||
|
|
||||||
|
tests := []func(c *client.Client) bool{
|
||||||
|
TestPodUpdate,
|
||||||
|
}
|
||||||
|
|
||||||
|
passed := true
|
||||||
|
for _, test := range tests {
|
||||||
|
testPassed := test(c)
|
||||||
|
if !testPassed {
|
||||||
|
passed = false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !passed {
|
||||||
|
glog.Fatalf("Tests failed")
|
||||||
|
}
|
||||||
|
}
|
|
@ -157,6 +157,7 @@ kube::default_build_targets() {
|
||||||
echo "cmd/proxy"
|
echo "cmd/proxy"
|
||||||
echo "cmd/apiserver"
|
echo "cmd/apiserver"
|
||||||
echo "cmd/controller-manager"
|
echo "cmd/controller-manager"
|
||||||
|
echo "cmd/e2e"
|
||||||
echo "cmd/kubelet"
|
echo "cmd/kubelet"
|
||||||
echo "cmd/kubecfg"
|
echo "cmd/kubecfg"
|
||||||
echo "plugin/cmd/scheduler"
|
echo "plugin/cmd/scheduler"
|
||||||
|
|
|
@ -346,6 +346,10 @@ func ValidatePod(pod *api.Pod) errs.ErrorList {
|
||||||
func ValidatePodUpdate(newPod, oldPod *api.Pod) errs.ErrorList {
|
func ValidatePodUpdate(newPod, oldPod *api.Pod) errs.ErrorList {
|
||||||
allErrs := errs.ErrorList{}
|
allErrs := errs.ErrorList{}
|
||||||
|
|
||||||
|
if newPod.ID != oldPod.ID {
|
||||||
|
allErrs = append(allErrs, errs.NewFieldInvalid("ID", newPod.ID))
|
||||||
|
}
|
||||||
|
|
||||||
if len(newPod.DesiredState.Manifest.Containers) != len(oldPod.DesiredState.Manifest.Containers) {
|
if len(newPod.DesiredState.Manifest.Containers) != len(oldPod.DesiredState.Manifest.Containers) {
|
||||||
allErrs = append(allErrs, errs.NewFieldInvalid("DesiredState.Manifest.Containers", newPod.DesiredState.Manifest.Containers))
|
allErrs = append(allErrs, errs.NewFieldInvalid("DesiredState.Manifest.Containers", newPod.DesiredState.Manifest.Containers))
|
||||||
return allErrs
|
return allErrs
|
||||||
|
@ -360,7 +364,7 @@ func ValidatePodUpdate(newPod, oldPod *api.Pod) errs.ErrorList {
|
||||||
newContainers = append(newContainers, container)
|
newContainers = append(newContainers, container)
|
||||||
}
|
}
|
||||||
pod.DesiredState.Manifest.Containers = newContainers
|
pod.DesiredState.Manifest.Containers = newContainers
|
||||||
if !reflect.DeepEqual(&pod, oldPod) {
|
if !reflect.DeepEqual(pod.DesiredState.Manifest, oldPod.DesiredState.Manifest) {
|
||||||
allErrs = append(allErrs, errs.NewFieldInvalid("DesiredState.Manifest.Containers", newPod.DesiredState.Manifest.Containers))
|
allErrs = append(allErrs, errs.NewFieldInvalid("DesiredState.Manifest.Containers", newPod.DesiredState.Manifest.Containers))
|
||||||
}
|
}
|
||||||
return allErrs
|
return allErrs
|
||||||
|
|
Loading…
Reference in New Issue