feat(cli): add unit tests for the whoami command

BRAVO68WEB · web-flow · commit 26691912fc8a · 2024-07-08T18:09:07.000Z
diff --git a/cli/whoami_test.go b/cli/whoami_test.go
@@ -0,0 +1,37 @@
+package cli_test
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/v2/cli/clitest"
+	"github.com/coder/coder/v2/coderd/coderdtest"
+)
+
+func TestWhoami(t *testing.T) {
+	t.Parallel()
+
+	t.Run("InitialUserNoTTY", func(t *testing.T) {
+		t.Parallel()
+		client := coderdtest.New(t, nil)
+		root, _ := clitest.New(t, "login", client.URL.String())
+		err := root.Run()
+		require.Error(t, err)
+	})
+
+	t.Run("OK", func(t *testing.T) {
+		t.Parallel()
+		client := coderdtest.New(t, nil)
+		_ = coderdtest.CreateFirstUser(t, client)
+		inv, root := clitest.New(t, "whoami")
+		clitest.SetupConfig(t, client, root)
+		buf := new(bytes.Buffer)
+		inv.Stdout = buf
+		err := inv.Run()
+		require.NoError(t, err)
+		whoami := buf.String()
+		require.NotEmpty(t, whoami)
+	})
+}
