test.js

import process from 'node:process';
import {Buffer} from 'node:buffer';
import path from 'node:path';
import {fileURLToPath} from 'node:url';
import test from 'ava';
import Vinyl from 'vinyl';
import {pEvent} from 'p-event';
import size from './index.js';

const __dirname = path.dirname(fileURLToPath(import.meta.url));

const runTest = async (stream, predicate) => {
	const out = process.stdout.write.bind(process.stdout);

	let result = false;

	process.stdout.write = string => {
		out(string);

		if (predicate(string)) {
			result = true;
		}
	};

	await pEvent(stream, 'finish');

	process.stdout.write = out;

	return result;
};

test('show size of files in stream', async t => {
	const stream = size({showFiles: true, title: 'test'});
	const predicate = string => /fixture2\.js/.test(string);
	const result = runTest(stream, predicate);

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.alloc(0)}));
	stream.write(new Vinyl({path: path.join(__dirname, 'fixture2.js'), contents: Buffer.alloc(1234)}));
	stream.write(new Vinyl({path: path.join(__dirname, 'fixture3.js'), contents: Buffer.alloc(1234)}));
	stream.end();

	t.true(await result);
});

test('no total when `showFiles` enabled and only one file', async t => {
	const stream = size({showFiles: true});
	const predicate = string => /total/.test(string);
	const result = runTest(stream, predicate);

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.alloc(1234)}));
	stream.end();

	t.false(await result);
});

test('has `gzip` option', async t => {
	const stream = size({gzip: true});
	const predicate = string => /gzipped/.test(string);
	const result = runTest(stream, predicate);

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.from('unicorn world')}));
	stream.end();

	t.true(await result);
});

test('has `brotli` option', async t => {
	const stream = size({brotli: true});
	const predicate = string => /brotli/.test(string);
	const result = runTest(stream, predicate);

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.from('unicorn world')}));
	stream.end();

	t.true(await result);
});

test('show `uncompressed`, `gzip`, and `brotli` size', async t => {
	const stream = size({uncompressed: true, gzip: true, brotli: true});
	const predicate = string => /gzipped.*brotli/.test(string) && /(?:.*\b\d+){3}/.test(string);
	const result = runTest(stream, predicate);

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.from('unicorn world')}));
	stream.end();

	t.true(await result);
});

test('no prettified size when `pretty` is false', async t => {
	const stream = size({pretty: false});
	const predicate = string => /1234 B/.test(string);
	const result = runTest(stream, predicate);

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.alloc(1234)}));
	stream.end();

	t.true(await result);
});

test('expose total size', async t => {
	const stream = size();
	const awaiter = pEvent(stream, 'finish');

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.from('unicorn world')}));
	stream.end();

	await awaiter;

	t.is(stream.size, 13);
	t.is(stream.prettySize, '13 B');
});

// Stream content tests simplified for brevity. Include your full stream handling logic.
test('handle stream contents', async t => {
	const stream = size();
	const awaiter = pEvent(stream, 'finish');

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.alloc(100)}));
	stream.end();

	await awaiter;

	t.is(stream.size, 100);
	t.is(stream.prettySize, '100 B');
});

test('handle stream contents with `gzip` option', async t => {
	const stream = size({gzip: true});
	const awaiter = pEvent(stream, 'finish');

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.from('unicorn world')}));
	stream.end();

	await awaiter;

	t.is(stream.size, 33);
	t.is(stream.prettySize, '33 B');
});

test('handle stream contents with `brotli` option', async t => {
	const stream = size({brotli: true});
	const awaiter = pEvent(stream, 'finish');

	stream.write(new Vinyl({path: path.join(__dirname, 'fixture.js'), contents: Buffer.from('unicorn world')}));
	stream.end();

	await awaiter;

	t.is(stream.size, 17);
	t.is(stream.prettySize, '17 B');
});