@@ -29,11 +29,9 @@ import type { TaskFlowRecord } from "./task-flow-registry.types.js";
|
29 | 29 | import { |
30 | 30 | cancelTaskById, |
31 | 31 | createTaskRecord as createTaskRecordOrNull, |
32 | | -findLatestTaskForOwnerKey, |
33 | 32 | findLatestTaskForRelatedSessionKey, |
34 | 33 | findTaskByRunId, |
35 | 34 | getTaskById, |
36 | | -getTaskRegistrySummary, |
37 | 35 | isParentFlowLinkError, |
38 | 36 | listTasksForAgentId, |
39 | 37 | listTasksForOwnerKey, |
@@ -71,6 +69,7 @@ import {
|
71 | 69 | sweepTaskRegistry, |
72 | 70 | } from "./task-registry.maintenance.js"; |
73 | 71 | import { configureTaskRegistryRuntime } from "./task-registry.store.js"; |
| 72 | +import { summarizeTaskRecords } from "./task-registry.summary.js"; |
74 | 73 | import type { TaskDeliveryState, TaskRecord } from "./task-registry.types.js"; |
75 | 74 | |
76 | 75 | const DEFAULT_TASK_RETENTION_MS = 7 * 24 * 60 * 60_000; |
@@ -867,7 +866,7 @@ describe("task-registry", () => {
|
867 | 866 | deliveryStatus: "session_queued", |
868 | 867 | }); |
869 | 868 | |
870 | | -expect(getTaskRegistrySummary()).toEqual({ |
| 869 | +expect(summarizeTaskRecords(listTaskRecords())).toEqual({ |
871 | 870 | total: 3, |
872 | 871 | active: 2, |
873 | 872 | terminal: 1, |
@@ -2205,7 +2204,7 @@ describe("task-registry", () => {
|
2205 | 2204 | }); |
2206 | 2205 | nowSpy.mockRestore(); |
2207 | 2206 | |
2208 | | -expect(findLatestTaskForOwnerKey("agent:main:main")?.taskId).toBe(latest.taskId); |
| 2207 | +expect(listTasksForOwnerKey("agent:main:main")[0]?.taskId).toBe(latest.taskId); |
2209 | 2208 | expect(listTasksForOwnerKey("agent:main:main").map((task) => task.taskId)).toEqual([ |
2210 | 2209 | latest.taskId, |
2211 | 2210 | older.taskId, |
@@ -3237,7 +3236,7 @@ describe("task-registry", () => {
|
3237 | 3236 | startedAt: now - 60_000, |
3238 | 3237 | lastEventAt: now - 60_000, |
3239 | 3238 | }); |
3240 | | -expect(getTaskRegistrySummary().active).toBe(1); |
| 3239 | +expect(summarizeTaskRecords(listTaskRecords()).active).toBe(1); |
3241 | 3240 | |
3242 | 3241 | durableTasks = new Map([ |
3243 | 3242 | [ |
@@ -3268,7 +3267,7 @@ describe("task-registry", () => {
|
3268 | 3267 | taskId: "task-durable", |
3269 | 3268 | status: "cancelled", |
3270 | 3269 | }); |
3271 | | -expect(getTaskRegistrySummary().active).toBe(0); |
| 3270 | +expect(summarizeTaskRecords(listTaskRecords()).active).toBe(0); |
3272 | 3271 | }); |
3273 | 3272 | }); |
3274 | 3273 | |
|