with-server.ts

import {promisify} from 'util';
import http = require('http');
import tempy = require('tempy');
import createTestServer = require('create-test-server');
import lolex = require('lolex');
import got from '../../source';

const generateHook = ({install}) => async (t, run) => {
	const clock = install ? lolex.install() : lolex.createClock();

	// @ts-ignore
	const server = await createTestServer({
		bodyParser: {
			type: () => false
		}
	});

	const options = {
		avaTest: t.title,
		handlers: [
			(options, next) => {
				const result = next(options);

				clock.tick(0);

				result.on('response', () => {
					clock.tick(0);
				});

				return result;
			}
		]
	};

	// @ts-ignore Ignore errors for extending got, for the tests
	const preparedGot = got.extend({prefixUrl: server.url, ...options});
	// @ts-ignore Ignore errors for extending got, for the tests
	preparedGot.secure = got.extend({prefixUrl: server.sslUrl, ...options});

	server.hostname = (new URL(server.url)).hostname;
	server.sslHostname = (new URL(server.sslUrl)).hostname;

	try {
		await run(t, server, preparedGot, clock);
	} finally {
		await server.close();
	}

	if (install) {
		// @ts-ignore This is a global clock.
		clock.uninstall();
	}
};

export default generateHook({install: false});

export const withServerAndLolex = generateHook({install: true});

// TODO: remove this when `create-test-server` supports custom listen
export const withSocketServer = async (t, run) => {
	const socketPath = tempy.file({extension: 'socket'});

	const server = http.createServer((request, response) => {
		server.emit(request.url, request, response);
	}) as any;

	server.socketPath = socketPath;

	await promisify(server.listen.bind(server))(socketPath);

	try {
		await run(t, server);
	} finally {
		await promisify(server.close.bind(server))();
	}
};