mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-10-30 21:30:16 +00:00
449 lines
12 KiB
Go
449 lines
12 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 cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/apimachinery/pkg/util/intstr"
|
|
restclient "k8s.io/client-go/rest"
|
|
"k8s.io/client-go/rest/fake"
|
|
"k8s.io/kubernetes/pkg/api"
|
|
"k8s.io/kubernetes/pkg/api/v1"
|
|
cmdtesting "k8s.io/kubernetes/pkg/kubectl/cmd/testing"
|
|
cmdutil "k8s.io/kubernetes/pkg/kubectl/cmd/util"
|
|
)
|
|
|
|
func TestGetRestartPolicy(t *testing.T) {
|
|
tests := []struct {
|
|
input string
|
|
interactive bool
|
|
expected api.RestartPolicy
|
|
expectErr bool
|
|
}{
|
|
{
|
|
input: "",
|
|
expected: api.RestartPolicyAlways,
|
|
},
|
|
{
|
|
input: "",
|
|
interactive: true,
|
|
expected: api.RestartPolicyOnFailure,
|
|
},
|
|
{
|
|
input: string(api.RestartPolicyAlways),
|
|
interactive: true,
|
|
expected: api.RestartPolicyAlways,
|
|
},
|
|
{
|
|
input: string(api.RestartPolicyNever),
|
|
interactive: true,
|
|
expected: api.RestartPolicyNever,
|
|
},
|
|
{
|
|
input: string(api.RestartPolicyAlways),
|
|
expected: api.RestartPolicyAlways,
|
|
},
|
|
{
|
|
input: string(api.RestartPolicyNever),
|
|
expected: api.RestartPolicyNever,
|
|
},
|
|
{
|
|
input: "foo",
|
|
expectErr: true,
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("restart", "", "dummy restart flag")
|
|
cmd.Flags().Lookup("restart").Value.Set(test.input)
|
|
policy, err := getRestartPolicy(cmd, test.interactive)
|
|
if test.expectErr && err == nil {
|
|
t.Error("unexpected non-error")
|
|
}
|
|
if !test.expectErr && err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if !test.expectErr && policy != test.expected {
|
|
t.Errorf("expected: %s, saw: %s (%s:%v)", test.expected, policy, test.input, test.interactive)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGetEnv(t *testing.T) {
|
|
test := struct {
|
|
input []string
|
|
expected []string
|
|
}{
|
|
input: []string{"a=b", "c=d"},
|
|
expected: []string{"a=b", "c=d"},
|
|
}
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().StringSlice("env", test.input, "")
|
|
|
|
envStrings := cmdutil.GetFlagStringSlice(cmd, "env")
|
|
if len(envStrings) != 2 || !reflect.DeepEqual(envStrings, test.expected) {
|
|
t.Errorf("expected: %s, saw: %s", test.expected, envStrings)
|
|
}
|
|
}
|
|
|
|
func TestRunArgsFollowDashRules(t *testing.T) {
|
|
one := int32(1)
|
|
rc := &v1.ReplicationController{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "rc1", Namespace: "test", ResourceVersion: "18"},
|
|
Spec: v1.ReplicationControllerSpec{
|
|
Replicas: &one,
|
|
},
|
|
}
|
|
|
|
tests := []struct {
|
|
args []string
|
|
argsLenAtDash int
|
|
expectError bool
|
|
name string
|
|
}{
|
|
{
|
|
args: []string{},
|
|
argsLenAtDash: -1,
|
|
expectError: true,
|
|
name: "empty",
|
|
},
|
|
{
|
|
args: []string{"foo"},
|
|
argsLenAtDash: -1,
|
|
expectError: false,
|
|
name: "no cmd",
|
|
},
|
|
{
|
|
args: []string{"foo", "sleep"},
|
|
argsLenAtDash: -1,
|
|
expectError: false,
|
|
name: "cmd no dash",
|
|
},
|
|
{
|
|
args: []string{"foo", "sleep"},
|
|
argsLenAtDash: 1,
|
|
expectError: false,
|
|
name: "cmd has dash",
|
|
},
|
|
{
|
|
args: []string{"foo", "sleep"},
|
|
argsLenAtDash: 0,
|
|
expectError: true,
|
|
name: "no name",
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
f, tf, codec, ns := cmdtesting.NewAPIFactory()
|
|
tf.Client = &fake.RESTClient{
|
|
APIRegistry: api.Registry,
|
|
NegotiatedSerializer: ns,
|
|
Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
|
|
if req.URL.Path == "/namespaces/test/replicationcontrollers" {
|
|
return &http.Response{StatusCode: 201, Header: defaultHeader(), Body: objBody(codec, rc)}, nil
|
|
} else {
|
|
return &http.Response{
|
|
StatusCode: http.StatusOK,
|
|
Body: ioutil.NopCloser(&bytes.Buffer{}),
|
|
}, nil
|
|
}
|
|
}),
|
|
}
|
|
tf.Namespace = "test"
|
|
tf.ClientConfig = &restclient.Config{}
|
|
cmd := NewCmdRun(f, os.Stdin, os.Stdout, os.Stderr)
|
|
cmd.Flags().Set("image", "nginx")
|
|
cmd.Flags().Set("generator", "run/v1")
|
|
err := Run(f, os.Stdin, os.Stdout, os.Stderr, cmd, test.args, test.argsLenAtDash)
|
|
if test.expectError && err == nil {
|
|
t.Errorf("unexpected non-error (%s)", test.name)
|
|
}
|
|
if !test.expectError && err != nil {
|
|
t.Errorf("unexpected error: %v (%s)", err, test.name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGenerateService(t *testing.T) {
|
|
|
|
tests := []struct {
|
|
port string
|
|
args []string
|
|
serviceGenerator string
|
|
params map[string]interface{}
|
|
expectErr bool
|
|
name string
|
|
service api.Service
|
|
expectPOST bool
|
|
}{
|
|
{
|
|
port: "80",
|
|
args: []string{"foo"},
|
|
serviceGenerator: "service/v2",
|
|
params: map[string]interface{}{
|
|
"name": "foo",
|
|
},
|
|
expectErr: false,
|
|
name: "basic",
|
|
service: api.Service{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo",
|
|
},
|
|
Spec: api.ServiceSpec{
|
|
Ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
Protocol: "TCP",
|
|
TargetPort: intstr.FromInt(80),
|
|
},
|
|
},
|
|
Selector: map[string]string{
|
|
"run": "foo",
|
|
},
|
|
Type: api.ServiceTypeClusterIP,
|
|
SessionAffinity: api.ServiceAffinityNone,
|
|
},
|
|
},
|
|
expectPOST: true,
|
|
},
|
|
{
|
|
port: "80",
|
|
args: []string{"foo"},
|
|
serviceGenerator: "service/v2",
|
|
params: map[string]interface{}{
|
|
"name": "foo",
|
|
"labels": "app=bar",
|
|
},
|
|
expectErr: false,
|
|
name: "custom labels",
|
|
service: api.Service{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo",
|
|
Labels: map[string]string{"app": "bar"},
|
|
},
|
|
Spec: api.ServiceSpec{
|
|
Ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
Protocol: "TCP",
|
|
TargetPort: intstr.FromInt(80),
|
|
},
|
|
},
|
|
Selector: map[string]string{
|
|
"app": "bar",
|
|
},
|
|
Type: api.ServiceTypeClusterIP,
|
|
SessionAffinity: api.ServiceAffinityNone,
|
|
},
|
|
},
|
|
expectPOST: true,
|
|
},
|
|
{
|
|
expectErr: true,
|
|
name: "missing port",
|
|
expectPOST: false,
|
|
},
|
|
{
|
|
port: "80",
|
|
args: []string{"foo"},
|
|
serviceGenerator: "service/v2",
|
|
params: map[string]interface{}{
|
|
"name": "foo",
|
|
},
|
|
expectErr: false,
|
|
name: "dry-run",
|
|
expectPOST: false,
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
sawPOST := false
|
|
f, tf, codec, ns := cmdtesting.NewAPIFactory()
|
|
tf.ClientConfig = &restclient.Config{ContentConfig: restclient.ContentConfig{GroupVersion: &api.Registry.GroupOrDie(api.GroupName).GroupVersion}}
|
|
tf.Printer = &testPrinter{}
|
|
tf.Client = &fake.RESTClient{
|
|
APIRegistry: api.Registry,
|
|
NegotiatedSerializer: ns,
|
|
Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
|
|
switch p, m := req.URL.Path, req.Method; {
|
|
case test.expectPOST && m == "POST" && p == "/namespaces/namespace/services":
|
|
sawPOST = true
|
|
body := objBody(codec, &test.service)
|
|
data, err := ioutil.ReadAll(req.Body)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
t.FailNow()
|
|
}
|
|
defer req.Body.Close()
|
|
svc := &api.Service{}
|
|
if err := runtime.DecodeInto(codec, data, svc); err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
t.FailNow()
|
|
}
|
|
// Copy things that are defaulted by the system
|
|
test.service.Annotations = svc.Annotations
|
|
|
|
if !reflect.DeepEqual(&test.service, svc) {
|
|
t.Errorf("expected:\n%v\nsaw:\n%v\n", &test.service, svc)
|
|
}
|
|
return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: body}, nil
|
|
default:
|
|
// Ensures no GET is performed when deleting by name
|
|
t.Errorf("%s: unexpected request: %s %#v\n%#v", test.name, req.Method, req.URL, req)
|
|
return nil, fmt.Errorf("unexpected request")
|
|
}
|
|
}),
|
|
}
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool(cmdutil.ApplyAnnotationsFlag, false, "")
|
|
cmd.Flags().Bool("record", false, "Record current kubectl command in the resource annotation. If set to false, do not record the command. If set to true, record the command. If not set, default to updating the existing annotation value only if one already exists.")
|
|
cmdutil.AddPrinterFlags(cmd)
|
|
cmdutil.AddInclude3rdPartyFlags(cmd)
|
|
addRunFlags(cmd)
|
|
|
|
if !test.expectPOST {
|
|
cmd.Flags().Set("dry-run", "true")
|
|
}
|
|
|
|
if len(test.port) > 0 {
|
|
cmd.Flags().Set("port", test.port)
|
|
test.params["port"] = test.port
|
|
}
|
|
|
|
buff := &bytes.Buffer{}
|
|
err := generateService(f, cmd, test.args, test.serviceGenerator, test.params, "namespace", buff)
|
|
if test.expectErr {
|
|
if err == nil {
|
|
t.Error("unexpected non-error")
|
|
}
|
|
continue
|
|
}
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if test.expectPOST != sawPOST {
|
|
t.Errorf("expectPost: %v, sawPost: %v", test.expectPOST, sawPOST)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRunValidations(t *testing.T) {
|
|
tests := []struct {
|
|
args []string
|
|
flags map[string]string
|
|
expectedErr string
|
|
}{
|
|
{
|
|
expectedErr: "NAME is required",
|
|
},
|
|
{
|
|
args: []string{"test"},
|
|
expectedErr: "Invalid image name",
|
|
},
|
|
{
|
|
args: []string{"test"},
|
|
flags: map[string]string{
|
|
"image": "busybox",
|
|
"stdin": "true",
|
|
"replicas": "2",
|
|
},
|
|
expectedErr: "stdin requires that replicas is 1",
|
|
},
|
|
{
|
|
args: []string{"test"},
|
|
flags: map[string]string{
|
|
"image": "busybox",
|
|
"rm": "true",
|
|
},
|
|
expectedErr: "rm should only be used for attached containers",
|
|
},
|
|
{
|
|
args: []string{"test"},
|
|
flags: map[string]string{
|
|
"image": "busybox",
|
|
"attach": "true",
|
|
"dry-run": "true",
|
|
},
|
|
expectedErr: "can't be used with attached containers options",
|
|
},
|
|
{
|
|
args: []string{"test"},
|
|
flags: map[string]string{
|
|
"image": "busybox",
|
|
"stdin": "true",
|
|
"dry-run": "true",
|
|
},
|
|
expectedErr: "can't be used with attached containers options",
|
|
},
|
|
{
|
|
args: []string{"test"},
|
|
flags: map[string]string{
|
|
"image": "busybox",
|
|
"tty": "true",
|
|
"stdin": "true",
|
|
"dry-run": "true",
|
|
},
|
|
expectedErr: "can't be used with attached containers options",
|
|
},
|
|
{
|
|
args: []string{"test"},
|
|
flags: map[string]string{
|
|
"image": "busybox",
|
|
"tty": "true",
|
|
},
|
|
expectedErr: "stdin is required for containers with -t/--tty",
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
f, tf, codec, ns := cmdtesting.NewTestFactory()
|
|
tf.Printer = &testPrinter{}
|
|
tf.Client = &fake.RESTClient{
|
|
APIRegistry: api.Registry,
|
|
NegotiatedSerializer: ns,
|
|
Resp: &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, cmdtesting.NewInternalType("", "", ""))},
|
|
}
|
|
tf.Namespace = "test"
|
|
tf.ClientConfig = &restclient.Config{ContentConfig: restclient.ContentConfig{GroupVersion: &schema.GroupVersion{Version: "v1"}}}
|
|
inBuf := bytes.NewReader([]byte{})
|
|
outBuf := bytes.NewBuffer([]byte{})
|
|
errBuf := bytes.NewBuffer([]byte{})
|
|
|
|
cmd := NewCmdRun(f, inBuf, outBuf, errBuf)
|
|
for flagName, flagValue := range test.flags {
|
|
cmd.Flags().Set(flagName, flagValue)
|
|
}
|
|
err := Run(f, inBuf, outBuf, errBuf, cmd, test.args, cmd.ArgsLenAtDash())
|
|
if err != nil && len(test.expectedErr) > 0 {
|
|
if !strings.Contains(err.Error(), test.expectedErr) {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|