Function TestService_FinalizeStale

coderd/x/chatd/chatdebug/service_test.go:683–729

(t *testing.T)

}

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

	ctx := testutil.Context(t, testutil.WaitLong)
	db, _ := dbtestutil.NewDB(t)
	_, owner, chat, model := seedChat(t, db)
	require.NotEqual(t, uuid.Nil, owner.ID)

	staleTime := time.Now().Add(-10 * time.Minute).UTC().Round(time.Microsecond)
	run, err := db.InsertChatDebugRun(ctx, database.InsertChatDebugRunParams{
		ChatID:        chat.ID,
		ModelConfigID: uuid.NullUUID{UUID: model.ID, Valid: true},
		Kind:          string(chatdebug.KindChatTurn),
		Status:        string(chatdebug.StatusInProgress),
		StartedAt:     sql.NullTime{Time: staleTime, Valid: true},
		UpdatedAt:     sql.NullTime{Time: staleTime, Valid: true},
	})
	require.NoError(t, err)
	step, err := db.InsertChatDebugStep(ctx, database.InsertChatDebugStepParams{
		RunID:      run.ID,
		StepNumber: 1,
		Operation:  string(chatdebug.OperationStream),
		Status:     string(chatdebug.StatusInProgress),
		StartedAt:  sql.NullTime{Time: staleTime, Valid: true},
		UpdatedAt:  sql.NullTime{Time: staleTime, Valid: true},
		ChatID:     chat.ID,
	})
	require.NoError(t, err)

	svc := chatdebug.NewService(db, testutil.Logger(t), nil)
	result, err := svc.FinalizeStale(ctx)
	require.NoError(t, err)
	require.EqualValues(t, 1, result.RunsFinalized)
	require.EqualValues(t, 1, result.StepsFinalized)

	storedRun, err := db.GetChatDebugRunByID(ctx, run.ID)
	require.NoError(t, err)
	require.Equal(t, string(chatdebug.StatusInterrupted), storedRun.Status)
	require.True(t, storedRun.FinishedAt.Valid)

	storedSteps, err := db.GetChatDebugStepsByRunID(ctx, run.ID)
	require.NoError(t, err)
	require.Len(t, storedSteps, 1)
	require.Equal(t, step.ID, storedSteps[0].ID)
	require.Equal(t, string(chatdebug.StatusInterrupted), storedSteps[0].Status)
	require.True(t, storedSteps[0].FinishedAt.Valid)
}

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