loadtest/lib/headers.js

Run load tests for your web application. Mostly ab-compatible interface, with an option to force requests per second. Includes an API for automated load testing.

/**
 * Add all raw headers given to the given array.
 */
export function addHeaders(rawHeaders, headers) {
	if (Array.isArray(rawHeaders)) {
		rawHeaders.forEach(function(header) {
			addTextHeader(header, headers);
		});
	} else if (typeof rawHeaders == 'string') {
		addTextHeader(rawHeaders, headers);
	} else if (typeof rawHeaders == 'object') {
		for (const key of Object.keys(rawHeaders)) {
			addHeader(key, rawHeaders[key], headers)
		}
	} else {
		console.error('Invalid header structure %j, it should be an array', rawHeaders);
	}
}

function addTextHeader(rawHeader, headers) {
	if (!rawHeader.includes(':')) {
		return console.error('Invalid header %s, it should be in the form -H key:value', rawHeader);
	}
	const index = rawHeader.indexOf(':');
	const key = rawHeader.substr(0, index);
	const value = rawHeader.substr(index + 1);
	addHeader(key, value, headers)
}

function addHeader(key, value, headers) {
	headers[key.toLowerCase()] = value;
}

/**
 * Add a user-agent header if not present.
 */
export function addUserAgent(headers) {
	if(!headers['user-agent']) {
		addHeader('user-agent', 'node.js loadtest bot', headers)
	}
}