feat: Add password prompt for local activation

open-amt-cloud-toolkit · Sep 6, 2023 · 0ffb8a9 · 0ffb8a9
1 parent 09fdd03
commit 0ffb8a9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/internal/flags/deactivate.go b/internal/flags/deactivate.go
@@ -3,10 +3,16 @@ package flags
 import (
 	"fmt"
 	"rpc/pkg/utils"
+
+	log "github.com/sirupsen/logrus"
 )
 
 func (f *Flags) handleDeactivateCommand() int {
 	f.amtDeactivateCommand.BoolVar(&f.Local, "local", false, "Execute command to AMT directly without cloud interaction")
+
+	if len(f.commandLineArgs) == 5 && f.commandLineArgs[3] == "-password" && !f.UseACM {
+		log.Warn("Password not required for CCM deactivation")
+	}
 	if len(f.commandLineArgs) == 2 {
 		f.amtDeactivateCommand.PrintDefaults()
 		return utils.IncorrectCommandLineParameters

diff --git a/internal/flags/deactivate_test.go b/internal/flags/deactivate_test.go
@@ -75,12 +75,11 @@ func TestHandleDeactivateCommandWithForce(t *testing.T) {
 }
 
 func TestHandleLocalDeactivation(t *testing.T) {
-	args := []string{"./rpc", "deactivate", "-local"}
+	args := []string{"./rpc", "deactivate", "-local", "password"}
 	flags := NewFlags(args)
 	errCode := flags.ParseFlags()
 	assert.Equal(t, errCode, utils.Success)
 }
-
 func TestParseFlagsDeactivate(t *testing.T) {
 	args := []string{"./rpc", "deactivate"}
 	flags := NewFlags(args)