Fix embedder tests after telemetry removal

Author: mrubens

src/services/code-index/embedders/__tests__/bedrock.spec.ts

@@ -5,6 +5,20 @@ import { BedrockEmbedder } from "../bedrock"
 import { MAX_ITEM_TOKENS, INITIAL_RETRY_DELAY_MS } from "../../constants"
 
 // Mock the AWS SDK
+vitest.mock("@aws-sdk/client-bedrock-runtime", () => {
+	return {
+		BedrockRuntimeClient: vitest.fn().mockImplementation(() => ({
+			send: vitest.fn(),
+		})),
+		InvokeModelCommand: vitest.fn().mockImplementation((input) => ({
+			input,
+		})),
+	}
+})
+vitest.mock("@aws-sdk/credential-providers", () => ({
+	fromEnv: vitest.fn().mockReturnValue(Promise.resolve({})),
+	fromIni: vitest.fn().mockReturnValue(Promise.resolve({})),
+}))
 
 // Mock i18n
 vitest.mock("../../../../i18n", () => ({

src/services/code-index/embedders/__tests__/gemini.spec.ts

@@ -4,6 +4,7 @@ import { GeminiEmbedder } from "../gemini"
 import { OpenAICompatibleEmbedder } from "../openai-compatible"
 
 // Mock the OpenAICompatibleEmbedder
+vitest.mock("../openai-compatible")
 
 const MockedOpenAICompatibleEmbedder = OpenAICompatibleEmbedder as MockedClass<typeof OpenAICompatibleEmbedder>
 

src/services/code-index/embedders/__tests__/mistral.spec.ts

@@ -4,6 +4,7 @@ import { MistralEmbedder } from "../mistral"
 import { OpenAICompatibleEmbedder } from "../openai-compatible"
 
 // Mock the OpenAICompatibleEmbedder
+vitest.mock("../openai-compatible")
 
 const MockedOpenAICompatibleEmbedder = OpenAICompatibleEmbedder as MockedClass<typeof OpenAICompatibleEmbedder>
 

src/services/code-index/embedders/__tests__/openai-compatible-rate-limit.spec.ts

@@ -4,6 +4,7 @@ import { OpenAI } from "openai"
 import { OpenAICompatibleEmbedder } from "../openai-compatible"
 
 // Mock the OpenAI SDK
+vi.mock("openai")
 
 // Mock i18n
 vi.mock("../../../../i18n", () => ({

src/services/code-index/embedders/__tests__/openai-compatible.spec.ts

@@ -4,6 +4,7 @@ import { OpenAICompatibleEmbedder } from "../openai-compatible"
 import { MAX_ITEM_TOKENS, INITIAL_RETRY_DELAY_MS } from "../../constants"
 
 // Mock the OpenAI SDK
+vitest.mock("openai")
 
 // Mock global fetch
 global.fetch = vitest.fn()

src/services/code-index/embedders/__tests__/openai.spec.ts

@@ -5,6 +5,7 @@ import { OpenAiEmbedder } from "../openai"
 import { MAX_ITEM_TOKENS, INITIAL_RETRY_DELAY_MS } from "../../constants"
 
 // Mock the OpenAI SDK
+vitest.mock("openai")
 
 // Mock i18n
 vitest.mock("../../../../i18n", () => ({
@@ -435,9 +436,6 @@ describe("OpenAiEmbedder", () => {
 
 			it("should handle errors with failing toString method", async () => {
 				const testTexts = ["Hello world"]
-				// When vitest tries to display the error object in test output,
-				// it calls toString which throws "toString failed"
-				// This happens before our error handling code runs
 				const errorWithFailingToString = {
 					toString: () => {
 						throw new Error("toString failed")
@@ -446,9 +444,9 @@ describe("OpenAiEmbedder", () => {
 
 				mockEmbeddingsCreate.mockRejectedValue(errorWithFailingToString)
 
-				// The test framework itself throws "toString failed" when trying to
-				// display the error, so we need to expect that specific error
-				await expect(embedder.createEmbeddings(testTexts)).rejects.toThrow("toString failed")
+				await expect(embedder.createEmbeddings(testTexts)).rejects.toThrow(
+					"Failed to create embeddings after 3 attempts: Unknown error",
+				)
 			})
 
 			it("should handle errors from response.status property", async () => {

src/services/code-index/embedders/__tests__/openrouter.spec.ts

@@ -5,6 +5,7 @@ import { OpenRouterEmbedder, OPENROUTER_DEFAULT_PROVIDER_NAME } from "../openrou
 import { getModelDimension, getDefaultModelId } from "../../../../shared/embeddingModels"
 
 // Mock the OpenAI SDK
+vi.mock("openai")
 
 // Mock i18n
 vi.mock("../../../../i18n", () => ({

src/services/code-index/embedders/__tests__/vercel-ai-gateway.spec.ts

@@ -4,6 +4,9 @@ import { VercelAiGatewayEmbedder } from "../vercel-ai-gateway"
 import { OpenAICompatibleEmbedder } from "../openai-compatible"
 
 // Mock the OpenAICompatibleEmbedder
+vi.mock("../openai-compatible", () => ({
+	OpenAICompatibleEmbedder: vi.fn(),
+}))
 
 const MockedOpenAICompatibleEmbedder = vi.mocked(OpenAICompatibleEmbedder)