1 files changed, 39 insertions, 1 deletions
diff --git a/src/features/chat/test-helpers.ts b/src/features/chat/test-helpers.ts
index d37b59e..07dad26 100644
--- a/src/features/chat/test-helpers.ts
+++ b/src/features/chat/test-helpers.ts
@@ -1,6 +1,6 @@
 import type { StoredChunk } from "@dispatch/wire";
 import type { ConversationCache } from "../conversation-cache";
-import type { ChatTransport, HistorySync } from "./ports";
+import type { ChatTransport, HistorySync, MetricsSync } from "./ports";
 
 export interface FakeTransport {
 	readonly sent: import("@dispatch/transport-contract").ChatSendMessage[];
@@ -46,6 +46,44 @@ export function createFakeHistorySync(): FakeHistorySync {
 	};
 }
 
+export interface FakeMetricsSync {
+	readonly calls: string[];
+	returnTurns: import("@dispatch/wire").TurnMetrics[];
+	/** If set, the next call will reject with this error. */
+	nextError: string | undefined;
+	readonly impl: MetricsSync;
+}
+
+export function createFakeMetricsSync(): FakeMetricsSync {
+	const calls: string[] = [];
+	let returnTurns: import("@dispatch/wire").TurnMetrics[] = [];
+	let nextError: string | undefined;
+	return {
+		calls,
+		get returnTurns() {
+			return returnTurns;
+		},
+		set returnTurns(v: import("@dispatch/wire").TurnMetrics[]) {
+			returnTurns = v;
+		},
+		get nextError() {
+			return nextError;
+		},
+		set nextError(v: string | undefined) {
+			nextError = v;
+		},
+		impl: async (conversationId) => {
+			calls.push(conversationId);
+			if (nextError !== undefined) {
+				const err = nextError;
+				nextError = undefined;
+				throw new Error(err);
+			}
+			return { turns: returnTurns };
+		},
+	};
+}
+
 export interface FakeCache {
 	readonly store: Map<string, StoredChunk[]>;
 	readonly impl: ConversationCache;