From a062152d4a08b09007200d76e20279a0285a1838 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 11 Apr 2024 11:08:07 +0800
Subject: [PATCH] fix tests

---
 packages/inbound-email-handler/mocha-config.json        | 4 +++-
 packages/inbound-email-handler/src/redis_data_source.ts | 9 +++------
 packages/inbound-email-handler/test/global-teardown.ts  | 5 +++++
 3 files changed, 11 insertions(+), 7 deletions(-)
 create mode 100644 packages/inbound-email-handler/test/global-teardown.ts

diff --git a/packages/inbound-email-handler/mocha-config.json b/packages/inbound-email-handler/mocha-config.json
index 5a2631f42..897a3bb1e 100644
--- a/packages/inbound-email-handler/mocha-config.json
+++ b/packages/inbound-email-handler/mocha-config.json
@@ -1,4 +1,6 @@
 {
     "extension": ["ts"],
-    "spec": "test/**/*.test.ts"
+    "spec": "test/**/*.test.ts",
+    "require": ["test/global-teardown.ts"],
+    "timeout": 10000
   }
diff --git a/packages/inbound-email-handler/src/redis_data_source.ts b/packages/inbound-email-handler/src/redis_data_source.ts
index b8ca0ce6b..05bf8d26c 100644
--- a/packages/inbound-email-handler/src/redis_data_source.ts
+++ b/packages/inbound-email-handler/src/redis_data_source.ts
@@ -81,12 +81,9 @@ export const redisDataSource = new RedisDataSource({
 const gracefulShutdown = async (signal: string) => {
   console.log(`Received ${signal}, shutting down gracefully...`)
 
-  try {
-    await redisDataSource.shutdown()
-    console.log('redis shutdown successfully')
-  } catch (error) {
-    console.error('error while shutting down redis', error)
-  }
+  await redisDataSource.shutdown()
+  console.log('redis shutdown successfully')
+
   process.exit(0)
 }
 
diff --git a/packages/inbound-email-handler/test/global-teardown.ts b/packages/inbound-email-handler/test/global-teardown.ts
new file mode 100644
index 000000000..cb733d056
--- /dev/null
+++ b/packages/inbound-email-handler/test/global-teardown.ts
@@ -0,0 +1,5 @@
+import { redisDataSource } from '../src/redis_data_source'
+
+export const mochaGlobalTeardown = async () => {
+  await redisDataSource.shutdown()
+}