mirror of
https://github.com/coder/coder.git
synced 2026-06-03 04:58:23 +00:00
Danielle Maywood
2521e732be
Instead of generating the fallback task name on the website, we instead generate it on coderd.
57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
package taskname_test
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/v2/coderd/taskname"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/coder/v2/testutil"
|
|
)
|
|
|
|
const (
|
|
anthropicEnvVar = "ANTHROPIC_API_KEY"
|
|
)
|
|
|
|
func TestGenerateFallback(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
name := taskname.GenerateFallback()
|
|
err := codersdk.NameValid(name)
|
|
require.NoErrorf(t, err, "expected fallback to be valid workspace name, instead found %s", name)
|
|
}
|
|
|
|
func TestGenerateTaskName(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("Fallback", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx := testutil.Context(t, testutil.WaitShort)
|
|
|
|
name, err := taskname.Generate(ctx, "Some random prompt")
|
|
require.ErrorIs(t, err, taskname.ErrNoAPIKey)
|
|
require.Equal(t, "", name)
|
|
})
|
|
|
|
t.Run("Anthropic", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
apiKey := os.Getenv(anthropicEnvVar)
|
|
if apiKey == "" {
|
|
t.Skipf("Skipping test as %s not set", anthropicEnvVar)
|
|
}
|
|
|
|
ctx := testutil.Context(t, testutil.WaitShort)
|
|
|
|
name, err := taskname.Generate(ctx, "Create a finance planning app", taskname.WithAPIKey(apiKey))
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, "", name)
|
|
|
|
err = codersdk.NameValid(name)
|
|
require.NoError(t, err, "name should be valid")
|
|
})
|
|
}
|