(t *testing.T)
| 275 | } |
| 276 | |
| 277 | func TestService_CreateStep(t *testing.T) { |
| 278 | t.Parallel() |
| 279 | |
| 280 | fixture := newFixture(t) |
| 281 | run := createRun(t, fixture) |
| 282 | historyTipMsg := insertMessage(t, fixture.db, fixture.chat.ID, |
| 283 | fixture.owner.ID, fixture.model.ID, database.ChatMessageRoleAssistant, |
| 284 | "history-tip") |
| 285 | |
| 286 | step, err := fixture.svc.CreateStep(fixture.ctx, chatdebug.CreateStepParams{ |
| 287 | RunID: run.ID, |
| 288 | ChatID: fixture.chat.ID, |
| 289 | StepNumber: 1, |
| 290 | Operation: chatdebug.OperationStream, |
| 291 | Status: chatdebug.StatusInProgress, |
| 292 | HistoryTipMessageID: historyTipMsg.ID, |
| 293 | NormalizedRequest: map[string]any{ |
| 294 | "messages": []string{"hello"}, |
| 295 | }, |
| 296 | }) |
| 297 | require.NoError(t, err) |
| 298 | require.Equal(t, fixture.chat.ID, step.ChatID) |
| 299 | require.Equal(t, run.ID, step.RunID) |
| 300 | require.EqualValues(t, 1, step.StepNumber) |
| 301 | require.Equal(t, string(chatdebug.OperationStream), step.Operation) |
| 302 | require.Equal(t, string(chatdebug.StatusInProgress), step.Status) |
| 303 | require.True(t, step.HistoryTipMessageID.Valid) |
| 304 | require.Equal(t, historyTipMsg.ID, step.HistoryTipMessageID.Int64) |
| 305 | require.JSONEq(t, `{"messages":["hello"]}`, string(step.NormalizedRequest)) |
| 306 | |
| 307 | steps, err := fixture.db.GetChatDebugStepsByRunID(fixture.ctx, run.ID) |
| 308 | require.NoError(t, err) |
| 309 | require.Len(t, steps, 1) |
| 310 | require.Equal(t, step.ID, steps[0].ID) |
| 311 | } |
| 312 | |
| 313 | func TestService_CreateStep_RetriesDuplicateStepNumbers(t *testing.T) { |
| 314 | t.Parallel() |
nothing calls this directly
no test coverage detected