SylarLong/iztro

import { EarthlyBranchName } from '../../i18n';
import {
  earthlyBranchIndexToPalaceIndex,
  fixEarthlyBranchIndex,
  fixIndex,
  getAgeIndex,
  getBrightness,
  timeToIndex,
} from '../../utils';

describe('Utils', () => {
  test('fixIndex() should return correct index.', () => {
    expect(fixIndex(1)).toBe(1);
    expect(fixIndex(11)).toBe(11);
    expect(fixIndex(15, 20)).toBe(15);
    expect(fixIndex(-2)).toBe(10);
    expect(fixIndex(-3)).toBe(9);
    expect(fixIndex(-13)).toBe(11);
    expect(fixIndex(-2, 10)).toBe(8);
    expect(fixIndex(-3, 10)).toBe(7);
    expect(fixIndex(-15, 10)).toBe(5);
    expect(fixIndex(-27, 10)).toBe(3);
    expect(fixIndex(12)).toBe(0);
    expect(fixIndex(13)).toBe(1);
    expect(fixIndex(23)).toBe(11);
    expect(fixIndex(23, 10)).toBe(3);
    expect(fixIndex(37, 10)).toBe(7);
  });

  test('getBrightness() should return correct value', () => {
    expect(getBrightness('破军', fixEarthlyBranchIndex('午'))).toBe('庙');
    expect(getBrightness('太阴', fixEarthlyBranchIndex('酉'))).toBe('不');
    expect(getBrightness('天机', fixEarthlyBranchIndex('未'))).toBe('陷');
    expect(getBrightness('天府', fixEarthlyBranchIndex('申'))).toBe('得');
    expect(getBrightness('廉贞', fixEarthlyBranchIndex('子'))).toBe('平');
    expect(getBrightness('陀罗', fixEarthlyBranchIndex('亥'))).toBe('陷');
    expect(getBrightness('擎羊', fixEarthlyBranchIndex('亥'))).toBe('');
    expect(getBrightness('擎羊', fixEarthlyBranchIndex('酉'))).toBe('陷');
  });

  test('timeToIndex()', () => {
    const data = [
      [0, 0],
      [1, 1],
      [2, 1],
      [3, 2],
      [4, 2],
      [5, 3],
      [6, 3],
      [7, 4],
      [8, 4],
      [9, 5],
      [10, 5],
      [11, 6],
      [12, 6],
      [13, 7],
      [14, 7],
      [15, 8],
      [16, 8],
      [17, 9],
      [18, 9],
      [19, 10],
      [20, 10],
      [21, 11],
      [22, 11],
      [23, 12],
    ];

    data.forEach(([key, value]) => {
      expect(timeToIndex(key)).toEqual(value);
    });
  });

  test('getAgeIndex()', () => {
    const data = {
      yin: 2,
      woo: 2,
      xu: 2,
      shen: 8,
      zi: 8,
      chen: 8,
      si: 5,
      you: 5,
      chou: 5,
      hai: 11,
      mao: 11,
      wei: 11,
    };

    Object.entries(data).forEach(([key, value]) => {
      expect(getAgeIndex(key as EarthlyBranchName)).toEqual(value);
    });
  });

  test('earthlyBranchIndexToPalaceIndex()', () => {
    const data = {
      yin: 0,
      mao: 1,
      chen: 2,
      si: 3,
      woo: 4,
      wei: 5,
      shen: 6,
      you: 7,
      xu: 8,
      hai: 9,
      zi: 10,
      chou: 11,
    };

    Object.entries(data).forEach(([key, value]) => {
      expect(earthlyBranchIndexToPalaceIndex(key as EarthlyBranchName)).toEqual(value);
    });
  });
});