Merge pull request #779 from Ramkishor-Chaladi/newfeature_775

added new feature for vpn disable command to user

Author: allmightyspiff

plugin/commands/user/user.go

@@ -29,6 +29,7 @@ func SetupCobraCommands(sl *metadata.SoftlayerCommand) *cobra.Command {
 	cobraCmd.AddCommand(NewVpnSubnetCommand(sl).Command)
 	cobraCmd.AddCommand(NewVpnManualCommand(sl).Command)
 	cobraCmd.AddCommand(NewVpnPasswordCommand(sl).Command)
+	cobraCmd.AddCommand(NewVpnDisableCommand(sl).Command)
 	cobraCmd.AddCommand(NewVpnEnableCommand(sl).Command)
 	cobraCmd.AddCommand(NewApikeyCommand(sl).Command)
 	return cobraCmd

plugin/commands/user/user_suite_test.go

@@ -35,6 +35,7 @@ var availableCommands = []string{
 	"vpn-manual",
 	"vpn-password",
 	"vpn-subnet",
+	"vpn-disable",
 	"vpn-enable",
 }
 

plugin/commands/user/vpn_disable.go

@@ -0,0 +1,57 @@
+package user
+
+import (
+	"strconv"
+
+	"github.com/softlayer/softlayer-go/datatypes"
+	"github.com/softlayer/softlayer-go/sl"
+	"github.com/spf13/cobra"
+
+	slErrors "github.ibm.com/SoftLayer/softlayer-cli/plugin/errors"
+	. "github.ibm.com/SoftLayer/softlayer-cli/plugin/i18n"
+	"github.ibm.com/SoftLayer/softlayer-cli/plugin/managers"
+	"github.ibm.com/SoftLayer/softlayer-cli/plugin/metadata"
+)
+
+type VpnDisableCommand struct {
+	*metadata.SoftlayerCommand
+	UserManager managers.UserManager
+	Command     *cobra.Command
+}
+
+func NewVpnDisableCommand(sl *metadata.SoftlayerCommand) (cmd *VpnDisableCommand) {
+	thisCmd := &VpnDisableCommand{
+		SoftlayerCommand: sl,
+		UserManager:      managers.NewUserManager(sl.Session),
+	}
+
+	cobraCmd := &cobra.Command{
+		Use:   "vpn-disable " + T("USER_ID"),
+		Short: T("Disable vpn for a user."),
+		Args:  metadata.OneArgs,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return thisCmd.Run(args)
+		},
+	}
+
+	thisCmd.Command = cobraCmd
+	return thisCmd
+}
+
+func (cmd *VpnDisableCommand) Run(args []string) error {
+	userID, err := strconv.Atoi(args[0])
+	if err != nil {
+		return slErrors.NewInvalidSoftlayerIdInputError("User ID")
+	}
+	userTemplate := datatypes.User_Customer{
+		SslVpnAllowedFlag: sl.Bool(false),
+	}
+	success, err := cmd.UserManager.EditUser(userTemplate, userID)
+	if err != nil {
+		return slErrors.NewAPIError(T(""), err.Error(), 2)
+	}
+	if success {
+		cmd.UI.Ok()
+	}
+	return nil
+}

plugin/commands/user/vpn_disable_test.go

@@ -0,0 +1,76 @@
+package user_test
+
+import (
+	"errors"
+
+	"github.com/IBM-Cloud/ibm-cloud-cli-sdk/testhelpers/terminal"
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+	"github.com/softlayer/softlayer-go/datatypes"
+	"github.com/softlayer/softlayer-go/session"
+	"github.com/softlayer/softlayer-go/sl"
+
+	"github.ibm.com/SoftLayer/softlayer-cli/plugin/commands/user"
+	"github.ibm.com/SoftLayer/softlayer-cli/plugin/metadata"
+	"github.ibm.com/SoftLayer/softlayer-cli/plugin/testhelpers"
+)
+
+var _ = Describe("user vpn-enable", func() {
+	var (
+		fakeUI          *terminal.FakeUI
+		cliCommand      *user.VpnDisableCommand
+		fakeSession     *session.Session
+		slCommand       *metadata.SoftlayerCommand
+		fakeUserManager *testhelpers.FakeUserManager
+	)
+	BeforeEach(func() {
+		fakeUI = terminal.NewFakeUI()
+		fakeSession = testhelpers.NewFakeSoftlayerSession([]string{})
+		fakeUserManager = new(testhelpers.FakeUserManager)
+		slCommand = metadata.NewSoftlayerCommand(fakeUI, fakeSession)
+		cliCommand = user.NewVpnDisableCommand(slCommand)
+		cliCommand.Command.PersistentFlags().Var(cliCommand.OutputFlag, "output", "--output=JSON for json output.")
+		cliCommand.UserManager = fakeUserManager
+
+		testUser = datatypes.User_Customer{
+			SslVpnAllowedFlag: sl.Bool(false),
+		}
+	})
+
+	Describe("user vpn-enable", func() {
+
+		Context("Return error", func() {
+			It("Set command without Argument", func() {
+				err := testhelpers.RunCobraCommand(cliCommand.Command)
+				Expect(err).To(HaveOccurred())
+				Expect(err.Error()).To(ContainSubstring("Incorrect Usage: This command requires one argument."))
+			})
+
+			It("Set command with an invalid user Id", func() {
+				err := testhelpers.RunCobraCommand(cliCommand.Command, "abcde")
+				Expect(err).To(HaveOccurred())
+				Expect(err.Error()).To(ContainSubstring("Invalid input for 'User ID'. It must be a positive integer."))
+			})
+		})
+
+		Context("Account cancel-item, softlayer errors", func() {
+			It("Set command with unknow item ID", func() {
+				fakeUserManager.EditUserReturns(true, errors.New("SoftLayer_Exception_ObjectNotFound: Unable to find object with id of '123'. (HTTP 404)"))
+				err := testhelpers.RunCobraCommand(cliCommand.Command, "123")
+				Expect(err).To(HaveOccurred())
+				Expect(err.Error()).To(ContainSubstring("SoftLayer_Exception_ObjectNotFound: Unable to find object with id of '123'. (HTTP 404)"))
+			})
+		})
+
+		Context("Return no error", func() {
+			BeforeEach(func() {
+				fakeUserManager.EditUserReturns(true, nil)
+			})
+			It("enable", func() {
+				err := testhelpers.RunCobraCommand(cliCommand.Command, "111111")
+				Expect(err).NotTo(HaveOccurred())
+				Expect(fakeUI.Outputs()).To(ContainSubstring("OK"))
+			})
+		})
+	})
+})