Merge pull request #412 from jgilbert01/force-number-for-circuit-breaker

Force number for circuit breaker

Author: ja-stevenson

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "aws-lambda-stream",
-  "version": "1.1.2",
+  "version": "1.1.3",
   "description": "Create stream processors with AWS Lambda functions.",
   "keywords": [
     "aws",

package.json

@@ -15,6 +15,9 @@ const toUpdateRequest = (opt) => faulty((uow) => ({
     UUID: process.env.ESM_ID, // Ref: TriggerEventSourceMappingDynamodbEntitiesTable
     Enabled: uow.event.alarmData.state.value !== 'ALARM',
     BatchSize: uow.event.alarmData.state.value === 'INSUFFICIENT_DATA'
-      ? 1 : process.env.BATCH_SIZE || 100,
+      ? 1 : Number(opt.circuitBreakerMaxBatchSize)
+      || Number(process.env.CIRCUIT_BREAKER_MAX_BATCH_SIZE)
+      || Number(process.env.BATCH_SIZE)
+      || 100,
   },
 }));

src/flavors/circuitbreaker.js

@@ -2,7 +2,7 @@ import 'mocha';
 import { expect } from 'chai';
 import sinon from 'sinon';
 
-import { initialize } from '../../../src';
+import { initialize, initializeFrom } from '../../../src';
 
 import { fromAlarm } from '../../../src/from/cw';
 
@@ -125,4 +125,85 @@ describe('flavors/circuitbreaker.js', () => {
       })
       .done(done);
   });
+
+  it('should enable event source mapping with batch size from opt)', (done) => {
+    const rules = [
+      {
+        id: 'circuitBreaker',
+        flavor: circuitBreaker,
+        circuitBreakerMaxBatchSize: '10',
+      },
+    ];
+    const event = {
+      source: 'aws.cloudwatch',
+      alarmArn:
+        'arn:aws:cloudwatch:us-east-1:444455556666:alarm:lambda-demo-metric-alarm',
+      accountId: '444455556666',
+      time: '2023-08-04T12:36:15.490+0000',
+      region: 'us-east-1',
+      alarmData: {
+        alarmName: 'lambda-demo-metric-alarm',
+        state: {
+          value: 'OK',
+        },
+        previousState: {
+          value: 'INSUFFICIENT_DATA',
+        },
+      },
+    };
+
+    initialize({
+      ...initializeFrom(rules),
+    })
+      .assemble(fromAlarm(event), false)
+      .collect()
+      // .tap((collected) => console.log(JSON.stringify(collected, null, 2)))
+      .tap((collected) => {
+        expect(collected.length).to.equal(1);
+        expect(collected[0].pipeline).to.equal('circuitBreaker');
+        expect(collected[0].updateRequest).to.deep.equal({
+          UUID: 'a092f90d-9948-4964-95b5-32c46093f734',
+          Enabled: true,
+          BatchSize: 10,
+        });
+      })
+      .done(done);
+  });
+  it('should enable event source mapping and force Number type for env var of CIRCUIT_BREAKER_MAX_BATCH_SIZE', (done) => {
+    process.env.CIRCUIT_BREAKER_MAX_BATCH_SIZE = '10';
+    const event = {
+      source: 'aws.cloudwatch',
+      alarmArn:
+        'arn:aws:cloudwatch:us-east-1:444455556666:alarm:lambda-demo-metric-alarm',
+      accountId: '444455556666',
+      time: '2023-08-04T12:36:15.490+0000',
+      region: 'us-east-1',
+      alarmData: {
+        alarmName: 'lambda-demo-metric-alarm',
+        state: {
+          value: 'OK',
+        },
+        previousState: {
+          value: 'INSUFFICIENT_DATA',
+        },
+      },
+    };
+
+    initialize({
+      circuitBreaker,
+    })
+      .assemble(fromAlarm(event), false)
+      .collect()
+      // .tap((collected) => console.log(JSON.stringify(collected, null, 2)))
+      .tap((collected) => {
+        expect(collected.length).to.equal(1);
+        expect(collected[0].pipeline).to.equal('circuitBreaker');
+        expect(collected[0].updateRequest).to.deep.equal({
+          UUID: 'a092f90d-9948-4964-95b5-32c46093f734',
+          Enabled: true,
+          BatchSize: 10,
+        });
+      })
+      .done(done);
+  });
 });