faux-lambda 0.5.1 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 87764a85a01fcb5d17b6f530c6bd937351b250ba
-  data.tar.gz: 812b5811990bfe757370e139303d4635ad3261c6
+  metadata.gz: b3624a8f9bbfff9b594edf213d4f9879c1b38cba
+  data.tar.gz: ecc9b19c5c464d049ec7fc6b3281fda34d1b56d1
 SHA512:
-  metadata.gz: bc6796056d6448cc034e558d353804526ea77e6c33550a575caacd7f20b0ca2a4895efd8eff99523172dcd8d7f38ca1006e96de72a8e23ff4649cdca28e784e8
-  data.tar.gz: 9c1b1505f25c727db6f60636d6bf80c70bfd4bed1102aa05eef05b91f710c25349fb20e16a5a4a8e68be044ae4ac9c46d988f2559ea5ba0d8d2808ab1bb9c03b
+  metadata.gz: c585de977a32487ef44156bda0feeca5b195b58b2b7966c231ac7f366241f4f2240f0d56338f6ae492651c81c22775530b548f999c751eff824912304948cc18
+  data.tar.gz: 24a329756a002fc97ed78127248f90df81e68b059c074535295c20c71e268143bc5cdf2c4cbc8c85cbea550f106641b4d3bb220a150f7275db9df5ab8ac3f567

data/bin/faux-lambda

@@ -1,92 +1,14 @@
 #!/usr/bin/env ruby
 
-require 'rack'
-require 'webrick'
-require 'optparse'
-require_relative '../lib/faux_lambda/version'
+require 'faux_lambda'
+require 'faux_lambda/cli_handler'
 
-options = {
-  port: 9123,
-  bind: '127.0.0.1',
-}
+cli_handler = FauxLambda::CliHandler.new
+cli_handler.parse_options(ARGV)
 
-functions = {
-  default: {
-    invocation_type: nil,
-    replies: []
-  }
-}
-
-current_function = :default
-parser = OptionParser.new do |opts|
-  opts.banner = "AWS Lambda debugging endpoint, version #{FauxLambda::VERSION}."
-  opts.separator('')
-  opts.separator('Usage: faux-lambda --reply "Hello world!"')
-  opts.separator('Query specifiers:')
-  opts.on('--function name', '-f name', 'Name of function to expect, optionally with :<qualifier>') do |function_name|
-    current_function = function_name
-    functions[current_function] = {replies: []}
-  end
-  opts.separator('Reply specifiers:')
-  opts.on('--reply payload', '-r payload', 'Data to respond with') do |payload|
-    functions[current_function][:replies] << payload
-  end
-  opts.on('--fail', 'AWS Lambda framework gives 400') do
-    functions[current_function][:replies] << :fail
-  end
-end
-parser.parse!
-
-def reply_from(replies)
-  if replies.size > 1
-    replies.shift
-  elsif replies.size == 1
-    replies.last
-  else
-    nil
-  end
-end
-
-def function_data(functions, function_name, qualifier)
-  qualified_function_name = if qualifier
-    "#{function_name}:#{qualifier}"
-  end
-  functions[qualified_function_name] || functions[function_name] || functions[:default]
-end
-
-def log(function_name, qualifier, payload, status_code, reply)
-  qualifier ||= "$LATEST"
-  puts "#{function_name}:#{qualifier} called with #{payload}, replying #{status_code}: #{reply}"
-end
-
-app = Proc.new do |env|
-  _, version, _, function_name, _ = env['REQUEST_PATH'].split('/')
-  raise "Unknown version #{version}" unless version == '2015-03-31'
-  qs = Rack::Utils.parse_nested_query(env["QUERY_STRING"])
-  qualifier = qs["Qualifier"]
-  payload = env['rack.input'].read
-
-  data = function_data(functions, function_name, qualifier)
-  reply = reply_from(data[:replies])
-  if reply.nil?
-    status_code = '404'
-    reply = 'Not found'
-  elsif reply == :fail
-    status_code = '400'
-    reply = 'Failed'
-  else
-    status_code = '200'
-  end
-
-  log(function_name, qualifier, payload, status_code, reply)
-  headers = {'Content-Type' => 'application/octet-stream'}
-  [status_code, headers, [reply]]
+begin
+  faux = FauxLambda.new(cli_handler.options)
+  faux.handle(&cli_handler.method(:handler))
+rescue Interrupt
+  exit 130
 end
-
-Rack::Handler::WEBrick.run(
-  app,
-  Port: options[:port],
-  BindAddress: options[:bind],
-  Logger: WEBrick::Log.new($stderr, WEBrick::Log::ERROR),
-  AccessLog: [['/dev/null', WEBrick::AccessLog::COMMON_LOG_FORMAT]]
-)

