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 TestGenerate(t *testing.T) {

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

// Ensure no API key in env for this test

t.Setenv("ANTHROPIC_API_KEY", "")

ctx := testutil.Context(t, testutil.WaitShort)

taskName := taskname.Generate(ctx, testutil.Logger(t), "Create a finance planning app")

// Should succeed via prompt sanitization

require.NoError(t, codersdk.NameValid(taskName.Name))

require.Contains(t, taskName.Name, "create-a-finance-planning-")

require.NotEmpty(t, taskName.DisplayName)

require.Equal(t, "Create a finance planning app", taskName.DisplayName)

})

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

apiKey := os.Getenv(anthropicEnvVar)

if apiKey == "" {

t.Skipf("Skipping test as %s not set", anthropicEnvVar)

}

// Set API key for this test

t.Setenv("ANTHROPIC_API_KEY", apiKey)

ctx := testutil.Context(t, testutil.WaitShort)

taskName := taskname.Generate(ctx, testutil.Logger(t), "Create a finance planning app")

// Should succeed with Claude-generated names

require.NoError(t, codersdk.NameValid(taskName.Name))

require.NotEmpty(t, taskName.DisplayName)

})

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

t.Setenv("ANTHROPIC_API_KEY", "")

ctx := testutil.Context(t, testutil.WaitShort)

taskName := taskname.Generate(ctx, testutil.Logger(t), "über cool feature")

require.NoError(t, codersdk.NameValid(taskName.Name))

require.True(t, len(taskName.DisplayName) > 0)

// The display name must start with "Ü", not corrupted bytes.

require.Equal(t, "Über cool feature", taskName.DisplayName)

})

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

// Ensure no API key

t.Setenv("ANTHROPIC_API_KEY", "")

ctx := testutil.Context(t, testutil.WaitShort)

// Use a prompt that can't be sanitized (only special chars)

taskName := taskname.Generate(ctx, testutil.Logger(t), "!@#$%^&*()")

// Should fall back to random name

require.NoError(t, codersdk.NameValid(taskName.Name))

require.NotEmpty(t, taskName.DisplayName)

})

}