Add statusCodes option (#18)

Co-authored-by: Sindre Sorhus <sindresorhus@gmail.com>

Author: fehmer

index.d.ts

@@ -19,6 +19,13 @@ export interface Options {
 	@default false
 	*/
 	useGet?: boolean;
+
+	/**
+	HTTP status codes to consider as successful responses.
+
+	@default [200]
+	*/
+	statusCodes?: readonly number[];
 }
 
 /**

index.js

@@ -1,6 +1,6 @@
 import http from 'node:http';
 
-export default function waitForLocalhost({port, path, useGet} = {}) {
+export default function waitForLocalhost({port, path, useGet, statusCodes = [200]} = {}) {
 	return new Promise(resolve => {
 		const retry = () => {
 			setTimeout(main, 200);
@@ -10,7 +10,7 @@ export default function waitForLocalhost({port, path, useGet} = {}) {
 
 		const doRequest = (ipVersion, next) => {
 			const request = http.request({method, port, path, family: ipVersion}, response => {
-				if (response.statusCode === 200) {
+				if (statusCodes.includes(response.statusCode)) {
 					resolve({ipVersion});
 					return;
 				}

readme.md

@@ -52,6 +52,13 @@ Default: `false`
 
 Use the `GET` HTTP-method instead of `HEAD` to check if the server is running.
 
+##### statusCodes
+
+Type: `number[]`\
+Default: `[200]`
+
+HTTP status codes to consider as successful responses.
+
 ## Related
 
 - [wait-for-localhost-cli](https://github.com/sindresorhus/wait-for-localhost-cli) - CLI for this module

test.js

@@ -59,3 +59,23 @@ test('use custom path', async t => {
 
 	await server.close();
 });
+
+test('use custom statusCodes', async t => {
+	t.plan(2);
+
+	const server = await createTestServer();
+	server.get('/', async (request, response) => {
+		await delay(1000);
+		response.status(202).end();
+		t.pass();
+	});
+
+	await waitForLocalhost({
+		port: server.port,
+		statusCodes: [201, 202],
+	});
+
+	t.pass();
+
+	await server.close();
+});