import { describe, expect, test } from "vitest";
import {
  JsonRpcErrorCodes,
  McpError,
  McpInvalidParamsError,
  McpMethodNotFoundError,
  McpPromptNotFoundError,
  McpResourceNotFoundError,
  McpToolNotFoundError,
} from "../index.ts";

// ---------------------------------------------------------------------------------------------------------------------

describe("McpError", () => {
  test("should be an instance of Error", () => {
    const error = new McpError("Test error");
    expect(error).toBeInstanceOf(Error);
  });

  test("should have correct name", () => {
    const error = new McpError("Test error");
    expect(error.name).toBe("McpError");
  });

  test("should have correct message", () => {
    const error = new McpError("Test error message");
    expect(error.message).toBe("Test error message");
  });

  test("should default to INTERNAL_ERROR code", () => {
    const error = new McpError("Test error");
    expect(error.code).toBe(JsonRpcErrorCodes.INTERNAL_ERROR);
  });

  test("should accept custom error code", () => {
    const error = new McpError("Test error", -32000);
    expect(error.code).toBe(-32000);
  });
});

// ---------------------------------------------------------------------------------------------------------------------

describe("McpMethodNotFoundError", () => {
  test("should be an instance of McpError", () => {
    const error = new McpMethodNotFoundError("unknown/method");
    expect(error).toBeInstanceOf(McpError);
    expect(error).toBeInstanceOf(Error);
  });

  test("should have correct name", () => {
    const error = new McpMethodNotFoundError("unknown/method");
    expect(error.name).toBe("McpMethodNotFoundError");
  });

  test("should include method name in message", () => {
    const error = new McpMethodNotFoundError("custom/method");
    expect(error.message).toBe("Method not found: custom/method");
  });

  test("should have METHOD_NOT_FOUND code", () => {
    const error = new McpMethodNotFoundError("test");
    expect(error.code).toBe(JsonRpcErrorCodes.METHOD_NOT_FOUND);
  });
});

// ---------------------------------------------------------------------------------------------------------------------

describe("McpToolNotFoundError", () => {
  test("should be an instance of McpError", () => {
    const error = new McpToolNotFoundError("unknown-tool");
    expect(error).toBeInstanceOf(McpError);
  });

  test("should have correct name", () => {
    const error = new McpToolNotFoundError("test-tool");
    expect(error.name).toBe("McpToolNotFoundError");
  });

  test("should include tool name in message", () => {
    const error = new McpToolNotFoundError("my-tool");
    expect(error.message).toBe("Tool not found: my-tool");
  });

  test("should store tool name as property", () => {
    const error = new McpToolNotFoundError("calculator");
    expect(error.tool).toBe("calculator");
  });

  test("should have METHOD_NOT_FOUND code", () => {
    const error = new McpToolNotFoundError("test");
    expect(error.code).toBe(JsonRpcErrorCodes.METHOD_NOT_FOUND);
  });
});

// ---------------------------------------------------------------------------------------------------------------------

describe("McpResourceNotFoundError", () => {
  test("should be an instance of McpError", () => {
    const error = new McpResourceNotFoundError("file:///test");
    expect(error).toBeInstanceOf(McpError);
  });

  test("should have correct name", () => {
    const error = new McpResourceNotFoundError("file:///test");
    expect(error.name).toBe("McpResourceNotFoundError");
  });

  test("should include URI in message", () => {
    const error = new McpResourceNotFoundError("config://app");
    expect(error.message).toBe("Resource not found: config://app");
  });

  test("should store URI as property", () => {
    const error = new McpResourceNotFoundError("db://users/123");
    expect(error.uri).toBe("db://users/123");
  });

  test("should have METHOD_NOT_FOUND code", () => {
    const error = new McpResourceNotFoundError("test://");
    expect(error.code).toBe(JsonRpcErrorCodes.METHOD_NOT_FOUND);
  });
});

// ---------------------------------------------------------------------------------------------------------------------

describe("McpPromptNotFoundError", () => {
  test("should be an instance of McpError", () => {
    const error = new McpPromptNotFoundError("unknown-prompt");
    expect(error).toBeInstanceOf(McpError);
  });

  test("should have correct name", () => {
    const error = new McpPromptNotFoundError("test-prompt");
    expect(error.name).toBe("McpPromptNotFoundError");
  });

  test("should include prompt name in message", () => {
    const error = new McpPromptNotFoundError("greeting");
    expect(error.message).toBe("Prompt not found: greeting");
  });

  test("should store prompt name as property", () => {
    const error = new McpPromptNotFoundError("code-review");
    expect(error.prompt).toBe("code-review");
  });

  test("should have METHOD_NOT_FOUND code", () => {
    const error = new McpPromptNotFoundError("test");
    expect(error.code).toBe(JsonRpcErrorCodes.METHOD_NOT_FOUND);
  });
});

// ---------------------------------------------------------------------------------------------------------------------

describe("McpInvalidParamsError", () => {
  test("should be an instance of McpError", () => {
    const error = new McpInvalidParamsError("Invalid params");
    expect(error).toBeInstanceOf(McpError);
  });

  test("should have correct name", () => {
    const error = new McpInvalidParamsError("test");
    expect(error.name).toBe("McpInvalidParamsError");
  });

  test("should have provided message", () => {
    const error = new McpInvalidParamsError("Missing required parameter: name");
    expect(error.message).toBe("Missing required parameter: name");
  });

  test("should have INVALID_PARAMS code", () => {
    const error = new McpInvalidParamsError("test");
    expect(error.code).toBe(JsonRpcErrorCodes.INVALID_PARAMS);
  });
});