From e3ca55e94aeb2dac79cef7ced232320c99323a0e Mon Sep 17 00:00:00 2001
From: matt <matt@fiddaman.net>
Date: Thu, 21 Apr 2022 14:48:49 +0000
Subject: [PATCH] Add ability to insert test data with custom domain email
 addresses

---
 gulpfile.js                  | 6 +++++-
 utility/db/insertTestData.js | 8 +++++++-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/gulpfile.js b/gulpfile.js
index b8b9474..45460ee 100644
--- a/gulpfile.js
+++ b/gulpfile.js
@@ -16,7 +16,7 @@ const dbModule = (name) => path.join(__dirname, 'utility/db', name);
 
 const cleanDb = require(dbModule('cleanDb'));
 const initDb = require(dbModule('initDb'));
-const insertTestData = require(dbModule('insertTestData'));
+const insertTestDataModule = require(dbModule('insertTestData'));
 
 // Set src and destination paths for css compilation
 const cssPaths = {
@@ -129,6 +129,10 @@ function setConfig(cb) {
 	cb();
 }
 
+async function insertTestData() {
+	await insertTestDataModule(process.env.CUSTOM_DOMAIN);
+}
+
 // Task to build stylesheet from start to finish
 exports.styles = series(cleanStyles, compileStyles);
 
diff --git a/utility/db/insertTestData.js b/utility/db/insertTestData.js
index 75a17b7..7ada1d8 100644
--- a/utility/db/insertTestData.js
+++ b/utility/db/insertTestData.js
@@ -18,7 +18,7 @@ const { data, details } = require('./testData');
  *
  * @return {void}
  */
-async function insertTestData() {
+async function insertTestData(customEmailDomain = false) {
 	const conn = await new DatabaseConnectionPool();
 
 	// Run the creation statment for each table
@@ -71,6 +71,12 @@ async function insertTestData() {
 
 			console.log(sql.trim());
 
+			if (customEmailDomain && dataToInsert.email) {
+				const split = dataToInsert.email.split('@');
+				split[1] = customEmailDomain;
+				dataToInsert.email = split.join('@');
+			}
+
 			try {
 				await conn.runQuery(sql,
 					Object.values(dataToInsert));