Function TestChatDebugStep

coderd/database/db2sdk/db2sdk_test.go:269–333

(t *testing.T)

}

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

	startedAt := time.Now().UTC().Round(time.Second)
	finishedAt := startedAt.Add(2 * time.Second)
	attempts := json.RawMessage(`[
		{
			"attempt_number": 1,
			"status": "completed",
			"raw_request": {"url": "https://example.com"},
			"raw_response": {"status": "200"},
			"duration_ms": 123,
			"started_at": "2026-03-01T10:00:01Z",
			"finished_at": "2026-03-01T10:00:02Z"
		}
	]`)
	step := database.ChatDebugStep{
		ID:                uuid.New(),
		RunID:             uuid.New(),
		ChatID:            uuid.New(),
		StepNumber:        1,
		Operation:         "stream",
		Status:            "completed",
		NormalizedRequest: json.RawMessage(`{"messages":[]}`),
		Attempts:          attempts,
		Metadata:          json.RawMessage(`{"provider":"openai"}`),
		StartedAt:         startedAt,
		UpdatedAt:         finishedAt,
		FinishedAt:        sql.NullTime{Time: finishedAt, Valid: true},
	}

	sdk := db2sdk.ChatDebugStep(step)

	// Verify all scalar fields are mapped correctly.
	require.Equal(t, step.ID, sdk.ID)
	require.Equal(t, step.RunID, sdk.RunID)
	require.Equal(t, step.ChatID, sdk.ChatID)
	require.Equal(t, step.StepNumber, sdk.StepNumber)
	require.Equal(t, codersdk.ChatDebugStepOperationStream, sdk.Operation)
	require.Equal(t, codersdk.ChatDebugStatusCompleted, sdk.Status)
	require.Equal(t, startedAt, sdk.StartedAt)
	require.Equal(t, finishedAt, sdk.UpdatedAt)
	require.Equal(t, &finishedAt, sdk.FinishedAt)

	// Verify JSON object fields are deserialized.
	require.NotNil(t, sdk.NormalizedRequest)
	require.Equal(t, map[string]any{"messages": []any{}}, sdk.NormalizedRequest)
	require.NotNil(t, sdk.Metadata)
	require.Equal(t, map[string]any{"provider": "openai"}, sdk.Metadata)

	// Verify nullable fields are nil when the DB row has NULL values.
	require.Nil(t, sdk.HistoryTipMessageID, "NULL HistoryTipMessageID should map to nil")
	require.Nil(t, sdk.AssistantMessageID, "NULL AssistantMessageID should map to nil")
	require.Nil(t, sdk.NormalizedResponse, "NULL NormalizedResponse should map to nil")
	require.Nil(t, sdk.Usage, "NULL Usage should map to nil")
	require.Nil(t, sdk.Error, "NULL Error should map to nil")

	// Verify attempts are preserved with all fields.