mirror of https://github.com/k3s-io/k3s
758 lines
20 KiB
Go
758 lines
20 KiB
Go
/*
|
|
Copyright 2014 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 util
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"os/user"
|
|
"path"
|
|
"reflect"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"k8s.io/kubernetes/pkg/api"
|
|
"k8s.io/kubernetes/pkg/api/meta"
|
|
"k8s.io/kubernetes/pkg/api/testapi"
|
|
"k8s.io/kubernetes/pkg/api/unversioned"
|
|
"k8s.io/kubernetes/pkg/api/validation"
|
|
"k8s.io/kubernetes/pkg/apimachinery/registered"
|
|
"k8s.io/kubernetes/pkg/apis/extensions"
|
|
"k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/fake"
|
|
testcore "k8s.io/kubernetes/pkg/client/testing/core"
|
|
"k8s.io/kubernetes/pkg/client/unversioned/clientcmd"
|
|
clientcmdapi "k8s.io/kubernetes/pkg/client/unversioned/clientcmd/api"
|
|
manualfake "k8s.io/kubernetes/pkg/client/unversioned/fake"
|
|
"k8s.io/kubernetes/pkg/controller"
|
|
"k8s.io/kubernetes/pkg/kubectl"
|
|
"k8s.io/kubernetes/pkg/kubectl/resource"
|
|
"k8s.io/kubernetes/pkg/labels"
|
|
"k8s.io/kubernetes/pkg/runtime"
|
|
"k8s.io/kubernetes/pkg/util/flag"
|
|
"k8s.io/kubernetes/pkg/watch"
|
|
)
|
|
|
|
func TestNewFactoryDefaultFlagBindings(t *testing.T) {
|
|
factory := NewFactory(nil)
|
|
|
|
if !factory.FlagSet().HasFlags() {
|
|
t.Errorf("Expected flags, but didn't get any")
|
|
}
|
|
}
|
|
|
|
func TestNewFactoryNoFlagBindings(t *testing.T) {
|
|
clientConfig := clientcmd.NewDefaultClientConfig(*clientcmdapi.NewConfig(), &clientcmd.ConfigOverrides{})
|
|
factory := NewFactory(clientConfig)
|
|
|
|
if factory.FlagSet().HasFlags() {
|
|
t.Errorf("Expected zero flags, but got %v", factory.FlagSet())
|
|
}
|
|
}
|
|
|
|
func TestPortsForObject(t *testing.T) {
|
|
f := NewFactory(nil)
|
|
|
|
pod := &api.Pod{
|
|
ObjectMeta: api.ObjectMeta{Name: "baz", Namespace: "test", ResourceVersion: "12"},
|
|
Spec: api.PodSpec{
|
|
Containers: []api.Container{
|
|
{
|
|
Ports: []api.ContainerPort{
|
|
{
|
|
ContainerPort: 101,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
expected := []string{"101"}
|
|
got, err := f.PortsForObject(pod)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
if len(expected) != len(got) {
|
|
t.Fatalf("Ports size mismatch! Expected %d, got %d", len(expected), len(got))
|
|
}
|
|
|
|
sort.Strings(expected)
|
|
sort.Strings(got)
|
|
|
|
for i, port := range got {
|
|
if port != expected[i] {
|
|
t.Fatalf("Port mismatch! Expected %s, got %s", expected[i], port)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestProtocolsForObject(t *testing.T) {
|
|
f := NewFactory(nil)
|
|
|
|
pod := &api.Pod{
|
|
ObjectMeta: api.ObjectMeta{Name: "baz", Namespace: "test", ResourceVersion: "12"},
|
|
Spec: api.PodSpec{
|
|
Containers: []api.Container{
|
|
{
|
|
Ports: []api.ContainerPort{
|
|
{
|
|
ContainerPort: 101,
|
|
Protocol: api.ProtocolTCP,
|
|
},
|
|
{
|
|
ContainerPort: 102,
|
|
Protocol: api.ProtocolUDP,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
expected := "101/TCP,102/UDP"
|
|
protocolsMap, err := f.ProtocolsForObject(pod)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
got := kubectl.MakeProtocols(protocolsMap)
|
|
expectedSlice := strings.Split(expected, ",")
|
|
gotSlice := strings.Split(got, ",")
|
|
|
|
sort.Strings(expectedSlice)
|
|
sort.Strings(gotSlice)
|
|
|
|
for i, protocol := range gotSlice {
|
|
if protocol != expectedSlice[i] {
|
|
t.Fatalf("Protocols mismatch! Expected %s, got %s", expectedSlice[i], protocol)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestLabelsForObject(t *testing.T) {
|
|
f := NewFactory(nil)
|
|
|
|
tests := []struct {
|
|
name string
|
|
object runtime.Object
|
|
expected string
|
|
err error
|
|
}{
|
|
{
|
|
name: "successful re-use of labels",
|
|
object: &api.Service{
|
|
ObjectMeta: api.ObjectMeta{Name: "baz", Namespace: "test", Labels: map[string]string{"svc": "test"}},
|
|
TypeMeta: unversioned.TypeMeta{Kind: "Service", APIVersion: "v1"},
|
|
},
|
|
expected: "svc=test",
|
|
err: nil,
|
|
},
|
|
{
|
|
name: "empty labels",
|
|
object: &api.Service{
|
|
ObjectMeta: api.ObjectMeta{Name: "foo", Namespace: "test", Labels: map[string]string{}},
|
|
TypeMeta: unversioned.TypeMeta{Kind: "Service", APIVersion: "v1"},
|
|
},
|
|
expected: "",
|
|
err: nil,
|
|
},
|
|
{
|
|
name: "nil labels",
|
|
object: &api.Service{
|
|
ObjectMeta: api.ObjectMeta{Name: "zen", Namespace: "test", Labels: nil},
|
|
TypeMeta: unversioned.TypeMeta{Kind: "Service", APIVersion: "v1"},
|
|
},
|
|
expected: "",
|
|
err: nil,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
gotLabels, err := f.LabelsForObject(test.object)
|
|
if err != test.err {
|
|
t.Fatalf("%s: Error mismatch: Expected %v, got %v", test.name, test.err, err)
|
|
}
|
|
got := kubectl.MakeLabels(gotLabels)
|
|
if test.expected != got {
|
|
t.Fatalf("%s: Labels mismatch! Expected %s, got %s", test.name, test.expected, got)
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
func TestCanBeExposed(t *testing.T) {
|
|
factory := NewFactory(nil)
|
|
tests := []struct {
|
|
kind unversioned.GroupKind
|
|
expectErr bool
|
|
}{
|
|
{
|
|
kind: api.Kind("ReplicationController"),
|
|
expectErr: false,
|
|
},
|
|
{
|
|
kind: api.Kind("Node"),
|
|
expectErr: true,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
err := factory.CanBeExposed(test.kind)
|
|
if test.expectErr && err == nil {
|
|
t.Error("unexpected non-error")
|
|
}
|
|
if !test.expectErr && err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestFlagUnderscoreRenaming(t *testing.T) {
|
|
factory := NewFactory(nil)
|
|
|
|
factory.FlagSet().SetNormalizeFunc(flag.WordSepNormalizeFunc)
|
|
factory.FlagSet().Bool("valid_flag", false, "bool value")
|
|
|
|
// In case of failure of this test check this PR: spf13/pflag#23
|
|
if factory.FlagSet().Lookup("valid_flag").Name != "valid-flag" {
|
|
t.Fatalf("Expected flag name to be valid-flag, got %s", factory.FlagSet().Lookup("valid_flag").Name)
|
|
}
|
|
}
|
|
|
|
func loadSchemaForTest() (validation.Schema, error) {
|
|
pathToSwaggerSpec := "../../../../api/swagger-spec/" + registered.GroupOrDie(api.GroupName).GroupVersion.Version + ".json"
|
|
data, err := ioutil.ReadFile(pathToSwaggerSpec)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return validation.NewSwaggerSchemaFromBytes(data, nil)
|
|
}
|
|
|
|
func header() http.Header {
|
|
header := http.Header{}
|
|
header.Set("Content-Type", runtime.ContentTypeJSON)
|
|
return header
|
|
}
|
|
|
|
func TestRefetchSchemaWhenValidationFails(t *testing.T) {
|
|
schema, err := loadSchemaForTest()
|
|
if err != nil {
|
|
t.Errorf("Error loading schema: %v", err)
|
|
t.FailNow()
|
|
}
|
|
output, err := json.Marshal(schema)
|
|
if err != nil {
|
|
t.Errorf("Error serializing schema: %v", err)
|
|
t.FailNow()
|
|
}
|
|
requests := map[string]int{}
|
|
|
|
c := &manualfake.RESTClient{
|
|
NegotiatedSerializer: testapi.Default.NegotiatedSerializer(),
|
|
Client: manualfake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
|
|
switch p, m := req.URL.Path, req.Method; {
|
|
case strings.HasPrefix(p, "/swaggerapi") && m == "GET":
|
|
requests[p] = requests[p] + 1
|
|
return &http.Response{StatusCode: 200, Header: header(), Body: ioutil.NopCloser(bytes.NewBuffer(output))}, nil
|
|
default:
|
|
t.Fatalf("unexpected request: %#v\n%#v", req.URL, req)
|
|
return nil, nil
|
|
}
|
|
}),
|
|
}
|
|
dir := os.TempDir() + "/schemaCache"
|
|
os.RemoveAll(dir)
|
|
|
|
fullDir, err := substituteUserHome(dir)
|
|
if err != nil {
|
|
t.Errorf("Error getting fullDir: %v", err)
|
|
t.FailNow()
|
|
}
|
|
cacheFile := path.Join(fullDir, "foo", "bar", schemaFileName)
|
|
err = writeSchemaFile(output, fullDir, cacheFile, "foo", "bar")
|
|
if err != nil {
|
|
t.Errorf("Error building old cache schema: %v", err)
|
|
t.FailNow()
|
|
}
|
|
|
|
obj := &extensions.Deployment{}
|
|
data, err := runtime.Encode(testapi.Extensions.Codec(), obj)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
t.FailNow()
|
|
}
|
|
|
|
// Re-get request, should use HTTP and write
|
|
if getSchemaAndValidate(c, data, "foo", "bar", dir, nil); err != nil {
|
|
t.Errorf("unexpected error validating: %v", err)
|
|
}
|
|
if requests["/swaggerapi/foo/bar"] != 1 {
|
|
t.Errorf("expected 1 schema request, saw: %d", requests["/swaggerapi/foo/bar"])
|
|
}
|
|
}
|
|
|
|
func TestValidateCachesSchema(t *testing.T) {
|
|
schema, err := loadSchemaForTest()
|
|
if err != nil {
|
|
t.Errorf("Error loading schema: %v", err)
|
|
t.FailNow()
|
|
}
|
|
output, err := json.Marshal(schema)
|
|
if err != nil {
|
|
t.Errorf("Error serializing schema: %v", err)
|
|
t.FailNow()
|
|
}
|
|
requests := map[string]int{}
|
|
|
|
c := &manualfake.RESTClient{
|
|
NegotiatedSerializer: testapi.Default.NegotiatedSerializer(),
|
|
Client: manualfake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
|
|
switch p, m := req.URL.Path, req.Method; {
|
|
case strings.HasPrefix(p, "/swaggerapi") && m == "GET":
|
|
requests[p] = requests[p] + 1
|
|
return &http.Response{StatusCode: 200, Header: header(), Body: ioutil.NopCloser(bytes.NewBuffer(output))}, nil
|
|
default:
|
|
t.Fatalf("unexpected request: %#v\n%#v", req.URL, req)
|
|
return nil, nil
|
|
}
|
|
}),
|
|
}
|
|
dir := os.TempDir() + "/schemaCache"
|
|
os.RemoveAll(dir)
|
|
|
|
obj := &api.Pod{}
|
|
data, err := runtime.Encode(testapi.Default.Codec(), obj)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
t.FailNow()
|
|
}
|
|
|
|
// Initial request, should use HTTP and write
|
|
if getSchemaAndValidate(c, data, "foo", "bar", dir, nil); err != nil {
|
|
t.Errorf("unexpected error validating: %v", err)
|
|
}
|
|
if _, err := os.Stat(path.Join(dir, "foo", "bar", schemaFileName)); err != nil {
|
|
t.Errorf("unexpected missing cache file: %v", err)
|
|
}
|
|
if requests["/swaggerapi/foo/bar"] != 1 {
|
|
t.Errorf("expected 1 schema request, saw: %d", requests["/swaggerapi/foo/bar"])
|
|
}
|
|
|
|
// Same version and group, should skip HTTP
|
|
if getSchemaAndValidate(c, data, "foo", "bar", dir, nil); err != nil {
|
|
t.Errorf("unexpected error validating: %v", err)
|
|
}
|
|
if requests["/swaggerapi/foo/bar"] != 2 {
|
|
t.Errorf("expected 1 schema request, saw: %d", requests["/swaggerapi/foo/bar"])
|
|
}
|
|
|
|
// Different API group, should go to HTTP and write
|
|
if getSchemaAndValidate(c, data, "foo", "baz", dir, nil); err != nil {
|
|
t.Errorf("unexpected error validating: %v", err)
|
|
}
|
|
if _, err := os.Stat(path.Join(dir, "foo", "baz", schemaFileName)); err != nil {
|
|
t.Errorf("unexpected missing cache file: %v", err)
|
|
}
|
|
if requests["/swaggerapi/foo/baz"] != 1 {
|
|
t.Errorf("expected 1 schema request, saw: %d", requests["/swaggerapi/foo/baz"])
|
|
}
|
|
|
|
// Different version, should go to HTTP and write
|
|
if getSchemaAndValidate(c, data, "foo2", "bar", dir, nil); err != nil {
|
|
t.Errorf("unexpected error validating: %v", err)
|
|
}
|
|
if _, err := os.Stat(path.Join(dir, "foo2", "bar", schemaFileName)); err != nil {
|
|
t.Errorf("unexpected missing cache file: %v", err)
|
|
}
|
|
if requests["/swaggerapi/foo2/bar"] != 1 {
|
|
t.Errorf("expected 1 schema request, saw: %d", requests["/swaggerapi/foo2/bar"])
|
|
}
|
|
|
|
// No cache dir, should go straight to HTTP and not write
|
|
if getSchemaAndValidate(c, data, "foo", "blah", "", nil); err != nil {
|
|
t.Errorf("unexpected error validating: %v", err)
|
|
}
|
|
if requests["/swaggerapi/foo/blah"] != 1 {
|
|
t.Errorf("expected 1 schema request, saw: %d", requests["/swaggerapi/foo/blah"])
|
|
}
|
|
if _, err := os.Stat(path.Join(dir, "foo", "blah", schemaFileName)); err == nil || !os.IsNotExist(err) {
|
|
t.Errorf("unexpected cache file error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestSubstitueUser(t *testing.T) {
|
|
usr, err := user.Current()
|
|
if err != nil {
|
|
t.Logf("SKIPPING TEST: unexpected error: %v", err)
|
|
return
|
|
}
|
|
tests := []struct {
|
|
input string
|
|
expected string
|
|
expectErr bool
|
|
}{
|
|
{input: "~/foo", expected: path.Join(os.Getenv("HOME"), "foo")},
|
|
{input: "~" + usr.Username + "/bar", expected: usr.HomeDir + "/bar"},
|
|
{input: "/foo/bar", expected: "/foo/bar"},
|
|
{input: "~doesntexit/bar", expectErr: true},
|
|
}
|
|
for _, test := range tests {
|
|
output, err := substituteUserHome(test.input)
|
|
if test.expectErr {
|
|
if err == nil {
|
|
t.Error("unexpected non-error")
|
|
}
|
|
continue
|
|
}
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if output != test.expected {
|
|
t.Errorf("expected: %s, saw: %s", test.expected, output)
|
|
}
|
|
}
|
|
}
|
|
|
|
func newPodList(count, isUnready, isUnhealthy int, labels map[string]string) *api.PodList {
|
|
pods := []api.Pod{}
|
|
for i := 0; i < count; i++ {
|
|
newPod := api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: fmt.Sprintf("pod-%d", i+1),
|
|
Namespace: api.NamespaceDefault,
|
|
CreationTimestamp: unversioned.Date(2016, time.April, 1, 1, 0, i, 0, time.UTC),
|
|
Labels: labels,
|
|
},
|
|
Status: api.PodStatus{
|
|
Conditions: []api.PodCondition{
|
|
{
|
|
Status: api.ConditionTrue,
|
|
Type: api.PodReady,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
pods = append(pods, newPod)
|
|
}
|
|
if isUnready > -1 && isUnready < count {
|
|
pods[isUnready].Status.Conditions[0].Status = api.ConditionFalse
|
|
}
|
|
if isUnhealthy > -1 && isUnhealthy < count {
|
|
pods[isUnhealthy].Status.ContainerStatuses = []api.ContainerStatus{{RestartCount: 5}}
|
|
}
|
|
return &api.PodList{
|
|
Items: pods,
|
|
}
|
|
}
|
|
|
|
func TestGetFirstPod(t *testing.T) {
|
|
labelSet := map[string]string{"test": "selector"}
|
|
tests := []struct {
|
|
name string
|
|
|
|
podList *api.PodList
|
|
watching []watch.Event
|
|
sortBy func([]*api.Pod) sort.Interface
|
|
|
|
expected *api.Pod
|
|
expectedNum int
|
|
expectedErr bool
|
|
}{
|
|
{
|
|
name: "kubectl logs - two ready pods",
|
|
podList: newPodList(2, -1, -1, labelSet),
|
|
sortBy: func(pods []*api.Pod) sort.Interface { return controller.ByLogging(pods) },
|
|
expected: &api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "pod-1",
|
|
Namespace: api.NamespaceDefault,
|
|
CreationTimestamp: unversioned.Date(2016, time.April, 1, 1, 0, 0, 0, time.UTC),
|
|
Labels: map[string]string{"test": "selector"},
|
|
},
|
|
Status: api.PodStatus{
|
|
Conditions: []api.PodCondition{
|
|
{
|
|
Status: api.ConditionTrue,
|
|
Type: api.PodReady,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedNum: 2,
|
|
},
|
|
{
|
|
name: "kubectl logs - one unhealthy, one healthy",
|
|
podList: newPodList(2, -1, 1, labelSet),
|
|
sortBy: func(pods []*api.Pod) sort.Interface { return controller.ByLogging(pods) },
|
|
expected: &api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "pod-2",
|
|
Namespace: api.NamespaceDefault,
|
|
CreationTimestamp: unversioned.Date(2016, time.April, 1, 1, 0, 1, 0, time.UTC),
|
|
Labels: map[string]string{"test": "selector"},
|
|
},
|
|
Status: api.PodStatus{
|
|
Conditions: []api.PodCondition{
|
|
{
|
|
Status: api.ConditionTrue,
|
|
Type: api.PodReady,
|
|
},
|
|
},
|
|
ContainerStatuses: []api.ContainerStatus{{RestartCount: 5}},
|
|
},
|
|
},
|
|
expectedNum: 2,
|
|
},
|
|
{
|
|
name: "kubectl attach - two ready pods",
|
|
podList: newPodList(2, -1, -1, labelSet),
|
|
sortBy: func(pods []*api.Pod) sort.Interface { return sort.Reverse(controller.ActivePods(pods)) },
|
|
expected: &api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "pod-1",
|
|
Namespace: api.NamespaceDefault,
|
|
CreationTimestamp: unversioned.Date(2016, time.April, 1, 1, 0, 0, 0, time.UTC),
|
|
Labels: map[string]string{"test": "selector"},
|
|
},
|
|
Status: api.PodStatus{
|
|
Conditions: []api.PodCondition{
|
|
{
|
|
Status: api.ConditionTrue,
|
|
Type: api.PodReady,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedNum: 2,
|
|
},
|
|
{
|
|
name: "kubectl attach - wait for ready pod",
|
|
podList: newPodList(1, 1, -1, labelSet),
|
|
watching: []watch.Event{
|
|
{
|
|
Type: watch.Modified,
|
|
Object: &api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "pod-1",
|
|
Namespace: api.NamespaceDefault,
|
|
CreationTimestamp: unversioned.Date(2016, time.April, 1, 1, 0, 0, 0, time.UTC),
|
|
Labels: map[string]string{"test": "selector"},
|
|
},
|
|
Status: api.PodStatus{
|
|
Conditions: []api.PodCondition{
|
|
{
|
|
Status: api.ConditionTrue,
|
|
Type: api.PodReady,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
sortBy: func(pods []*api.Pod) sort.Interface { return sort.Reverse(controller.ActivePods(pods)) },
|
|
expected: &api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "pod-1",
|
|
Namespace: api.NamespaceDefault,
|
|
CreationTimestamp: unversioned.Date(2016, time.April, 1, 1, 0, 0, 0, time.UTC),
|
|
Labels: map[string]string{"test": "selector"},
|
|
},
|
|
Status: api.PodStatus{
|
|
Conditions: []api.PodCondition{
|
|
{
|
|
Status: api.ConditionTrue,
|
|
Type: api.PodReady,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedNum: 1,
|
|
},
|
|
}
|
|
|
|
for i := range tests {
|
|
test := tests[i]
|
|
fake := fake.NewSimpleClientset(test.podList)
|
|
if len(test.watching) > 0 {
|
|
watcher := watch.NewFake()
|
|
for _, event := range test.watching {
|
|
switch event.Type {
|
|
case watch.Added:
|
|
go watcher.Add(event.Object)
|
|
case watch.Modified:
|
|
go watcher.Modify(event.Object)
|
|
}
|
|
}
|
|
fake.PrependWatchReactor("pods", testcore.DefaultWatchReactor(watcher, nil))
|
|
}
|
|
selector := labels.Set(labelSet).AsSelector()
|
|
|
|
pod, numPods, err := GetFirstPod(fake.Core(), api.NamespaceDefault, selector, 1*time.Minute, test.sortBy)
|
|
if !test.expectedErr && err != nil {
|
|
t.Errorf("%s: unexpected error: %v", test.name, err)
|
|
continue
|
|
}
|
|
if test.expectedErr && err == nil {
|
|
t.Errorf("%s: expected an error", test.name)
|
|
continue
|
|
}
|
|
if test.expectedNum != numPods {
|
|
t.Errorf("%s: expected %d pods, got %d", test.name, test.expectedNum, numPods)
|
|
continue
|
|
}
|
|
if !reflect.DeepEqual(test.expected, pod) {
|
|
t.Errorf("%s:\nexpected pod:\n%#v\ngot:\n%#v\n\n", test.name, test.expected, pod)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPrintObjectSpecificMessage(t *testing.T) {
|
|
f := NewFactory(nil)
|
|
tests := []struct {
|
|
obj runtime.Object
|
|
expectOutput bool
|
|
}{
|
|
{
|
|
obj: &api.Service{},
|
|
expectOutput: false,
|
|
},
|
|
{
|
|
obj: &api.Pod{},
|
|
expectOutput: false,
|
|
},
|
|
{
|
|
obj: &api.Service{Spec: api.ServiceSpec{Type: api.ServiceTypeLoadBalancer}},
|
|
expectOutput: false,
|
|
},
|
|
{
|
|
obj: &api.Service{Spec: api.ServiceSpec{Type: api.ServiceTypeNodePort}},
|
|
expectOutput: true,
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
buff := &bytes.Buffer{}
|
|
f.PrintObjectSpecificMessage(test.obj, buff)
|
|
if test.expectOutput && buff.Len() == 0 {
|
|
t.Errorf("Expected output, saw none for %v", test.obj)
|
|
}
|
|
if !test.expectOutput && buff.Len() > 0 {
|
|
t.Errorf("Expected no output, saw %s for %v", buff.String(), test.obj)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestMakePortsString(t *testing.T) {
|
|
tests := []struct {
|
|
ports []api.ServicePort
|
|
useNodePort bool
|
|
expectedOutput string
|
|
}{
|
|
{ports: nil, expectedOutput: ""},
|
|
{ports: []api.ServicePort{}, expectedOutput: ""},
|
|
{ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
Protocol: "TCP",
|
|
},
|
|
},
|
|
expectedOutput: "tcp:80",
|
|
},
|
|
{ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
Protocol: "TCP",
|
|
},
|
|
{
|
|
Port: 8080,
|
|
Protocol: "UDP",
|
|
},
|
|
{
|
|
Port: 9000,
|
|
Protocol: "TCP",
|
|
},
|
|
},
|
|
expectedOutput: "tcp:80,udp:8080,tcp:9000",
|
|
},
|
|
{ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
NodePort: 9090,
|
|
Protocol: "TCP",
|
|
},
|
|
{
|
|
Port: 8080,
|
|
NodePort: 80,
|
|
Protocol: "UDP",
|
|
},
|
|
},
|
|
useNodePort: true,
|
|
expectedOutput: "tcp:9090,udp:80",
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
output := makePortsString(test.ports, test.useNodePort)
|
|
if output != test.expectedOutput {
|
|
t.Errorf("expected: %s, saw: %s.", test.expectedOutput, output)
|
|
}
|
|
}
|
|
}
|
|
|
|
func fakeClient() resource.ClientMapper {
|
|
return resource.ClientMapperFunc(func(*meta.RESTMapping) (resource.RESTClient, error) {
|
|
return &manualfake.RESTClient{}, nil
|
|
})
|
|
}
|
|
|
|
func TestDiscoveryReplaceAliases(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
arg string
|
|
expected string
|
|
}{
|
|
{
|
|
name: "no-replacement",
|
|
arg: "service",
|
|
expected: "service",
|
|
},
|
|
{
|
|
name: "all-replacement",
|
|
arg: "all",
|
|
expected: "pods,replicationcontrollers,services,petsets,horizontalpodautoscalers,jobs,deployments,replicasets",
|
|
},
|
|
{
|
|
name: "alias-in-comma-separated-arg",
|
|
arg: "all,secrets",
|
|
expected: "pods,replicationcontrollers,services,petsets,horizontalpodautoscalers,jobs,deployments,replicasets,secrets",
|
|
},
|
|
}
|
|
|
|
mapper := NewShortcutExpander(testapi.Default.RESTMapper(), nil)
|
|
b := resource.NewBuilder(mapper, api.Scheme, fakeClient(), testapi.Default.Codec())
|
|
|
|
for _, test := range tests {
|
|
replaced := b.ReplaceAliases(test.arg)
|
|
if replaced != test.expected {
|
|
t.Errorf("%s: unexpected argument: expected %s, got %s", test.name, test.expected, replaced)
|
|
}
|
|
}
|
|
}
|