mirror of
https://github.com/Unleash/unleash.git
synced 2025-01-01 00:08:27 +01:00
dfc0c3c63f
- Refactors data processing and overage calculations to separate hooks - Adds support for estimating traffic costs based on monthly usage up to current point - Adds accrued traffic charges to the billing page  
91 lines
2.8 KiB
TypeScript
91 lines
2.8 KiB
TypeScript
import {
|
|
toSelectablePeriod,
|
|
calculateOverageCost,
|
|
calculateEstimatedMonthlyCost,
|
|
calculateProjectedUsage,
|
|
} from './useTrafficData';
|
|
|
|
const testData4Days = [
|
|
{
|
|
label: 'Frontend',
|
|
data: [23_000_000, 22_000_000, 24_000_000, 21_000_000],
|
|
backgroundColor: 'red',
|
|
hoverBackgroundColor: 'red',
|
|
},
|
|
{
|
|
label: 'Admin',
|
|
data: [23_000_000, 22_000_000, 24_000_000, 21_000_000],
|
|
backgroundColor: 'red',
|
|
hoverBackgroundColor: 'red',
|
|
},
|
|
{
|
|
label: 'SDK',
|
|
data: [23_000_000, 22_000_000, 24_000_000, 21_000_000],
|
|
backgroundColor: 'red',
|
|
hoverBackgroundColor: 'red',
|
|
},
|
|
];
|
|
|
|
describe('traffic overage calculation', () => {
|
|
it('should return 0 if there is no overage this month', () => {
|
|
const dataUsage = 52_900_000;
|
|
const includedTraffic = 53_000_000;
|
|
const result = calculateOverageCost(dataUsage, includedTraffic);
|
|
expect(result).toBe(0);
|
|
});
|
|
|
|
it('should return 5 if overage this month is atleast 1 request above included', () => {
|
|
const dataUsage = 53_000_001;
|
|
const includedTraffic = 53_000_000;
|
|
const result = calculateOverageCost(dataUsage, includedTraffic);
|
|
expect(result).toBe(5);
|
|
});
|
|
|
|
it('doesnt estimate when having less than 5 days worth of data', () => {
|
|
const now = new Date();
|
|
const period = toSelectablePeriod(now);
|
|
const testNow = new Date(now.getFullYear(), now.getMonth(), 4);
|
|
const result = calculateEstimatedMonthlyCost(
|
|
period.key,
|
|
testData4Days,
|
|
53_000_000,
|
|
testNow,
|
|
);
|
|
expect(result).toBe(0);
|
|
});
|
|
|
|
it('needs 5 days or more to estimate for the month', () => {
|
|
const testData = testData4Days;
|
|
testData[0].data.push(23_000_000);
|
|
testData[1].data.push(23_000_000);
|
|
testData[2].data.push(23_000_000);
|
|
const now = new Date();
|
|
const period = toSelectablePeriod(now);
|
|
const testNow = new Date(now.getFullYear(), now.getMonth(), 5);
|
|
const result = calculateEstimatedMonthlyCost(
|
|
period.key,
|
|
testData,
|
|
53_000_000,
|
|
testNow,
|
|
);
|
|
expect(result).toBeGreaterThan(1430);
|
|
});
|
|
|
|
it('estimates projected data usage', () => {
|
|
const testData = testData4Days;
|
|
testData[0].data.push(22_500_000);
|
|
testData[1].data.push(22_500_000);
|
|
testData[2].data.push(22_500_000);
|
|
// Testing April 5th of 2024 (30 days)
|
|
const now = new Date(2024, 3, 5);
|
|
const period = toSelectablePeriod(now);
|
|
const result = calculateProjectedUsage(
|
|
now.getDate(),
|
|
testData,
|
|
period.dayCount,
|
|
);
|
|
// 22_500_000 * 3 * 30 = 2_025_000_000
|
|
expect(result).toBe(2_025_000_000);
|
|
});
|
|
});
|