@@ -54,6 +54,10 @@ function gatewayRequest(method: string): Record<string, unknown> {
|
54 | 54 | return requireRecord(request); |
55 | 55 | } |
56 | 56 | |
| 57 | +function firstRegisteredSubagentRun(): Record<string, unknown> { |
| 58 | +return requireRecord(hoisted.registerSubagentRunMock.mock.calls[0]?.[0]); |
| 59 | +} |
| 60 | + |
57 | 61 | describe("spawnSubagentDirect seam flow", () => { |
58 | 62 | beforeAll(async () => { |
59 | 63 | ({ resetSubagentRegistryForTests, spawnSubagentDirect } = await loadSubagentSpawnModuleForTest({ |
@@ -215,7 +219,7 @@ describe("spawnSubagentDirect seam flow", () => {
|
215 | 219 | const childSessionKey = result.childSessionKey as string; |
216 | 220 | expect(hoisted.pruneLegacyStoreKeysMock).toHaveBeenCalledTimes(3); |
217 | 221 | expect(hoisted.updateSessionStoreMock).toHaveBeenCalledTimes(3); |
218 | | -const registerInput = requireRecord(hoisted.registerSubagentRunMock.mock.calls.at(0)?.[0]); |
| 222 | +const registerInput = firstRegisteredSubagentRun(); |
219 | 223 | const requesterOrigin = requireRecord(registerInput.requesterOrigin); |
220 | 224 | expect(registerInput.runId).toBe("run-1"); |
221 | 225 | expect(registerInput.childSessionKey).toBe(childSessionKey); |
@@ -281,7 +285,7 @@ describe("spawnSubagentDirect seam flow", () => {
|
281 | 285 | ); |
282 | 286 | |
283 | 287 | expect(result.status).toBe("accepted"); |
284 | | -const registerInput = requireRecord(hoisted.registerSubagentRunMock.mock.calls.at(0)?.[0]); |
| 288 | +const registerInput = firstRegisteredSubagentRun(); |
285 | 289 | const requesterOrigin = requireRecord(registerInput.requesterOrigin); |
286 | 290 | expect(requesterOrigin.channel).toBe("discord"); |
287 | 291 | expect(requesterOrigin.accountId).toBe("acct-1"); |
|
