kubevela/pkg/commands/exec_test.go

package commands

import (
	"context"
	"fmt"
	"os"
	"testing"

	"github.com/oam-dev/kubevela/pkg/appfile/api"

	"github.com/spf13/cobra"
	"github.com/stretchr/testify/assert"
	corev1 "k8s.io/api/core/v1"
	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/cli-runtime/pkg/genericclioptions"
	"k8s.io/client-go/kubernetes/fake"
	"k8s.io/kubectl/pkg/cmd/exec"
	cmdtesting "k8s.io/kubectl/pkg/cmd/testing"
	k8scmdutil "k8s.io/kubectl/pkg/cmd/util"

	"github.com/oam-dev/kubevela/apis/types"
	cmdutil "github.com/oam-dev/kubevela/pkg/commands/util"
	"github.com/oam-dev/kubevela/pkg/oam"
)

func TestExecCommand(t *testing.T) {
	tf := cmdtesting.NewTestFactory()
	defer tf.Cleanup()
	io := cmdutil.IOStreams{In: os.Stdin, Out: os.Stdout, ErrOut: os.Stderr}
	fakeC := types.Args{
		Config: tf.ClientConfigVal,
	}
	cmd := NewExecCommand(fakeC, io)
	cmd.PersistentFlags().StringP("env", "e", "", "")
	o := &VelaExecOptions{
		kcExecOptions: &exec.ExecOptions{},
		f:             tf,
		ClientSet: fake.NewSimpleClientset(&corev1.PodList{
			Items: []corev1.Pod{
				{
					ObjectMeta: v1.ObjectMeta{
						Name:      "fakePod",
						Namespace: "default",
						Labels: map[string]string{
							oam.LabelAppName:      "fakeApp",
							oam.LabelAppComponent: "fakeComp",
						}},
				},
			},
		}),
	}
	err := o.Init(context.Background(), cmd, []string{"fakeApp"})
	errString := fmt.Sprintf(`application "%s" not found`, "fakeApp")
	assert.EqualError(t, err, errString)
	fakeApp := &api.Application{
		AppFile: &api.AppFile{
			Name: "fakeApp",
			Services: map[string]api.Service{
				"fakeComp": map[string]interface{}{},
			},
		},
	}
	o.App = fakeApp

	cf := genericclioptions.NewConfigFlags(true)
	cf.Namespace = &o.Env.Namespace
	o.f = k8scmdutil.NewFactory(k8scmdutil.NewMatchVersionFlags(cf))
	err = o.Complete()
	assert.NoError(t, err)
}

func TestExecCommandPersistentPreRunE(t *testing.T) {
	io := cmdutil.IOStreams{In: os.Stdin, Out: os.Stdout, ErrOut: os.Stderr}
	fakeC := types.Args{}
	cmd := NewExecCommand(fakeC, io)
	assert.Nil(t, cmd.PersistentPreRunE(new(cobra.Command), []string{}))
}

func TestGetComponent(t *testing.T) {
	o := &VelaExecOptions{
		App: &api.Application{
			AppFile: &api.AppFile{
				Name: "fakeApp",
				Services: map[string]api.Service{
					"fakeComp1": map[string]interface{}{},
					"fakeComp2": map[string]interface{}{},
				},
			},
		},
	}

	o.ServiceName = "fakeComp1"
	svcName, err := o.getComponentName()
	assert.NoError(t, err)
	assert.Equal(t, o.ServiceName, svcName)

	o.ServiceName = "fakeComp2"
	svcName, err = o.getComponentName()
	assert.NoError(t, err)
	assert.Equal(t, o.ServiceName, svcName)
}