Merge 716f545 into b0f0f0d

cwhenderson20 · Apr 3, 2018 · 94420e0 · 94420e0
2 parents b0f0f0d + 716f545
commit 94420e0
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/mocksConfig.js b/lib/mocksConfig.js
@@ -2,15 +2,16 @@
 
 function generateConfig() {
 	const env = process.env.NODE_ENV || "development";
+	const host = process.env.MONGODB_HOST || "localhost";
+	const port = process.env.MONGODB_PORT || 27017;
+	let db = process.env.MONGODB_DB || "mockSQS";
 
 	if (env === "test") {
-		return {
-			db: process.env.MOCK_SQS_DB || "mongodb://localhost:27017/mockSQS_test",
-		};
+		db = `${db}_${env}`;
 	}
 
 	return {
-		db: process.env.MOCK_SQS_DB || "mongodb://localhost:27017/mockSQS",
+		db: process.env.MOCK_SQS_DB || `mongodb://${host}:${port}/${db}`,
 	};
 }