fog-aws 0.4.1 → 0.5.0

data/lib/fog/aws/requests/ecs/register_task_definition.rb

@@ -0,0 +1,68 @@
+module Fog
+  module AWS
+    class ECS
+      class Real
+        require 'fog/aws/parsers/ecs/register_task_definition'
+
+        # Registers a new task definition from the supplied family and containerDefinitions.
+        # http://docs.aws.amazon.com/AmazonECS/latest/APIReference/API_RegisterTaskDefinition.html
+        # ==== Parameters
+        # * containerDefinitions <~Array> - list of container definitions in JSON format that describe the different containers that make up your task.
+        # * family <~String> - family for a task definition, which allows you to track multiple versions of the same task definition.
+        # * volumes <~String> - list of volume definitions in JSON format that containers in your task may use.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'TaskDefinition' <~Array> - full task definition description registered
+        def register_task_definition(params={})
+          serialized_params = {}
+          params.each_pair do |k,v|
+            serialized_params.merge!(Fog::AWS.serialize_keys(k, v))
+          end
+          request({
+            'Action'  => 'RegisterTaskDefinition',
+            :parser   => Fog::Parsers::AWS::ECS::RegisterTaskDefinition.new
+          }.merge(serialized_params))
+        end
+      end
+
+      class Mock
+        def register_task_definition(params={})
+          response = Excon::Response.new
+          response.status = 200
+
+          family_error = 'ClientException => Family can not be blank.'
+          container_error = 'ClientException => Container list cannot be empty.'
+          raise Fog::AWS::ECS::Error, family_error    unless params['family']
+          raise Fog::AWS::ECS::Error, container_error unless params['containerDefinitions']
+
+          owner_id = Fog::AWS::Mock.owner_id
+          taskdef_name = params['family']
+          taskdef_rev = (1..9).to_a.shuffle.first
+          taskdef_path = "task-definition/#{taskdef_name}:#{taskdef_rev}"
+          taskdef_arn = Fog::AWS::Mock.arn('ecs', owner_id, taskdef_path, region)
+
+          task_definition = {
+            'revision'             => taskdef_rev,
+            'taskDefinitionArn'    => taskdef_arn,
+            'family'               => params['family'],
+            'containerDefinitions' => params['containerDefinitions']
+          }
+          task_definition['volumes'] = params['volumes'] if params['volumes']
+
+          self.data[:task_definitions] << task_definition
+
+          response.body = {
+            'ResponseMetadata' => {
+              'RequestId' => Fog::AWS::Mock.request_id
+            },
+            'RegisterTaskDefinitionResult' => {
+              'taskDefinition' => task_definition
+            }
+          }
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/ecs/run_task.rb

