aws-lambda-runner 1.4.5 → 1.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9deef3294eaceddab1deed86424fd6125708787c
-  data.tar.gz: e2f1e5c9186acc26de923ec024cf8ee7aa8700e6
+  metadata.gz: c5f197efab92eddcd95b273da0e8b246b1e6f6aa
+  data.tar.gz: 6d5bc45491b52c7cce37100c25766c41e3a1e7ad
 SHA512:
-  metadata.gz: 3953eea6de12d1bd95403d1734ed2ca885ce0f329f7dc48b029e51d50c6a42a271bf77a8c8dd1c6c8413e737e02bcec0a11aedaff1fac5e5af6381d6d945a1be
-  data.tar.gz: 08ba985ea2454f49586071c800d2116943d35e1f610c260c7a2219e8cb816007120e08880217f3a4d0e40b56b79d9f5e80a26d2d9c1451300f26ab75ab8fc243
+  metadata.gz: 383a7964ea00cf6f5df36b43d0d52a06d1455b46d748a74177a927b71202a96c9e2fa738e1b510af05b1c175d9a10eaba18c791256b24d57dad4de9d494a2e63
+  data.tar.gz: 6d7435e028758058c7fd7a0aa0e723aaeb5cd2d464fd197fd01d95e79d157cfded220049fc03732cc8663955fa8fa1f249ad9861183cc045c6d7d223d38723e2

data/js/request.js

@@ -136,7 +136,7 @@ var getJobStatus = function (result) {
   return { status: status, data: responseBody };
 };
 
