unit test, process event

Author: mwunderl

sample-apps/blank/2-build-layer.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 set -eo pipefail
 mkdir -p lib/nodejs
-rm -rf lib/nodejs/node_modules
-npm install
+rm -rf node_modules lib/nodejs/node_modules
+npm install --production
 mv node_modules lib/nodejs/

sample-apps/blank/5-cleanup.sh

@@ -14,4 +14,4 @@ if [ -f bucket-name.txt ]; then
     done
 fi
 rm -f out.yml out.json
-rm -rf function/node_modules function/package-lock.json
+rm -rf lib package-lock.json

sample-apps/blank/event.json

@@ -1,3 +1,20 @@
 {
-  "KEY": "VALUE"
+  "Records": [
+    {
+      "messageId": "19dd0b57-b21e-4ac1-bd88-01bbb068cb78",
+      "receiptHandle": "MessageReceiptHandle",
+      "body": "Hello from SQS!",
+      "attributes": {
+        "ApproximateReceiveCount": "1",
+        "SentTimestamp": "1523232000000",
+        "SenderId": "123456789012",
+        "ApproximateFirstReceiveTimestamp": "1523232000001"
+      },
+      "messageAttributes": {},
+      "md5OfBody": "7b270e59b47ff90a553787216d55d91d",
+      "eventSource": "aws:sqs",
+      "eventSourceARN": "arn:aws:sqs:us-west-2:123456789012:MyQueue",
+      "awsRegion": "us-west-2"
+    }
+  ]
 }

sample-apps/blank/function/index.js

@@ -1,19 +1,26 @@
-const AWS = require('aws-sdk')
-// Read environment variable
-let region = process.env.AWS_REGION
-if (process.env.region)
-  region = process.env.region
+const AWSXRay = require('aws-xray-sdk-core')
+const AWS = AWSXRay.captureAWS(require('aws-sdk'))
+
 // Create client outside of handler to reuse
-const lambda = new AWS.Lambda({region: region})
+const lambda = new AWS.Lambda()
 
 // Handler
-exports.handler = function(event, context) {
-  console.log('Region: ' + region)
-  console.log('Event: ' + JSON.stringify(event, null, 2))
+exports.handler = async function(event, context) {
+  event.Records.forEach(record => {
+    console.log(record.body)
+  })
+  console.log('## ENVIRONMENT VARIABLES: ' + serialize(process.env))
+  console.log('## CONTEXT: ' + serialize(context))
+  console.log('## EVENT: ' + serialize(event))
+  
   return getAccountSettings()
 }
 
 // Use SDK client
 var getAccountSettings = function(){
   return lambda.getAccountSettings().promise()
 }
+
+var serialize = function(object) {
+  return JSON.stringify(object, null, 2)
+}

sample-apps/blank/function/index.test.js

@@ -0,0 +1,12 @@
+const index = require('./index')
+const fs = require('fs')
+const AWSXRay = require('aws-xray-sdk-core')
+AWSXRay.setContextMissingStrategy('LOG_ERROR')
+
+test('Runs function handler', async () => {
+    let eventFile = fs.readFileSync('event.json')
+    let event = JSON.parse(eventFile)
+    let response = await index.handler(event, null)
+    expect(JSON.stringify(response)).toContain('AccountLimit')
+  }
+)

sample-apps/blank/package.json

@@ -3,7 +3,7 @@
   "version": "1.0.0",
   "private": true,
   "devDependencies": {
-    "aws-sdk": "2.585.0"
+    "aws-sdk": "2.631.0"
   },
   "dependencies": {
     "aws-xray-sdk-core": "1.1.2"

sample-apps/error-processor/package.json

@@ -2,11 +2,11 @@
   "name": "error-processor",
   "version": "1.0.0",
   "private": true,
+  "devDependencies": {
+    "aws-sdk": "2.631.0"
+  },
   "dependencies": {
     "aws-xray-sdk-core": "1.1.2",
     "chance": "1.0.16"
-  },
-  "scripts": {
-    "start": "node app.js"
   }
 }

sample-apps/error-processor/random-error/index.js

@@ -1,8 +1,8 @@
-var AWSXRay = require('aws-xray-sdk-core')
-var aws = AWSXRay.captureAWS(require('aws-sdk'))
+const AWSXRay = require('aws-xray-sdk-core')
+const aws = AWSXRay.captureAWS(require('aws-sdk'))
 
-var Chance = require('chance')
-var lambda = new aws.Lambda()
+const Chance = require('chance')
+const lambda = new aws.Lambda()
 
 var myFunction = async function(event, context) {
   var chance = new Chance()

sample-apps/nodejs-apig/2-deploy.sh

@@ -1,8 +1,5 @@
 #!/bin/bash
 set -eo pipefail
 ARTIFACT_BUCKET=$(cat bucket-name.txt)
-cd function 
-npm install --production
-cd ../
 aws cloudformation package --template-file template.yml --s3-bucket $ARTIFACT_BUCKET --output-template-file out.yml
 aws cloudformation deploy --template-file out.yml --stack-name nodejs-apig --capabilities CAPABILITY_NAMED_IAM

sample-apps/nodejs-apig/function/index.js

@@ -4,9 +4,9 @@ const lambda = new AWS.Lambda()
 
 // Handler
 exports.handler = async function(event, context) {
-  console.log('ENVIRONMENT VARIABLES: ' + serialize(process.env))
-  console.log('CONTEXT: ' + serialize(context))
-  console.log('EVENT: ' + serialize(event))
+  console.log('## ENVIRONMENT VARIABLES: ' + serialize(process.env))
+  console.log('## CONTEXT: ' + serialize(context))
+  console.log('## EVENT: ' + serialize(event))
   try {
     let accountSettings = await getAccountSettings()
     return formatResponse(serialize(accountSettings.AccountUsage))

sample-apps/nodejs-apig/function/package.json renamed to sample-apps/nodejs-apig/package.json

@@ -3,9 +3,6 @@
   "version": "1.0.0",
   "private": true,
   "devDependencies": {
-    "aws-sdk": "2.585.0"
-  },
-  "scripts": {
-    "start": "node app.js"
+    "aws-sdk": "2.631.0"
   }
 }