Check attempts before dispatching

Author: danieljackins

packages/browser/src/plugins/segmentio/__tests__/retries.test.ts

@@ -36,7 +36,7 @@ describe('Segment.io retries 500s and 429', () => {
     const ctx = await analytics.track('event')
     jest.runAllTimers()
 
-    expect(ctx.attempts).toBeGreaterThanOrEqual(3) // Gets incremented after use
+    expect(ctx.attempts).toBeGreaterThanOrEqual(2) // Gets incremented after use
     expect(fetch.mock.calls.length).toBeGreaterThanOrEqual(2)
     expect(fetch.mock.lastCall[1].body).toContain('"retryCount":')
   })
@@ -131,3 +131,54 @@ describe('Batches retry 500s and 429', () => {
     expect(fetch.mock.lastCall[1].body).toContain('"retryCount":2')
   })
 })
+
+describe('retryQueue', () => {
+  let options: SegmentioSettings
+  let analytics: Analytics
+  let segment: Plugin
+  beforeEach(async () => {
+    jest.useRealTimers()
+    jest.resetAllMocks()
+    jest.restoreAllMocks()
+
+    options = {
+      apiKey: 'foo',
+    }
+
+    fetch.mockReturnValue(createError({ status: 500 }))
+  })
+
+  it('Only attempts once if retryQueue is false', async () => {
+    analytics = new Analytics(
+      { writeKey: options.apiKey },
+      { retryQueue: false }
+    )
+    segment = await segmentio(
+      analytics,
+      options,
+      cdnSettingsMinimal.integrations
+    )
+    await analytics.register(segment, envEnrichment)
+
+    await analytics.track('foo')
+    await new Promise((resolve) => setTimeout(resolve, 1000))
+    expect(fetch).toHaveBeenCalledTimes(1)
+  })
+
+  it('Attempts multiple times if retryQueue is not false', async () => {
+    analytics = new Analytics(
+      { writeKey: options.apiKey },
+      { retryQueue: true }
+    )
+    segment = await segmentio(
+      analytics,
+      options,
+      cdnSettingsMinimal.integrations
+    )
+    await analytics.register(segment, envEnrichment)
+
+    await analytics.track('foo')
+    await new Promise((resolve) => setTimeout(resolve, 1000))
+    expect(fetch.mock.calls.length).toBeGreaterThanOrEqual(2)
+  })
+})

packages/browser/src/plugins/segmentio/index.ts

@@ -112,6 +112,11 @@ export function segmentio(
       json = onAlias(analytics, json)
     }
 
+    if (buffer.getAttempts(ctx) >= buffer.maxAttempts) {
+      inflightEvents.delete(ctx)
+      return ctx
+    }
+
     return client
       .dispatch(
         `${remote}/${path}`,