@herdctl/core 0.0.1 → 0.1.0

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 (284) hide show
  1. package/dist/config/__tests__/agent.test.js +61 -13
  2. package/dist/config/__tests__/agent.test.js.map +1 -1
  3. package/dist/config/__tests__/merge.test.js +10 -3
  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 +841 -27
  12. package/dist/config/schema.d.ts.map +1 -1
  13. package/dist/config/schema.js +129 -10
  14. package/dist/config/schema.js.map +1 -1
  15. package/dist/fleet-manager/__tests__/coverage.test.js +14 -331
  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 +114 -0
  20. package/dist/fleet-manager/__tests__/integration.test.js.map +1 -1
  21. package/dist/fleet-manager/__tests__/job-control.test.js +13 -14
  22. package/dist/fleet-manager/__tests__/job-control.test.js.map +1 -1
  23. package/dist/fleet-manager/__tests__/reload.test.js +12 -2
  24. package/dist/fleet-manager/__tests__/reload.test.js.map +1 -1
  25. package/dist/fleet-manager/__tests__/status-queries.test.js +6 -0
  26. package/dist/fleet-manager/__tests__/status-queries.test.js.map +1 -1
  27. package/dist/fleet-manager/__tests__/trigger.test.js +10 -2
  28. package/dist/fleet-manager/__tests__/trigger.test.js.map +1 -1
  29. package/dist/fleet-manager/config-reload.d.ts +164 -0
  30. package/dist/fleet-manager/config-reload.d.ts.map +1 -0
  31. package/dist/fleet-manager/config-reload.js +445 -0
  32. package/dist/fleet-manager/config-reload.js.map +1 -0
  33. package/dist/fleet-manager/context.d.ts +76 -0
  34. package/dist/fleet-manager/context.d.ts.map +1 -0
  35. package/dist/fleet-manager/context.js +11 -0
  36. package/dist/fleet-manager/context.js.map +1 -0
  37. package/dist/fleet-manager/errors.d.ts +0 -25
  38. package/dist/fleet-manager/errors.d.ts.map +1 -1
  39. package/dist/fleet-manager/errors.js +0 -38
  40. package/dist/fleet-manager/errors.js.map +1 -1
  41. package/dist/fleet-manager/event-emitters.d.ts +123 -0
  42. package/dist/fleet-manager/event-emitters.d.ts.map +1 -0
  43. package/dist/fleet-manager/event-emitters.js +136 -0
  44. package/dist/fleet-manager/event-emitters.js.map +1 -0
  45. package/dist/fleet-manager/event-types.d.ts +0 -15
  46. package/dist/fleet-manager/event-types.d.ts.map +1 -1
  47. package/dist/fleet-manager/fleet-manager.d.ts +40 -653
  48. package/dist/fleet-manager/fleet-manager.d.ts.map +1 -1
  49. package/dist/fleet-manager/fleet-manager.js +95 -1720
  50. package/dist/fleet-manager/fleet-manager.js.map +1 -1
  51. package/dist/fleet-manager/index.d.ts +13 -2
  52. package/dist/fleet-manager/index.d.ts.map +1 -1
  53. package/dist/fleet-manager/index.js +19 -6
  54. package/dist/fleet-manager/index.js.map +1 -1
  55. package/dist/fleet-manager/job-control.d.ts +67 -0
  56. package/dist/fleet-manager/job-control.d.ts.map +1 -0
  57. package/dist/fleet-manager/job-control.js +333 -0
  58. package/dist/fleet-manager/job-control.js.map +1 -0
  59. package/dist/fleet-manager/log-streaming.d.ts +171 -0
  60. package/dist/fleet-manager/log-streaming.d.ts.map +1 -0
  61. package/dist/fleet-manager/log-streaming.js +503 -0
  62. package/dist/fleet-manager/log-streaming.js.map +1 -0
  63. package/dist/fleet-manager/schedule-executor.d.ts +63 -0
  64. package/dist/fleet-manager/schedule-executor.d.ts.map +1 -0
  65. package/dist/fleet-manager/schedule-executor.js +209 -0
  66. package/dist/fleet-manager/schedule-executor.js.map +1 -0
  67. package/dist/fleet-manager/schedule-management.d.ts +71 -0
  68. package/dist/fleet-manager/schedule-management.d.ts.map +1 -0
  69. package/dist/fleet-manager/schedule-management.js +171 -0
  70. package/dist/fleet-manager/schedule-management.js.map +1 -0
  71. package/dist/fleet-manager/status-queries.d.ts +105 -0
  72. package/dist/fleet-manager/status-queries.d.ts.map +1 -0
  73. package/dist/fleet-manager/status-queries.js +247 -0
  74. package/dist/fleet-manager/status-queries.js.map +1 -0
  75. package/dist/fleet-manager/types.d.ts +0 -39
  76. package/dist/fleet-manager/types.d.ts.map +1 -1
  77. package/dist/runner/__tests__/job-executor.test.js +206 -1
  78. package/dist/runner/__tests__/job-executor.test.js.map +1 -1
  79. package/dist/runner/job-executor.d.ts +9 -0
  80. package/dist/runner/job-executor.d.ts.map +1 -1
  81. package/dist/runner/job-executor.js +78 -4
  82. package/dist/runner/job-executor.js.map +1 -1
  83. package/dist/runner/message-processor.d.ts.map +1 -1
  84. package/dist/runner/message-processor.js +53 -0
  85. package/dist/runner/message-processor.js.map +1 -1
  86. package/dist/runner/types.d.ts +3 -1
  87. package/dist/runner/types.d.ts.map +1 -1
  88. package/dist/scheduler/__tests__/cron.test.d.ts +2 -0
  89. package/dist/scheduler/__tests__/cron.test.d.ts.map +1 -0
  90. package/dist/scheduler/__tests__/cron.test.js +867 -0
  91. package/dist/scheduler/__tests__/cron.test.js.map +1 -0
  92. package/dist/scheduler/__tests__/scheduler.test.js +164 -5
  93. package/dist/scheduler/__tests__/scheduler.test.js.map +1 -1
  94. package/dist/scheduler/cron.d.ts +126 -0
  95. package/dist/scheduler/cron.d.ts.map +1 -0
  96. package/dist/scheduler/cron.js +390 -0
  97. package/dist/scheduler/cron.js.map +1 -0
  98. package/dist/scheduler/errors.d.ts +81 -1
  99. package/dist/scheduler/errors.d.ts.map +1 -1
  100. package/dist/scheduler/errors.js +81 -6
  101. package/dist/scheduler/errors.js.map +1 -1
  102. package/dist/scheduler/index.d.ts +1 -0
  103. package/dist/scheduler/index.d.ts.map +1 -1
  104. package/dist/scheduler/index.js +2 -0
  105. package/dist/scheduler/index.js.map +1 -1
  106. package/dist/scheduler/schedule-runner.d.ts +2 -2
  107. package/dist/scheduler/schedule-runner.d.ts.map +1 -1
  108. package/dist/scheduler/schedule-runner.js +20 -8
  109. package/dist/scheduler/schedule-runner.js.map +1 -1
  110. package/dist/scheduler/scheduler.d.ts +4 -4
  111. package/dist/scheduler/scheduler.d.ts.map +1 -1
  112. package/dist/scheduler/scheduler.js +95 -20
  113. package/dist/scheduler/scheduler.js.map +1 -1
  114. package/dist/scheduler/types.d.ts +1 -1
  115. package/dist/scheduler/types.d.ts.map +1 -1
  116. package/dist/state/schemas/job-metadata.d.ts +2 -2
  117. package/package.json +33 -8
  118. package/.turbo/turbo-build.log +0 -4
  119. package/.turbo/turbo-test.log +0 -219
  120. package/.turbo/turbo-typecheck.log +0 -4
  121. package/coverage/base.css +0 -224
  122. package/coverage/block-navigation.js +0 -87
  123. package/coverage/coverage-final.json +0 -51
  124. package/coverage/favicon.png +0 -0
  125. package/coverage/index.html +0 -251
  126. package/coverage/prettify.css +0 -1
  127. package/coverage/prettify.js +0 -2
  128. package/coverage/sort-arrow-sprite.png +0 -0
  129. package/coverage/sorter.js +0 -210
  130. package/coverage/src/config/index.html +0 -191
  131. package/coverage/src/config/index.ts.html +0 -442
  132. package/coverage/src/config/interpolate.ts.html +0 -652
  133. package/coverage/src/config/loader.ts.html +0 -1501
  134. package/coverage/src/config/merge.ts.html +0 -823
  135. package/coverage/src/config/parser.ts.html +0 -1213
  136. package/coverage/src/config/schema.ts.html +0 -1123
  137. package/coverage/src/fleet-manager/errors.ts.html +0 -2326
  138. package/coverage/src/fleet-manager/event-types.ts.html +0 -1219
  139. package/coverage/src/fleet-manager/fleet-manager.ts.html +0 -7030
  140. package/coverage/src/fleet-manager/index.html +0 -206
  141. package/coverage/src/fleet-manager/index.ts.html +0 -469
  142. package/coverage/src/fleet-manager/job-manager.ts.html +0 -2074
  143. package/coverage/src/fleet-manager/job-queue.ts.html +0 -2479
  144. package/coverage/src/fleet-manager/types.ts.html +0 -2602
  145. package/coverage/src/index.html +0 -116
  146. package/coverage/src/index.ts.html +0 -181
  147. package/coverage/src/runner/errors.ts.html +0 -1006
  148. package/coverage/src/runner/index.html +0 -191
  149. package/coverage/src/runner/index.ts.html +0 -256
  150. package/coverage/src/runner/job-executor.ts.html +0 -1429
  151. package/coverage/src/runner/message-processor.ts.html +0 -1150
  152. package/coverage/src/runner/sdk-adapter.ts.html +0 -658
  153. package/coverage/src/runner/types.ts.html +0 -559
  154. package/coverage/src/scheduler/errors.ts.html +0 -388
  155. package/coverage/src/scheduler/index.html +0 -206
  156. package/coverage/src/scheduler/index.ts.html +0 -244
  157. package/coverage/src/scheduler/interval.ts.html +0 -652
  158. package/coverage/src/scheduler/schedule-runner.ts.html +0 -1411
  159. package/coverage/src/scheduler/schedule-state.ts.html +0 -718
  160. package/coverage/src/scheduler/scheduler.ts.html +0 -1795
  161. package/coverage/src/scheduler/types.ts.html +0 -733
  162. package/coverage/src/state/directory.ts.html +0 -736
  163. package/coverage/src/state/errors.ts.html +0 -376
  164. package/coverage/src/state/fleet-state.ts.html +0 -937
  165. package/coverage/src/state/index.html +0 -221
  166. package/coverage/src/state/index.ts.html +0 -322
  167. package/coverage/src/state/job-metadata.ts.html +0 -1420
  168. package/coverage/src/state/job-output.ts.html +0 -1033
  169. package/coverage/src/state/schemas/fleet-state.ts.html +0 -445
  170. package/coverage/src/state/schemas/index.html +0 -176
  171. package/coverage/src/state/schemas/index.ts.html +0 -286
  172. package/coverage/src/state/schemas/job-metadata.ts.html +0 -628
  173. package/coverage/src/state/schemas/job-output.ts.html +0 -616
  174. package/coverage/src/state/schemas/session-info.ts.html +0 -361
  175. package/coverage/src/state/session.ts.html +0 -844
  176. package/coverage/src/state/types.ts.html +0 -262
  177. package/coverage/src/state/utils/atomic.ts.html +0 -748
  178. package/coverage/src/state/utils/index.html +0 -146
  179. package/coverage/src/state/utils/index.ts.html +0 -103
  180. package/coverage/src/state/utils/reads.ts.html +0 -1621
  181. package/coverage/src/work-sources/adapters/github.ts.html +0 -3583
  182. package/coverage/src/work-sources/adapters/index.html +0 -131
  183. package/coverage/src/work-sources/adapters/index.ts.html +0 -277
  184. package/coverage/src/work-sources/errors.ts.html +0 -298
  185. package/coverage/src/work-sources/index.html +0 -176
  186. package/coverage/src/work-sources/index.ts.html +0 -529
  187. package/coverage/src/work-sources/manager.ts.html +0 -1324
  188. package/coverage/src/work-sources/registry.ts.html +0 -619
  189. package/coverage/src/work-sources/types.ts.html +0 -568
  190. package/dist/fleet-manager/__tests__/event-helpers.test.d.ts +0 -7
  191. package/dist/fleet-manager/__tests__/event-helpers.test.d.ts.map +0 -1
  192. package/dist/fleet-manager/__tests__/event-helpers.test.js +0 -368
  193. package/dist/fleet-manager/__tests__/event-helpers.test.js.map +0 -1
  194. package/src/config/__tests__/agent.test.ts +0 -864
  195. package/src/config/__tests__/interpolate.test.ts +0 -644
  196. package/src/config/__tests__/loader.test.ts +0 -784
  197. package/src/config/__tests__/merge.test.ts +0 -751
  198. package/src/config/__tests__/parser.test.ts +0 -533
  199. package/src/config/__tests__/schema.test.ts +0 -873
  200. package/src/config/index.ts +0 -119
  201. package/src/config/interpolate.ts +0 -189
  202. package/src/config/loader.ts +0 -472
  203. package/src/config/merge.ts +0 -246
  204. package/src/config/parser.ts +0 -376
  205. package/src/config/schema.ts +0 -346
  206. package/src/fleet-manager/__tests__/coverage.test.ts +0 -2869
  207. package/src/fleet-manager/__tests__/errors.test.ts +0 -660
  208. package/src/fleet-manager/__tests__/event-helpers.test.ts +0 -448
  209. package/src/fleet-manager/__tests__/integration.test.ts +0 -1209
  210. package/src/fleet-manager/__tests__/job-control.test.ts +0 -283
  211. package/src/fleet-manager/__tests__/job-manager.test.ts +0 -869
  212. package/src/fleet-manager/__tests__/job-queue.test.ts +0 -401
  213. package/src/fleet-manager/__tests__/reload.test.ts +0 -751
  214. package/src/fleet-manager/__tests__/status-queries.test.ts +0 -595
  215. package/src/fleet-manager/__tests__/trigger.test.ts +0 -601
  216. package/src/fleet-manager/errors.ts +0 -747
  217. package/src/fleet-manager/event-types.ts +0 -378
  218. package/src/fleet-manager/fleet-manager.ts +0 -2315
  219. package/src/fleet-manager/index.ts +0 -128
  220. package/src/fleet-manager/job-manager.ts +0 -663
  221. package/src/fleet-manager/job-queue.ts +0 -798
  222. package/src/fleet-manager/types.ts +0 -839
  223. package/src/index.ts +0 -32
  224. package/src/runner/__tests__/errors.test.ts +0 -382
  225. package/src/runner/__tests__/job-executor.test.ts +0 -1708
  226. package/src/runner/__tests__/message-processor.test.ts +0 -960
  227. package/src/runner/__tests__/sdk-adapter.test.ts +0 -626
  228. package/src/runner/errors.ts +0 -307
  229. package/src/runner/index.ts +0 -57
  230. package/src/runner/job-executor.ts +0 -448
  231. package/src/runner/message-processor.ts +0 -355
  232. package/src/runner/sdk-adapter.ts +0 -191
  233. package/src/runner/types.ts +0 -158
  234. package/src/scheduler/__tests__/errors.test.ts +0 -159
  235. package/src/scheduler/__tests__/interval.test.ts +0 -515
  236. package/src/scheduler/__tests__/schedule-runner.test.ts +0 -798
  237. package/src/scheduler/__tests__/schedule-state.test.ts +0 -671
  238. package/src/scheduler/__tests__/scheduler.test.ts +0 -1280
  239. package/src/scheduler/errors.ts +0 -101
  240. package/src/scheduler/index.ts +0 -53
  241. package/src/scheduler/interval.ts +0 -189
  242. package/src/scheduler/schedule-runner.ts +0 -442
  243. package/src/scheduler/schedule-state.ts +0 -211
  244. package/src/scheduler/scheduler.ts +0 -570
  245. package/src/scheduler/types.ts +0 -216
  246. package/src/state/__tests__/directory.test.ts +0 -595
  247. package/src/state/__tests__/fleet-state.test.ts +0 -868
  248. package/src/state/__tests__/job-metadata-schema.test.ts +0 -414
  249. package/src/state/__tests__/job-metadata.test.ts +0 -831
  250. package/src/state/__tests__/job-output.test.ts +0 -856
  251. package/src/state/__tests__/session-schema.test.ts +0 -378
  252. package/src/state/__tests__/session.test.ts +0 -604
  253. package/src/state/directory.ts +0 -217
  254. package/src/state/errors.ts +0 -97
  255. package/src/state/fleet-state.ts +0 -284
  256. package/src/state/index.ts +0 -79
  257. package/src/state/job-metadata.ts +0 -445
  258. package/src/state/job-output.ts +0 -316
  259. package/src/state/schemas/__tests__/job-output.test.ts +0 -338
  260. package/src/state/schemas/fleet-state.ts +0 -120
  261. package/src/state/schemas/index.ts +0 -67
  262. package/src/state/schemas/job-metadata.ts +0 -181
  263. package/src/state/schemas/job-output.ts +0 -177
  264. package/src/state/schemas/session-info.ts +0 -92
  265. package/src/state/session.ts +0 -253
  266. package/src/state/types.ts +0 -59
  267. package/src/state/utils/__tests__/atomic.test.ts +0 -723
  268. package/src/state/utils/__tests__/reads.test.ts +0 -1071
  269. package/src/state/utils/atomic.ts +0 -221
  270. package/src/state/utils/index.ts +0 -6
  271. package/src/state/utils/reads.ts +0 -512
  272. package/src/work-sources/__tests__/github.test.ts +0 -1800
  273. package/src/work-sources/__tests__/manager.test.ts +0 -529
  274. package/src/work-sources/__tests__/registry.test.ts +0 -477
  275. package/src/work-sources/__tests__/types.test.ts +0 -479
  276. package/src/work-sources/adapters/github.ts +0 -1166
  277. package/src/work-sources/adapters/index.ts +0 -64
  278. package/src/work-sources/errors.ts +0 -71
  279. package/src/work-sources/index.ts +0 -148
  280. package/src/work-sources/manager.ts +0 -413
  281. package/src/work-sources/registry.ts +0 -178
  282. package/src/work-sources/types.ts +0 -161
  283. package/tsconfig.json +0 -9
  284. 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
- });