e709599854
Since PodInterface.Create returns the server's representation of the pod, which may differ from the api.Pod object passed to Create, we do the same from the framework's pod creation functions. This is useful if e.g. you create pods using Pod.GenerateName rather than Pod.Name, and you still want to refer to pods by name later on (e.g. for deletion).
107 lines
3.3 KiB
Go
107 lines
3.3 KiB
Go
/*
|
|
Copyright 2016 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 framework
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
|
|
"k8s.io/kubernetes/pkg/api"
|
|
"k8s.io/kubernetes/pkg/api/errors"
|
|
"k8s.io/kubernetes/pkg/client/unversioned"
|
|
"k8s.io/kubernetes/pkg/util/wait"
|
|
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
// TODO: Consolidate pod-specific framework functions here.
|
|
|
|
// Convenience method for getting a pod client interface in the framework's namespace.
|
|
func (f *Framework) PodClient() unversioned.PodInterface {
|
|
return f.Client.Pods(f.Namespace.Name)
|
|
}
|
|
|
|
// Create a new pod according to the framework specifications, and wait for it to start.
|
|
// Returns the server's representation of the pod.
|
|
func (f *Framework) CreatePod(pod *api.Pod) *api.Pod {
|
|
p := f.CreatePodAsync(pod)
|
|
ExpectNoError(f.WaitForPodRunning(p.Name))
|
|
return p
|
|
}
|
|
|
|
// Create a new pod according to the framework specifications (don't wait for it to start).
|
|
// Returns the server's representation of the pod.
|
|
func (f *Framework) CreatePodAsync(pod *api.Pod) *api.Pod {
|
|
f.MungePodSpec(pod)
|
|
p, err := f.PodClient().Create(pod)
|
|
ExpectNoError(err, "Error creating Pod")
|
|
return p
|
|
}
|
|
|
|
// Batch version of CreatePod. All pods are created before waiting.
|
|
// Returns a slice, in the same order as pods, containing the server's representations of the pods.
|
|
func (f *Framework) CreatePods(pods []*api.Pod) []*api.Pod {
|
|
ps := make([]*api.Pod, len(pods))
|
|
for i, pod := range pods {
|
|
ps[i] = f.CreatePodAsync(pod)
|
|
}
|
|
var wg sync.WaitGroup
|
|
for _, pod := range ps {
|
|
wg.Add(1)
|
|
podName := pod.Name
|
|
go func() {
|
|
ExpectNoError(f.WaitForPodRunning(podName))
|
|
wg.Done()
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
return ps
|
|
}
|
|
|
|
// Apply test-suite specific transformations to the pod spec.
|
|
// TODO: figure out a nicer, more generic way to tie this to framework instances.
|
|
func (f *Framework) MungePodSpec(pod *api.Pod) {
|
|
if TestContext.NodeName != "" {
|
|
Expect(pod.Spec.NodeName).To(Or(BeZero(), Equal(TestContext.NodeName)), "Test misconfigured")
|
|
pod.Spec.NodeName = TestContext.NodeName
|
|
}
|
|
}
|
|
|
|
// UpdatePod updates the pod object. It retries if there is a conflict, throw out error if
|
|
// there is any other errors. name is the pod name, updateFn is the function updating the
|
|
// pod object.
|
|
func (f *Framework) UpdatePod(name string, updateFn func(pod *api.Pod)) {
|
|
ExpectNoError(wait.Poll(time.Millisecond*500, time.Second*30, func() (bool, error) {
|
|
pod, err := f.PodClient().Get(name)
|
|
if err != nil {
|
|
return false, fmt.Errorf("failed to get pod %q: %v", name, err)
|
|
}
|
|
updateFn(pod)
|
|
_, err = f.PodClient().Update(pod)
|
|
if err == nil {
|
|
Logf("Successfully updated pod %q", name)
|
|
return true, nil
|
|
}
|
|
if errors.IsConflict(err) {
|
|
Logf("Conflicting update to pod %q, re-get and re-update: %v", name, err)
|
|
return false, nil
|
|
}
|
|
return false, fmt.Errorf("failed to update pod %q: %v", name, err)
|
|
}))
|
|
}
|