import { describe, it, expect } from 'vitest';
import { buildColorScale, FALLBACK_COLORS } from '../src/shared/color-scale';
import type { DayEntry } from '../src/types';

describe('buildColorScale', () => {
  const days: DayEntry[] = [
    { date: '2025-01-01', hours: 2, count: 3 },
    { date: '2025-01-02', hours: 8, count: 5 },
    { date: '2025-01-03', hours: 0, count: 0 },
  ];

  it('builds scale for hours metric', () => {
    const scale = buildColorScale(days, 'hours');
    // max hours is 8, so domain is [0, 8]
    expect(scale(0)).toBe(FALLBACK_COLORS[0]);
    expect(scale(8)).toBe(FALLBACK_COLORS[3]);
  });

  it('builds scale for count metric', () => {
    const scale = buildColorScale(days, 'count');
    // max count is 5, so domain is [0, 5]
    expect(scale(0)).toBe(FALLBACK_COLORS[0]);
    expect(scale(5)).toBe(FALLBACK_COLORS[3]);
  });

  it('defaults to hours metric', () => {
    const scale = buildColorScale(days);
    expect(scale(8)).toBe(FALLBACK_COLORS[3]);
  });

  it('handles empty array with domain [0, 1]', () => {
    const scale = buildColorScale([]);
    expect(scale.domain()).toEqual([0, 1]);
  });
});