Skip to content
Snippets Groups Projects
test.js 3.7 KiB
Newer Older
import stream from 'stream';
import http from 'http';
import test from 'ava';
import createTestServer from 'create-test-server';
import pify from 'pify';
Yadd's avatar
Yadd committed
import pEvent from 'p-event';
Yadd's avatar
Yadd committed
import mimicResponse from '.';
Yadd's avatar
Yadd committed
const nodejsMajorVersion = process.versions.node.split('.')[0];
let server;

test.before(async () => {
	server = await createTestServer();

Yadd's avatar
Yadd committed
	server.get('/', (_request, response) => {
		response.write('a');

		setTimeout(() => {
			response.end('sdf');
		}, 2);
	});

	server.get('/aborted', (_request, response) => {
		response.write('a');

		setTimeout(() => {
			response.socket.destroy();
		}, 2);
Yadd's avatar
Yadd committed
test('normal', async t => {
	const response = await pify(http.get, {errorFirst: false})(server.url);
	response.unicorn = '🦄';
	response.getContext = function () {
		return this;
	};

Yadd's avatar
Yadd committed
	const toStream = new stream.PassThrough({autoDestroy: false});
Yadd's avatar
Yadd committed
	mimicResponse(response, toStream);

	t.is(toStream.statusCode, 200);
	t.is(toStream.unicorn, '🦄');
	t.is(toStream.getContext(), response.getContext());
Yadd's avatar
Yadd committed

	response.resume();
	await pEvent(response, 'end');

	t.true(toStream.complete);
Yadd's avatar
Yadd committed

test('do not overwrite prototype properties', async t => {
	const response = await pify(http.get, {errorFirst: false})(server.url);
	response.unicorn = '🦄';
	response.getContext = function () {
		return this;
	};

	const origOn = response.on;
	response.on = function (name, handler) {
		return origOn.call(this, name, handler);
	};

Yadd's avatar
Yadd committed
	const toStream = new stream.PassThrough({autoDestroy: false});
Yadd's avatar
Yadd committed
	mimicResponse(response, toStream);

	t.false(Object.keys(toStream).includes('on'));
	t.is(toStream.statusCode, 200);
	t.is(toStream.unicorn, '🦄');
	t.is(toStream.getContext(), response.getContext());
Yadd's avatar
Yadd committed
	t.false(toStream.complete);

	response.resume();
	await pEvent(response, 'end');

	await new Promise(resolve => setImmediate(resolve));

	t.true(toStream.complete);
});

test('`aborted` event', async t => {
	const response = await pify(http.get, {errorFirst: false})(`${server.url}/aborted`);

	const toStream = new stream.PassThrough({autoDestroy: false});
	mimicResponse(response, toStream);

	await pEvent(toStream, 'aborted');

	t.true(toStream.destroyed);
});

test('autoDestroy must be false', async t => {
	const response = await pify(http.get, {errorFirst: false})(`${server.url}/aborted`);

	const toStream = new stream.PassThrough({autoDestroy: true});

	t.throws(() => mimicResponse(response, toStream), {
		message: 'The second stream must have the `autoDestroy` option set to `false`'
	});
});

test('`close` event', async t => {
	{
		const response = await pify(http.get, {errorFirst: false})(server.url);

		const toStream = new stream.PassThrough({autoDestroy: false});
		mimicResponse(response, toStream);

		t.true(response.readable);

		if (nodejsMajorVersion > 11) {
			t.false(response.readableEnded);
		}

		response.pipe(toStream);
		toStream.resume();

		await pEvent(toStream, 'close');

		t.false(response.readable);
		t.false(toStream.readable);

		if (nodejsMajorVersion > 11) {
			t.true(response.readableEnded);
			t.true(toStream.readableEnded);
		}
	}

	{
		const response = await pify(http.get, {errorFirst: false})(`${server.url}/aborted`);

		const toStream = new stream.PassThrough({autoDestroy: false});
		mimicResponse(response, toStream);

		t.true(response.readable);

		if (nodejsMajorVersion > 11) {
			t.false(response.readableEnded);
		}

		response.pipe(toStream);
		toStream.resume();

		await pEvent(toStream, 'close');

		if (nodejsMajorVersion < 12) {
			t.false(response.readable);
			t.true(toStream.readable);
		} else if (nodejsMajorVersion < 13) {
			t.true(response.readableEnded);
			t.false(toStream.readableEnded);
		} else {
			t.false(response.readableEnded);
			t.false(toStream.readableEnded);
		}
	}
Yadd's avatar
Yadd committed
});