fix: add missing arg length check to fix runtime panic (#568)

* Add missing arg length check to fix runtime panic on "repo branch delete <reponame>" command with missing branch arg * Add testscript case for missing argument --------- Co-authored-by: x2 <[email protected]>
charmbracelet · Sep 9, 2024 · 85b4625 · 85b4625
1 parent c354d5f
commit 85b4625
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pkg/ssh/cmd/branch.go b/pkg/ssh/cmd/branch.go
@@ -148,6 +148,7 @@ func branchDeleteCommand() *cobra.Command {
 		Use:               "delete REPOSITORY BRANCH",
 		Aliases:           []string{"remove", "rm", "del"},
 		Short:             "Delete a branch",
+		Args:              cobra.ExactArgs(2),
 		PersistentPreRunE: checkIfCollab,
 		RunE: func(cmd *cobra.Command, args []string) error {
 			ctx := cmd.Context()

diff --git a/testscript/testdata/repo-delete.txtar b/testscript/testdata/repo-delete.txtar
@@ -10,6 +10,11 @@ soft repo create repo-to-delete
 soft repo delete repo-to-delete
 ! soft repo delete nope
 stderr '.*not found.*'
+
+# missing argument should fail
+! soft repo branch delete repo1
+stderr 'Error.*accepts 2 arg.*'
+
 soft repo list
 stdout 'repo1'