Function assertToolTurnOrdering

nodejs/test/e2e/multi_turn.e2e.test.ts:20–97

(turnEvents: SessionEvent[], turnDescription: string)

    }

    function assertToolTurnOrdering(turnEvents: SessionEvent[], turnDescription: string): void {
        const types = turnEvents.map((e) => e.type);
        const observedTypes = types.join(", ");

        const userMsgIdx = types.indexOf("user.message");
        expect(
            userMsgIdx,
            `Expected user.message in ${turnDescription}. Observed: ${observedTypes}`
        ).toBeGreaterThanOrEqual(0);

        const toolStarts = turnEvents
            .map((e, i) => ({ e, i }))
            .filter(({ e }) => e.type === "tool.execution_start");
        const toolCompletes = turnEvents
            .map((e, i) => ({ e, i }))
            .filter(({ e }) => e.type === "tool.execution_complete");

        expect(
            toolStarts.length,
            `Expected tool starts in ${turnDescription}. Observed: ${observedTypes}`
        ).toBeGreaterThan(0);
        expect(
            toolCompletes.length,
            `Expected tool completes in ${turnDescription}. Observed: ${observedTypes}`
        ).toBeGreaterThan(0);

        const firstToolStartIdx = Math.min(...toolStarts.map(({ i }) => i));
        expect(
            userMsgIdx,
            `Expected user.message before first tool start in ${turnDescription}. Observed: ${observedTypes}`
        ).toBeLessThan(firstToolStartIdx);

        for (const { e: complete, i: completeIdx } of toolCompletes) {
            const matchingStart = toolStarts.find(
                ({ e: start, i: startIdx }) =>
                    start.data.toolCallId === complete.data.toolCallId && startIdx < completeIdx
            );
            expect(
                matchingStart,
                `Expected matching tool start for tool complete with id ${complete.data.toolCallId}`
            ).toBeDefined();
        }

        const lastToolCompleteIdx = Math.max(...toolCompletes.map(({ i }) => i));
        let assistantAfterToolsIdx = -1;
        for (let i = lastToolCompleteIdx + 1; i < turnEvents.length; i++) {
            if (turnEvents[i]!.type === "assistant.message") {
                assistantAfterToolsIdx = i;
                break;
            }
        }

        let sessionIdleIdx = -1;
        const searchFrom = assistantAfterToolsIdx >= 0 ? assistantAfterToolsIdx + 1 : 0;
        for (let i = searchFrom; i < turnEvents.length; i++) {
            if (turnEvents[i]!.type === "session.idle") {
                sessionIdleIdx = i;
                break;