data/lib/faux_lambda.rb

@@ -0,0 +1,59 @@
+require 'rack'
+require 'webrick'
+
+class FauxLambda
+  Call = Struct.new(
+    :function_name,
+    :qualifier,
+    :payload
+  )
+
+  def initialize(options)
+    @bind = options[:bind] || '127.0.0.1'
+    @port = options[:port] || 9123
+  end
+
+  def handle(&block)
+    Rack::Handler::WEBrick.run(
+      lambda do |env|
+        _, version, _, function_name, _ = env['REQUEST_PATH'].split('/')
+        raise "Unknown version #{version}" unless version == '2015-03-31'
+        qs = Rack::Utils.parse_nested_query(env["QUERY_STRING"])
+        qualifier = qs["Qualifier"]
+        payload = env['rack.input'].read
+
+        call = Call.new(function_name, qualifier, payload)
+        begin
+          reply = block.call(call)
+          if reply.nil?
+            status_code = '404'
+            reply = 'Not found'
+          elsif reply == :fail
+            status_code = '400'
+            reply = 'Failed'
+          else
+            status_code = '200'
+          end
+          log(function_name, qualifier, payload, status_code, reply)
+        rescue => e
+          reply = ''
+          log(function_name, qualifier, payload, '200', e)
+        end
+
+        headers = {'Content-Type' => 'application/octet-stream'}
+        [status_code, headers, [reply]]
+      end,
+      Host: @bind,
+      Port: @port,
+      Logger: WEBrick::Log.new($stderr, WEBrick::Log::ERROR),
+      AccessLog: [['/dev/null', WEBrick::AccessLog::COMMON_LOG_FORMAT]]
+    )
+  end
+
+  private
+
+  def log(function_name, qualifier, payload, status_code, reply)
+    qualifier ||= "$LATEST"
+    puts "#{function_name}:#{qualifier} called with '#{payload}', replying #{status_code}: '#{reply}'"
+  end
+end

data/lib/faux_lambda/cli_handler.rb

@@ -0,0 +1,78 @@
+require 'optparse'
+
+class FauxLambda::CliHandler
+  attr_reader :options
+
+  def initialize
+    @functions = {
+      default: {
+        invocation_type: nil,
+        replies: []
+      }
+    }
+    @options = {}
+  end
+
+  def parse_options(argv)
+    current_function = :default
+    parser = OptionParser.new do |opts|
+      opts.banner = "AWS Lambda debugging endpoint, version #{FauxLambda::VERSION}."
+      opts.separator('')
+      opts.separator('Usage: faux-lambda --reply "Hello world!"')
+      opts.separator('Query specifiers:')
+      opts.on('--function name', '-f name', 'Name of function to expect, optionally with :<qualifier>') do |function_name|
+        current_function = function_name
+        @functions[current_function] = {replies: []}
+      end
+      opts.separator('Reply specifiers:')
+      opts.on('--reply payload', '-r payload', 'Data to respond with') do |payload|
+        @functions[current_function][:replies] << lambda {|_| payload }
+      end
+      opts.on('--handler script.rb', '-h script.rb', 'Ruby script is eval:ed to produce reply') do |script|
+        @functions[current_function][:replies] << make_handler(script)
+      end
+      opts.on('--fail', 'AWS Lambda framework gives 400') do
+        @functions[current_function][:replies] << lambda {|_| :fail }
+      end
+      opts.separator('Control options')
+      opts.on('--port port', 'TCP port to bind, 9123 by default') do |port|
+        @options[:port] = port
+      end
+      opts.on('--bind address', 'Interface to bind to, localhost by default') do |bindaddress|
+        @options[:bind] = bindaddress
+      end
+    end
+    parser.parse!(argv)
+  end
+
+  def handler(call)
+    data = function_data(call)
+    reply_from(data[:replies], call)
+  end
+
+  private
+
+  def make_handler(script)
+    code = File.read(script)
+    lambda do |call|
+      eval(code, binding(), script)
+    end
+  end
+
+  def reply_from(replies, call)
+    if replies.size > 1
+      replies.shift.call(call)
+    elsif replies.size == 1
+      replies.last.call(call)
+    else
+      nil
+    end
+  end
+
+  def function_data(call)
+    qualified_function_name = if call.qualifier
+      "#{call.function_name}:#{call.qualifier}"
+    end
+    @functions[qualified_function_name] || @functions[call.function_name] || @functions[:default]
+  end
+end

data/lib/faux_lambda/version.rb

@@ -1,3 +1,3 @@
-module FauxLambda
-  VERSION = '0.5.1'
+class FauxLambda
+  VERSION = '0.6.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: faux-lambda
 version: !ruby/object:Gem::Version