-exports.request = function(req, res, opts, handler) {
+exports.request = function(req, res, opts, handler, server) {
 
   if (req.method === 'POST') {
 
@@ -150,7 +150,9 @@ exports.request = function(req, res, opts, handler) {
       var terminationMessage = 'Terminating server at http://[localhost]:' + opts.port + ' for ' + opts['module-path'] + ' / ' + opts.handler;
       res.end(terminationMessage + '\n');
       console.info(terminationMessage);
-      server.close();
+      if (server) {
+        server.close();
+      }
     })();
 
   } else if (req.method === 'GET') {

data/js/startup.js

@@ -13,7 +13,11 @@ var opts = stdio.getopt({
 console.info(opts);
 
 var module = require(opts['module-path']);
-var server = http.createServer(function (req, res) {
-  request.request(req, res, opts, module[opts.handler]);
-}).listen(opts.port);
+var server = http.createServer();
+
+server.on('request', function (req, res) {
+  request.request(req, res, opts, module[opts.handler], server);
+});
+
+server.listen(opts.port);
 console.info('Server running at http://[localhost]:' + opts.port + ' for ' + opts['module-path'] + ' / ' + opts.handler);

data/lib/lambda_runner.rb

@@ -52,7 +52,7 @@ module LambdaRunner
       payload = JSON.generate({ event: event, context: context })
       id = RestClient.post(url, payload, content_type: :json).to_str
       loop do
-        response = RestClient.get(url, params: { id: id })
+        response = RestClient.get(url, params: { id: id }) {|response, request, res| response}
         data = JSON.parse('['+response.body+']').first
 
         case response.code
@@ -99,6 +99,28 @@ module LambdaRunner
       event
     end
 
+    def self.dynamodb_event(key, new_image = nil, old_image = nil)
+      event = load_json('sample_dynamodb_event.json')
+      get_dynamodb_record(event['Records'][0], key, new_image, old_image)
+      event
+    end
+
+    private_class_method def self.get_dynamodb_record(record, key, new_image, old_image)
+      if new_image && old_image
+        record['dynamodb']['NewImage'] = new_image['NewImage']
+        record['dynamodb']['OldImage'] = old_image['OldImage']
+        record['eventName'] = 'MODIFY'
+      elsif new_image
+        record['dynamodb']['NewImage'] = new_image['NewImage']
+        record['eventName'] = 'INSERT'
+      else
+        record['dynamodb']['OldImage'] = old_image['OldImage']
+        record['eventName'] = 'REMOVE'
+      end
+      record['dynamodb']['Keys'] = key
+      record
+    end
+
     private
 
     def self.load_json(name)

data/samples/sample_dynamodb_event.json

@@ -0,0 +1,19 @@
+{
+  "Records": [
+    {
+      "eventID": "e837fca4eb43e83b3ffb681d1daa701d",
+      "eventName": "INSERT",
+      "eventVersion": "1.0",
+      "eventSource": "aws:dynamodb",
+      "awsRegion": "eu-west-1",
+      "dynamodb": {
+        "Keys": {
+        },
+       "SequenceNumber": "0",
+       "SizeBytes": 149,
+       "StreamViewType": "NEW_AND_OLD_IMAGES"
+      },
+      "eventSourceARN": "arn:aws:dynamodb:eu-west-1:123456789:table/dynamodb-table/stream/2015-09-30T16:34:32.317"
+    }
+  ]
+}

data/spec/lambdarunner_events_spec.rb

@@ -0,0 +1,69 @@
+require 'rspec'
+require_relative '../lib/lambda_runner'
+
+describe LambdaRunner::Events do
+
+  it "should generate an S3 event" do
+    data = LambdaRunner::Events.s3_event("some-bucket", "some/key")
+    expect(data["Records"][0]["s3"]["bucket"]["name"]).to eq("some-bucket")
+    expect(data["Records"][0]["s3"]["bucket"]["arn"]).to eq("arn:aws:s3:::some-bucket")
+    expect(data["Records"][0]["s3"]["object"]["key"]).to eq("some/key")
+  end
+
+  it "should generate an SNS event (data body)" do
+    message = { "this_is" => "the message" }
+    data = LambdaRunner::Events.sns_event("some-arn", "some-id", "some-timestamp", message)
+    expect(data["Records"][0]["Sns"]["TopicArn"]).to eq("some-arn")
+    expect(data["Records"][0]["Sns"]["MessageId"]).to eq("some-id")
+    expect(data["Records"][0]["Sns"]["Timestamp"]).to eq("some-timestamp")
+    expect(JSON.parse data["Records"][0]["Sns"]["Message"]).to eq(message)
+  end
+
+  it "should generate an SNS event (string body)" do
+    message = { "this_is" => "the message" }
+    data = LambdaRunner::Events.sns_event("some-arn", "some-id", "some-timestamp", JSON.generate(message))
+    expect(data["Records"][0]["Sns"]["TopicArn"]).to eq("some-arn")
+    expect(data["Records"][0]["Sns"]["MessageId"]).to eq("some-id")
+    expect(data["Records"][0]["Sns"]["Timestamp"]).to eq("some-timestamp")
+    expect(JSON.parse data["Records"][0]["Sns"]["Message"]).to eq(message)
+  end
+
+  it "should generate an DynamoDB event for insert (data body)" do
+    new_image = { "NewImage" => { "id" => { "S" => "1234567ABC" }, "message" => { "S" => "updated message" } } }
+    key = { "id" => { "S" => "1234567ABC" } }
+
+    data = LambdaRunner::Events.dynamodb_event(key, new_image)
+
+    expect(data["Records"][0]["dynamodb"]["NewImage"]).to eq(new_image["NewImage"])
+    expect(data["Records"][0]["dynamodb"]["Keys"]).to eq(key)
+    expect(data["Records"][0]["eventName"]).to eq("INSERT")
+    expect(data["Records"][0]["dynamodb"]["SequenceNumber"]).to eq("0")
+  end
+
+  it "should generate an DynamoDB event for remove (data body)" do
+    old_image = { "OldImage" => { "id" => { "S" => "1234567ABC" }, "message" => { "S" => "message" } } }
+    key = { "id" => { "S" => "1234567ABC" } }
+
+    data = LambdaRunner::Events.dynamodb_event(key, nil, old_image)
+
+    expect(data["Records"][0]["dynamodb"]["OldImage"]).to eq(old_image["OldImage"])
+    expect(data["Records"][0]["dynamodb"]["Keys"]).to eq(key)
+    expect(data["Records"][0]["eventName"]).to eq("REMOVE")
+    expect(data["Records"][0]["dynamodb"]["SequenceNumber"]).to eq("0")
+  end
+
+  it "should generate an DynamoDB event for modify (data body)" do
+    old_image = { "OldImage" => { "id" => { "S" => "1234567ABC" }, "message" => { "S" => "message" } } }
+    new_image = { "NewImage" => { "id" => { "S" => "1234567ABC" }, "message" => { "S" => "updated message" } } }
+    key = { "id" => { "S" => "1234567ABC" } }
+
+    data = LambdaRunner::Events.dynamodb_event(key, new_image, old_image)
+
+    expect(data["Records"][0]["dynamodb"]["OldImage"]).to eq(old_image["OldImage"])
+    expect(data["Records"][0]["dynamodb"]["NewImage"]).to eq(new_image["NewImage"])
+    expect(data["Records"][0]["dynamodb"]["Keys"]).to eq(key)
+    expect(data["Records"][0]["eventName"]).to eq("MODIFY")
+    expect(data["Records"][0]["dynamodb"]["SequenceNumber"]).to eq("0")
+  end
+
+end

data/spec/lambdarunner_runner_spec.rb

@@ -0,0 +1,27 @@
+require 'rspec'
+require_relative '../lib/lambda_runner'
+
+describe LambdaRunner::Runner do
+
+  before(:all) do
+    test_js = File.expand_path("test.js", File.dirname(__FILE__))
+    @under_test = LambdaRunner::Runner.new(test_js, "node_0_10_42_handler")
+    @under_test.start
+  end
+
+  after(:all) do
+    @under_test.stop
+  end
+
+  it "should emulate AWS Lambda (node.js v0.10.42) - success" do
+    r = @under_test.process_event({ succeed: { delay: 50, result: "ohai" } })
+    expect(r).to eq("ohai")
+  end
+
+  it "should emulate AWS Lambda (node.js v0.10.42) - failure" do
+    expect {
+      @under_test.process_event({ fail: { delay: 50, err: "oh noes" } })
+    }.to raise_error("oh noes")
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aws-lambda-runner
 version: !ruby/object:Gem::Version
-  version: 1.4.5
+  version: 1.5.0
 platform: ruby
 authors:
 - andrew wheat
@@ -11,7 +11,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-07-31 00:00:00.000000000 Z
+date: 2017-02-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: process-helper
@@ -150,14 +150,16 @@ files:
 - js/startup.js
 - lib/lambda_runner.rb
 - samples/sample_cloudwatch_event.json
+- samples/sample_dynamodb_event.json
 - samples/sample_s3_event.json
 - samples/sample_s3_test_event.json
 - samples/sample_scheduled_event.json
 - samples/sample_sns_event.json
-- spec/lambda-runner_spec.rb
+- spec/lambdarunner_events_spec.rb
+- spec/lambdarunner_runner_spec.rb
 homepage: https://github.com/bbc/aws-lambda-runner
 licenses:
-- Apache 2
+- Apache-2.0
 metadata: {}
 post_install_message: 
 rdoc_options: []

data/spec/lambda-runner_spec.rb

@@ -1,31 +0,0 @@
-require 'rspec'
-require_relative '../lib/lambda_runner'
-
-describe LambdaRunner::Events do
-
-  it "should generate an S3 event" do
-    data = LambdaRunner::Events.s3_event("some-bucket", "some/key")
-    expect(data["Records"][0]["s3"]["bucket"]["name"]).to eq("some-bucket")
-    expect(data["Records"][0]["s3"]["bucket"]["arn"]).to eq("arn:aws:s3:::some-bucket")
-    expect(data["Records"][0]["s3"]["object"]["key"]).to eq("some/key")
-  end
-
-  it "should generate an SNS event (data body)" do
-    message = { "this_is" => "the message" }
-    data = LambdaRunner::Events.sns_event("some-arn", "some-id", "some-timestamp", message)
-    expect(data["Records"][0]["Sns"]["TopicArn"]).to eq("some-arn")
-    expect(data["Records"][0]["Sns"]["MessageId"]).to eq("some-id")
-    expect(data["Records"][0]["Sns"]["Timestamp"]).to eq("some-timestamp")
-    expect(JSON.parse data["Records"][0]["Sns"]["Message"]).to eq(message)
-  end
-
-  it "should generate an SNS event (string body)" do
-    message = { "this_is" => "the message" }
-    data = LambdaRunner::Events.sns_event("some-arn", "some-id", "some-timestamp", JSON.generate(message))
-    expect(data["Records"][0]["Sns"]["TopicArn"]).to eq("some-arn")
-    expect(data["Records"][0]["Sns"]["MessageId"]).to eq("some-id")
-    expect(data["Records"][0]["Sns"]["Timestamp"]).to eq("some-timestamp")
-    expect(JSON.parse data["Records"][0]["Sns"]["Message"]).to eq(message)
-  end
-
-end