import { assert, beforeAll, expect, test } from 'vite-plus/test';
import { ArrayBuffers } from './ArrayBuffers';

beforeAll(() => {
	expect(ArrayBuffers.isNativeBufferAvailable()).toBeTruthy();
	ArrayBuffers.setNativeBufferAllowed(false);
});

test('base64: ignore whitespace', () => {
	const text = '\n   YW9ldQ==  ';
	const buf = ArrayBuffers.from(text, 'base64');
	expect(ArrayBuffers.toString(buf)).toBe('aoeu');
});

test('base64: strings without padding', () => {
	expect(ArrayBuffers.toString(ArrayBuffers.from('YW9ldQ', 'base64'))).toBe('aoeu');
});

test('base64: newline in utf8 -- should not be an issue', () => {
	assert.equal(
		ArrayBuffers.toString(
			ArrayBuffers.from('LS0tCnRpdGxlOiBUaHJlZSBkYXNoZXMgbWFya3MgdGhlIHNwb3QKdGFnczoK', 'base64'),
			'utf8',
		),
		'---\ntitle: Three dashes marks the spot\ntags:\n',
	);
});

test('base64: newline in base64 -- should get stripped', () => {
	assert.equal(
		ArrayBuffers.toString(
			ArrayBuffers.from(
				'LS0tCnRpdGxlOiBUaHJlZSBkYXNoZXMgbWFya3MgdGhlIHNwb3QKdGFnczoK\nICAtIHlhbWwKICAtIGZyb250LW1hdHRlcgogIC0gZGFzaGVzCmV4cGFuZWQt',
				'base64',
			),
			'utf8',
		),
		'---\ntitle: Three dashes marks the spot\ntags:\n  - yaml\n  - front-matter\n  - dashes\nexpaned-',
	);
});

test('base64: tab characters in base64 - should get stripped', () => {
	assert.equal(
		ArrayBuffers.toString(
			ArrayBuffers.from(
				'LS0tCnRpdGxlOiBUaHJlZSBkYXNoZXMgbWFya3MgdGhlIHNwb3QKdGFnczoK\t\t\t\tICAtIHlhbWwKICAtIGZyb250LW1hdHRlcgogIC0gZGFzaGVzCmV4cGFuZWQt',
				'base64',
			),
			'utf8',
		),
		'---\ntitle: Three dashes marks the spot\ntags:\n  - yaml\n  - front-matter\n  - dashes\nexpaned-',
	);
});

test('base64: invalid non-alphanumeric characters -- should be stripped', () => {
	expect(ArrayBuffers.toString(ArrayBuffers.from('!"#$%&\'()*,.:;<=>?@[\\]^`{|}~', 'base64'), 'utf8')).toBe('');
});

test('base64: high byte', () => {
	const highByte = ArrayBuffers.from([128]);
	assert.deepEqual(ArrayBuffers.alloc(1, ArrayBuffers.toString(highByte, 'base64'), 'base64'), highByte);
});

test('fromBase64', () => {
	const text = 'aoeu';
	const buf = ArrayBuffers.from(text, 'utf8', Uint8Array);
	expect(ArrayBuffers.fromBase64(ArrayBuffers.toBase64(buf))).toEqual(buf);
});

test('fromHex', () => {
	const text = 'aoeu';
	const buf = ArrayBuffers.from(text, 'utf8', Uint8Array);
	expect(ArrayBuffers.fromHex(ArrayBuffers.toHex(buf))).toEqual(buf);
});

test('toHex', () => {
	const text = 'aoeu';
	const buf = ArrayBuffers.from(text, 'utf8', Uint8Array);
	expect(ArrayBuffers.toHex(buf)).toEqual('616f6575');
});

test('toBase64', () => {
	const text = 'aoeu';
	const buf = ArrayBuffers.from(text, 'utf8', Uint8Array);
	expect(ArrayBuffers.toBase64(buf)).toEqual('YW9ldQ==');
});

test('base64: utf8 string', () => {
	const text = '你好';
	// latin1 only
	expect(() => btoa(text)).toThrow('Invalid character');
	expect(ArrayBuffers.toBase64(text)).toEqual('5L2g5aW9');
});