-  version: 0.5.1
+  version: 0.6.0
 platform: ruby
 authors:
 - Anders Qvist
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-11-01 00:00:00.000000000 Z
+date: 2017-11-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rack
@@ -33,12 +33,9 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
-- ".gitignore"
-- Gemfile
-- Gemfile.lock
-- README.md
 - bin/faux-lambda
-- faux-lambda.gemspec
+- lib/faux_lambda.rb
+- lib/faux_lambda/cli_handler.rb
 - lib/faux_lambda/version.rb
 homepage: https://github.com/bittrance/faux-lambda
 licenses:

data/.gitignore

@@ -1,2 +0,0 @@
-.bundle/
-.ruby-version

data/Gemfile

@@ -1,4 +0,0 @@
-# frozen_string_literal: true
-source "https://rubygems.org"
-
-gemspec

data/Gemfile.lock

@@ -1,19 +0,0 @@
-PATH
-  remote: .
-  specs:
-    faux-lambda (0.5.1)
-      rack (~> 2.0)
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    rack (2.0.3)
-
-PLATFORMS
-  ruby
-
-DEPENDENCIES
-  faux-lambda!
-
-BUNDLED WITH
-   1.15.4

data/README.md

@@ -1,78 +0,0 @@
-# Faux Lambda server for testing
-
-Lambda service mock implementation for testing and exploration. It starts a simple web server (default is http://localhost:9123) and processes Lambda invocation requests. It can be used both programmatically and as a command line tool.
-
-## As a command line tool
-
-In its simplest form:
-```
-gem install faux-lambda
-faux-lambda --reply='{}' &
-aws --endpoint http://localhost:9123 lambda invoke --function-name whatever /dev/stdout
-```
-
-The CLI can also simulate specific scenarios:
-```
-$ faux-lambda --function=foo --reply='{}' --function=bar --reply='{}' --reply='{}' --fail &
-$ aws --endpoint http://localhost:12345 lambda invoke --function-name foo /dev/stdout
-Called foo with ...
-{}
-```
-Here, each call to function `foo` gets reply `{}`, first call to function `bar` gets `{}`, second gets `{}` and third will fail.
-
-You can also pipe in replies from `stdin`, one reply per object (may contain newlines):
-```
-faux-lambda --function=foo <<<EOF
-{}
-{}
-EOF
-```
-In this mode, the CLI will accept only one function, but may prove useful where there are numerous or large replies.
-
-```
-Usage: faux-lambda [options]
-Query specifiers:
-  --function-name=regex
-  --async       Require invocation to be async
-  --sync        Require invocation to be sync
-Reply specifiers:
-  --reply=json  Send this reply
-  --fail        ...
-Various:
-  --quiet       Don't log sent and received messages
-  --version
-```
-
-## faux-lambda as a library
-
-You can use `faux-lambda` programmatically too, like so:
-```
-require `faux-lambda`
-lambda = FauxLambda.new(port: 12345).handle do |call|
-  return if call.async
-  if call.function_name == 'foo'
-    '{}'
-  end
-end
-
-Aws::Lambda.new(endpoint: lambda.endpoint).invoke(...)
-```
-
-## faux-lambda in your specs
-
-Or in your specs as if FauxLambda were an `rspec-mocks` double:
-```
-let(:lambda) { FauxLambda.new }
-before do
-  allow(lambda).to receive(:incrementer).and_return(43)
-end
-
-subject do
-  ThingUnderTest.new(lambda.endpoint).do_math(42)
-end
-
-it 'invokes lambda and returns its value' do
-  expect(subject).to equal(43)
-  expect(lambda).to have_received(:incrementer).with(value: 42)
-end
-```

data/faux-lambda.gemspec

@@ -1,19 +0,0 @@
-require File.expand_path('../lib/faux_lambda/version', __FILE__)
-
-Gem::Specification.new do |spec|
-  spec.name        = 'faux-lambda'
-  spec.version     = FauxLambda::VERSION
-  spec.summary     = 'faux-lambda provides a simple tool to mock AWS Lambda endpoint'
-  spec.description = 'faux-lambda is a toolbox to make it easier to develop client that call AWS Lambda functions. A simple CLI lets you mock AWS Lambda and specify replies for specific function as well as simulate AWS Lambda framework failures.'
-  spec.authors     = ['Anders Qvist']
-  spec.email       = 'quest@lysator.liu.se'
-  spec.homepage    = 'https://github.com/bittrance/faux-lambda'
-  spec.license     = 'MIT'
-
-  spec.executables = ['faux-lambda']
-  spec.files       = `git ls-files -z`.split("\x0").reject { |f| f.match(/^spec/) }
-
-  spec.required_ruby_version = '>= 2.0.0'
-
-  spec.add_runtime_dependency 'rack', '~> 2.0'
-end