fog-aws 0.5.0 → 0.6.0

data/lib/fog/aws/requests/lambda/create_event_source_mapping.rb

@@ -0,0 +1,94 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        # Identifies a stream as an event source for a Lambda function.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_CreateEventSourceMapping.html
+        # ==== Parameters
+        # * BatchSize <~Integer> - largest number of records that AWS Lambda will retrieve from your event source at the time of invoking your function.
+        # * Enabled <~Boolean> - indicates whether AWS Lambda should begin polling the event source.
+        # * EventSourceArn <~String> - Amazon Resource Name (ARN) of the stream that is the event source
+        # * FunctionName <~String> - Lambda function to invoke when AWS Lambda detects an event on the stream.
+        # * StartingPosition <~String> - position in the stream where AWS Lambda should start reading.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'BatchSize' <~Integer> - largest number of records that AWS Lambda will retrieve from your event source at the time of invoking your function.
+        #     * 'EventSourceArn' <~String> - Amazon Resource Name (ARN) of the stream that is the source of events.
+        #     * 'FunctionArn' <~String> - Lambda function to invoke when AWS Lambda detects an event on the stream.
+        #     * 'LastModified' <~Time> - UTC time string indicating the last time the event mapping was updated.
+        #     * 'LastProcessingResult' <~String> - result of the last AWS Lambda invocation of your Lambda function.
+        #     * 'State' <~String> - state of the event source mapping.
+        #     * 'StateTransitionReason' <~String> - reason the event source mapping is in its current state.
+        #     * 'UUID' <~String> - AWS Lambda assigned opaque identifier for the mapping.
+        def create_event_source_mapping(params={})
+          enabled          = params.delete('Enabled')
+          batch_size       = params.delete('BatchSize')
+          event_source_arn = params.delete('EventSourceArn')
+          function_name    = params.delete('FunctionName')
+          starting_pos     = params.delete('StartingPosition')
+
+          data = {
+            'EventSourceArn'   => event_source_arn,
+            'FunctionName'     => function_name,
+            'StartingPosition' => starting_pos
+          }
+          data.merge!('BatchSize' => batch_size) if batch_size
+          data.merge!('Enabled'   => enabled)    if !enabled.nil?
+
+          request({
+            :method  => 'POST',
+            :path    => '/event-source-mappings/',
+            :expects => 202,
+            :body    => Fog::JSON.encode(data)
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def create_event_source_mapping(params={})
+          enabled          = params.delete('Enabled')   || false
+          batch_size       = params.delete('BatchSize') || 100
+          event_source_arn = params.delete('EventSourceArn')
+          function_name    = params.delete('FunctionName')
+          starting_pos     = params.delete('StartingPosition')
+
+          function = self.get_function_configuration('FunctionName' => function_name).body
+
+          unless event_source_arn
+            message  = "ValidationException => "
+            message << "'eventSourceArn' cannot be blank"
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          unless starting_pos
+            message  = "ValidationException => "
+            message << "'startingPosition' cannot be blank"
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          event_source_mapping_id = UUID.uuid
+          event_source_mapping = {
+            'BatchSize'             => batch_size,
+            'EventSourceArn'        => event_source_arn,
+            'FunctionArn'           => function['FunctionArn'],
+            'LastModified'          => Time.now.to_f,
+            'LastProcessingResult'  => 'No records processed',
+            'State'                 => 'Creating',
+            'StateTransitionReason' => 'User action',
+            'UUID'                  => event_source_mapping_id
+          }
+
+          self.data[:event_source_mappings].merge!(
+            event_source_mapping_id => event_source_mapping
+          )
+
+          response = Excon::Response.new
+          response.body = event_source_mapping
+          response.status = 202
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/create_function.rb

@@ -0,0 +1,146 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        require 'fog/aws/parsers/lambda/base'
+
+        # Creates a new Lambda function.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_CreateFunction.html
+        # ==== Parameters
+        # * Code <~Hash> - code for the Lambda function.
+        # * Description <~String> - short, user-defined function description.
+        # * FunctionName <~String> - name you want to assign to the function you are uploading.
+        # * Handler <~String> - function within your code that Lambda calls to begin execution.
+        # * MemorySize <~Integer> - amount of memory, in MB, your Lambda function is given.
+        # * Role <~String> - ARN of the IAM role that Lambda assumes when it executes your function to access any other AWS resources.
+        # * Runtime <~String> - runtime environment for the Lambda function you are uploading.
+        # * Timeout <~Integer> - function execution time at which Lambda should terminate the function.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'CodeSize' <~Integer> - size, in bytes, of the function .zip file you uploaded.
+        #     * 'Description' <~String> - user-provided description.
+        #     * 'FunctionArn' <~String> - Amazon Resource Name (ARN) assigned to the function.
+        #     * 'FunctionName' <~String> - name of the function.
+        #     * 'Handler' <~String> - function Lambda calls to begin executing your function.
+        #     * 'LastModified' <~Time> - timestamp of the last time you updated the function.
+        #     * 'MemorySize' <~Integer> - memory size, in MB, you configured for the function.
+        #     * 'Role' <~String> - ARN of the IAM role that Lambda assumes when it executes your function to access any other AWS resources.
+        #     * 'Runtime' <~String> - runtime environment for the Lambda function.
+        #     * 'Timeout' <~Integer> - function execution time at which Lambda should terminate the function.
+        def create_function(params={})
+          runtime       = params.delete('Runtime') || 'nodejs'
+          code          = params.delete('Code')
+          function_name = params.delete('FunctionName')
+          handler       = params.delete('Handler')
+          role          = params.delete('Role')
+
+          data = {
+            'Runtime'      => runtime,
+            'Code'         => code,
+            'FunctionName' => function_name,
+            'Handler'      => handler,
+            'Role'         => role
+          }
+
+          description = params.delete('Description')
+          data.merge!('Description' => description) if description
+
+          memory_size = params.delete('MemorySize')
+          data.merge!('MemorySize' => memory_size)  if memory_size
+
+          timeout = params.delete('Timeout')
+          data.merge!('Timeout' => timeout) if timeout
+
+          request({
+            :method  => 'POST',
+            :path    => '/functions',
+            :expects => 201,
+            :body    => Fog::JSON.encode(data),
+            :parser  => Fog::AWS::Parsers::Lambda::Base.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def create_function(params={})
+          response = Excon::Response.new
+
+          runtime = params.delete('Runtime') || 'nodejs'
+          if !%w(nodejs java8).include?(runtime)
+            message = 'ValidationException: Runtime must be nodejs or java8.'
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          unless code = params.delete('Code')
+            message = 'ValidationException: Code cannot be blank.'
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          unless function_name = params.delete('FunctionName')
+            message = 'ValidationException: Function name cannot be blank.'
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          unless handler = params.delete('Handler')
+            message = 'ValidationException: Handler cannot be blank.'
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          unless role = params.delete('Role')
+            message = 'ValidationException: Role cannot be blank.'
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          code_size = if code.has_key?('ZipFile')
+            Base64.decode64(code['ZipFile']).length
+          else
+            Fog::Mock.random_numbers(5).to_i
+          end
+
+          description = params.delete('Description')
+
+          function = {}
+          begin
+            opts     = { 'FunctionName' => function_name }
+            function = self.get_function_configuration(opts).body
+          rescue Fog::AWS::Lambda::Error => e
+            # ignore: if the function doesn't exist we are OK.
+          end
+
+          if !function.empty?
+            message  = "ResourceConflictException => "
+            message << "Function already exist: #{function_name}"
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          function_path = "function:#{function_name}"
+          function_arn = Fog::AWS::Mock.arn(
+            'lambda',
+            self.account_id,
+            function_path,
+            self.region
+          )
+
+          function = {
+            'CodeSize'     => code_size,
+            'FunctionArn'  => function_arn,
+            'FunctionName' => function_name,
+            'Handler'      => handler,
+            'LastModified' => Time.now.utc,
+            'MemorySize'   => params.delete('MemorySize') || 128,
+            'Timeout'      => params.delete('Timeout')    || 3,
+            'Role'         => role,
+            'Runtime'      => runtime
+          }
+          function['Description'] = description if description
+
+          self.data[:functions][function_arn] = function
+          response.body   = function
+          response.status = 200
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/delete_event_source_mapping.rb

@@ -0,0 +1,46 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+
+        # Removes an event source mapping.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_DeleteEventSourceMapping.html
+        # ==== Parameters
+        # * UUID <~String> - event source mapping ID.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~String>:
+        def delete_event_source_mapping(params={})
+          mapping_id = params.delete('UUID')
+          request({
+            :method  => 'DELETE',
+            :path    => "/event-source-mappings/#{mapping_id}",
+            :expects => 202
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def delete_event_source_mapping(params={})
+          mapping = self.get_event_source_mapping(params).body
+
+          unless mapping
+            message  = "ResourceNotFoundException => "
+            message << "The resource you requested does not exist."
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          mapping_id = mapping['UUID']
+          self.data[:event_source_mappings].delete(mapping_id)
+
+          mapping['State'] = 'Deleting'
+
+          response = Excon::Response.new
+          response.status = 202
+          response.body = mapping
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/delete_function.rb

@@ -0,0 +1,43 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+
+        # Deletes the specified Lambda function code and configuration.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_DeleteFunction.html
+        # ==== Parameters
+        # * FunctionName <~String> - Lambda function to delete.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~String>:
+        def delete_function(params={})
+          function_name = params.delete('FunctionName')
+          request({
+            :method  => 'DELETE',
+            :path    => "/functions/#{function_name}",
+            :expects => 204
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def delete_function(params={})
+          response = Excon::Response.new
+          response.status = 204
+          response.body = ''
+
+          function = self.get_function_configuration(params).body
+          function_id = function['FunctionArn']
+
+          self.data[:functions].delete function_id
+          self.data[:permissions].delete function_id
+          self.data[:event_source_mappings].delete_if do |m,f|
+            f['FunctionArn'].eql?(function_id)
+          end
+
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/get_event_source_mapping.rb

@@ -0,0 +1,54 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        # Returns configuration information for the specified event source mapping.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_GetEventSourceMapping.html
+        # ==== Parameters
+        # * UUID <~String> - AWS Lambda assigned ID of the event source mapping.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'BatchSize' <~Integer> - largest number of records that AWS Lambda will retrieve from your event source at the time of invoking your function.
+        #     * 'EventSourceArn' <~String> - Amazon Resource Name (ARN) of the stream that is the source of events.
+        #     * 'FunctionArn' <~String> - Lambda function to invoke when AWS Lambda detects an event on the stream.
+        #     * 'LastModified' <~Time> - UTC time string indicating the last time the event mapping was updated.
+        #     * 'LastProcessingResult' <~String> - result of the last AWS Lambda invocation of your Lambda function.
+        #     * 'State' <~String> - state of the event source mapping.
+        #     * 'StateTransitionReason' <~String> - reason the event source mapping is in its current state.
+        #     * 'UUID' <~String> - AWS Lambda assigned opaque identifier for the mapping.
+        #     * 'Code' <~Hash> - object for the Lambda function location.
+        #     * 'Configuration' <~Hash> - function metadata description.
+        def get_event_source_mapping(params={})
+          mapping_id = params.delete('UUID')
+          request({
+            :method  => 'GET',
+            :path    => "/event-source-mappings/#{mapping_id}"
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def get_event_source_mapping(params={})
+          mapping_id = params.delete('UUID')
+
+          unless mapping = self.data[:event_source_mappings][mapping_id]
+            message  = 'ResourceNotFoundException => '
+            message << 'The resource you requested does not exist.'
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          if mapping['State'].eql?('Creating')
+            mapping['LastProcessingResult'] = 'OK'
+            mapping['State'] = 'Enabled'
+          end
+
+          response = Excon::Response.new
+          response.status = 200
+          response.body = mapping
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/get_function.rb

@@ -0,0 +1,74 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        require 'fog/aws/parsers/lambda/base'
+
+        # Returns the configuration information of the Lambda function.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_GetFunction.html
+        # ==== Parameters
+        # * FunctionName <~String> - Lambda function name.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'Code' <~Hash> - object for the Lambda function location.
+        #     * 'Configuration' <~Hash> - function metadata description.
+        def get_function(params={})
+          function_name = params.delete('FunctionName')
+          request({
+            :method  => 'GET',
+            :path    => "/functions/#{function_name}/versions/HEAD",
+            :parser  => Fog::AWS::Parsers::Lambda::Base.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def get_function(params={})
+          response = Excon::Response.new
+          response.status = 200
+          response.body = ''
+
+          unless function_id = params.delete('FunctionName')
+            raise Fog::AWS::Lambda::Error, 'Function name cannot be blank.'
+          end
+
+          if function_id.match(/^arn:aws:lambda:.+:function:.+/)
+            function = self.data[:functions][function_id]
+          else
+            search_function = Hash[
+              self.data[:functions].select do |f,v|
+                v['FunctionName'].eql?(function_id)
+              end
+            ]
+            function = search_function.values.first
+          end
+
+          msg = 'The resource you requested does not exist.'
+          raise Fog::AWS::Lambda::Error, msg if (function.nil? || function.empty?)
+
+          location = "https://awslambda-#{self.region}-tasks.s3-#{self.region}"
+          location << ".amazonaws.com/snapshot/#{self.account_id}/"
+          location << "#{function['FunctionName']}-#{UUID.uuid}"
+          location << '?x-amz-security-token='
+          location << Fog::Mock.random_base64(718)
+          location << "&AWSAccessKeyId=#{self.aws_access_key_id}"
+          location << "&Expires=#{Time.now.to_i + 60*10}"
+          location << '&Signature='
+          location << Fog::Mock.random_base64(28)
+
+          body = {
+            'Code' => {
+              'Location'       => location,
+              'RepositoryType' => 'S3'
+            },
+            'Configuration' => function
+          }
+          response.body = body
+
+          response
+        end
+      end
+    end
+  end
+end