@herdctl/core 0.0.1 → 0.0.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (275) hide show
  1. package/dist/config/__tests__/agent.test.js +31 -13
  2. package/dist/config/__tests__/agent.test.js.map +1 -1
  3. package/dist/config/__tests__/merge.test.js +9 -2
  4. package/dist/config/__tests__/merge.test.js.map +1 -1
  5. package/dist/config/__tests__/schema.test.js +350 -1
  6. package/dist/config/__tests__/schema.test.js.map +1 -1
  7. package/dist/config/index.d.ts +1 -1
  8. package/dist/config/index.d.ts.map +1 -1
  9. package/dist/config/index.js +3 -1
  10. package/dist/config/index.js.map +1 -1
  11. package/dist/config/schema.d.ts +828 -24
  12. package/dist/config/schema.d.ts.map +1 -1
  13. package/dist/config/schema.js +118 -6
  14. package/dist/config/schema.js.map +1 -1
  15. package/dist/fleet-manager/__tests__/coverage.test.js +11 -332
  16. package/dist/fleet-manager/__tests__/coverage.test.js.map +1 -1
  17. package/dist/fleet-manager/__tests__/errors.test.js +1 -49
  18. package/dist/fleet-manager/__tests__/errors.test.js.map +1 -1
  19. package/dist/fleet-manager/__tests__/integration.test.js +109 -0
  20. package/dist/fleet-manager/__tests__/integration.test.js.map +1 -1
  21. package/dist/fleet-manager/__tests__/reload.test.js +1 -1
  22. package/dist/fleet-manager/__tests__/reload.test.js.map +1 -1
  23. package/dist/fleet-manager/config-reload.d.ts +164 -0
  24. package/dist/fleet-manager/config-reload.d.ts.map +1 -0
  25. package/dist/fleet-manager/config-reload.js +445 -0
  26. package/dist/fleet-manager/config-reload.js.map +1 -0
  27. package/dist/fleet-manager/context.d.ts +76 -0
  28. package/dist/fleet-manager/context.d.ts.map +1 -0
  29. package/dist/fleet-manager/context.js +11 -0
  30. package/dist/fleet-manager/context.js.map +1 -0
  31. package/dist/fleet-manager/errors.d.ts +0 -25
  32. package/dist/fleet-manager/errors.d.ts.map +1 -1
  33. package/dist/fleet-manager/errors.js +0 -38
  34. package/dist/fleet-manager/errors.js.map +1 -1
  35. package/dist/fleet-manager/event-emitters.d.ts +123 -0
  36. package/dist/fleet-manager/event-emitters.d.ts.map +1 -0
  37. package/dist/fleet-manager/event-emitters.js +136 -0
  38. package/dist/fleet-manager/event-emitters.js.map +1 -0
  39. package/dist/fleet-manager/event-types.d.ts +0 -15
  40. package/dist/fleet-manager/event-types.d.ts.map +1 -1
  41. package/dist/fleet-manager/fleet-manager.d.ts +40 -653
  42. package/dist/fleet-manager/fleet-manager.d.ts.map +1 -1
  43. package/dist/fleet-manager/fleet-manager.js +95 -1720
  44. package/dist/fleet-manager/fleet-manager.js.map +1 -1
  45. package/dist/fleet-manager/index.d.ts +13 -2
  46. package/dist/fleet-manager/index.d.ts.map +1 -1
  47. package/dist/fleet-manager/index.js +19 -6
  48. package/dist/fleet-manager/index.js.map +1 -1
  49. package/dist/fleet-manager/job-control.d.ts +64 -0
  50. package/dist/fleet-manager/job-control.d.ts.map +1 -0
  51. package/dist/fleet-manager/job-control.js +296 -0
  52. package/dist/fleet-manager/job-control.js.map +1 -0
  53. package/dist/fleet-manager/log-streaming.d.ts +171 -0
  54. package/dist/fleet-manager/log-streaming.d.ts.map +1 -0
  55. package/dist/fleet-manager/log-streaming.js +503 -0
  56. package/dist/fleet-manager/log-streaming.js.map +1 -0
  57. package/dist/fleet-manager/schedule-executor.d.ts +63 -0
  58. package/dist/fleet-manager/schedule-executor.d.ts.map +1 -0
  59. package/dist/fleet-manager/schedule-executor.js +209 -0
  60. package/dist/fleet-manager/schedule-executor.js.map +1 -0
  61. package/dist/fleet-manager/schedule-management.d.ts +71 -0
  62. package/dist/fleet-manager/schedule-management.d.ts.map +1 -0
  63. package/dist/fleet-manager/schedule-management.js +171 -0
  64. package/dist/fleet-manager/schedule-management.js.map +1 -0
  65. package/dist/fleet-manager/status-queries.d.ts +105 -0
  66. package/dist/fleet-manager/status-queries.d.ts.map +1 -0
  67. package/dist/fleet-manager/status-queries.js +247 -0
  68. package/dist/fleet-manager/status-queries.js.map +1 -0
  69. package/dist/fleet-manager/types.d.ts +0 -39
  70. package/dist/fleet-manager/types.d.ts.map +1 -1
  71. package/dist/runner/__tests__/job-executor.test.js +206 -1
  72. package/dist/runner/__tests__/job-executor.test.js.map +1 -1
  73. package/dist/runner/job-executor.d.ts +9 -0
  74. package/dist/runner/job-executor.d.ts.map +1 -1
  75. package/dist/runner/job-executor.js +78 -4
  76. package/dist/runner/job-executor.js.map +1 -1
  77. package/dist/runner/types.d.ts +2 -0
  78. package/dist/runner/types.d.ts.map +1 -1
  79. package/dist/scheduler/__tests__/cron.test.d.ts +2 -0
  80. package/dist/scheduler/__tests__/cron.test.d.ts.map +1 -0
  81. package/dist/scheduler/__tests__/cron.test.js +867 -0
  82. package/dist/scheduler/__tests__/cron.test.js.map +1 -0
  83. package/dist/scheduler/__tests__/scheduler.test.js +164 -5
  84. package/dist/scheduler/__tests__/scheduler.test.js.map +1 -1
  85. package/dist/scheduler/cron.d.ts +126 -0
  86. package/dist/scheduler/cron.d.ts.map +1 -0
  87. package/dist/scheduler/cron.js +390 -0
  88. package/dist/scheduler/cron.js.map +1 -0
  89. package/dist/scheduler/errors.d.ts +81 -1
  90. package/dist/scheduler/errors.d.ts.map +1 -1
  91. package/dist/scheduler/errors.js +81 -6
  92. package/dist/scheduler/errors.js.map +1 -1
  93. package/dist/scheduler/index.d.ts +1 -0
  94. package/dist/scheduler/index.d.ts.map +1 -1
  95. package/dist/scheduler/index.js +2 -0
  96. package/dist/scheduler/index.js.map +1 -1
  97. package/dist/scheduler/schedule-runner.d.ts +2 -2
  98. package/dist/scheduler/schedule-runner.d.ts.map +1 -1
  99. package/dist/scheduler/schedule-runner.js +20 -8
  100. package/dist/scheduler/schedule-runner.js.map +1 -1
  101. package/dist/scheduler/scheduler.d.ts +4 -4
  102. package/dist/scheduler/scheduler.d.ts.map +1 -1
  103. package/dist/scheduler/scheduler.js +86 -20
  104. package/dist/scheduler/scheduler.js.map +1 -1
  105. package/dist/scheduler/types.d.ts +1 -1
  106. package/dist/scheduler/types.d.ts.map +1 -1
  107. package/dist/state/schemas/job-metadata.d.ts +2 -2
  108. package/package.json +33 -8
  109. package/.turbo/turbo-build.log +0 -4
  110. package/.turbo/turbo-test.log +0 -219
  111. package/.turbo/turbo-typecheck.log +0 -4
  112. package/coverage/base.css +0 -224
  113. package/coverage/block-navigation.js +0 -87
  114. package/coverage/coverage-final.json +0 -51
  115. package/coverage/favicon.png +0 -0
  116. package/coverage/index.html +0 -251
  117. package/coverage/prettify.css +0 -1
  118. package/coverage/prettify.js +0 -2
  119. package/coverage/sort-arrow-sprite.png +0 -0
  120. package/coverage/sorter.js +0 -210
  121. package/coverage/src/config/index.html +0 -191
  122. package/coverage/src/config/index.ts.html +0 -442
  123. package/coverage/src/config/interpolate.ts.html +0 -652
  124. package/coverage/src/config/loader.ts.html +0 -1501
  125. package/coverage/src/config/merge.ts.html +0 -823
  126. package/coverage/src/config/parser.ts.html +0 -1213
  127. package/coverage/src/config/schema.ts.html +0 -1123
  128. package/coverage/src/fleet-manager/errors.ts.html +0 -2326
  129. package/coverage/src/fleet-manager/event-types.ts.html +0 -1219
  130. package/coverage/src/fleet-manager/fleet-manager.ts.html +0 -7030
  131. package/coverage/src/fleet-manager/index.html +0 -206
  132. package/coverage/src/fleet-manager/index.ts.html +0 -469
  133. package/coverage/src/fleet-manager/job-manager.ts.html +0 -2074
  134. package/coverage/src/fleet-manager/job-queue.ts.html +0 -2479
  135. package/coverage/src/fleet-manager/types.ts.html +0 -2602
  136. package/coverage/src/index.html +0 -116
  137. package/coverage/src/index.ts.html +0 -181
  138. package/coverage/src/runner/errors.ts.html +0 -1006
  139. package/coverage/src/runner/index.html +0 -191
  140. package/coverage/src/runner/index.ts.html +0 -256
  141. package/coverage/src/runner/job-executor.ts.html +0 -1429
  142. package/coverage/src/runner/message-processor.ts.html +0 -1150
  143. package/coverage/src/runner/sdk-adapter.ts.html +0 -658
  144. package/coverage/src/runner/types.ts.html +0 -559
  145. package/coverage/src/scheduler/errors.ts.html +0 -388
  146. package/coverage/src/scheduler/index.html +0 -206
  147. package/coverage/src/scheduler/index.ts.html +0 -244
  148. package/coverage/src/scheduler/interval.ts.html +0 -652
  149. package/coverage/src/scheduler/schedule-runner.ts.html +0 -1411
  150. package/coverage/src/scheduler/schedule-state.ts.html +0 -718
  151. package/coverage/src/scheduler/scheduler.ts.html +0 -1795
  152. package/coverage/src/scheduler/types.ts.html +0 -733
  153. package/coverage/src/state/directory.ts.html +0 -736
  154. package/coverage/src/state/errors.ts.html +0 -376
  155. package/coverage/src/state/fleet-state.ts.html +0 -937
  156. package/coverage/src/state/index.html +0 -221
  157. package/coverage/src/state/index.ts.html +0 -322
  158. package/coverage/src/state/job-metadata.ts.html +0 -1420
  159. package/coverage/src/state/job-output.ts.html +0 -1033
  160. package/coverage/src/state/schemas/fleet-state.ts.html +0 -445
  161. package/coverage/src/state/schemas/index.html +0 -176
  162. package/coverage/src/state/schemas/index.ts.html +0 -286
  163. package/coverage/src/state/schemas/job-metadata.ts.html +0 -628
  164. package/coverage/src/state/schemas/job-output.ts.html +0 -616
  165. package/coverage/src/state/schemas/session-info.ts.html +0 -361
  166. package/coverage/src/state/session.ts.html +0 -844
  167. package/coverage/src/state/types.ts.html +0 -262
  168. package/coverage/src/state/utils/atomic.ts.html +0 -748
  169. package/coverage/src/state/utils/index.html +0 -146
  170. package/coverage/src/state/utils/index.ts.html +0 -103
  171. package/coverage/src/state/utils/reads.ts.html +0 -1621
  172. package/coverage/src/work-sources/adapters/github.ts.html +0 -3583
  173. package/coverage/src/work-sources/adapters/index.html +0 -131
  174. package/coverage/src/work-sources/adapters/index.ts.html +0 -277
  175. package/coverage/src/work-sources/errors.ts.html +0 -298
  176. package/coverage/src/work-sources/index.html +0 -176
  177. package/coverage/src/work-sources/index.ts.html +0 -529
  178. package/coverage/src/work-sources/manager.ts.html +0 -1324
  179. package/coverage/src/work-sources/registry.ts.html +0 -619
  180. package/coverage/src/work-sources/types.ts.html +0 -568
  181. package/dist/fleet-manager/__tests__/event-helpers.test.d.ts +0 -7
  182. package/dist/fleet-manager/__tests__/event-helpers.test.d.ts.map +0 -1
  183. package/dist/fleet-manager/__tests__/event-helpers.test.js +0 -368
  184. package/dist/fleet-manager/__tests__/event-helpers.test.js.map +0 -1
  185. package/src/config/__tests__/agent.test.ts +0 -864
  186. package/src/config/__tests__/interpolate.test.ts +0 -644
  187. package/src/config/__tests__/loader.test.ts +0 -784
  188. package/src/config/__tests__/merge.test.ts +0 -751
  189. package/src/config/__tests__/parser.test.ts +0 -533
  190. package/src/config/__tests__/schema.test.ts +0 -873
  191. package/src/config/index.ts +0 -119
  192. package/src/config/interpolate.ts +0 -189
  193. package/src/config/loader.ts +0 -472
  194. package/src/config/merge.ts +0 -246
  195. package/src/config/parser.ts +0 -376
  196. package/src/config/schema.ts +0 -346
  197. package/src/fleet-manager/__tests__/coverage.test.ts +0 -2869
  198. package/src/fleet-manager/__tests__/errors.test.ts +0 -660
  199. package/src/fleet-manager/__tests__/event-helpers.test.ts +0 -448
  200. package/src/fleet-manager/__tests__/integration.test.ts +0 -1209
  201. package/src/fleet-manager/__tests__/job-control.test.ts +0 -283
  202. package/src/fleet-manager/__tests__/job-manager.test.ts +0 -869
  203. package/src/fleet-manager/__tests__/job-queue.test.ts +0 -401
  204. package/src/fleet-manager/__tests__/reload.test.ts +0 -751
  205. package/src/fleet-manager/__tests__/status-queries.test.ts +0 -595
  206. package/src/fleet-manager/__tests__/trigger.test.ts +0 -601
  207. package/src/fleet-manager/errors.ts +0 -747
  208. package/src/fleet-manager/event-types.ts +0 -378
  209. package/src/fleet-manager/fleet-manager.ts +0 -2315
  210. package/src/fleet-manager/index.ts +0 -128
  211. package/src/fleet-manager/job-manager.ts +0 -663
  212. package/src/fleet-manager/job-queue.ts +0 -798
  213. package/src/fleet-manager/types.ts +0 -839
  214. package/src/index.ts +0 -32
  215. package/src/runner/__tests__/errors.test.ts +0 -382
  216. package/src/runner/__tests__/job-executor.test.ts +0 -1708
  217. package/src/runner/__tests__/message-processor.test.ts +0 -960
  218. package/src/runner/__tests__/sdk-adapter.test.ts +0 -626
  219. package/src/runner/errors.ts +0 -307
  220. package/src/runner/index.ts +0 -57
  221. package/src/runner/job-executor.ts +0 -448
  222. package/src/runner/message-processor.ts +0 -355
  223. package/src/runner/sdk-adapter.ts +0 -191
  224. package/src/runner/types.ts +0 -158
  225. package/src/scheduler/__tests__/errors.test.ts +0 -159
  226. package/src/scheduler/__tests__/interval.test.ts +0 -515
  227. package/src/scheduler/__tests__/schedule-runner.test.ts +0 -798
  228. package/src/scheduler/__tests__/schedule-state.test.ts +0 -671
  229. package/src/scheduler/__tests__/scheduler.test.ts +0 -1280
  230. package/src/scheduler/errors.ts +0 -101
  231. package/src/scheduler/index.ts +0 -53
  232. package/src/scheduler/interval.ts +0 -189
  233. package/src/scheduler/schedule-runner.ts +0 -442
  234. package/src/scheduler/schedule-state.ts +0 -211
  235. package/src/scheduler/scheduler.ts +0 -570
  236. package/src/scheduler/types.ts +0 -216
  237. package/src/state/__tests__/directory.test.ts +0 -595
  238. package/src/state/__tests__/fleet-state.test.ts +0 -868
  239. package/src/state/__tests__/job-metadata-schema.test.ts +0 -414
  240. package/src/state/__tests__/job-metadata.test.ts +0 -831
  241. package/src/state/__tests__/job-output.test.ts +0 -856
  242. package/src/state/__tests__/session-schema.test.ts +0 -378
  243. package/src/state/__tests__/session.test.ts +0 -604
  244. package/src/state/directory.ts +0 -217
  245. package/src/state/errors.ts +0 -97
  246. package/src/state/fleet-state.ts +0 -284
  247. package/src/state/index.ts +0 -79
  248. package/src/state/job-metadata.ts +0 -445
  249. package/src/state/job-output.ts +0 -316
  250. package/src/state/schemas/__tests__/job-output.test.ts +0 -338
  251. package/src/state/schemas/fleet-state.ts +0 -120
  252. package/src/state/schemas/index.ts +0 -67
  253. package/src/state/schemas/job-metadata.ts +0 -181
  254. package/src/state/schemas/job-output.ts +0 -177
  255. package/src/state/schemas/session-info.ts +0 -92
  256. package/src/state/session.ts +0 -253
  257. package/src/state/types.ts +0 -59
  258. package/src/state/utils/__tests__/atomic.test.ts +0 -723
  259. package/src/state/utils/__tests__/reads.test.ts +0 -1071
  260. package/src/state/utils/atomic.ts +0 -221
  261. package/src/state/utils/index.ts +0 -6
  262. package/src/state/utils/reads.ts +0 -512
  263. package/src/work-sources/__tests__/github.test.ts +0 -1800
  264. package/src/work-sources/__tests__/manager.test.ts +0 -529
  265. package/src/work-sources/__tests__/registry.test.ts +0 -477
  266. package/src/work-sources/__tests__/types.test.ts +0 -479
  267. package/src/work-sources/adapters/github.ts +0 -1166
  268. package/src/work-sources/adapters/index.ts +0 -64
  269. package/src/work-sources/errors.ts +0 -71
  270. package/src/work-sources/index.ts +0 -148
  271. package/src/work-sources/manager.ts +0 -413
  272. package/src/work-sources/registry.ts +0 -178
  273. package/src/work-sources/types.ts +0 -161
  274. package/tsconfig.json +0 -9
  275. package/vitest.config.ts +0 -19
