added new feature vpn enable command to user

Author: Ramkishor Chaladi

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(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-enable",
 }
 
 // This test suite exists to make sure commands don't get accidently removed from the SetupCobraCommands

plugin/commands/user/vpn_enable.go

@@ -0,0 +1,72 @@
+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 VpnEnableCommand struct {
+	*metadata.SoftlayerCommand
+	UserManager managers.UserManager
+	Command     *cobra.Command
+}
+
+func NewVpnEnableCommand(sl *metadata.SoftlayerCommand) (cmd *VpnEnableCommand) {
+	thisCmd := &VpnEnableCommand{
+		SoftlayerCommand: sl,
+		UserManager:      managers.NewUserManager(sl.Session),
+	}
+
+	cobraCmd := &cobra.Command{
+		Use:   "vpn-enable " + T("USER_ID"),
+		Short: T("Enable 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 *VpnEnableCommand) 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(true),
+	}
+
+	subs := map[string]interface{}{"UserId": userID}
+	confirm, err := cmd.UI.Confirm(T("This will Enable vpn for the user id: {{.UserId}}. Continue?", subs))
+	if err != nil {
+		return err
+	}
+	if !confirm {
+		cmd.UI.Print(T("Aborted."))
+		return nil
+	}
+
+	success, err := cmd.UserManager.EditUser(userTemplate, userID)
+	if err != nil {
+		return slErrors.NewAPIError(T("Failed to enable vpn for the 'User ID' \n", subs), err.Error(), 2)
+	}
+
+	if success {
+		cmd.UI.Ok()
+		cmd.UI.Print(T("{{.UserId}} is successfully enabled.", subs))
+	}
+	return nil
+}

plugin/commands/user/vpn_enable_test.go

@@ -0,0 +1,84 @@
+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/session"
+
+	"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.VpnEnableCommand
+		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.NewVpnEnableCommand(slCommand)
+		cliCommand.Command.PersistentFlags().Var(cliCommand.OutputFlag, "output", "--output=JSON for json output.")
+		cliCommand.UserManager = fakeUserManager
+	})
+
+	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("Vpn enable with correct vpn ID but continue", func() {
+			It("return no error", func() {
+				fakeUI.Inputs("No")
+				err := testhelpers.RunCobraCommand(cliCommand.Command, "1234")
+				Expect(err).NotTo(HaveOccurred())
+				Expect(fakeUI.Outputs()).To(ContainSubstring("This will Enable vpn for the user id: 1234. Continue?"))
+				Expect(fakeUI.Outputs()).To(ContainSubstring("Aborted."))
+			})
+		})
+
+		Context("Return error", func() {
+			BeforeEach(func() {
+				fakeUserManager.EditUserReturns(false, errors.New("Failed"))
+			})
+			It("Failed edit user", func() {
+				err := testhelpers.RunCobraCommand(cliCommand.Command, "111111")
+				Expect(err).To(HaveOccurred())
+				Expect(err.Error()).To(ContainSubstring("Failed to enable vpn for the 'User ID' "))
+				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("vpn enable", func() {
+				err := testhelpers.RunCobraCommand(cliCommand.Command, "111111")
+				Expect(err).NotTo(HaveOccurred())
+				Expect(fakeUI.Outputs()).To(ContainSubstring("OK"))
+				Expect(fakeUI.Outputs()).To(ContainSubstring("1234 is successfully enabled."))
+			})
+		})
+	})
+})