Explicit capture

supergoodsystems · Apr 22, 2024 · dd168c3 · dd168c3
1 parent d36fc26
commit dd168c3
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -488,9 +488,22 @@ const Supergood = () => {
     supergoodAsyncLocalStorage.disable();
   }
 
+  const getAsyncLocalStorage = () => supergoodAsyncLocalStorage.getStore();
+
   // Set up cleanup catch for exit signals
   onExit(() => close(), { alwaysLast: true });
-  return { close, flushCache, waitAndFlushCache, withTags, init, withCapture, startCapture, stopCapture };
+
+  return {
+    close,
+    flushCache,
+    waitAndFlushCache,
+    withTags,
+    init,
+    withCapture,
+    startCapture,
+    stopCapture,
+    getAsyncLocalStorage
+  };
 };
 
 export = Supergood();
diff --git a/test/e2e/explicit-capture.e2e.test.ts b/test/e2e/explicit-capture.e2e.test.ts
@@ -118,7 +118,6 @@ describe('capture functionality', () => {
         await axios.get(`${MOCK_DATA_SERVER}/posts`), 250
       });
       getInterval.unref();
-
       await Supergood.startCapture(
         {
           config: { ...SUPERGOOD_CONFIG, allowLocalUrls: true },
@@ -133,6 +132,7 @@ describe('capture functionality', () => {
       }
 
       await Supergood.stopCapture();
+      await axios.get(`${MOCK_DATA_SERVER}/posts`);
 
       clearInterval(getInterval);
 
@@ -147,5 +147,39 @@ describe('capture functionality', () => {
     });
   });
 
+  it('should capture calls without an async kickoff, with useRemoteConfig set to false', async () => {
+    const numberOfHttpCalls = 5;
+
+    const getInterval = setInterval(async () => {
+      await axios.get(`${MOCK_DATA_SERVER}/posts`), 250
+    });
+    getInterval.unref();
+    Supergood.startCapture(
+      {
+        config: { ...SUPERGOOD_CONFIG, allowLocalUrls: true, useRemoteConfig: false },
+        clientId: SUPERGOOD_CLIENT_ID,
+        clientSecret: SUPERGOOD_CLIENT_SECRET,
+        baseUrl: SUPERGOOD_SERVER
+      });
+
+
+    for (let i = 0; i < numberOfHttpCalls; i++) {
+      await axios.get(`${MOCK_DATA_SERVER}/posts`);
+    }
+
+    await Supergood.stopCapture();
+    await axios.get(`${MOCK_DATA_SERVER}/posts`);
 
+    clearInterval(getInterval);
+
+    checkPostedEvents(postEventsMock, numberOfHttpCalls, {
+      request: expect.objectContaining({
+        requestedAt: expect.any(Date)
+      }),
+      response: expect.objectContaining({
+        respondedAt: expect.any(Date)
+      })
+    });
+  });
 });
+