From a5c1afbcb48eeca28ac7c33ac5f8279d5148985e Mon Sep 17 00:00:00 2001
From: Innei <tukon479@gmail.com>
Date: Sun, 24 Apr 2022 22:48:26 +0800
Subject: [PATCH] fix: test failed

---
 test/helper/db-mock.helper.ts                 |  2 +-
 .../serverless/serverless.service.spec.ts     | 21 ++++++++++++++-----
 2 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/test/helper/db-mock.helper.ts b/test/helper/db-mock.helper.ts
index d769192c..b5c09132 100644
--- a/test/helper/db-mock.helper.ts
+++ b/test/helper/db-mock.helper.ts
@@ -19,7 +19,7 @@ const connect = async () => {
   mongod = await MongoMemoryServer.create()
   const uri = mongod.getUri()
 
-  await mongoose.connect(uri, {
+  return await mongoose.connect(uri, {
     autoIndex: true,
     maxPoolSize: 10,
   })
diff --git a/test/src/modules/serverless/serverless.service.spec.ts b/test/src/modules/serverless/serverless.service.spec.ts
index 93e47511..4a1a3b8a 100644
--- a/test/src/modules/serverless/serverless.service.spec.ts
+++ b/test/src/modules/serverless/serverless.service.spec.ts
@@ -1,7 +1,9 @@
+import { dbHelper } from 'test/helper/db-mock.helper'
+import { redisHelper } from 'test/helper/redis-mock.helper'
+
 import { Test } from '@nestjs/testing'
 import { getModelForClass } from '@typegoose/typegoose'
-import { dbHelper } from 'test/helper/db-mock.helper'
-import { getModelToken } from '~/transformers/model.transformer'
+
 import { createMockedContextResponse } from '~/modules/serverless/mock-response.util'
 import { ServerlessService } from '~/modules/serverless/serverless.service'
 import { SnippetModel, SnippetType } from '~/modules/snippet/snippet.model'
@@ -9,12 +11,15 @@ import { CacheService } from '~/processors/cache/cache.service'
 import { DatabaseService } from '~/processors/database/database.service'
 import { AssetService } from '~/processors/helper/helper.asset.service'
 import { HttpService } from '~/processors/helper/helper.http.service'
+import { getModelToken } from '~/transformers/model.transformer'
 
 describe('test serverless function service', () => {
   let service: ServerlessService
 
   beforeAll(async () => {
-    await dbHelper.connect()
+    const connection = await dbHelper.connect()
+
+    await (await redisHelper).connect()
     const moduleRef = Test.createTestingModule({
       providers: [
         ServerlessService,
@@ -22,9 +27,14 @@ describe('test serverless function service', () => {
         AssetService,
         {
           provide: CacheService,
-          useValue: {},
+          useValue: (await redisHelper).CacheService,
+        },
+        {
+          provide: DatabaseService,
+          useValue: {
+            db: connection.connection.db,
+          },
         },
-        { provide: DatabaseService, useValue: {} },
 
         {
           provide: getModelToken('SnippetModel'),
@@ -41,6 +51,7 @@ describe('test serverless function service', () => {
   afterAll(async () => {
     await dbHelper.clear()
     await dbHelper.close()
+    ;(await redisHelper).close()
   })
 
   describe('run serverless function', () => {