add: test

test/holidays/arrangement.test.ts

@@ -15,25 +15,25 @@ describe('Arrangement class', () => {
 
   it('should mark holiday correctly', () => {
     arrangement.nyd();
-    expect((arrangement as any).dayDetails.holiday).toBe(Holiday.NewYearsDay);
+    expect((arrangement as any).dayDetails.holiday).toBe(Holiday.NY);
   });
 
   it('should save holiday correctly', () => {
     arrangement.yearAt(2024).nyd().rest(1, 1);
     const date = dayjs('2024-01-01').format('YYYY-MM-DD');
-    expect(arrangement.holidays[date]).toBe(Holiday.NewYearsDay);
+    expect(arrangement.holidays[date]).toBe(Holiday.NY);
   });
 
   it('should save workday correctly', () => {
     arrangement.yearAt(2024).sf().work(2, 4);
     const date = dayjs('2024-02-04').format('YYYY-MM-DD');
-    expect(arrangement.workdays[date]).toBe(Holiday.SpringFestival);
+    expect(arrangement.workdays[date]).toBe(Holiday.S);
   });
 
   it('should save in-lieu day correctly', () => {
     arrangement.yearAt(2024).maf().inLieu(9, 16);
     const date = dayjs('2024-09-16').format('YYYY-MM-DD');
-    expect(arrangement.inLieuDays[date]).toBe(Holiday.MidAutumnFestival);
+    expect(arrangement.inLieuDays[date]).toBe(Holiday.M);
   });
 
   it('should save holiday range correctly', () => {
@@ -42,7 +42,7 @@ describe('Arrangement class', () => {
       dayjs(date).format('YYYY-MM-DD')
     );
     dates.forEach(date => {
-      expect(arrangement.holidays[date]).toBe(Holiday.SpringFestival);
+      expect(arrangement.holidays[date]).toBe(Holiday.S);
     });
   });
 

test/holidays/index.test.ts

@@ -0,0 +1,99 @@
+import dayjs from 'dayjs';
+import { wrapDate, getDates } from '../../src/utils';
+import { holidays, workdays, inLieuDays } from '../../src/holidays/constants';
+import Arrangement, { Holiday } from '../../src/holidays/arrangement';
+import {
+  isHoliday,
+  isWorkday,
+  isInLieu,
+  getDayDetail,
+  getHolidays,
+  getWorkdays,
+  findWorkday,
+} from '../../src/holidays';
+
+describe('Holiday Functions', () => {
+  test('isHoliday should return correct boolean values', () => {
+    const date1 = '2024-05-01';
+    const date2 = '2024-05-06';
+
+    expect(isHoliday(date1)).toBe(true); 
+    expect(isHoliday(date2)).toBe(false);
+  });
+
+  test('isWorkday should return correct boolean values', () => {
+    const date1 = '2024-05-01';
+    const date2 = '2024-05-06';
+
+    expect(isWorkday(date1)).toBe(false);
+    expect(isWorkday(date2)).toBe(true); 
+  });
+
+  test('isInLieu should return correct boolean values', () => {
+    const date1 = '2024-05-01';
+    const date2 = '2024-05-03';
+
+    expect(isInLieu(date1)).toBe(false);
+    expect(isInLieu(date2)).toBe(true);
+  });
+
+  test('getDayDetail should return correct details', () => {
+    const date = '2024-05-01';
+    const detail = getDayDetail(date);
+
+    expect(detail).toEqual({
+      date: '2024-05-01',
+      work: false,
+      name: holidays['2024-05-01'],
+    });
+  });
+
+  test('getHolidays should return correct holidays within a range', () => {
+    const start = '2024-05-01';
+    const end = '2024-05-31';
+    const holidaysInRange = getHolidays(start, end, true);
+
+    expect(holidaysInRange).toContain('2024-05-01');
+  });
+
+  test('getWorkdays should return correct workdays within a range', () => {
+    const start = '2024-05-01';
+    const end = '2024-05-31';
+    const workdaysInRange = getWorkdays(start, end, true);
+
+    expect(workdaysInRange).toContain('2024-05-06');
+  });
+
+  test('findWorkday should return correct workday', () => {
+    const date = '2024-05-01';
+    const nextWorkday = findWorkday(1, date);
+
+    expect(nextWorkday).toBe('2024-05-06');
+  });
+});
+
+describe('Arrangement Class', () => {
+  let arrangement: Arrangement;
+
+  beforeEach(() => {
+    arrangement = new Arrangement();
+  });
+
+  test('should correctly handle 2023 holidays', () => {
+    arrangement.yearAt(2024)
+    .nyd().rest(1, 1)
+    .sf().rest(2, 10).to(2, 17).work(2, 4).work(2, 18).inLieu(2, 15).to(2, 16)
+    .tsd().rest(4, 4).to(4, 6).work(4, 7).inLieu(4, 5)
+    .ld().rest(5, 1).to(5, 5).work(4, 28).work(5, 11).inLieu(5, 2).to(5, 3)
+    .dbf().rest(6, 10)
+    .maf().rest(9, 15).to(9, 17).work(9, 14).inLieu(9, 16)
+    .nd().rest(10, 1).to(10, 7).work(9, 29).work(10, 12).inLieu(10, 4).inLieu(10, 7)
+
+    expect(arrangement.holidays).toHaveProperty('2024-05-01');
+    expect(arrangement.holidays).toHaveProperty('2024-05-02');
+    expect(arrangement.holidays).toHaveProperty('2024-05-04');
+    expect(arrangement.holidays).toHaveProperty('2024-05-05');
+    expect(arrangement.workdays).toHaveProperty('2024-04-28');
+    expect(arrangement.workdays).toHaveProperty('2024-05-11');
+  });
+});