Mirrors/unleash.unleash
1
0
Fork 0
mirror of https://github.com/Unleash/unleash.git synced 2025-10-27 11:02:16 +01:00
Code Issues Projects Releases Wiki Activity

feat(1-3260): update test to add assertions

Browse Source
This commit is contained in:
Thomas Heartman 2025-01-21 15:26:31 +01:00
parent ac7ddb9519
commit 44ee534b9c
No known key found for this signature in database
GPG Key ID: BD1F880DAED1EE78
2 changed files with 32 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/lib/features/traffic-data-usage/traffic-data-usage-store-type.ts
View File

@ -8,7 +8,7 @@ export type IStatTrafficUsage = {
}; };
export type IStatMonthlyTrafficUsage = { export type IStatMonthlyTrafficUsage = {
month: Date; month: string;
trafficGroup: string; trafficGroup: string;
statusCodeSeries: number; statusCodeSeries: number;
count: number; count: number;

41
src/lib/features/traffic-data-usage/traffic-data-usage-store.test.ts
View File

@ -1,4 +1,4 @@
import { subMonths } from 'date-fns'; import { differenceInCalendarMonths, subMonths } from 'date-fns';
import dbInit, { type ITestDb } from '../../../test/e2e/helpers/database-init'; import dbInit, { type ITestDb } from '../../../test/e2e/helpers/database-init';
import getLogger from '../../../test/fixtures/no-logger'; import getLogger from '../../../test/fixtures/no-logger';
import type { ITrafficDataUsageStore, IUnleashStores } from '../../types'; import type { ITrafficDataUsageStore, IUnleashStores } from '../../types';
@ -200,24 +200,37 @@ test('can query for data from specific periods', async () => {
test('can query for monthly aggregation of data for a specified range', async () => { test('can query for monthly aggregation of data for a specified range', async () => {
const now = new Date(); const now = new Date();
const expectedValues: number[] = []; const expectedValues: { groupA: number; groupB: number }[] = [];
// fill in with data for the last 13 months // fill in with data for the last 13 months
for (let i = 0; i <= 12; i++) { for (let i = 0; i <= 12; i++) {
const month = subMonths(now, i).getMonth(); const month = subMonths(now, i).getMonth();
let monthAggregate = 0; let monthAggregateA = 0;
let monthAggregateB = 0;
for (let day = 1; day <= 5; day++) { for (let day = 1; day <= 5; day++) {
const dayValue = i + day; const dayValue = i + day;
monthAggregate += dayValue; const dayValueB = dayValue * 2;
const data = { monthAggregateA += dayValue;
monthAggregateB += dayValueB;
const dataA = {
day: new Date(2024, month, day), day: new Date(2024, month, day),
trafficGroup: 'default-period-query', trafficGroup: 'groupA',
statusCodeSeries: 200, statusCodeSeries: 200,
count: dayValue, count: dayValue,
}; };
await trafficDataUsageStore.upsert(data); await trafficDataUsageStore.upsert(dataA);
const dataB = {
day: new Date(2024, month, day),
trafficGroup: 'groupB',
statusCodeSeries: 200,
count: dayValueB,
};
await trafficDataUsageStore.upsert(dataB);
} }
expectedValues.push(monthAggregate); expectedValues.push({
groupA: monthAggregateA,
groupB: monthAggregateB,
});
} }
console.log(expectedValues); console.log(expectedValues);
@ -229,7 +242,15 @@ test('can query for monthly aggregation of data for a specified range', async ()
// should have the current month and the preceding n months // should have the current month and the preceding n months
expect(result.length).toBe(monthsBack + 1); expect(result.length).toBe(monthsBack + 1);
// the data should be aggregated correctly for (const entry of result) {
// expect(result.every(data, index) => { data.property === expectedValues[index] }) const index = differenceInCalendarMonths(
now,
new Date(entry.month),
);
const expected = expectedValues[index];
expect(entry.count).toBe(expected[entry.trafficGroup]);
}
} }
}); });