Files
grafana/packages/grafana-data/src/datetime/datemath.test.ts
2026-03-12 14:35:34 +02:00

290 lines
10 KiB
TypeScript

import { each } from 'lodash';
import * as dateMath from './datemath';
import { dateTime, DurationUnit, DateTime } from './moment_wrapper';
describe('DateMath', () => {
const spans: DurationUnit[] = ['s', 'm', 'h', 'd', 'w', 'M', 'y'];
const anchor = '2014-01-01T06:06:06.666Z';
const unix = dateTime(anchor).valueOf();
const format = 'YYYY-MM-DDTHH:mm:ss.SSSZ';
describe('errors', () => {
it('should return undefined if passed empty string', () => {
expect(dateMath.parse('')).toBe(undefined);
});
it('should return undefined if I pass an operator besides [+-/]', () => {
expect(dateMath.parse('now&1d')).toBe(undefined);
});
it('should return undefined if I pass a unit besides' + spans.toString(), () => {
expect(dateMath.parse('now+5f')).toBe(undefined);
});
it('should return undefined if rounding unit is not 1', () => {
expect(dateMath.parse('now/2y')).toBe(undefined);
expect(dateMath.parse('now/0.5y')).toBe(undefined);
});
it('should not go into an infinite loop when missing a unit', () => {
expect(dateMath.parse('now-0')).toBe(undefined);
expect(dateMath.parse('now-00')).toBe(undefined);
});
});
it('now/d should set to start of current day', () => {
const expected = new Date();
expected.setHours(0);
expected.setMinutes(0);
expected.setSeconds(0);
expected.setMilliseconds(0);
const startOfDay = dateMath.parse('now/d', false)!.valueOf();
expect(startOfDay).toBe(expected.getTime());
});
it('now/d on a utc dashboard should be start of the current day in UTC time', () => {
const today = new Date();
const expected = new Date(Date.UTC(today.getUTCFullYear(), today.getUTCMonth(), today.getUTCDate(), 0, 0, 0, 0));
const startOfDay = dateMath.parse('now/d', false, 'utc')!.valueOf();
expect(startOfDay).toBe(expected.getTime());
});
describe('with fiscal quarters', () => {
beforeEach(() => {
const fixedTime = dateTime('2023-07-05T06:06:06.666Z').valueOf();
jest.useFakeTimers({ now: fixedTime });
});
afterEach(() => {
jest.useRealTimers();
});
it('should parse current fiscal quarter correctly', () => {
const today = new Date();
const expected = new Date(Date.UTC(today.getUTCFullYear(), 6, 1, 0, 0, 0, 0));
const startOfDay = dateMath.parse('now/fQ', false, 'utc', 0)!.valueOf();
expect(startOfDay).toBe(expected.getTime());
});
it('should parse previous fiscal quarter correctly', () => {
const today = new Date();
const expected = new Date(Date.UTC(today.getUTCFullYear(), 3, 1, 0, 0, 0, 0));
const startOfDay = dateMath.parse('now-1Q/fQ', false, 'utc', 0)!.valueOf();
expect(startOfDay).toBe(expected.getTime());
});
describe('with a custom fiscal year start month', () => {
const FISCAL_YEAR_START_MONTH = 7; // August
it('should parse current fiscal quarter correctly', () => {
const today = new Date();
const expected = new Date(Date.UTC(today.getUTCFullYear(), 4, 1, 0, 0, 0, 0));
const startOfDay = dateMath.parse('now/fQ', false, 'utc', FISCAL_YEAR_START_MONTH)!.valueOf();
expect(startOfDay).toBe(expected.getTime());
});
it('should parse previous fiscal quarter correctly', () => {
const today = new Date();
const expected = new Date(Date.UTC(today.getUTCFullYear(), 1, 1, 0, 0, 0, 0));
const startOfDay = dateMath.parse('now-1Q/fQ', false, 'utc', FISCAL_YEAR_START_MONTH)!.valueOf();
expect(startOfDay).toBe(expected.getTime());
});
});
});
describe('subtraction', () => {
let now: DateTime;
let anchored: DateTime;
beforeEach(() => {
jest.useFakeTimers({ now: unix });
now = dateTime();
anchored = dateTime(anchor);
});
each(spans, (span) => {
const nowEx = 'now-5' + span;
const thenEx = anchor + '||-5' + span;
it('should return 5' + span + ' ago', () => {
expect(dateMath.parse(nowEx)!.format(format)).toEqual(now.subtract(5, span).format(format));
});
it('should return 5' + span + ' before ' + anchor, () => {
expect(dateMath.parse(thenEx)!.format(format)).toEqual(anchored.subtract(5, span).format(format));
});
});
afterEach(() => {
jest.useRealTimers();
});
});
describe('rounding', () => {
let now: DateTime;
beforeEach(() => {
jest.useFakeTimers({ now: unix });
now = dateTime();
});
each(spans, (span) => {
it('should round now to the beginning of the ' + span, () => {
expect(dateMath.parse('now/' + span)!.format(format)).toEqual(now.startOf(span).format(format));
});
it('should round now to the end of the ' + span, () => {
expect(dateMath.parse('now/' + span, true)!.format(format)).toEqual(now.endOf(span).format(format));
});
});
afterEach(() => {
jest.useRealTimers();
});
});
describe('isValid', () => {
it('should return false when invalid date text', () => {
expect(dateMath.isValid('asd')).toBe(false);
});
it('should return true when valid date text', () => {
expect(dateMath.isValid('now-1h')).toBe(true);
});
});
describe('relative time to date parsing', () => {
it('should handle negative time', () => {
const date = dateMath.parseDateMath('-2d', dateTime([2014, 1, 5]));
expect(date!.valueOf()).toEqual(dateTime([2014, 1, 3]).valueOf());
});
it('should handle multiple math expressions', () => {
const date = dateMath.parseDateMath('-2d-6h', dateTime([2014, 1, 5]));
expect(date!.valueOf()).toEqual(dateTime([2014, 1, 2, 18]).valueOf());
});
it('should return false when invalid expression', () => {
const date = dateMath.parseDateMath('2', dateTime([2014, 1, 5]));
expect(date).toEqual(undefined);
});
it('should strip whitespace from string', () => {
const date = dateMath.parseDateMath(' - 2d', dateTime([2014, 1, 5]));
expect(date!.valueOf()).toEqual(dateTime([2014, 1, 3]).valueOf());
});
it('should not mutate dateTime passed in', () => {
const dateInput = dateTime([2014, 1, 5]);
dateMath.parseDateMath(' - 2d', dateInput);
expect(dateInput.valueOf()).toEqual(dateTime([2014, 1, 5]).valueOf());
});
});
describe('isMathString', () => {
it('should return true when valid date text', () => {
expect(dateMath.isMathString('now-1h')).toBe(true);
});
it('should return false when an absolute date is inserted', () => {
const date = new Date();
const result = dateMath.isMathString(date);
expect(result).toBe(false);
});
it('should return false when invalid date text', () => {
expect(dateMath.isMathString('2 + 2')).toBe(false);
});
it('should return false if no text is passed', () => {
expect(dateMath.isMathString('')).toBe(false);
});
it('should return false if nothing is passed ', () => {
expect(dateMath.isMathString(' ')).toBe(false);
});
});
describe('Round to fiscal start/end', () => {
it('Should round to start of fiscal year when datetime is the same year as the start of the fiscal year', () => {
let date = dateMath.roundToFiscal(1, dateTime([2021, 3, 5]), 'y', false);
let expected = dateTime([2021, 1, 1]);
expect(date!.valueOf()).toEqual(expected.valueOf());
});
it('Should round to start of fiscal year when datetime is the next year from the start of the fiscal year', () => {
let date = dateMath.roundToFiscal(1, dateTime([2022, 0, 2]), 'y', false);
let expected = dateTime([2021, 1, 1]);
expect(date!.valueOf()).toEqual(expected.valueOf());
});
it('Should round to start of fiscal year when datetime is on a leap day', () => {
let date = dateMath.roundToFiscal(1, dateTime([2020, 1, 29]), 'y', false);
let expected = dateTime([2020, 1, 1]);
expect(date!.valueOf()).toEqual(expected.valueOf());
});
it('Should round to end of fiscal year when datetime is the same year as the start of the fiscal year', () => {
let date = dateMath.roundToFiscal(1, dateTime([2021, 5, 2]), 'y', true);
let expected = dateTime([2022, 0, 1]).endOf('M');
expect(date!.valueOf()).toEqual(expected.valueOf());
});
it('Should round to end of fiscal year when datetime is the next year from the start of the fiscal year', () => {
let date = dateMath.roundToFiscal(1, dateTime([2022, 0, 1]), 'y', true);
let expected = dateTime([2022, 0, 1]).endOf('M');
expect(date!.valueOf()).toEqual(expected.valueOf());
});
it('Should round to end of fiscal year when datetime is on a leap day', () => {
let date = dateMath.roundToFiscal(1, dateTime([2020, 1, 29]), 'y', true);
let expected = dateTime([2021, 0, 1]).endOf('M');
expect(date!.valueOf()).toEqual(expected.valueOf());
});
//fq1 = 2021-02-01 - 2021-04-30
//fq2 = 2021-05-01 - 2021-07-31
//fq3 = 2021-08-01 - 2021-10-31
//fq4 = 2021-11-01 - 2022-01-31
it('Should round to start of q2 when one month into q2', () => {
let date = dateMath.roundToFiscal(1, dateTime([2021, 6, 1]), 'Q', false);
let expected = dateTime([2021, 4, 1]);
expect(date!.valueOf()).toEqual(expected.valueOf());
});
it('Should round to start of q4 when datetime is in next year from fiscal year start', () => {
let date = dateMath.roundToFiscal(1, dateTime([2022, 0, 1]), 'Q', false);
let expected = dateTime([2021, 10, 1]);
expect(date!.valueOf()).toEqual(expected.valueOf());
});
it('Should round to end of q2 when one month into q2', () => {
let date = dateMath.roundToFiscal(1, dateTime([2021, 6, 1]), 'Q', true);
let expected = dateTime([2021, 6, 1]).endOf('M');
expect(date!.valueOf()).toEqual(expected.valueOf());
});
it('Should round to end of q4 when datetime is in next year from fiscal year start', () => {
let date = dateMath.roundToFiscal(1, dateTime([2022, 0, 1]), 'Q', true);
let expected = dateTime([2022, 0, 31]).endOf('M');
expect(date!.valueOf()).toEqual(expected.valueOf());
});
it('should handle fyStartMonths set later in the year correctly', () => {
// Use fyStartMonth to 10 (November)
// Fiscal quarters are then Nov-Jan, Feb-Apr, May-Jul, Aug-Oct
// Use 1st Jan as the date to round
let date = dateMath.roundToFiscal(10, dateTime([2022, 0, 1]), 'Q', false);
// This should round back to 1st Nov 2021
let expected = dateTime([2021, 10, 1]);
expect(date!.valueOf()).toEqual(expected.valueOf());
});
});
});