feat: Allow custom servers to be injected.

sngular · Apr 5, 2021 · b909180 · b909180
1 parent 5c87eb2
commit b909180
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/lib/open-api-mocker.js b/lib/open-api-mocker.js
@@ -12,11 +12,15 @@ class OpenApiMocker {
 		return 5000;
 	}
 
-	constructor({ port, server = 'express' }) {
-		// eslint-disable-next-line global-require, import/no-dynamic-require
-		const Server = require(`./mocker/${server}/server`);
-
-		this.server = new Server();
+	constructor({ port, server }) {
+		// Legacy handling of string based server that loads a server implementation based on folder name
+		if(typeof server === 'string' || server == null) {
+			// eslint-disable-next-line global-require, import/no-dynamic-require
+			const Server = require(`./mocker/${server || 'express'}/server`);
+			server = new Server();
+		}
+
+		this.server = server;
 		this.port = port || this.constructor.defaultPort;
 	}