Throw error if mock not found

firebase · Jan 29, 2025 · c68a16d · c68a16d
1 parent dafae52
commit c68a16d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/packages/vertexai/test-utils/mock-response.ts b/packages/vertexai/test-utils/mock-response.ts
@@ -49,6 +49,9 @@ export function getMockResponseStreaming(
   filename: string,
   chunkLength: number = 20
 ): Partial<Response> {
+  if (!(filename in mocksLookup)) {
+    throw Error(`Mock response file '${filename}' not found.`);
+  }
   const fullText = mocksLookup[filename];
 
   return {
@@ -57,6 +60,9 @@ export function getMockResponseStreaming(
 }
 
 export function getMockResponse(filename: string): Partial<Response> {
+  if (!(filename in mocksLookup)) {
+    throw Error(`Mock response file '${filename}' not found.`);
+  }
   const fullText = mocksLookup[filename];
   return {
     ok: true,