@@ -751,8 +751,12 @@ describe("generateAndAppendDreamNarrative", () => {
|
751 | 751 | limit: 5, |
752 | 752 | }); |
753 | 753 | expect(subagent.deleteSession).toHaveBeenCalledTimes(2); |
754 | | -expect(subagent.deleteSession.mock.calls[0]?.[0]).toEqual({ sessionKey: expectedSessionKey }); |
755 | | -expect(subagent.deleteSession.mock.calls[1]?.[0]).toEqual({ sessionKey: retrySessionKey }); |
| 754 | +expect(mockObjectArg(subagent.deleteSession, "delete session")).toEqual({ |
| 755 | +sessionKey: expectedSessionKey, |
| 756 | +}); |
| 757 | +expect(mockObjectArg(subagent.deleteSession, "delete session", 1)).toEqual({ |
| 758 | +sessionKey: retrySessionKey, |
| 759 | +}); |
756 | 760 | expectLogIncludes(logger.warn, "unknown model"); |
757 | 761 | }); |
758 | 762 | |
@@ -1038,15 +1042,19 @@ describe("generateAndAppendDreamNarrative", () => {
|
1038 | 1042 | logger, |
1039 | 1043 | }); |
1040 | 1044 | |
1041 | | -const firstSessionKey = subagent.run.mock.calls[0]?.[0]?.sessionKey; |
1042 | | -const secondSessionKey = subagent.run.mock.calls[1]?.[0]?.sessionKey; |
| 1045 | +const firstSessionKey = mockObjectArg(subagent.run, "subagent run").sessionKey; |
| 1046 | +const secondSessionKey = mockObjectArg(subagent.run, "subagent run", 1).sessionKey; |
1043 | 1047 | expect(firstSessionKey).toBeTypeOf("string"); |
1044 | 1048 | expect(secondSessionKey).toBeTypeOf("string"); |
1045 | 1049 | expect(firstSessionKey).not.toBe(secondSessionKey); |
1046 | 1050 | expect(firstSessionKey).toContain("dreaming-narrative-light-"); |
1047 | 1051 | expect(secondSessionKey).toContain("dreaming-narrative-light-"); |
1048 | | -expect(subagent.deleteSession.mock.calls[0]?.[0]?.sessionKey).toBe(firstSessionKey); |
1049 | | -expect(subagent.deleteSession.mock.calls[1]?.[0]?.sessionKey).toBe(secondSessionKey); |
| 1052 | +expect(mockObjectArg(subagent.deleteSession, "delete session").sessionKey).toBe( |
| 1053 | +firstSessionKey, |
| 1054 | +); |
| 1055 | +expect(mockObjectArg(subagent.deleteSession, "delete session", 1).sessionKey).toBe( |
| 1056 | +secondSessionKey, |
| 1057 | +); |
1050 | 1058 | }); |
1051 | 1059 | }); |
1052 | 1060 | |
|
