import { describe, expect, test } from "bun:test";
import {
  canonicalExchange,
  canonicalTickerKey,
  CANONICAL_EXCHANGE_ALIASES,
  EXCHANGE_TIME_ZONES,
  parsePublicTickerKey,
  publicTickerKey,
  resolveExchangeTimeZone,
} from "./exchanges";

describe("exchange metadata", () => {
  test("has a valid timezone for every canonical exchange", () => {
    const canonicalExchanges = [...new Set(Object.values(CANONICAL_EXCHANGE_ALIASES))].sort();
    const missing = canonicalExchanges.filter((exchange) => !EXCHANGE_TIME_ZONES[exchange]);

    expect(missing).toEqual([]);
    for (const exchange of canonicalExchanges) {
      expect(() => new Intl.DateTimeFormat("en-US", {
        timeZone: EXCHANGE_TIME_ZONES[exchange],
      })).not.toThrow();
    }
  });

  test("normalizes broker venue aliases used by portfolio rows", () => {
    expect(canonicalExchange("LSEETF")).toBe("LSE");
    expect(canonicalExchange("TSE")).toBe("TSX");
    expect(canonicalExchange("AEB")).toBe("AMS");
    expect(canonicalExchange("EURONEXT")).toBe("AMS");
    expect(resolveExchangeTimeZone("LSEETF")).toBe("Europe/London");
    expect(resolveExchangeTimeZone("TSE")).toBe("America/Toronto");
    expect(resolveExchangeTimeZone("AEB")).toBe("Europe/Amsterdam");
  });

  test("round-trips public exchange-qualified ticker keys", () => {
    expect(publicTickerKey("3hnx", "LSE")).toBe("3HNX:XLON");
    expect(parsePublicTickerKey("3hnx:xlon")).toEqual({ symbol: "3HNX", exchange: "LSE" });
  });
});


test("listing keys remain idempotent across persisted, public, and canonical forms", () => {
  for (const exchange of ["LSE", "NASDAQ", "HKEX", "TSXV", "CSE", "XETRA"]) {
    const publicKey = publicTickerKey("VOD", exchange);
    const canonicalKey = canonicalTickerKey("VOD", exchange);
    expect(publicTickerKey(publicKey, exchange)).toBe(publicKey);
    expect(publicTickerKey(canonicalKey)).toBe(publicKey);
    expect(canonicalTickerKey(publicKey, exchange)).toBe(canonicalKey);
    expect(canonicalTickerKey(canonicalKey)).toBe(canonicalKey);
    expect(parsePublicTickerKey(publicKey)).toEqual({ symbol: "VOD", exchange: canonicalExchange(exchange) });
  }
  // A saved UK listing cannot switch back to a remembered US listing.
  expect(publicTickerKey("VOD:XLON", "NASDAQ")).toBe("VOD:XLON");
  expect(canonicalTickerKey("VOD:XLON", "NASDAQ")).toBe("VOD:LSE");
  expect(publicTickerKey("BRK.B")).toBe("BRK.B");
});

test("repeated exchange suffixes are repaired before provider and snapshot lookup", () => {
  expect(parsePublicTickerKey("VOD:XLON:XLON")).toEqual({ symbol: "VOD", exchange: "LSE" });
  expect(publicTickerKey("VOD:LSE:XLON", "LSE")).toBe("VOD:XLON");
  expect(canonicalTickerKey("VOD:XLON:LSE")).toBe("VOD:LSE");
});