@@ -1,798 +0,0 @@
1
- import {
2
- describe,
3
- it,
4
- expect,
5
- beforeEach,
6
- afterEach,
7
- vi,
8
- } from "vitest";
9
- import { mkdir, rm, realpath } from "node:fs/promises";
10
- import { join } from "node:path";
11
- import { tmpdir } from "node:os";
12
- import {
13
- runSchedule,
14
- buildSchedulePrompt,
15
- type RunScheduleOptions,
16
- type ScheduleRunnerLogger,
17
- } from "../schedule-runner.js";
18
- import type { ResolvedAgent, Schedule } from "../../config/index.js";
19
- import type { ScheduleState } from "../../state/schemas/fleet-state.js";
20
- import type { SDKQueryFunction, SDKMessage } from "../../runner/index.js";
21
- import type {
22
- WorkSourceManager,
23
- WorkItem,
24
- GetNextWorkItemResult,
25
- WorkResult,
26
- } from "../../work-sources/index.js";
27
- import { readFleetState } from "../../state/fleet-state.js";
28
-
29
- // Helper to create a temp directory
30
- async function createTempDir(): Promise<string> {
31
- const baseDir = join(
32
- tmpdir(),
33
- `herdctl-schedule-runner-test-${Date.now()}-${Math.random().toString(36).slice(2)}`
34
- );
35
- await mkdir(baseDir, { recursive: true });
36
- return await realpath(baseDir);
37
- }
38
-
39
- // Helper to create a mock logger
40
- function createMockLogger(): ScheduleRunnerLogger & {
41
- debugs: string[];
42
- infos: string[];
43
- warnings: string[];
44
- errors: string[];
45
- } {
46
- const debugs: string[] = [];
47
- const infos: string[] = [];
48
- const warnings: string[] = [];
49
- const errors: string[] = [];
50
- return {
51
- debugs,
52
- infos,
53
- warnings,
54
- errors,
55
- debug: (message: string) => debugs.push(message),
56
- info: (message: string) => infos.push(message),
57
- warn: (message: string) => warnings.push(message),
58
- error: (message: string) => errors.push(message),
59
- };
60
- }
61
-
62
- // Helper to create a test agent
63
- function createTestAgent(
64
- name: string,
65
- overrides?: Partial<ResolvedAgent>
66
- ): ResolvedAgent {
67
- return {
68
- name,
69
- configPath: `/fake/path/${name}.yaml`,
70
- ...overrides,
71
- } as ResolvedAgent;
72
- }
73
-
74
- // Helper to create a test schedule
75
- function createTestSchedule(overrides?: Partial<Schedule>): Schedule {
76
- return {
77
- type: "interval",
78
- interval: "1h",
79
- prompt: "Default test prompt",
80
- ...overrides,
81
- } as Schedule;
82
- }
83
-
84
- // Helper to create a test schedule state
85
- function createTestScheduleState(overrides?: Partial<ScheduleState>): ScheduleState {
86
- return {
87
- status: "idle",
88
- last_run_at: null,
89
- next_run_at: null,
90
- last_error: null,
91
- ...overrides,
92
- };
93
- }
94
-
95
- // Helper to create a test work item
96
- function createTestWorkItem(overrides?: Partial<WorkItem>): WorkItem {
97
- return {
98
- id: "github:123",
99
- source: "github",
100
- externalId: "123",
101
- title: "Fix authentication bug",
102
- description: "Users are getting logged out unexpectedly",
103
- priority: "high",
104
- labels: ["bug", "auth"],
105
- metadata: {},
106
- url: "https://github.com/org/repo/issues/123",
107
- createdAt: new Date("2024-01-01"),
108
- updatedAt: new Date("2024-01-02"),
109
- ...overrides,
110
- };
111
- }
112
-
113
- // Helper to create a mock SDK query function
114
- function createMockSDKQuery(
115
- messages: SDKMessage[] = []
116
- ): SDKQueryFunction {
117
- return async function* mockQuery() {
118
- // Emit init message with session_id
119
- yield {
120
- type: "system" as const,
121
- subtype: "init",
122
- session_id: "test-session-123",
123
- };
124
-
125
- // Emit any provided messages
126
- for (const message of messages) {
127
- yield message;
128
- }
129
-
130
- // Emit result message
131
- yield {
132
- type: "assistant" as const,
133
- content: "Task completed successfully",
134
- };
135
- };
136
- }
137
-
138
- // Helper to create a mock work source manager
139
- function createMockWorkSourceManager(
140
- nextWorkResult?: GetNextWorkItemResult
141
- ): WorkSourceManager & {
142
- getNextWorkItemCalls: Array<{ agent: ResolvedAgent }>;
143
- reportOutcomeCalls: Array<{ taskId: string; result: WorkResult }>;
144
- releaseWorkItemCalls: Array<{ taskId: string; reason?: string }>;
145
- } {
146
- const getNextWorkItemCalls: Array<{ agent: ResolvedAgent }> = [];
147
- const reportOutcomeCalls: Array<{ taskId: string; result: WorkResult }> = [];
148
- const releaseWorkItemCalls: Array<{ taskId: string; reason?: string }> = [];
149
-
150
- return {
151
- getNextWorkItemCalls,
152
- reportOutcomeCalls,
153
- releaseWorkItemCalls,
154
- getNextWorkItem: vi.fn(async (agent) => {
155
- getNextWorkItemCalls.push({ agent });
156
- return nextWorkResult ?? { item: null, claimed: false };
157
- }),
158
- reportOutcome: vi.fn(async (taskId, result) => {
159
- reportOutcomeCalls.push({ taskId, result });
160
- }),
161
- releaseWorkItem: vi.fn(async (taskId, options) => {
162
- releaseWorkItemCalls.push({ taskId, reason: options?.reason });
163
- return { success: true };
164
- }),
165
- getAdapter: vi.fn(async () => null),
166
- clearCache: vi.fn(),
167
- };
168
- }
169
-
170
- describe("buildSchedulePrompt", () => {
171
- describe("without work item", () => {
172
- it("returns schedule prompt when configured", () => {
173
- const schedule = createTestSchedule({ prompt: "Check for updates" });
174
- const result = buildSchedulePrompt(schedule);
175
- expect(result).toBe("Check for updates");
176
- });
177
-
178
- it("returns default prompt when no schedule prompt", () => {
179
- const schedule = createTestSchedule({ prompt: undefined });
180
- const result = buildSchedulePrompt(schedule);
181
- expect(result).toBe("Execute scheduled task.");
182
- });
183
- });
184
-
185
- describe("with work item", () => {
186
- it("combines schedule prompt and work item", () => {
187
- const schedule = createTestSchedule({ prompt: "Process this issue:" });
188
- const workItem = createTestWorkItem();
189
-
190
- const result = buildSchedulePrompt(schedule, workItem);
191
-
192
- expect(result).toContain("Process this issue:");
193
- expect(result).toContain("## Work Item: Fix authentication bug");
194
- expect(result).toContain("Users are getting logged out unexpectedly");
195
- expect(result).toContain("**Source:** github");
196
- expect(result).toContain("**Priority:** high");
197
- expect(result).toContain("**Labels:** bug, auth");
198
- });
199
-
200
- it("works with work item alone (no schedule prompt)", () => {
201
- const schedule = createTestSchedule({ prompt: undefined });
202
- const workItem = createTestWorkItem({
203
- title: "Add new feature",
204
- description: "Implement the widget",
205
- });
206
-
207
- const result = buildSchedulePrompt(schedule, workItem);
208
-
209
- expect(result).toContain("## Work Item: Add new feature");
210
- expect(result).toContain("Implement the widget");
211
- expect(result).not.toContain("undefined");
212
- });
213
-
214
- it("includes work item URL", () => {
215
- const schedule = createTestSchedule();
216
- const workItem = createTestWorkItem({
217
- url: "https://github.com/org/repo/issues/456",
218
- });
219
-
220
- const result = buildSchedulePrompt(schedule, workItem);
221
-
222
- expect(result).toContain("**URL:** https://github.com/org/repo/issues/456");
223
- });
224
-
225
- it("handles work item without labels", () => {
226
- const schedule = createTestSchedule();
227
- const workItem = createTestWorkItem({ labels: [] });
228
-
229
- const result = buildSchedulePrompt(schedule, workItem);
230
-
231
- expect(result).not.toContain("**Labels:**");
232
- });
233
- });
234
- });
235
-
236
- describe("runSchedule", () => {
237
- let tempDir: string;
238
- let mockLogger: ReturnType<typeof createMockLogger>;
239
-
240
- beforeEach(async () => {
241
- tempDir = await createTempDir();
242
- // Create jobs and sessions directories for job executor
243
- await mkdir(join(tempDir, "jobs"), { recursive: true });
244
- await mkdir(join(tempDir, "sessions"), { recursive: true });
245
- mockLogger = createMockLogger();
246
- });
247
-
248
- afterEach(async () => {
249
- await rm(tempDir, { recursive: true, force: true });
250
- });
251
-
252
- describe("basic execution", () => {
253
- it("executes a schedule successfully", async () => {
254
- const agent = createTestAgent("test-agent");
255
- const schedule = createTestSchedule({ prompt: "Do the thing" });
256
- const sdkQuery = createMockSDKQuery();
257
-
258
- const result = await runSchedule({
259
- agent,
260
- scheduleName: "hourly",
261
- schedule,
262
- scheduleState: createTestScheduleState(),
263
- stateDir: tempDir,
264
- sdkQuery,
265
- logger: mockLogger,
266
- });
267
-
268
- expect(result.success).toBe(true);
269
- expect(result.jobId).toBeDefined();
270
- expect(result.sessionId).toBe("test-session-123");
271
- expect(result.processedWorkItem).toBe(false);
272
- expect(result.workItem).toBeUndefined();
273
- });
274
-
275
- it("updates schedule state to running during execution", async () => {
276
- const agent = createTestAgent("test-agent");
277
- const schedule = createTestSchedule();
278
-
279
- // Track state during execution
280
- let stateWhileRunning: ScheduleState | undefined;
281
-
282
- const sdkQuery: SDKQueryFunction = async function* () {
283
- // Read state during execution
284
- const fleetState = await readFleetState(join(tempDir, "state.yaml"));
285
- stateWhileRunning = fleetState.agents["test-agent"]?.schedules?.hourly;
286
-
287
- yield { type: "system" as const, subtype: "init", session_id: "test" };
288
- yield { type: "assistant" as const, content: "Done" };
289
- };
290
-
291
- await runSchedule({
292
- agent,
293
- scheduleName: "hourly",
294
- schedule,
295
- scheduleState: createTestScheduleState(),
296
- stateDir: tempDir,
297
- sdkQuery,
298
- logger: mockLogger,
299
- });
300
-
301
- expect(stateWhileRunning?.status).toBe("running");
302
- });
303
-
304
- it("updates schedule state with last_run_at after completion", async () => {
305
- const agent = createTestAgent("test-agent");
306
- const schedule = createTestSchedule({ interval: "1h" });
307
- const sdkQuery = createMockSDKQuery();
308
-
309
- const beforeRun = new Date();
310
-
311
- await runSchedule({
312
- agent,
313
- scheduleName: "hourly",
314
- schedule,
315
- scheduleState: createTestScheduleState(),
316
- stateDir: tempDir,
317
- sdkQuery,
318
- logger: mockLogger,
319
- });
320
-
321
- const fleetState = await readFleetState(join(tempDir, "state.yaml"));
322
- const scheduleState = fleetState.agents["test-agent"]?.schedules?.hourly;
323
-
324
- expect(scheduleState?.status).toBe("idle");
325
- expect(scheduleState?.last_run_at).toBeDefined();
326
- expect(new Date(scheduleState!.last_run_at!).getTime()).toBeGreaterThanOrEqual(
327
- beforeRun.getTime()
328
- );
329
- });
330
-
331
- it("calculates next_run_at based on interval", async () => {
332
- const agent = createTestAgent("test-agent");
333
- const schedule = createTestSchedule({ interval: "1h" });
334
- const sdkQuery = createMockSDKQuery();
335
-
336
- await runSchedule({
337
- agent,
338
- scheduleName: "hourly",
339
- schedule,
340
- scheduleState: createTestScheduleState(),
341
- stateDir: tempDir,
342
- sdkQuery,
343
- logger: mockLogger,
344
- });
345
-
346
- const fleetState = await readFleetState(join(tempDir, "state.yaml"));
347
- const scheduleState = fleetState.agents["test-agent"]?.schedules?.hourly;
348
-
349
- expect(scheduleState?.next_run_at).toBeDefined();
350
-
351
- // Next run should be approximately 1 hour from now
352
- const nextRun = new Date(scheduleState!.next_run_at!);
353
- const now = new Date();
354
- const diffMs = nextRun.getTime() - now.getTime();
355
- const diffHours = diffMs / (1000 * 60 * 60);
356
-
357
- expect(diffHours).toBeGreaterThan(0.9);
358
- expect(diffHours).toBeLessThan(1.1);
359
- });
360
-
361
- it("passes correct trigger type and schedule name to executor", async () => {
362
- const agent = createTestAgent("test-agent");
363
- const schedule = createTestSchedule();
364
-
365
- // We can't easily inspect what was passed to the executor,
366
- // but we can verify the job was created correctly by checking logs
367
- const sdkQuery = createMockSDKQuery();
368
-
369
- const result = await runSchedule({
370
- agent,
371
- scheduleName: "my-schedule",
372
- schedule,
373
- scheduleState: createTestScheduleState(),
374
- stateDir: tempDir,
375
- sdkQuery,
376
- logger: mockLogger,
377
- });
378
-
379
- expect(result.success).toBe(true);
380
- expect(mockLogger.infos.some((m) => m.includes("my-schedule"))).toBe(true);
381
- });
382
- });
383
-
384
- describe("with work source", () => {
385
- it("fetches work item when schedule has work_source", async () => {
386
- const agent = createTestAgent("test-agent");
387
- const schedule = createTestSchedule({
388
- prompt: "Process issue:",
389
- work_source: { type: "github", repo: "org/repo" } as const,
390
- });
391
- const workItem = createTestWorkItem();
392
- const workSourceManager = createMockWorkSourceManager({
393
- item: workItem,
394
- claimed: true,
395
- claimResult: { success: true, workItem },
396
- });
397
- const sdkQuery = createMockSDKQuery();
398
-
399
- const result = await runSchedule({
400
- agent,
401
- scheduleName: "hourly",
402
- schedule,
403
- scheduleState: createTestScheduleState(),
404
- stateDir: tempDir,
405
- sdkQuery,
406
- workSourceManager,
407
- logger: mockLogger,
408
- });
409
-
410
- expect(result.success).toBe(true);
411
- expect(result.processedWorkItem).toBe(true);
412
- expect(result.workItem).toBe(workItem);
413
- expect(workSourceManager.getNextWorkItemCalls).toHaveLength(1);
414
- });
415
-
416
- it("reports outcome to work source after successful execution", async () => {
417
- const agent = createTestAgent("test-agent");
418
- const schedule = createTestSchedule({
419
- work_source: { type: "github", repo: "org/repo" } as const,
420
- });
421
- const workItem = createTestWorkItem();
422
- const workSourceManager = createMockWorkSourceManager({
423
- item: workItem,
424
- claimed: true,
425
- claimResult: { success: true, workItem },
426
- });
427
- const sdkQuery = createMockSDKQuery();
428
-
429
- await runSchedule({
430
- agent,
431
- scheduleName: "hourly",
432
- schedule,
433
- scheduleState: createTestScheduleState(),
434
- stateDir: tempDir,
435
- sdkQuery,
436
- workSourceManager,
437
- logger: mockLogger,
438
- });
439
-
440
- expect(workSourceManager.reportOutcomeCalls).toHaveLength(1);
441
- expect(workSourceManager.reportOutcomeCalls[0].taskId).toBe(workItem.id);
442
- expect(workSourceManager.reportOutcomeCalls[0].result.outcome).toBe("success");
443
- });
444
-
445
- it("reports failure outcome when job fails", async () => {
446
- const agent = createTestAgent("test-agent");
447
- const schedule = createTestSchedule({
448
- work_source: { type: "github", repo: "org/repo" } as const,
449
- });
450
- const workItem = createTestWorkItem();
451
- const workSourceManager = createMockWorkSourceManager({
452
- item: workItem,
453
- claimed: true,
454
- claimResult: { success: true, workItem },
455
- });
456
-
457
- // Create SDK query that produces an error
458
- const sdkQuery: SDKQueryFunction = async function* () {
459
- yield { type: "system" as const, subtype: "init", session_id: "test" };
460
- yield { type: "error" as const, message: "API error", code: "API_ERROR" };
461
- };
462
-
463
- await runSchedule({
464
- agent,
465
- scheduleName: "hourly",
466
- schedule,
467
- scheduleState: createTestScheduleState(),
468
- stateDir: tempDir,
469
- sdkQuery,
470
- workSourceManager,
471
- logger: mockLogger,
472
- });
473
-
474
- expect(workSourceManager.reportOutcomeCalls).toHaveLength(1);
475
- expect(workSourceManager.reportOutcomeCalls[0].result.outcome).toBe("failure");
476
- });
477
-
478
- it("continues when no work is available", async () => {
479
- const agent = createTestAgent("test-agent");
480
- const schedule = createTestSchedule({
481
- work_source: { type: "github", repo: "org/repo" } as const,
482
- });
483
- const workSourceManager = createMockWorkSourceManager({
484
- item: null,
485
- claimed: false,
486
- });
487
- const sdkQuery = createMockSDKQuery();
488
-
489
- const result = await runSchedule({
490
- agent,
491
- scheduleName: "hourly",
492
- schedule,
493
- scheduleState: createTestScheduleState(),
494
- stateDir: tempDir,
495
- sdkQuery,
496
- workSourceManager,
497
- logger: mockLogger,
498
- });
499
-
500
- // Still runs with schedule prompt, just no work item
501
- expect(result.success).toBe(true);
502
- expect(result.processedWorkItem).toBe(false);
503
- expect(result.workItem).toBeUndefined();
504
- expect(workSourceManager.reportOutcomeCalls).toHaveLength(0);
505
- });
506
-
507
- it("handles claim failure gracefully", async () => {
508
- const agent = createTestAgent("test-agent");
509
- const schedule = createTestSchedule({
510
- work_source: { type: "github", repo: "org/repo" } as const,
511
- });
512
- const workItem = createTestWorkItem();
513
- const workSourceManager = createMockWorkSourceManager({
514
- item: workItem,
515
- claimed: false,
516
- claimResult: { success: false, reason: "already_claimed" },
517
- });
518
- const sdkQuery = createMockSDKQuery();
519
-
520
- const result = await runSchedule({
521
- agent,
522
- scheduleName: "hourly",
523
- schedule,
524
- scheduleState: createTestScheduleState(),
525
- stateDir: tempDir,
526
- sdkQuery,
527
- workSourceManager,
528
- logger: mockLogger,
529
- });
530
-
531
- // Should still run, just without work item
532
- expect(result.success).toBe(true);
533
- expect(result.processedWorkItem).toBe(false);
534
- expect(mockLogger.warnings.some((m) => m.includes("claim failed"))).toBe(true);
535
- });
536
-
537
- it("skips work source when no manager provided", async () => {
538
- const agent = createTestAgent("test-agent");
539
- const schedule = createTestSchedule({
540
- prompt: "Test",
541
- work_source: { type: "github", repo: "org/repo" } as const,
542
- });
543
- const sdkQuery = createMockSDKQuery();
544
-
545
- const result = await runSchedule({
546
- agent,
547
- scheduleName: "hourly",
548
- schedule,
549
- scheduleState: createTestScheduleState(),
550
- stateDir: tempDir,
551
- sdkQuery,
552
- // No workSourceManager provided
553
- logger: mockLogger,
554
- });
555
-
556
- expect(result.success).toBe(true);
557
- expect(result.processedWorkItem).toBe(false);
558
- });
559
- });
560
-
561
- describe("error handling", () => {
562
- it("reports failure outcome when SDK throws", async () => {
563
- const agent = createTestAgent("test-agent");
564
- const schedule = createTestSchedule({
565
- work_source: { type: "github", repo: "org/repo" } as const,
566
- });
567
- const workItem = createTestWorkItem();
568
- const workSourceManager = createMockWorkSourceManager({
569
- item: workItem,
570
- claimed: true,
571
- claimResult: { success: true, workItem },
572
- });
573
-
574
- // Create SDK query that throws - JobExecutor catches this and returns failed result
575
- const sdkQuery: SDKQueryFunction = async function* () {
576
- throw new Error("Unexpected SDK failure");
577
- };
578
-
579
- const result = await runSchedule({
580
- agent,
581
- scheduleName: "hourly",
582
- schedule,
583
- scheduleState: createTestScheduleState(),
584
- stateDir: tempDir,
585
- sdkQuery,
586
- workSourceManager,
587
- logger: mockLogger,
588
- });
589
-
590
- // JobExecutor catches SDK errors and returns failed result
591
- expect(result.success).toBe(false);
592
- expect(result.error).toBeDefined();
593
-
594
- // Work outcome should still be reported as failure
595
- expect(workSourceManager.reportOutcomeCalls).toHaveLength(1);
596
- expect(workSourceManager.reportOutcomeCalls[0].result.outcome).toBe("failure");
597
- });
598
-
599
- it("records error in schedule state on SDK failure", async () => {
600
- const agent = createTestAgent("test-agent");
601
- const schedule = createTestSchedule();
602
-
603
- // Create SDK query that throws - JobExecutor catches this
604
- const sdkQuery: SDKQueryFunction = async function* () {
605
- throw new Error("Execution failed");
606
- };
607
-
608
- const result = await runSchedule({
609
- agent,
610
- scheduleName: "hourly",
611
- schedule,
612
- scheduleState: createTestScheduleState(),
613
- stateDir: tempDir,
614
- sdkQuery,
615
- logger: mockLogger,
616
- });
617
-
618
- expect(result.success).toBe(false);
619
-
620
- const fleetState = await readFleetState(join(tempDir, "state.yaml"));
621
- const scheduleState = fleetState.agents["test-agent"]?.schedules?.hourly;
622
-
623
- expect(scheduleState?.status).toBe("idle");
624
- // Error message contains the original error but may be wrapped
625
- expect(scheduleState?.last_error).toContain("Execution failed");
626
- });
627
-
628
- it("still calculates next_run_at on SDK failure", async () => {
629
- const agent = createTestAgent("test-agent");
630
- const schedule = createTestSchedule({ interval: "30m" });
631
-
632
- const sdkQuery: SDKQueryFunction = async function* () {
633
- throw new Error("Execution failed");
634
- };
635
-
636
- const result = await runSchedule({
637
- agent,
638
- scheduleName: "hourly",
639
- schedule,
640
- scheduleState: createTestScheduleState(),
641
- stateDir: tempDir,
642
- sdkQuery,
643
- logger: mockLogger,
644
- });
645
-
646
- expect(result.success).toBe(false);
647
-
648
- const fleetState = await readFleetState(join(tempDir, "state.yaml"));
649
- const scheduleState = fleetState.agents["test-agent"]?.schedules?.hourly;
650
-
651
- // Should still have calculated next_run_at
652
- expect(scheduleState?.next_run_at).toBeDefined();
653
- });
654
-
655
- it("continues if reporting outcome fails", async () => {
656
- const agent = createTestAgent("test-agent");
657
- const schedule = createTestSchedule({
658
- work_source: { type: "github", repo: "org/repo" } as const,
659
- });
660
- const workItem = createTestWorkItem();
661
- const workSourceManager = createMockWorkSourceManager({
662
- item: workItem,
663
- claimed: true,
664
- claimResult: { success: true, workItem },
665
- });
666
-
667
- // Make reportOutcome throw
668
- (workSourceManager.reportOutcome as ReturnType<typeof vi.fn>).mockRejectedValue(
669
- new Error("Report failed")
670
- );
671
-
672
- const sdkQuery = createMockSDKQuery();
673
-
674
- const result = await runSchedule({
675
- agent,
676
- scheduleName: "hourly",
677
- schedule,
678
- scheduleState: createTestScheduleState(),
679
- stateDir: tempDir,
680
- sdkQuery,
681
- workSourceManager,
682
- logger: mockLogger,
683
- });
684
-
685
- // Should still succeed overall
686
- expect(result.success).toBe(true);
687
- expect(mockLogger.errors.some((m) => m.includes("Report failed"))).toBe(true);
688
- });
689
- });
690
-
691
- describe("logging", () => {
692
- it("logs schedule start", async () => {
693
- const agent = createTestAgent("my-agent");
694
- const schedule = createTestSchedule();
695
- const sdkQuery = createMockSDKQuery();
696
-
697
- await runSchedule({
698
- agent,
699
- scheduleName: "my-schedule",
700
- schedule,
701
- scheduleState: createTestScheduleState(),
702
- stateDir: tempDir,
703
- sdkQuery,
704
- logger: mockLogger,
705
- });
706
-
707
- expect(
708
- mockLogger.infos.some(
709
- (m) => m.includes("Running") && m.includes("my-agent/my-schedule")
710
- )
711
- ).toBe(true);
712
- });
713
-
714
- it("logs schedule completion", async () => {
715
- const agent = createTestAgent("my-agent");
716
- const schedule = createTestSchedule();
717
- const sdkQuery = createMockSDKQuery();
718
-
719
- await runSchedule({
720
- agent,
721
- scheduleName: "my-schedule",
722
- schedule,
723
- scheduleState: createTestScheduleState(),
724
- stateDir: tempDir,
725
- sdkQuery,
726
- logger: mockLogger,
727
- });
728
-
729
- expect(
730
- mockLogger.infos.some(
731
- (m) => m.includes("Completed") && m.includes("my-agent/my-schedule")
732
- )
733
- ).toBe(true);
734
- });
735
-
736
- it("logs work item claim", async () => {
737
- const agent = createTestAgent("test-agent");
738
- const schedule = createTestSchedule({
739
- work_source: { type: "github", repo: "org/repo" } as const,
740
- });
741
- const workItem = createTestWorkItem({ title: "Important task" });
742
- const workSourceManager = createMockWorkSourceManager({
743
- item: workItem,
744
- claimed: true,
745
- claimResult: { success: true, workItem },
746
- });
747
- const sdkQuery = createMockSDKQuery();
748
-
749
- await runSchedule({
750
- agent,
751
- scheduleName: "hourly",
752
- schedule,
753
- scheduleState: createTestScheduleState(),
754
- stateDir: tempDir,
755
- sdkQuery,
756
- workSourceManager,
757
- logger: mockLogger,
758
- });
759
-
760
- expect(
761
- mockLogger.infos.some(
762
- (m) => m.includes("Claimed") && m.includes("Important task")
763
- )
764
- ).toBe(true);
765
- });
766
-
767
- it("logs outcome reporting", async () => {
768
- const agent = createTestAgent("test-agent");
769
- const schedule = createTestSchedule({
770
- work_source: { type: "github", repo: "org/repo" } as const,
771
- });
772
- const workItem = createTestWorkItem();
773
- const workSourceManager = createMockWorkSourceManager({
774
- item: workItem,
775
- claimed: true,
776
- claimResult: { success: true, workItem },
777
- });
778
- const sdkQuery = createMockSDKQuery();
779
-
780
- await runSchedule({
781
- agent,
782
- scheduleName: "hourly",
783
- schedule,
784
- scheduleState: createTestScheduleState(),
785
- stateDir: tempDir,
786
- sdkQuery,
787
- workSourceManager,
788
- logger: mockLogger,
789
- });
790
-
791
- expect(
792
- mockLogger.infos.some(
793
- (m) => m.includes("Reported outcome") && m.includes(workItem.id)
794
- )
795
- ).toBe(true);
796
- });
797
- });
798
- });