loadtest 8.0.9 → 8.1.1

package/lib/httpClient.js

@@ -3,7 +3,7 @@ import * as http from 'http'
 import * as https from 'https'
 import * as querystring from 'querystring'
 import * as websocket from 'websocket'
-import {addUserAgent} from './headers.js'
+import { addUserAgent } from './headers.js'
 import * as agentkeepalive from 'agentkeepalive'
 import * as HttpsProxyAgent from 'https-proxy-agent'
 
@@ -68,7 +68,7 @@ export class HttpClient {
 		}
 		if (this.options.cookies) {
 			if (Array.isArray(this.options.cookies)) {
-				this.params.headers.Cookie =  this.options.cookies.join('; ');
+				this.params.headers.Cookie = this.options.cookies.join('; ');
 			} else if (typeof this.options.cookies == 'string') {
 				this.params.headers.Cookie = this.options.cookies;
 			} else {
@@ -119,7 +119,7 @@ export class HttpClient {
 			return
 		}
 		const id = this.latency.begin();
-		const params = {...this.params, headers: {...this.params.headers}}
+		const params = { ...this.params, headers: { ...this.params.headers } }
 		this.customizeIndex(params)
 		const request = this.getRequest(id, params)
 		if (this.options.timeout) {
@@ -183,7 +183,7 @@ export class HttpClient {
 		if (this.params.protocol == 'https:') {
 			return https;
 		}
-		if (this.params.protocol == 'ws:') {
+		if (this.params.protocol == 'ws:' || this.params.protocol == 'wss:') {
 			return websocket;
 		}
 		return http;

package/lib/options.js

@@ -56,10 +56,10 @@ class Options {
 		if (!options.url) {
 			throw new Error('Missing URL in options')
 		}
-		if (!options.url.startsWith('http://') && !options.url.startsWith('https://') && !options.url.startsWith('ws://')) {
-			throw new Error(`Invalid URL ${options.url}, must be http://, https:// or ws://'`)
+		if (!options.url.startsWith('http://') && !options.url.startsWith('https://') && !options.url.startsWith('ws://') && !options.url.startsWith('wss://')) {
+			throw new Error(`Invalid URL ${options.url}, must be http://, https://, ws://, or wss://'`)
 		}
-		if (options.url.startsWith('ws:')) {
+		if (options.url.startsWith('ws:') || options.url.startsWith('wss:')) {
 			if (options.requestsPerSecond) {
 				throw new Error(`"requestsPerSecond" not supported for WebSockets`);
 			}

package/lib/pool.js

@@ -46,8 +46,7 @@ export class Pool {
 	}
 
 	createClient() {
-		// TODO: || this.options.url.startsWith('wss:'))
-		if (this.options.url.startsWith('ws:')) {
+		if (this.options.url.startsWith('ws:') || this.options.url.startsWith('wss:')) {
 			return new WebsocketClient(this.loadTest)
 		}
 		if (this.options.tcp) {
@@ -72,8 +71,11 @@ export class Pool {
 	 */
 	finishRequest(client, result, error) {
 		if (this.options.statusCallback) {
-			result.requestIndex = this.requestIndex++
-			result.instanceIndex = this.loadTest.instanceIndex
+			// result can be undefined if an error is thrown for a request
+			if (result) {
+				result.requestIndex = this.requestIndex++
+				result.instanceIndex = this.loadTest.instanceIndex
+			}
 			this.options.statusCallback(error, result);
 		}
 		if (this.loadTest.checkStop()) {

package/lib/websocket.js

@@ -1,5 +1,5 @@
 import websocket from 'websocket'
-import {BaseClient} from './baseClient.js'
+import { BaseClient } from './baseClient.js'
 
 let latency;
 
@@ -21,10 +21,13 @@ export class WebsocketClient extends BaseClient {
 	 * Start the websocket client.
 	 */
 	start() {
-		this.client = new websocket.client();
-		this.client.on('connectFailed', () => {});
+		this.client = new websocket.client({ rejectUnauthorized: false });
+		this.client.on('connectFailed', (e) => console.error(`Websocket connection error: ${e.message}`));
 		this.client.on('connect', connection => this.connect(connection));
-		this.client.connect(this.options.url, []);
+		if (this.options.insecure)
+			this.client.connect(this.options.url, undefined, undefined, undefined, { "rejectUnauthorized": false });
+		else
+			this.client.connect(this.options.url);
 	}
 
 	/**
@@ -84,7 +87,7 @@ export class WebsocketClient extends BaseClient {
 				try {
 					json = JSON.parse(message.utf8Data);
 				}
-				catch(e) {
+				catch (e) {
 					console.error('Invalid JSON: ' + message.utf8Data);
 					return;
 				}
@@ -104,7 +107,7 @@ export class WebsocketClient extends BaseClient {
 				requestFinished(null, json);
 			});
 
-			let message={some:"message"};
+			let message = { some: "message" };
 
 			if (this.generateMessage) {
 				message = this.generateMessage(id);

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "loadtest",
-	"version": "8.0.9",
+	"version": "8.1.1",
 	"type": "module",
 	"description": "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.",
 	"homepage": "https://github.com/alexfernandez/loadtest",