import { describe, expect, test } from "bun:test";

import {
  getAnyProviderEnvVar,
  getLlmProviderEnvVar,
  getSearchProviderEnvVar,
} from "../provider-env-vars.js";

describe("getLlmProviderEnvVar", () => {
  test("returns ANTHROPIC_API_KEY for anthropic", () => {
    expect(getLlmProviderEnvVar("anthropic")).toBe("ANTHROPIC_API_KEY");
  });

  test("returns OPENAI_API_KEY for openai", () => {
    expect(getLlmProviderEnvVar("openai")).toBe("OPENAI_API_KEY");
  });

  test("returns GEMINI_API_KEY for gemini", () => {
    expect(getLlmProviderEnvVar("gemini")).toBe("GEMINI_API_KEY");
  });

  test("returns FIREWORKS_API_KEY for fireworks", () => {
    expect(getLlmProviderEnvVar("fireworks")).toBe("FIREWORKS_API_KEY");
  });

  test("returns OPENROUTER_API_KEY for openrouter", () => {
    expect(getLlmProviderEnvVar("openrouter")).toBe("OPENROUTER_API_KEY");
  });

  test("returns undefined for ollama (keyless provider)", () => {
    expect(getLlmProviderEnvVar("ollama")).toBeUndefined();
  });

  test("returns undefined for search providers (out of scope)", () => {
    expect(getLlmProviderEnvVar("brave")).toBeUndefined();
    expect(getLlmProviderEnvVar("perplexity")).toBeUndefined();
    expect(getLlmProviderEnvVar("tavily")).toBeUndefined();
    expect(getLlmProviderEnvVar("firecrawl")).toBeUndefined();
  });

  test("returns undefined for unknown provider", () => {
    expect(getLlmProviderEnvVar("unknown-provider")).toBeUndefined();
  });
});

describe("getSearchProviderEnvVar", () => {
  test("returns BRAVE_API_KEY for brave", () => {
    expect(getSearchProviderEnvVar("brave")).toBe("BRAVE_API_KEY");
  });

  test("returns PERPLEXITY_API_KEY for perplexity", () => {
    expect(getSearchProviderEnvVar("perplexity")).toBe("PERPLEXITY_API_KEY");
  });

  test("returns TAVILY_API_KEY for tavily", () => {
    expect(getSearchProviderEnvVar("tavily")).toBe("TAVILY_API_KEY");
  });

  test("returns FIRECRAWL_API_KEY for firecrawl", () => {
    expect(getSearchProviderEnvVar("firecrawl")).toBe("FIRECRAWL_API_KEY");
  });

  test("returns undefined for LLM providers (out of scope)", () => {
    expect(getSearchProviderEnvVar("anthropic")).toBeUndefined();
    expect(getSearchProviderEnvVar("openai")).toBeUndefined();
  });

  test("returns undefined for unknown provider", () => {
    expect(getSearchProviderEnvVar("unknown-provider")).toBeUndefined();
  });
});

describe("getAnyProviderEnvVar", () => {
  test("returns LLM env var for LLM providers", () => {
    expect(getAnyProviderEnvVar("anthropic")).toBe("ANTHROPIC_API_KEY");
    expect(getAnyProviderEnvVar("openai")).toBe("OPENAI_API_KEY");
  });

  test("returns search env var for search providers", () => {
    expect(getAnyProviderEnvVar("brave")).toBe("BRAVE_API_KEY");
    expect(getAnyProviderEnvVar("perplexity")).toBe("PERPLEXITY_API_KEY");
    expect(getAnyProviderEnvVar("tavily")).toBe("TAVILY_API_KEY");
    expect(getAnyProviderEnvVar("firecrawl")).toBe("FIRECRAWL_API_KEY");
  });

  test("returns undefined for ollama (keyless LLM provider)", () => {
    expect(getAnyProviderEnvVar("ollama")).toBeUndefined();
  });

  test("returns undefined for unknown provider", () => {
    expect(getAnyProviderEnvVar("unknown")).toBeUndefined();
  });
});