Files
OpenFrame/tests/unit/lib/upload-size.test.ts
T
yusufipek 7b60f3bf76 feat(uploads): size one upload against the account's own quota
The per-file ceiling was a flat 5 GiB from the environment, which is both
too small for a paying account with 200 GB of storage and unaware of what
an upload actually costs. The provider derives its own renditions from the
file (1080p, 720p and down) and bills them to the same account, so a file
allowed to fill the quota exactly is over it by the time it finishes
processing.

The ceiling is now 80% of whatever limit the account is held to: 160 GB on
the plan, 2.4 GB on a cardless trial, and it moves on its own when either
number changes. OPENFRAME_MAX_VIDEO_UPLOAD_BYTES keeps working as an
absolute cap for a host that wants one, where the lower of the two applies,
and an instance running without billing has no quota to divide and falls
back to the flat 5 GiB. The refusal now names the ceiling, which the old
one left the client to guess.

Finalize re-checks only the host cap. Re-deriving the account's ceiling
there would delete a finished upload over a plan that lapsed while the
bytes were in flight, and an upload larger than what was declared is
already caught by the declared-size check beside it.
2026-08-20 08:52:26 +03:00

77 lines
2.6 KiB
TypeScript

import { describe, expect, it } from 'vitest';
import { formatSizeLimit, parseDeclaredUploadSize } from '@/lib/upload-size';
const MAX = BigInt(5) * BigInt(1024) * BigInt(1024) * BigInt(1024);
function size(result: ReturnType<typeof parseDeclaredUploadSize>): bigint | null {
return 'sizeBytes' in result ? result.sizeBytes : null;
}
describe('parseDeclaredUploadSize', () => {
it('accepts a size sent as a string, which is how a client sends bytes it cannot hold in a number', () => {
expect(size(parseDeclaredUploadSize('4294967296', MAX))).toBe(BigInt(4294967296));
});
it('accepts a size sent as a number', () => {
expect(size(parseDeclaredUploadSize(1024, MAX))).toBe(BigInt(1024));
});
it('rejects a missing size', () => {
expect(parseDeclaredUploadSize(undefined, MAX)).toEqual({
error: 'sizeBytes must be a positive integer',
});
});
// Zero was the old behaviour of every Bunny init: it asked the quota whether
// it could store nothing, and the answer was always yes.
it('rejects zero', () => {
expect(parseDeclaredUploadSize(0, MAX)).toEqual({
error: 'sizeBytes must be a positive integer',
});
});
it('rejects a negative size', () => {
expect(parseDeclaredUploadSize(-1, MAX)).toEqual({
error: 'sizeBytes must be a positive integer',
});
});
it('rejects a fractional size rather than rounding it', () => {
expect(parseDeclaredUploadSize(1.5, MAX)).toEqual({
error: 'sizeBytes must be a positive integer',
});
});
it('rejects text that is not a number', () => {
expect(parseDeclaredUploadSize('a lot', MAX)).toEqual({
error: 'sizeBytes must be a positive integer',
});
});
it('rejects a size over the ceiling, naming the ceiling so the client knows what fits', () => {
expect(parseDeclaredUploadSize(MAX + BigInt(1), MAX)).toEqual({
error: 'File exceeds the maximum allowed upload size (5 GB)',
});
});
it('accepts a size exactly at the ceiling', () => {
expect(size(parseDeclaredUploadSize(MAX, MAX))).toBe(MAX);
});
});
describe('formatSizeLimit', () => {
it('drops the decimal on a whole number of gigabytes', () => {
expect(formatSizeLimit(BigInt(160) * BigInt(1024) ** BigInt(3))).toBe('160 GB');
});
it('keeps one decimal where the share of a small quota is not whole', () => {
expect(
formatSizeLimit((BigInt(3) * BigInt(1024) ** BigInt(3) * BigInt(80)) / BigInt(100))
).toBe('2.4 GB');
});
it('falls back to megabytes below a gigabyte, where "0.0 GB" would say nothing', () => {
expect(formatSizeLimit(BigInt(500) * BigInt(1024) * BigInt(1024))).toBe('500 MB');
});
});