chore: clitest: fix TestCli

johnstcn · johnstcn · commit de6b5ed56165 · 2022-05-23T16:15:50.000+01:00
diff --git a/cli/clitest/clitest_test.go b/cli/clitest/clitest_test.go
@@ -1,11 +1,8 @@
 package clitest_test
 
 import (
-	"context"
 	"testing"
-	"time"
 
-	"github.com/stretchr/testify/require"
 	"go.uber.org/goleak"
 
 	"github.com/coder/coder/cli/clitest"
@@ -19,20 +16,15 @@ func TestMain(m *testing.M) {
 
 func TestCli(t *testing.T) {
 	t.Parallel()
-	ctx, cancelFunc := context.WithTimeout(context.Background(), 5*time.Second)
-	defer cancelFunc()
 	clitest.CreateTemplateVersionSource(t, nil)
 	client := coderdtest.New(t, nil)
 	cmd, config := clitest.New(t)
 	clitest.SetupConfig(t, client, config)
 	pty := ptytest.New(t)
 	cmd.SetIn(pty.Input())
 	cmd.SetOut(pty.Output())
-	errC := make(chan error)
 	go func() {
-		errC <- cmd.ExecuteContext(ctx)
+		_ = cmd.Execute()
 	}()
 	pty.ExpectMatch("coder")
-	cancelFunc()
-	require.NoError(t, <-errC)
 }
