Function TestGenerateFromAnthropicMock

coderd/taskname/taskname_internal_test.go:226–291

(t *testing.T)

}

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

	tests := []struct {
		name                string
		responseText        string
		expectedDisplayName string
		expectedNamePrefix  string
	}{
		{
			name:                "BareJSON",
			responseText:        `{"display_name": "Fix bug", "task_name": "fix-bug"}`,
			expectedDisplayName: "Fix bug",
			expectedNamePrefix:  "fix-bug-",
		},
		{
			name:                "FencedJSON",
			responseText:        "```json\n{\"display_name\": \"Debug auth\", \"task_name\": \"debug-auth\"}\n```",
			expectedDisplayName: "Debug auth",
			expectedNamePrefix:  "debug-auth-",
		},
		{
			name:                "FencedNoLanguage",
			responseText:        "```\n{\"display_name\": \"Setup CI\", \"task_name\": \"setup-ci\"}\n```",
			expectedDisplayName: "Setup CI",
			expectedNamePrefix:  "setup-ci-",
		},
		{
			name:                "FencedJSONWithTrailingText",
			responseText:        "```json\n{\"display_name\": \"Debug auth\", \"task_name\": \"debug-auth\"}\n```\n\nDone.",
			expectedDisplayName: "Debug auth",
			expectedNamePrefix:  "debug-auth-",
		},
		{
			name:                "BareJSONWithTrailingFence",
			responseText:        "{\"display_name\": \"Setup CI\", \"task_name\": \"setup-ci\"}\n```",
			expectedDisplayName: "Setup CI",
			expectedNamePrefix:  "setup-ci-",
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
				w.Header().Set("Content-Type", "text/event-stream")
				_, _ = w.Write([]byte(fakeAnthropicSSE(t, tc.responseText)))
			}))
			t.Cleanup(srv.Close)

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

			taskName, err := generateFromAnthropic(
				ctx, "test prompt", "fake-key",
				anthropic.ModelClaudeHaiku4_5,
				anthropicoption.WithBaseURL(srv.URL),
			)