import { describe, expect, it } from "vitest"

import type { SelectionChartPoint } from "../api/selection.types"
import { buildSelectionChartOption } from "./selection-chart-option"

function point(index: number): SelectionChartPoint {
  return {
    close: index + 10.5,
    d: index + 20,
    high: index + 11,
    j: index + 30,
    k: index + 10,
    low: index + 9,
    open: index + 10,
    trade_date: `2026-${String(Math.floor(index / 28) + 1).padStart(2, "0")}-${String((index % 28) + 1).padStart(2, "0")}`,
    volume: index + 1000,
  }
}

describe("selection chart option", () => {
  it("aligns all three grids and opens on exactly the last 120 points", () => {
    const points = Array.from({ length: 250 }, (_, index) => point(index))
    const option = buildSelectionChartOption(points)
    const dataZoom = option.dataZoom as Array<{
      endValue: string
      startValue: string
      xAxisIndex: number[]
    }>
    const series = option.series as Array<{
      data: unknown[]
      xAxisIndex: number
      yAxisIndex: number
    }>

    expect(dataZoom).toHaveLength(2)
    expect(dataZoom[0]).toMatchObject({
      endValue: points[249].trade_date,
      startValue: points[130].trade_date,
      xAxisIndex: [0, 1, 2],
    })
    expect(dataZoom[1].xAxisIndex).toEqual([0, 1, 2])
    expect(series[0]).toMatchObject({ xAxisIndex: 0, yAxisIndex: 0 })
    expect(series[1]).toMatchObject({ xAxisIndex: 1, yAxisIndex: 1 })
    expect(series.slice(2)).toEqual(
      expect.arrayContaining([
        expect.objectContaining({ xAxisIndex: 2, yAxisIndex: 2 }),
      ]),
    )
    expect(series[0].data[0]).toEqual([10, 10.5, 9, 11])
  })

  it("keeps missing candles and indicator values empty", () => {
    const incomplete = {
      ...point(0),
      close: null,
      d: null,
      high: null,
      j: null,
      k: null,
      volume: null,
    }
    const option = buildSelectionChartOption([incomplete])
    const series = option.series as Array<{
      connectNulls?: boolean
      data: unknown[]
    }>
    const candle = series[0].data[0] as number[]

    expect(candle).toHaveLength(4)
    expect(candle.every(Number.isNaN)).toBe(true)
    expect(series[1].data[0]).toMatchObject({ value: null })
    for (const indicator of series.slice(2)) {
      expect(indicator.connectNulls).toBe(false)
      expect(indicator.data).toEqual([null])
    }
  })
})