Function TestService_UpdateStep

coderd/x/chatd/chatdebug/service_test.go:443–497

(t *testing.T)

}

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

	fixture := newFixture(t)
	run := createRun(t, fixture)
	step, err := fixture.svc.CreateStep(fixture.ctx, chatdebug.CreateStepParams{
		RunID:      run.ID,
		ChatID:     fixture.chat.ID,
		StepNumber: 1,
		Operation:  chatdebug.OperationStream,
		Status:     chatdebug.StatusInProgress,
	})
	require.NoError(t, err)

	assistantMsg := insertMessage(t, fixture.db, fixture.chat.ID,
		fixture.owner.ID, fixture.model.ID, database.ChatMessageRoleAssistant,
		"assistant")
	finishedAt := time.Now().UTC().Round(time.Microsecond)
	updated, err := fixture.svc.UpdateStep(fixture.ctx, chatdebug.UpdateStepParams{
		ID:                 step.ID,
		ChatID:             fixture.chat.ID,
		Status:             chatdebug.StatusCompleted,
		AssistantMessageID: assistantMsg.ID,
		NormalizedResponse: map[string]any{"text": "done"},
		Usage:              map[string]any{"input_tokens": 10, "output_tokens": 5},
		Attempts: []chatdebug.Attempt{{
			Number:         1,
			ResponseStatus: 200,
			DurationMs:     25,
		}},
		Metadata:   map[string]any{"provider": fixture.model.Provider},
		FinishedAt: finishedAt,
	})
	require.NoError(t, err)
	require.Equal(t, string(chatdebug.StatusCompleted), updated.Status)
	require.True(t, updated.AssistantMessageID.Valid)
	require.Equal(t, assistantMsg.ID, updated.AssistantMessageID.Int64)
	require.True(t, updated.NormalizedResponse.Valid)
	require.JSONEq(t, `{"text":"done"}`,
		string(updated.NormalizedResponse.RawMessage))
	require.True(t, updated.Usage.Valid)
	require.JSONEq(t, `{"input_tokens":10,"output_tokens":5}`,
		string(updated.Usage.RawMessage))
	require.JSONEq(t,
		`[{"number":1,"response_status":200,"duration_ms":25}]`,
		string(updated.Attempts),
	)
	require.JSONEq(t, `{"provider":"`+fixture.model.Provider+`"}`,
		string(updated.Metadata))
	require.True(t, updated.FinishedAt.Valid)
	storedSteps, err := fixture.db.GetChatDebugStepsByRunID(fixture.ctx, run.ID)
	require.NoError(t, err)
	require.Len(t, storedSteps, 1)
	require.Equal(t, updated.ID, storedSteps[0].ID)
}

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