@@ -1045,13 +1045,13 @@ describe("runEmbeddedAttempt context engine sessionKey forwarding", () => {
|
1045 | 1045 | prePromptMessageCount: 1, |
1046 | 1046 | }); |
1047 | 1047 | |
1048 | | -expect(ingest).toHaveBeenCalled(); |
1049 | | -expect( |
1050 | | -ingest.mock.calls.every((call) => { |
1051 | | -const params = call[0]; |
1052 | | -return params.sessionKey === sessionKey; |
| 1048 | +expect(ingest).toHaveBeenCalledTimes(1); |
| 1049 | +expect(ingest).toHaveBeenCalledWith( |
| 1050 | +expect.objectContaining({ |
| 1051 | +message: doneMessage, |
| 1052 | + sessionKey, |
1053 | 1053 | }), |
1054 | | -).toBe(true); |
| 1054 | +); |
1055 | 1055 | }); |
1056 | 1056 | |
1057 | 1057 | it("forwards silentExpected to the embedded subscription", () => { |
@@ -1094,7 +1094,7 @@ describe("runEmbeddedAttempt context engine sessionKey forwarding", () => {
|
1094 | 1094 | |
1095 | 1095 | await finalizeTurn(sessionKey, createTestContextEngine({ bootstrap, assemble, afterTurn })); |
1096 | 1096 | |
1097 | | -expect(afterTurn).toHaveBeenCalled(); |
| 1097 | +expectCalledWithSessionKey(afterTurn, sessionKey); |
1098 | 1098 | expect( |
1099 | 1099 | hoisted.runContextEngineMaintenanceMock.mock.calls.some( |
1100 | 1100 | ([params]) => requireRecord(params, "maintenance params").reason === "turn", |
@@ -1294,7 +1294,7 @@ describe("runEmbeddedAttempt context engine sessionKey forwarding", () => {
|
1294 | 1294 | prePromptMessageCount: 1, |
1295 | 1295 | }); |
1296 | 1296 | |
1297 | | -expect(ingestBatch).toHaveBeenCalled(); |
| 1297 | +expectCalledWithSessionKey(ingestBatch, sessionKey); |
1298 | 1298 | expect( |
1299 | 1299 | hoisted.runContextEngineMaintenanceMock.mock.calls.some( |
1300 | 1300 | ([params]) => requireRecord(params, "maintenance params").reason === "turn", |
|
