test both confirmation of delete and --yes with no confirmation

ketang · ketang · commit bdeec4bb422b · 2022-07-10T15:21:15.000-05:00
diff --git a/cli/templatedelete_test.go b/cli/templatedelete_test.go
@@ -2,11 +2,14 @@ package cli_test
 
 import (
 	"context"
+	"fmt"
+	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/require"
 
 	"github.com/coder/coder/cli/clitest"
+	"github.com/coder/coder/cli/cliui"
 	"github.com/coder/coder/coderd/coderdtest"
 	"github.com/coder/coder/codersdk"
 	"github.com/coder/coder/pty/ptytest"
@@ -32,7 +35,7 @@ func TestTemplateDelete(t *testing.T) {
 		require.Error(t, err, "template should not exist")
 	})
 
-	t.Run("Multiple", func(t *testing.T) {
+	t.Run("Multiple --yes", func(t *testing.T) {
 		t.Parallel()
 
 		client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerD: true})
@@ -49,7 +52,7 @@ func TestTemplateDelete(t *testing.T) {
 			templateNames = append(templateNames, template.Name)
 		}
 
-		cmd, root := clitest.New(t, append([]string{"templates", "delete"}, templateNames...)...)
+		cmd, root := clitest.New(t, append([]string{"templates", "delete", "--yes"}, templateNames...)...)
 		clitest.SetupConfig(t, client, root)
 		require.NoError(t, cmd.Execute())
 
@@ -59,6 +62,45 @@ func TestTemplateDelete(t *testing.T) {
 		}
 	})
 
+	t.Run("Multiple prompted", func(t *testing.T) {
+		t.Parallel()
+
+		client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerD: true})
+		user := coderdtest.CreateFirstUser(t, client)
+		version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil)
+		_ = coderdtest.AwaitTemplateVersionJob(t, client, version.ID)
+		templates := []codersdk.Template{
+			coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID),
+			coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID),
+			coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID),
+		}
+		templateNames := []string{}
+		for _, template := range templates {
+			templateNames = append(templateNames, template.Name)
+		}
+
+		cmd, root := clitest.New(t, append([]string{"templates", "delete"}, templateNames...)...)
+		clitest.SetupConfig(t, client, root)
+		pty := ptytest.New(t)
+		cmd.SetIn(pty.Input())
+		cmd.SetOut(pty.Output())
+
+		execDone := make(chan error)
+		go func() {
+			execDone <- cmd.Execute()
+		}()
+
+		pty.ExpectMatch(fmt.Sprintf("Delete these templates: %s?", cliui.Styles.Code.Render(strings.Join(templateNames, ", "))))
+		pty.WriteLine("yes")
+
+		for _, template := range templates {
+			_, err := client.Template(context.Background(), template.ID)
+			require.Error(t, err, "template should not exist")
+		}
+
+		require.NoError(t, <-execDone)
+	})
+
 	t.Run("Selector", func(t *testing.T) {
 		t.Parallel()
 
