🌲 test: Add E2E Coverage for Message Tree Streaming (#13570)

* add e2e message tree stream coverage

* fix e2e message tree review findings

* expand message tree e2e recovery coverage

* fix stream-start failure recovery coverage
This commit is contained in:
Danny Avila 2026-06-07 09:21:27 -04:00 committed by GitHub
parent 1612dba353
commit 15108f0f2f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 878 additions and 28 deletions

View file

@ -17,6 +17,7 @@ import type { QueryClient } from '@tanstack/react-query';
import type { LocalizeFunction } from '~/common';
export const TEXT_KEY_DIVIDER = '|||';
export const STREAM_START_FAILED_METADATA_KEY = 'streamStartFailed';
type SiblingIndexLookup = (parentMessageId: string | null | undefined) => number;
@ -136,6 +137,16 @@ export const getAllContentText = (message?: TMessage | null): string => {
return '';
};
export const hasStreamStartFailed = (message?: Pick<TMessage, 'metadata'> | null): boolean =>
message?.metadata?.[STREAM_START_FAILED_METADATA_KEY] === true;
export const markStreamStartFailedMetadata = (
metadata?: TMessage['metadata'],
): TMessage['metadata'] => ({
...(metadata ?? {}),
[STREAM_START_FAILED_METADATA_KEY]: true,
});
const getLatestContentForKey = (message: TMessage): string => {
const formatText = (str: string, index: number): string => {
if (str.length === 0) {