coder/cli/exp_chat_test.go

package cli_test

import (
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/coder/coder/v2/cli/clitest"
)

func TestExpChatContextAdd(t *testing.T) {
	t.Parallel()

	t.Run("RequiresWorkspaceOrDir", func(t *testing.T) {
		t.Parallel()

		inv, _ := clitest.New(t, "exp", "chat", "context", "add")

		err := inv.Run()
		require.Error(t, err)
		require.Contains(t, err.Error(), "this command must be run inside a Coder workspace")
	})

	t.Run("AllowsExplicitDir", func(t *testing.T) {
		t.Parallel()

		inv, _ := clitest.New(t, "exp", "chat", "context", "add", "--dir", t.TempDir())

		err := inv.Run()
		if err != nil {
			require.NotContains(t, err.Error(), "this command must be run inside a Coder workspace")
		}
	})

	t.Run("AllowsWorkspaceEnv", func(t *testing.T) {
		t.Parallel()

		inv, _ := clitest.New(t, "exp", "chat", "context", "add")
		inv.Environ.Set("CODER", "true")

		err := inv.Run()
		if err != nil {
			require.NotContains(t, err.Error(), "this command must be run inside a Coder workspace")
		}
	})
}