fix tests roundTimeByTimeRange

Author: v-tarasevich

datahub-web-react/src/app/entityV2/shared/tabs/Dataset/Stats/StatsTabV2/graphs/QueryCountGraph/useQueryCountData.ts

@@ -81,7 +81,11 @@ export default function useQueryCountData(urn: string | undefined, timeRange?: T
 
     useEffect(() => {
         if (timeRange && urn !== undefined && canViewDatasetUsage) {
-            const startTime = roundTimeByTimeRange(getCalendarStartTimeByTimeRange(Date.now(), timeRange), timeRange);
+            const startTime = roundTimeByTimeRange(
+                getCalendarStartTimeByTimeRange(Date.now(), timeRange),
+                timeRange,
+                moment.tz.guess(),
+            );
 
             getTimeRangeUsageAggregations({
                 variables: {

datahub-web-react/src/app/entityV2/shared/tabs/Dataset/Stats/StatsTabV2/graphs/__tests__/utils/roundTimeByTimeRange.test.ts

@@ -2,46 +2,47 @@ import { roundTimeByTimeRange } from '@app/entityV2/shared/tabs/Dataset/Stats/St
 import { TimeRange } from '@src/types.generated';
 
 const MOCKED_NOW = 1739791888317;
+const MOCKED_TZ = 'UTC';
 
 describe('roundTimeByTimeRange', () => {
     it('should return correct value for Day of TimeRange', () => {
-        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Day);
+        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Day, MOCKED_TZ);
 
-        expect(response).toEqual(1739739600000);
+        expect(response).toEqual(1739750400000);
     });
 
     it('should return correct value for Week of TimeRange', () => {
-        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Week);
+        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Week, MOCKED_TZ);
 
-        expect(response).toEqual(1739739600000);
+        expect(response).toEqual(1739750400000);
     });
 
     it('should return correct value for Month of TimeRange', () => {
-        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Month);
+        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Month, MOCKED_TZ);
 
-        expect(response).toEqual(1739739600000);
+        expect(response).toEqual(1739750400000);
     });
 
     it('should return correct value for Quarter of TimeRange', () => {
-        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Quarter);
+        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Quarter, MOCKED_TZ);
 
-        expect(response).toEqual(1739653200000);
+        expect(response).toEqual(1739664000000);
     });
 
     it('should return correct value for HalfYear of TimeRange', () => {
-        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.HalfYear);
+        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.HalfYear, MOCKED_TZ);
 
-        expect(response).toEqual(1738357200000);
+        expect(response).toEqual(1738368000000);
     });
 
     it('should return correct value for Year of TimeRange', () => {
-        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Year);
+        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.Year, MOCKED_TZ);
 
-        expect(response).toEqual(1738357200000);
+        expect(response).toEqual(1738368000000);
     });
 
     it('should return the same value for All of TimeRange', () => {
-        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.All);
+        const response = roundTimeByTimeRange(MOCKED_NOW, TimeRange.All, MOCKED_TZ);
 
         expect(response).toEqual(MOCKED_NOW);
     });

datahub-web-react/src/app/entityV2/shared/tabs/Dataset/Stats/StatsTabV2/graphs/utils.ts

@@ -1,11 +1,13 @@
 import dayjs from 'dayjs';
+import timezone from 'dayjs/plugin/timezone';
 import utc from 'dayjs/plugin/utc';
 
 import { LookbackWindow } from '@app/entityV2/shared/tabs/Dataset/Stats/lookbackWindows';
 import { getTimeWindowStart } from '@src/app/shared/time/timeUtils';
 import { DateInterval, TimeRange } from '@src/types.generated';
 
 dayjs.extend(utc);
+dayjs.extend(timezone);
 
 export type TimeSeriesDatum = {
     time: number;
@@ -175,10 +177,11 @@ export function getCalendarStartTimeByTimeRange(time: number, range: TimeRange):
     }
 }
 
-export function roundTimeByTimeRange(time: number | undefined, range: TimeRange) {
+export function roundTimeByTimeRange(time: number | undefined, range: TimeRange, tz?: string) {
     if (time === undefined) return time;
 
-    const dayjsTime = dayjs(time);
+    let dayjsTime = dayjs(time);
+    if (tz) dayjsTime = dayjsTime.tz(tz);
 
     switch (range) {
         case TimeRange.Day: