mirror of https://github.com/k3s-io/k3s
116 lines
3.3 KiB
Go
116 lines
3.3 KiB
Go
/*
|
|
Copyright 2016 The Kubernetes Authors 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 e2e_node
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"k8s.io/kubernetes/pkg/api"
|
|
apierrs "k8s.io/kubernetes/pkg/api/errors"
|
|
client "k8s.io/kubernetes/pkg/client/unversioned"
|
|
|
|
"github.com/onsi/gomega/format"
|
|
"github.com/onsi/gomega/types"
|
|
)
|
|
|
|
//One pod one container
|
|
type ConformanceContainer struct {
|
|
Container api.Container
|
|
Client *client.Client
|
|
Phase api.PodPhase
|
|
NodeName string
|
|
}
|
|
|
|
type ConformanceContainerEqualMatcher struct {
|
|
Expected interface{}
|
|
}
|
|
|
|
func CContainerEqual(expected interface{}) types.GomegaMatcher {
|
|
return &ConformanceContainerEqualMatcher{
|
|
Expected: expected,
|
|
}
|
|
}
|
|
|
|
func (matcher *ConformanceContainerEqualMatcher) Match(actual interface{}) (bool, error) {
|
|
if actual == nil && matcher.Expected == nil {
|
|
return false, fmt.Errorf("Refusing to compare <nil> to <nil>.\nBe explicit and use BeNil() instead. This is to avoid mistakes where both sides of an assertion are erroneously uninitialized.")
|
|
}
|
|
val := api.Semantic.DeepDerivative(matcher.Expected, actual)
|
|
return val, nil
|
|
}
|
|
|
|
func (matcher *ConformanceContainerEqualMatcher) FailureMessage(actual interface{}) (message string) {
|
|
return format.Message(actual, "to equal", matcher.Expected)
|
|
}
|
|
|
|
func (matcher *ConformanceContainerEqualMatcher) NegatedFailureMessage(actual interface{}) (message string) {
|
|
return format.Message(actual, "not to equal", matcher.Expected)
|
|
}
|
|
|
|
func (cc *ConformanceContainer) Create() error {
|
|
pod := &api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
//Same with the container name
|
|
Name: cc.Container.Name,
|
|
Namespace: api.NamespaceDefault,
|
|
},
|
|
Spec: api.PodSpec{
|
|
NodeName: cc.NodeName,
|
|
RestartPolicy: api.RestartPolicyNever,
|
|
Containers: []api.Container{
|
|
cc.Container,
|
|
},
|
|
},
|
|
}
|
|
|
|
_, err := cc.Client.Pods(api.NamespaceDefault).Create(pod)
|
|
return err
|
|
}
|
|
|
|
//Same with 'delete'
|
|
func (cc *ConformanceContainer) Stop() error {
|
|
return cc.Client.Pods(api.NamespaceDefault).Delete(cc.Container.Name, &api.DeleteOptions{})
|
|
}
|
|
|
|
func (cc *ConformanceContainer) Delete() error {
|
|
return cc.Client.Pods(api.NamespaceDefault).Delete(cc.Container.Name, &api.DeleteOptions{})
|
|
}
|
|
|
|
func (cc *ConformanceContainer) Get() (ConformanceContainer, error) {
|
|
pod, err := cc.Client.Pods(api.NamespaceDefault).Get(cc.Container.Name)
|
|
if err != nil {
|
|
return ConformanceContainer{}, err
|
|
}
|
|
|
|
containers := pod.Spec.Containers
|
|
if containers == nil || len(containers) != 1 {
|
|
return ConformanceContainer{}, errors.New("Failed to get container")
|
|
}
|
|
return ConformanceContainer{containers[0], cc.Client, pod.Status.Phase, pod.Spec.NodeName}, nil
|
|
}
|
|
|
|
func (cc *ConformanceContainer) Present() (bool, error) {
|
|
_, err := cc.Client.Pods(api.NamespaceDefault).Get(cc.Container.Name)
|
|
if err == nil {
|
|
return true, nil
|
|
} else if err != nil && apierrs.IsNotFound(err) {
|
|
return false, nil
|
|
}
|
|
return false, err
|
|
}
|