@@ -74,6 +74,18 @@ function requireMockCallArg(
7474return requireRecord(call[0], `mock call ${index} arg`);
7575}
767677+function requireLastMockCallArg(
78+mock: { mock: { calls: unknown[][] } },
79+label: string,
80+): Record<string, unknown> {
81+const calls = mock.mock.calls;
82+const call = calls[calls.length - 1];
83+if (!call) {
84+throw new Error(`expected ${label} mock call`);
85+}
86+return requireRecord(call[0], `${label} mock call arg`);
87+}
88+7789function expectBlockReplyText(onBlockReply: { mock: { calls: unknown[][] } }, text: string): void {
7890expect(
7991onBlockReply.mock.calls.some(
@@ -567,12 +579,8 @@ describe("createFollowupRunner runtime config", () => {
567579}),
568580);
569581570-const call = runEmbeddedPiAgentMock.mock.calls.at(-1)?.[0] as
571-| {
572-config?: unknown;
573-}
574-| undefined;
575-expect(call?.config).toBe(runtimeConfig);
582+const call = requireLastMockCallArg(runEmbeddedPiAgentMock, "run embedded pi agent");
583+expect(call.config).toBe(runtimeConfig);
576584});
577585578586it("resolves queued embedded followups before preflight helpers read config", async () => {
@@ -626,12 +634,8 @@ describe("createFollowupRunner runtime config", () => {
626634627635expect(queued.run.config).toBe(runtimeConfig);
628636expect(requireMockCallArg(runPreflightCompactionIfNeededMock, 0).cfg).toBe(runtimeConfig);
629-const call = runEmbeddedPiAgentMock.mock.calls.at(-1)?.[0] as
630-| {
631-config?: unknown;
632-}
633-| undefined;
634-expect(call?.config).toBe(runtimeConfig);
637+const call = requireLastMockCallArg(runEmbeddedPiAgentMock, "run embedded pi agent");
638+expect(call.config).toBe(runtimeConfig);
635639});
636640637641it("passes queued origin scope into queued execution-config resolution", async () => {
@@ -691,14 +695,9 @@ describe("createFollowupRunner runtime config", () => {
691695}),
692696);
693697694-const call = runEmbeddedPiAgentMock.mock.calls.at(-1)?.[0] as
695-| {
696-images?: unknown;
697-imageOrder?: unknown;
698-}
699-| undefined;
700-expect(call?.images).toBe(images);
701-expect(call?.imageOrder).toBe(imageOrder);
698+const call = requireLastMockCallArg(runEmbeddedPiAgentMock, "run embedded pi agent");
699+expect(call.images).toBe(images);
700+expect(call.imageOrder).toBe(imageOrder);
702701});
703702});
704703@@ -1108,16 +1107,10 @@ describe("createFollowupRunner bootstrap warning dedupe", () => {
1108110711091108await runner(baseQueuedRun());
111011091111-const call = runEmbeddedPiAgentMock.mock.calls.at(-1)?.[0] as
1112-| {
1113-allowGatewaySubagentBinding?: boolean;
1114-bootstrapPromptWarningSignaturesSeen?: string[];
1115-bootstrapPromptWarningSignature?: string;
1116-}
1117-| undefined;
1118-expect(call?.allowGatewaySubagentBinding).toBe(true);
1119-expect(call?.bootstrapPromptWarningSignaturesSeen).toEqual(["sig-a", "sig-b"]);
1120-expect(call?.bootstrapPromptWarningSignature).toBe("sig-b");
1110+const call = requireLastMockCallArg(runEmbeddedPiAgentMock, "run embedded pi agent");
1111+expect(call.allowGatewaySubagentBinding).toBe(true);
1112+expect(call.bootstrapPromptWarningSignaturesSeen).toEqual(["sig-a", "sig-b"]);
1113+expect(call.bootstrapPromptWarningSignature).toBe("sig-b");
11211114});
11221115});
11231116@@ -1331,7 +1324,7 @@ describe("createFollowupRunner messaging delivery and dedupe", () => {
13311324).resolves.toBeUndefined();
1332132513331326expect(requireMockCallArg(persistSpy, 0).providerUsed).toBe("anthropic");
1334-expect(persistSpy.mock.calls.at(0)?.[0]?.usageIsContextSnapshot).toBeUndefined();
1327+expect(requireMockCallArg(persistSpy, 0).usageIsContextSnapshot).toBeUndefined();
13351328persistSpy.mockRestore();
13361329});
13371330@@ -1680,7 +1673,7 @@ describe("createFollowupRunner agentDir forwarding", () => {
16801673});
1681167416821675expect(runEmbeddedPiAgentMock).toHaveBeenCalledTimes(1);
1683-const call = runEmbeddedPiAgentMock.mock.calls.at(-1)?.[0] as { agentDir?: string };
1684-expect(call?.agentDir).toBe(agentDir);
1676+const call = requireLastMockCallArg(runEmbeddedPiAgentMock, "run embedded pi agent");
1677+expect(call.agentDir).toBe(agentDir);
16851678});
16861679});
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。