Expose flushed callback

croct-tech · Apr 19, 2020 · b89082a · b89082a
1 parent fa28c86
commit b89082a
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/plug.ts b/src/plug.ts
@@ -15,6 +15,7 @@ export interface Plug {
     readonly tracker: TrackerFacade;
     readonly user: UserFacade;
     readonly session: SessionFacade;
+    readonly flushed: Promise<void>;
 
     plug(configuration: Configuration): void;
 
@@ -52,6 +53,10 @@ class GlobalPlug implements Plug {
         this.facade = SdkFacade.init(configuration);
     }
 
+    public get flushed(): Promise<void> {
+        return this.tracker.flushed;
+    }
+
     private get instance(): SdkFacade {
         if (this.facade === undefined) {
             throw new Error('Croct is not plugged in.');

diff --git a/test/index.test.ts b/test/index.test.ts
@@ -41,6 +41,26 @@ describe('The Croct plug', () => {
         expect(initialize).toBeCalledTimes(1);
     });
 
+    test('should provide a callback that is called when the current pending events are flushed', async () => {
+        const config: SdkFacadeConfiguration = {appId: appId};
+
+        const sdkFacade = SdkFacade.init(config);
+
+        const flushed = jest.fn().mockResolvedValue(undefined);
+
+        Object.defineProperty(sdkFacade.tracker, 'flushed', {
+            get: flushed,
+        });
+
+        jest.spyOn(SdkFacade, 'init').mockReturnValue(sdkFacade);
+
+        croct.plug(config);
+
+        await expect(croct.flushed).resolves.toBeUndefined();
+
+        expect(flushed).toHaveBeenCalledTimes(1);
+    });
+
     test('should provide a tracker facade', () => {
         const config: SdkFacadeConfiguration = {appId: appId};
         const sdkFacade = SdkFacade.init(config);