|
| 1 | +import { describe, expect, test } from 'bun:test' |
| 2 | +import { randomBytes } from 'node:crypto' |
| 3 | +import { |
| 4 | + aesEcbPaddedSize, |
| 5 | + buildCdnDownloadUrl, |
| 6 | + buildCdnUploadUrl, |
| 7 | + decryptAesEcb, |
| 8 | + encryptAesEcb, |
| 9 | + guessMediaType, |
| 10 | + parseAesKey, |
| 11 | +} from '../media.js' |
| 12 | +import { UploadMediaType } from '../types.js' |
| 13 | + |
| 14 | +describe('AES-128-ECB', () => { |
| 15 | + test('encrypt then decrypt returns original data', () => { |
| 16 | + const key = randomBytes(16) |
| 17 | + const plaintext = Buffer.from('hello world test data!!') |
| 18 | + const ciphertext = encryptAesEcb(plaintext, key) |
| 19 | + expect(decryptAesEcb(ciphertext, key)).toEqual(plaintext) |
| 20 | + }) |
| 21 | + |
| 22 | + test('different keys produce different ciphertext', () => { |
| 23 | + const plaintext = Buffer.from('test data') |
| 24 | + expect( |
| 25 | + encryptAesEcb(plaintext, randomBytes(16)), |
| 26 | + ).not.toEqual(encryptAesEcb(plaintext, randomBytes(16))) |
| 27 | + }) |
| 28 | +}) |
| 29 | + |
| 30 | +describe('aesEcbPaddedSize', () => { |
| 31 | + test('pads to next 16-byte boundary', () => { |
| 32 | + expect(aesEcbPaddedSize(1)).toBe(16) |
| 33 | + expect(aesEcbPaddedSize(16)).toBe(32) |
| 34 | + expect(aesEcbPaddedSize(17)).toBe(32) |
| 35 | + expect(aesEcbPaddedSize(32)).toBe(48) |
| 36 | + }) |
| 37 | +}) |
| 38 | + |
| 39 | +describe('parseAesKey', () => { |
| 40 | + test('parses 16 raw bytes from base64', () => { |
| 41 | + const raw = randomBytes(16) |
| 42 | + expect(parseAesKey(raw.toString('base64'))).toEqual(raw) |
| 43 | + }) |
| 44 | + |
| 45 | + test('parses hex-encoded key from base64', () => { |
| 46 | + const raw = randomBytes(16) |
| 47 | + const b64 = Buffer.from(raw.toString('hex'), 'ascii').toString('base64') |
| 48 | + expect(parseAesKey(b64)).toEqual(raw) |
| 49 | + }) |
| 50 | + |
| 51 | + test('throws on invalid key length', () => { |
| 52 | + expect(() => parseAesKey(Buffer.from('short').toString('base64'))).toThrow( |
| 53 | + 'Invalid aes_key', |
| 54 | + ) |
| 55 | + }) |
| 56 | +}) |
| 57 | + |
| 58 | +describe('CDN URL builders', () => { |
| 59 | + test('buildCdnDownloadUrl encodes param', () => { |
| 60 | + expect(buildCdnDownloadUrl('abc=123', 'https://cdn.example.com')).toBe( |
| 61 | + 'https://cdn.example.com/download?encrypted_query_param=abc%3D123', |
| 62 | + ) |
| 63 | + }) |
| 64 | + |
| 65 | + test('buildCdnUploadUrl encodes params', () => { |
| 66 | + expect( |
| 67 | + buildCdnUploadUrl('https://cdn.example.com', 'param1', 'key1'), |
| 68 | + ).toBe( |
| 69 | + 'https://cdn.example.com/upload?encrypted_query_param=param1&filekey=key1', |
| 70 | + ) |
| 71 | + }) |
| 72 | +}) |
| 73 | + |
| 74 | +describe('guessMediaType', () => { |
| 75 | + test('detects image extensions', () => { |
| 76 | + expect(guessMediaType('photo.jpg')).toBe(UploadMediaType.IMAGE) |
| 77 | + expect(guessMediaType('photo.png')).toBe(UploadMediaType.IMAGE) |
| 78 | + expect(guessMediaType('photo.webp')).toBe(UploadMediaType.IMAGE) |
| 79 | + }) |
| 80 | + |
| 81 | + test('detects video extensions', () => { |
| 82 | + expect(guessMediaType('video.mp4')).toBe(UploadMediaType.VIDEO) |
| 83 | + expect(guessMediaType('video.mov')).toBe(UploadMediaType.VIDEO) |
| 84 | + }) |
| 85 | + |
| 86 | + test('defaults to FILE for unknown extensions', () => { |
| 87 | + expect(guessMediaType('doc.pdf')).toBe(UploadMediaType.FILE) |
| 88 | + expect(guessMediaType('archive.zip')).toBe(UploadMediaType.FILE) |
| 89 | + }) |
| 90 | +}) |
0 commit comments