frontend/test/common/datetime/calc_date.test.ts

77 lines
2.2 KiB
TypeScript

import type { HassConfig } from "home-assistant-js-websocket";
import { describe, it, expect } from "vitest";
import { addDays } from "date-fns";
import {
calcDate,
calcDateProperty,
calcDateDifferenceProperty,
shiftDateRange,
} from "../../../src/common/datetime/calc_date";
import {
type FrontendLocaleData,
TimeZone,
} from "../../../src/data/translation";
const locale: FrontendLocaleData = {
language: "en-US",
time_zone: TimeZone.local,
} as any;
const localeServer: FrontendLocaleData = {
language: "en-US",
time_zone: TimeZone.server,
} as any;
const config: HassConfig = { time_zone: "UTC" } as any;
describe("calcDate", () => {
it("should calculate date correctly", () => {
const date = new Date(2024, 1, 28);
const result = calcDate(date, addDays, locale, config, 1);
expect(result).toEqual(new Date(2024, 1, 29));
});
it("should calculate date correctly with server time zone", () => {
const date = new Date(2024, 1, 28);
const result = calcDate(date, addDays, localeServer, config, 1);
expect(result).toEqual(new Date(2024, 1, 29));
});
});
describe("calcDateProperty", () => {
it("should calculate date property correctly", () => {
const date = new Date(2023, 0, 1);
const options = "test-options";
const result = calcDateProperty(
date,
(d, o) => (o === options ? d.getDate() : false),
locale,
config,
options
);
expect(result).toBe(1);
});
});
describe("calcDateDifferenceProperty", () => {
it("should calculate date difference property correctly", () => {
const startDate = new Date(2023, 0, 1);
const endDate = new Date(2023, 0, 2);
const result = calcDateDifferenceProperty(
endDate,
startDate,
(d, o) => d.getDate() - o.getDate(),
locale,
config
);
expect(result).toBe(1);
});
});
describe("shiftDateRange", () => {
it("should shift date range correctly", () => {
const startDate = new Date(2024, 0, 1);
const endDate = new Date(2024, 0, 31);
const result = shiftDateRange(startDate, endDate, true, locale, config);
expect(result.start).toEqual(new Date(2024, 1, 1));
expect(result.end).toEqual(new Date(2024, 1, 29, 23, 59, 59, 999));
});
});