fix: fix FireormService provider

glebbash · Apr 5, 2021 · 8b55836 · 8b55836
1 parent c6e8f87
commit 8b55836
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/fireorm/fireorm.module.spec.ts b/src/fireorm/fireorm.module.spec.ts
@@ -1,5 +1,6 @@
 import { getRepositoryToken } from '../nestjs-fireorm';
 import { FireormModule, FireormSettings, firestoreProvider } from './fireorm.module';
+jest.mock('./fireorm.service');
 import { FireormService } from './fireorm.service';
 import 'jest-extended';
 jest.mock('@google-cloud/firestore');
@@ -41,7 +42,18 @@ describe('FireormModule', () => {
             inject: [FireormSettings],
             useFactory: firestoreProvider,
           },
-          FireormService,
+          {
+            provide: FireormService,
+            inject: [Firestore, FireormSettings],
+            useFactory: expect.toSatisfy((fun) => {
+              const fireormService = fun(1, 2);
+
+              expect(fireormService).toBeInstanceOf(FireormService);
+              expect(FireormService).toBeCalledWith(1, 2);
+
+              return true;
+            }),
+          },
         ],
         exports: [FireormService],
       });

diff --git a/src/fireorm/fireorm.module.ts b/src/fireorm/fireorm.module.ts
@@ -79,7 +79,12 @@ export class FireormModule {
           inject: [FireormSettings],
           useFactory: firestoreProvider,
         },
-        FireormService,
+        {
+          provide: FireormService,
+          inject: [Firestore, FireormSettings],
+          useFactory: (firestore: Firestore, settings: FireormSettings) =>
+            new FireormService(firestore, settings),
+        },
       ],
       exports: [FireormService],
     };