fix(core): /retry чистит postmortem-трассы — на новом прогоне постмортем запускается заново
This commit is contained in:
@@ -264,6 +264,26 @@ func TestUpdateTraceOutput(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
// TestDeleteTracesByAgent — удаление трасс по агенту (сброс маркера при /retry).
|
||||
func TestDeleteTracesByAgent(t *testing.T) {
|
||||
s, ctx := setupTestDB(t)
|
||||
task := &Task{ChatID: "tg://del", TaskTag: "delete-by-agent"}
|
||||
id, _ := s.CreateTask(ctx, task)
|
||||
|
||||
_, _ = s.AppendTrace(ctx, &Trace{TaskID: id, Agent: "dev"})
|
||||
_, _ = s.AppendTrace(ctx, &Trace{TaskID: id, Agent: "postmortem"})
|
||||
_, _ = s.AppendTrace(ctx, &Trace{TaskID: id, Agent: "postmortem"})
|
||||
|
||||
if err := s.DeleteTracesByAgent(ctx, id, "postmortem"); err != nil {
|
||||
t.Fatalf("DeleteTracesByAgent: %v", err)
|
||||
}
|
||||
|
||||
traces, _ := s.GetTraces(ctx, id)
|
||||
if len(traces) != 1 || traces[0].Agent != "dev" {
|
||||
t.Fatalf("traces = %d (agent %q), want только dev", len(traces), traces[0].Agent)
|
||||
}
|
||||
}
|
||||
|
||||
// хелперы для проверки классов ошибок
|
||||
func IsNotFound(err error) bool {
|
||||
return errors.Is(err, ErrNotFound)
|
||||
|
||||
Reference in New Issue
Block a user