Add test using cobra cmd

Emyrk · Emyrk · commit ad76f646b570 · 2022-05-20T11:30:21.000-05:00
diff --git a/pty/ptytest/ptytest_test.go b/pty/ptytest/ptytest_test.go
@@ -1,9 +1,11 @@
 package ptytest_test
 
 import (
+	"fmt"
 	"os/exec"
 	"testing"
 
+	"github.com/spf13/cobra"
 	"github.com/stretchr/testify/require"
 
 	"github.com/coder/coder/pty/ptytest"
@@ -18,6 +20,7 @@ func TestPtytest(t *testing.T) {
 		pty.ExpectMatch("write")
 		pty.WriteLine("read")
 	})
+
 	// nolint:paralleltest
 	t.Run("Do not hang on Intel macOS", func(t *testing.T) {
 		cmd := exec.Command("sh", "-c", "echo hi, I will cause a hang")
@@ -27,4 +30,23 @@ func TestPtytest(t *testing.T) {
 		err := cmd.Run()
 		require.NoError(t, err)
 	})
+
+	// nolint:paralleltest
+	t.Run("CobraCommandWorksLinux", func(t *testing.T) {
+		// Example with cobra command instead of exec. More abstractions, but
+		// for some reason works on linux.
+		cmd := cobra.Command{
+			Use: "test",
+			RunE: func(cmd *cobra.Command, args []string) error {
+				fmt.Println("Hello world")
+				return nil
+			},
+		}
+
+		pty := ptytest.New(t)
+		cmd.SetIn(pty.Input())
+		cmd.SetOut(pty.Output())
+		err := cmd.Execute()
+		require.NoError(t, err)
+	})
 }