@@ -0,0 +1,114 @@
+module Fog
+  module AWS
+    class ECS
+      class Real
+        require 'fog/aws/parsers/ecs/run_task'
+
+        # Start a task using random placement and the default Amazon ECS scheduler.
+        # http://docs.aws.amazon.com/AmazonECS/latest/APIReference/API_RunTask.html
+        # ==== Parameters
+        # * cluster <~String> - short name or full Amazon Resource Name (ARN) of the cluster that you want to run your task on.
+        # * count <~Integer> - number of instantiations of the specified task that you would like to place on your cluster.
+        # * overrides <~Hash> - list of container overrides. 
+        # * startedBy <~String> - optional tag specified when a task is started
+        # * taskDefinition <~String> - family and revision (family:revision) or full ARN of the task definition that you want to run.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'tasks' <~Array> - full description of the tasks that were run.
+        #     * 'failures' <~Array> - Any failed tasks from your RunTask action are listed here.
+        def run_task(params={})
+          if overrides = params.delete('overrides')
+            serialized_overrides = {}
+            if overrides.is_a?(Hash)
+              overrides.each_pair do |k,v|
+                serialized_overrides.merge!(Fog::AWS.serialize_keys(k, v))
+              end
+            end
+            params.merge!('overrides' => serialized_overrides)
+          end
+
+          request({
+            'Action'  => 'RunTask',
+            :parser   => Fog::Parsers::AWS::ECS::RunTask.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def run_task(params={})
+          response = Excon::Response.new
+          response.status = 200
+
+          unless task_def_id = params.delete('taskDefinition')
+            msg = 'ClientException => TaskDefinition cannot be empty.'
+            raise Fog::AWS::ECS::Error, msg
+          end
+
+          begin
+            result = describe_task_definition('taskDefinition' => task_def_id).body
+            task_def = result["DescribeTaskDefinitionResult"]["taskDefinition"]
+            task_def_arn = task_def["taskDefinitionArn"]
+          rescue Fog::AWS::ECS::Error => e
+            msg = 'ClientException => TaskDefinition not found.'
+            raise Fog::AWS::ECS::Error, msg
+          end
+
+          if %w(count overrides).any? { |k| params.has_key?(k) }
+            Fog::Logger.warning("you used parameters not mocked yet [light_black](#{caller.first})[/]")
+            Fog::Mock.not_implemented
+          end
+
+          cluster_id = params.delete('cluster') || 'default'
+          cluster_arn = nil
+          owner_id = Fog::AWS::Mock.owner_id
+
+          if cluster_id.match(/^arn:aws:ecs:.+:\d{1,12}:cluster\/(.+)$/)
+            cluster_arn = cluster_id
+          else
+            cluster_path = "cluster/#{cluster_id}"
+            cluster_arn = Fog::AWS::Mock.arn('ecs', owner_id, cluster_path, region)
+          end
+
+          task_path = "task/#{UUID.uuid}"
+          task_arn = Fog::AWS::Mock.arn('ecs', owner_id, task_path, region)
+          instance_path = "container-instance/#{UUID.uuid}"
+          container_instance_arn = Fog::AWS::Mock.arn('ecs', owner_id, instance_path, region)
+
+          containers = []
+          task_def["containerDefinitions"].each do |c|
+            container_path = "container/#{UUID.uuid}"
+            containers << {
+              'name'         => c['name'],
+              'taskArn'      => task_arn,
+              'lastStatus'   => 'PENDING',
+              'containerArn' => Fog::AWS::Mock.arn('ecs', owner_id, container_path, region)
+            }
+          end
+
+          task = {
+            'clusterArn'           => cluster_arn,
+            'desiredStatus'        => 'RUNNING',
+            'taskDefinitionArn'    => task_def_arn,
+            'lastStatus'           => 'PENDING',
+            'taskArn'              => task_arn,
+            'containerInstanceArn' => container_instance_arn,
+            'containers'           => containers
+          }
+          self.data[:tasks] << task
+
+          response.body = {
+            'RunTaskResult' => {
+              'failures' => [],
+              'tasks' => [] << task
+            },
+            'ResponseMetadata' => {
+              'RequestId' => Fog::AWS::Mock.request_id
+            }
+          }
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/ecs/start_task.rb

@@ -0,0 +1,130 @@
+module Fog
+  module AWS
+    class ECS
+      class Real
+        require 'fog/aws/parsers/ecs/start_task'
+
+        # Starts a new task from the specified task definition on the specified container instance or instances.
+        # http://docs.aws.amazon.com/AmazonECS/latest/APIReference/API_StartTask.html
+        # ==== Parameters
+        # * cluster <~String> - short name or full Amazon Resource Name (ARN) of the cluster that you want to start your task on.
+        # * containerInstances <~Array> - container instance UUIDs or full ARN entries for the container instances on which you would like to place your task.
+        # * overrides <~Hash> - list of container overrides.
+        # * startedBy <~String> - optional tag specified when a task is started
+        # * taskDefinition <~String> - family and revision (family:revision) or full ARN of the task definition that you want to start.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'tasks' <~Array> - full description of the tasks that were started.
+        #     * 'failures' <~Array> - Any failed tasks from your StartTask action are listed here.
+        def start_task(params={})
+          if container_instances = params.delete('containerInstances')
+            params.merge!(Fog::AWS.indexed_param('containerInstances.member', [*container_instances]))
+          end
+
+          if overrides = params.delete('overrides')
+            serialized_overrides = {}
+            if overrides.is_a?(Hash)
+              overrides.each_pair do |k,v|
+                serialized_overrides.merge!(Fog::AWS.serialize_keys(k, v))
+              end
+            end
+            params.merge!('overrides' => serialized_overrides)
+          end
+
+          request({
+            'Action'  => 'StartTask',
+            :parser   => Fog::Parsers::AWS::ECS::StartTask.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def start_task(params={})
+          response = Excon::Response.new
+          response.status = 200
+
+          unless task_def_id = params.delete('taskDefinition')
+            msg = 'ClientException => TaskDefinition cannot be empty.'
+            raise Fog::AWS::ECS::Error, msg
+          end
+
+          unless instances_id = params.delete('containerInstances')
+            msg = 'ClientException => Container instances cannot be empty.'
+            raise Fog::AWS::ECS::Error, msg
+          end
+
+          begin
+            result = describe_task_definition('taskDefinition' => task_def_id).body
+            task_def = result["DescribeTaskDefinitionResult"]["taskDefinition"]
+            task_def_arn = task_def["taskDefinitionArn"]
+          rescue Fog::AWS::ECS::Error => e
+            msg = 'ClientException => TaskDefinition not found.'
+            raise Fog::AWS::ECS::Error, msg
+          end
+
+          if %w(startedBy overrides).any? { |k| params.has_key?(k) }
+            Fog::Logger.warning("you used parameters not mocked yet [light_black](#{caller.first})[/]")
+            Fog::Mock.not_implemented
+          end
+
+          cluster_id = params.delete('cluster') || 'default'
+          cluster_arn = nil
+          owner_id = Fog::AWS::Mock.owner_id
+
+          if cluster_id.match(/^arn:aws:ecs:.+:\d{1,12}:cluster\/(.+)$/)
+            cluster_arn = cluster_id
+          else
+            cluster_path = "cluster/#{cluster_id}"
+            cluster_arn = Fog::AWS::Mock.arn('ecs', owner_id, cluster_path, region)
+          end
+
+          task_path = "task/#{UUID.uuid}"
+          task_arn = Fog::AWS::Mock.arn('ecs', owner_id, task_path, region)
+          instance_path = "container-instance/#{UUID.uuid}"
+
+          instance_id = [*instances_id].first
+          if instance_id.match(/^arn:aws:ecs:.+:\d{1,12}:container-instance\/(.+)$/)
+            container_instance_arn = instance_id
+          else
+            instance_path = "container-instance/#{instance_id}"
+            container_instance_arn = Fog::AWS::Mock.arn('ecs', owner_id, instance_path, region)
+          end
+
+          containers = []
+          task_def["containerDefinitions"].each do |c|
+            container_path = "container/#{UUID.uuid}"
+            containers << {
+              'name'         => c['name'],
+              'taskArn'      => task_arn,
+              'lastStatus'   => 'PENDING',
+              'containerArn' => Fog::AWS::Mock.arn('ecs', owner_id, container_path, region)
+            }
+          end
+
+          task = {
+            'clusterArn'           => cluster_arn,
+            'desiredStatus'        => 'RUNNING',
+            'taskDefinitionArn'    => task_def_arn,
+            'lastStatus'           => 'PENDING',
+            'taskArn'              => task_arn,
+            'containerInstanceArn' => container_instance_arn,
+            'containers'           => containers
+          }
+          self.data[:tasks] << task
+
+          response.body = {
+            'StartTaskResult' => {
+              'failures' => [],
+              'tasks' => [] << task
+            },
+            'ResponseMetadata' => {
+              'RequestId' => Fog::AWS::Mock.request_id
+            }
+          }
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/ecs/stop_task.rb

@@ -0,0 +1,64 @@
+module Fog
+  module AWS
+    class ECS
+      class Real
+        require 'fog/aws/parsers/ecs/stop_task'
+
+        # Stops a running task.
+        # http://docs.aws.amazon.com/AmazonECS/latest/APIReference/API_StopTask.html
+        # ==== Parameters
+        # * cluster <~String> - short name or full Amazon Resource Name (ARN) of the cluster that hosts the task you want to stop.
+        # * task <~String> - task UUIDs or full Amazon Resource Name (ARN) entry of the task you would like to stop.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'Task' <~Hash> - The full description of the stopped task.
+        def stop_task(params={})
+          request({
+            'Action' => 'StopTask',
+            :parser  => Fog::Parsers::AWS::ECS::StopTask.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def stop_task(params={})
+          response = Excon::Response.new
+          response.status = 200
+
+          unless task_id = params.delete('task')
+            msg = "InvalidParameterException => Task can not be blank."
+            raise Fog::AWS::ECS::Error, msg
+          end
+
+          if cluster = params.delete('cluster')
+            Fog::Logger.warning("you used parameters not mocked yet [light_black](#{caller.first})[/]")
+          end
+
+          if match = task_id.match(/^arn:aws:ecs:.+:\d{1,12}:task\/(.+)$/)
+            i = self.data[:tasks].index { |t| t['taskArn'].eql?(task_id) }
+          else
+            i = self.data[:tasks].index { |t| t['taskArn'].match(/#{task_id}$/) }
+          end
+
+          msg = "ClientException => The referenced task was not found."
+          raise Fog::AWS::ECS::Error, msg unless i
+
+          task = self.data[:tasks][i]
+          task['desiredStatus'] = 'STOPPED'
+          self.data[:tasks].delete_at(i)
+
+          response.body = {
+            'StopTaskResult' => {
+              'task' => task
+            },
+            'ResponseMetadata' => {
+              'RequestId' => Fog::AWS::Mock.request_id
+            }
+          }
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/ecs/update_service.rb

@@ -0,0 +1,81 @@
+module Fog
+  module AWS
+    class ECS
+      class Real
+        require 'fog/aws/parsers/ecs/update_service'
+
+        # Modify the desired count or task definition used in a service.
+        # http://docs.aws.amazon.com/AmazonECS/latest/APIReference/API_UpdateService.html
+        # ==== Parameters
+        # * cluster <~String> - short name or full Amazon Resource Name (ARN) of the cluster that your service is running on.
+        # * desiredCount <~Integer> - number of instantiations of the task that you would like to place and keep running in your service.
+        # * service <~String> - name of the service that you want to update.
+        # * taskDefinition <~String> - family and revision (family:revision) or full Amazon Resource Name (ARN) of the task definition that you want to run in your service.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'Service'<~Hash> - The full description of the updated cluster
+        def update_service(params={})
+          request({
+            'Action'  => 'UpdateService',
+            :parser   => Fog::Parsers::AWS::ECS::UpdateService.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def update_service(params={})
+          response = Excon::Response.new
+          response.status = 200
+
+          service_id = params.delete('service')
+          msg = 'ClientException => Service cannot be empty.'
+          raise Fog::AWS::ECS::Error, msg unless service_id
+
+          owner_id = Fog::AWS::Mock.owner_id
+
+          cluster = params.delete('cluster') || 'default'
+          if !cluster.match(/^arn:aws:ecs:.+:.+:cluster\/(.+)$/)
+            cluster_path = "cluster/#{cluster}"
+            cluster_arn = Fog::AWS::Mock.arn('ecs', owner_id, cluster_path, region)
+          else
+            cluster_arn = cluster
+          end
+
+          if match = service_id.match(/^arn:aws:ecs:.+:\d{1,12}:service\/(.+)$/)
+            i = self.data[:services].index do |s|
+              s['clusterArn'].eql?(cluster_arn) && s['serviceArn'].eql?(service_id)
+            end
+          else
+            i = self.data[:services].index do |s|
+              s['clusterArn'].eql?(cluster_arn) && s['serviceName'].eql?(service_id)
+            end
+          end
+
+          msg = "ServiceNotFoundException => Service not found."
+          raise Fog::AWS::ECS::Error, msg unless i
+
+          service = self.data[:services][i]
+
+          if desired_count = params.delete('desiredCount')
+            # ignore
+          end
+
+          if task_definition = params.delete('taskDefinition')
+            service['taskDefinition'] = task_definition
+          end
+
+          response.body = {
+            'UpdateServiceResult' => {
+              'service' => service
+            },
+            'ResponseMetadata' => {
+              'RequestId' => Fog::AWS::Mock.request_id
+            }
+          }
+          response
+        end
+      end
+    end
+  end
+end