Add unit tests for util/config.ts

1 files changed, 6 insertions, 6 deletions

diff --git a/src/util/config.ts b/src/util/config.ts
index 34a9a46..1eddda3 100644
--- a/src/util/config.ts
+++ b/src/util/config.ts
@@ -7,21 +7,21 @@ import loggerConfig from "./logger";
 const logger = winston.createLogger(loggerConfig("ROOT"));
 
 export const configOrError = (varName: string): string => {
-  if (!process.env[varName]) {
+  const value = process.env[varName];
+  if (value !== undefined) {
+    return value;
+  } else {
     logger.error("%s must be defined", varName);
     process.exit(1);
-    return "";
-  } else {
-    return process.env[varName] || "";
   }
 };
 
 export const handleShutdown = (mongo: MongoClient, server: Server) => (
   signal: NodeJS.Signals
-): void => {
+): Promise<void> => {
   logger.warn("%s signal received. Closing shop.", signal);
 
-  mongo
+  return mongo
     .close()
     .then(() => {
       logger.info("MongoDB client connection closed.");