kt-connect/pkg/kt/command/dashboard_test.go

package command

import (
	"flag"
	"io/ioutil"
	"testing"

	"github.com/alibaba/kt-connect/pkg/fake/kt/action"
	"github.com/golang/mock/gomock"

	fakeKt "github.com/alibaba/kt-connect/pkg/fake/kt"
	"github.com/alibaba/kt-connect/pkg/kt/options"
	"github.com/urfave/cli"
)

func Test_newDashboardCommand(t *testing.T) {
	ctl := gomock.NewController(t)
	fakeKtCli := fakeKt.NewMockCliInterface(ctl)

	mockAction := action.NewMockActionInterface(ctl)
	mockAction.EXPECT().OpenDashboard(gomock.Any(), gomock.Any()).Return(nil)
	mockAction.EXPECT().ApplyDashboard(gomock.Any(), gomock.Any()).Return(nil)

	cases := []struct {
		testArgs               []string
		skipFlagParsing        bool
		useShortOptionHandling bool
		expectedErr            error
	}{
		{testArgs: []string{"dashboard", "init"}, skipFlagParsing: false, useShortOptionHandling: false, expectedErr: nil},
		{testArgs: []string{"dashboard", "open"}, skipFlagParsing: false, useShortOptionHandling: false, expectedErr: nil},
	}

	for _, c := range cases {

		app := &cli.App{Writer: ioutil.Discard}
		set := flag.NewFlagSet("test", 0)
		_ = set.Parse(c.testArgs)

		context := cli.NewContext(app, set, nil)

		opts := options.NewDaemonOptions()
		opts.Debug = true
		command := newDashboardCommand(fakeKtCli, opts, mockAction)
		err := command.Run(context)

		if c.expectedErr != nil {
			if err.Error() != c.expectedErr.Error() {
				t.Errorf("expected %v but is %v", c.expectedErr, err)
			}
		} else if err != c.expectedErr {
			t.Errorf("expected %v but is %v", c.expectedErr, err)
		}

	}
}