aws-codedeploy-agent 0.0.1

data/test/instance_agent/codedeploy_plugin/request_helper_test.rb

@@ -0,0 +1,37 @@
+require 'instance_metadata'
+require 'aws/codedeploy_commands'
+
+class CodeDeployRequestHelperTest < InstanceAgentTestCase
+
+  include InstanceAgent::CodeDeployPlugin
+
+  context "The CodeDeploy request helper" do
+
+    context "when calling verify_clients_configuration" do
+     setup do
+        ENV['AWS_REGION'] = 'us-west-2'
+        @deploy_client = mock('InstanceAgent::CodeDeployPlugin::CodeDeployControl')
+        @deploy_client.stubs(:ssl_verify_peer).returns(true)
+        @deploy_client.stubs(:verify_cert_fields).returns(true)
+
+        @request_helper = RequestHelper.new(:deploy_control_client => @deploy_client)
+
+      end
+
+      should "successfully call verify_clients_configuration" do
+        assert_equal(true, @request_helper.verify_clients_configuration.empty?)
+      end
+
+      should "fail if deploy client doesnt do ssl verify peer" do
+        @deploy_client.stubs(:ssl_verify_peer).returns(false)
+        assert_equal(false, @request_helper.verify_clients_configuration.empty?)
+      end
+
+      should "fail if deploy client cert verify failed" do
+        @deploy_client.stubs(:verify_cert_fields).returns(false)
+        assert_equal(false, @request_helper.verify_clients_configuration.empty?)
+      end
+
+    end
+  end
+end

data/test/instance_agent/config_test.rb

@@ -0,0 +1,64 @@
+require 'test_helper'
+
+class InstanceAgentConfigTest < InstanceAgentTestCase
+  context 'The instance agent configuration' do
+
+    should 'have a default configuration' do
+      InstanceAgent::Config.init
+      assert_equal({
+        :wait_between_spawning_children => 1,
+        :log_dir => nil,
+        :pid_dir => nil,
+        :shared_dir => nil,
+        :user => nil,
+        :group=>nil,
+        :program_name => "codedeploy-agent",
+        :wait_after_throttle_error => 60,
+        :wait_between_runs => 30,
+        :verbose => false,
+        :config_file => nil,
+        :wait_after_connection_problem => 5,
+        :children => 1,
+        :max_runs_per_worker => 0,
+        :http_read_timeout => 80,
+        :instance_service_region => nil,
+        :instance_service_endpoint => nil,
+        :instance_service_port => nil,
+        :wait_between_runs => 30,
+        :wait_after_error => 30,
+        :codedeploy_test_profile => 'prod'
+      }, InstanceAgent::Config.config)
+    end
+
+    should 'be the same as the ProcessManager configuration for the current object' do
+      config = InstanceAgent::Config.instance_variable_get(:@config)
+      assert_equal config, ProcessManager::Config.instance_variable_get(:@config)
+    end
+
+    should 'execute all available validation methods' do
+      validations = sequence('validation')
+      err = []
+      InstanceAgent::Config.any_instance.expects(:validate_children).with(err).in_sequence(validations)
+      InstanceAgent::Config.validate_config
+    end
+
+    context 'validate configuration' do
+
+      setup do
+        InstanceAgent::Config.config[:instance_service_region] = 'eu-west-1'
+        InstanceAgent::Config.config[:instance_service_endpoint] = 'api-endpoint.example.com'
+        InstanceAgent::Config.config[:instance_service_port] = 123
+      end
+
+      should 'validate the children setting' do
+        InstanceAgent::Config.config[:children] = nil
+        puts InstanceAgent::Config.config.inspect
+        assert_equal 'children can only be set to 1', InstanceAgent::Config.validate_config.pop
+        InstanceAgent::Config.config[:children] = 2
+        assert_equal 'children can only be set to 1', InstanceAgent::Config.validate_config.pop
+        InstanceAgent::Config.config[:children] = 1
+        assert InstanceAgent::Config.validate_config.empty?, InstanceAgent::Config.validate_config.inspect
+      end
+    end
+  end
+end

data/test/instance_agent/runner/child_test.rb

@@ -0,0 +1,87 @@
+require 'test_helper'
+
+class RunnerChildTest < InstanceAgentTestCase
+  context 'The runner child' do
+    setup do
+      @dir = '/tmp'
+      @agent = mock()
+      InstanceAgent::CodeDeployPlugin::CommandPoller.stubs(:new).returns(@agent)
+      @agent.stubs(:description).returns("CommandPoller")
+      InstanceAgent::Runner::Child.any_instance.stubs(:trap_signals)
+      @child = InstanceAgent::Runner::Child.new(0, 777)
+      ProcessManager::Log.init(File.join(@dir, 'codedeploy-agent.log'))
+      ProcessManager.stubs(:set_program_name)
+      InstanceAgent::Config.config[:wait_after_throttle_error] = 0
+      InstanceAgent::Config.config[:wait_between_runs] = 0
+      InstanceAgent::Config.config[:wait_between_spawning_children] = 0
+      SimplePid.stubs(:drop)
+      ProcessManager.reset_on_error_callbacks
+    end
+
+    context 'when preparing the run' do
+      should 'load the correct runner' do
+        assert_nothing_raised do
+          @child = InstanceAgent::Runner::Child.new(0, 777)
+          @child.prepare_run
+          assert_equal @child.runner, @agent
+        end
+      end
+
+      should 'validate the index' do
+        assert_raise(ArgumentError) do
+          @child = InstanceAgent::Runner::Child.new(9, 777)
+          @child.prepare_run
+        end
+
+        assert_nothing_raised do
+          @child = InstanceAgent::Runner::Child.new(0, 777)
+          @child.prepare_run
+        end
+      end
+
+      context 'sets the process description' do
+        should 'set it for the running children' do
+          @child.stubs(:runner).returns(runner = mock('runner'))
+          runner.stubs(:description).returns 'master-process'
+          assert_equal 'master-process of master 777', @child.description
+        end
+
+        should 'set it for the booting children' do
+          assert_equal 'booting child', @child.description
+        end
+      end
+
+      context 'handle exceptions' do
+        setup do
+          @child.stubs(:runner).returns(runner = mock('runner'))
+          runner.stubs(:description).returns 'master-process'
+        end
+        should 'handle SocketErrors during the run and exit cleanly' do
+          InstanceAgent::Config.config[:wait_after_connection_problem] = 0
+          @child.expects(:runner).raises(SocketError)
+          ::ProcessManager::Log.expects(:info)
+          @child.expects(:sleep).with(0)
+          @child.expects(:exit).with(1)
+          @child.run
+        end
+
+        should 'handle throttling exceptions nicely' do
+          @child.expects(:runner).raises(Exception, 'throttle exception')
+          ::ProcessManager::Log.expects(:error)
+          @child.expects(:sleep).with(0)
+          @child.expects(:exit).with(1).at_least(1)
+
+          @child.run
+        end
+
+        should 'handle other exceptions nicely' do
+          @child.expects(:runner).raises(Exception, 'some exception')
+          ::ProcessManager::Log.expects(:error)
+          @child.expects(:exit).with(1)
+
+          @child.run
+        end
+      end
+    end
+  end
+end

data/test/instance_metadata_test.rb

@@ -0,0 +1,97 @@
+require 'test_helper'
+require 'json'
+
+class InstanceMetadataTest < InstanceAgentTestCase
+
+  def self.should_check_status_code(&blk)
+    should 'raise unless status code is 200' do
+      @response.stubs(:code).returns(503)
+      assert_raise(&blk)
+    end
+  end
+
+  context 'The instance metadata service' do
+    setup do
+      region = 'us-east-1'
+      account_id = '123456789012'
+      instance_id = 'i-deadbeef'
+      @host_identifier = "arn:aws:ec2:#{region}:#{account_id}:instance/#{instance_id}"
+      @instance_document = JSON.dump({"accountId" => account_id, "region" => region, "instanceId" => instance_id})
+      @http = mock()
+      @response = mock()
+      @response.stubs(:code).returns("200")
+      @http.stubs(:get).returns(@response)
+      Net::HTTP.stubs(:start).yields(@http)
+    end
+
+    context 'getting the host identifier' do
+
+      setup do
+        @response.stubs(:body).returns(@instance_document)
+      end
+
+      should 'connect to the right host' do
+        Net::HTTP.expects(:start).with('169.254.169.254', 80).yields(@http)
+        InstanceMetadata.host_identifier
+      end
+
+      should 'call the correct URL' do
+        @http.expects(:get).
+          with("/latest/dynamic/instance-identity/document").
+          returns(@response)
+        InstanceMetadata.host_identifier
+      end
+
+      should 'return the body' do
+        assert_equal(@host_identifier, InstanceMetadata.host_identifier)
+      end
+
+      should 'strip whitesace in the body' do
+        @response.stubs(:body).returns(" \t#{@instance_document}   ")
+        assert_equal(@host_identifier, InstanceMetadata.host_identifier)
+      end
+
+      should_check_status_code { InstanceMetadata.host_identifier }
+
+    end
+
+    context 'getting the region' do
+
+      setup do
+        @response.stubs(:body).returns("us-east-1a")
+      end
+
+      should 'connect to the right host' do
+        Net::HTTP.expects(:start).with('169.254.169.254', 80).yields(@http)
+        InstanceMetadata.region
+      end
+
+      should 'call the correct URL' do
+        @http.expects(:get).
+          with("/latest/meta-data/placement/availability-zone").
+          returns(@response)
+        InstanceMetadata.region
+      end
+
+      should 'return the region part of the AZ' do
+        @response.stubs(:body).returns("us-east-1a")
+        assert_equal("us-east-1", InstanceMetadata.region)
+      end
+
+      should 'raise an error if the response is not an AZ' do
+        @response.stubs(:body).returns("foobar")
+        assert_raise { InstanceMetadata.region }
+      end
+
+      should 'ignore whitespace in the body' do
+        @response.stubs(:body).returns(" \tus-east-1a   ")
+        assert_equal("us-east-1", InstanceMetadata.region)
+      end
+
+      should_check_status_code { InstanceMetadata.region }
+
+    end
+
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,16 @@
+# encoding: UTF-8
+require 'thread'
+require 'rubygems'
+require "bundler"
+Bundler.require(:default, :test)
+
+# test framework
+require 'test/unit'
+require 'active_support/testing/assertions'
+require 'shoulda'
+require 'mocha/setup'
+require 'base64'
+
+# require local test helpers. If you need a helper write,
+# keep this pattern or you'll be punished hard
+require 'helpers/instance_agent_helper'

data/vendor/gems/.codedeploy-commands-1.0.0.created.rid

@@ -0,0 +1 @@
+Tue, 04 Nov 2014 07:48:38 +0000

data/vendor/gems/codedeploy-commands/apis/CodeDeployCommand.api.json

@@ -0,0 +1,372 @@
+{
+  "metadata": {
+    "apiVersion": "2014-10-06",
+    "endpointPrefix": "codedeploy-commands",
+    "jsonVersion": "1.1",
+    "regionalEndpoints": {
+      "us-east-1": "codedeploy-commands.us-east-1.amazonaws.com",
+      "us-west-2": "codedeploy-commands.us-west-2.amazonaws.com"
+    },
+    "serviceAbbreviation": "CodeDeployCommand",
+    "serviceFullName": "AWS CodeDeploy Command Service",
+    "signatureVersion": "v4",
+    "targetPrefix": "CodeDeployCommandService_v20141006",
+    "protocol": "json"
+  },
+  "documentation": "Control plane for the CodeDeploy stack. Exposes APIs needed by CodeDeploy clients to run deployment commands.",
+  "operations": {
+    "GetDeploymentSpecification": {
+      "name": "GetDeploymentSpecification",
+      "http": {
+        "method": "POST",
+        "requestUri": "/"
+      },
+      "input": {
+        "shape": "GetDeploymentSpecificationInput"
+      },
+      "output": {
+        "shape": "GetDeploymentSpecificationOutput"
+      },
+      "errors": [
+        {
+          "shape": "ClientException",
+          "exception": true,
+          "documentation": "This exception indicates that the request failed due to the fault of the customer (either an invalid request was provided, referred to a non-existant object, or another reason within the client's control)."
+        },
+        {
+          "shape": "ServerException",
+          "exception": true,
+          "fault": true,
+          "documentation": "This exception indicates that the request failed due to a problem on the server, or with the server's dependencies."
+        }
+      ],
+      "documentation": "Retrieve the deployment specification for the deployment and host, consisting of the client metadata provided when the deployment was created. The generic client metadata will be provided, as well as the client metadata for the host's variant (if variant-specific metadata was provided). Throws DeploymentNotFoundException if the DeploymentExecutionId does not identify a current deployment. Throws HostNotFoundException if the host is not recognized by the deployment engine. Throws ServerException for failures caused by the deployment system or its dependencies."
+    },
+    "PollHostCommand": {
+      "name": "PollHostCommand",
+      "http": {
+        "method": "POST",
+        "requestUri": "/"
+      },
+      "input": {
+        "shape": "PollHostCommandInput"
+      },
+      "output": {
+        "shape": "PollHostCommandOutput"
+      },
+      "errors": [
+        {
+          "shape": "ClientException",
+          "exception": true,
+          "documentation": "This exception indicates that the request failed due to the fault of the customer (either an invalid request was provided, referred to a non-existant object, or another reason within the client's control)."
+        },
+        {
+          "shape": "ServerException",
+          "exception": true,
+          "fault": true,
+          "documentation": "This exception indicates that the request failed due to a problem on the server, or with the server's dependencies."
+        }
+      ],
+      "documentation": "This requests a command from the deployment workflow engine. If no command is ready to be dispatched, the output will be empty (HostCommand will be null). Throws HostNotFoundException if the host is not recognized by the deployment engine. Throws ServerException for failures caused by the deployment system or its dependencies."
+    },
+    "PostHostCommandUpdate": {
+      "name": "PostHostCommandUpdate",
+      "http": {
+        "method": "POST",
+        "requestUri": "/"
+      },
+      "input": {
+        "shape": "PostHostCommandUpdateInput"
+      },
+      "output": {
+        "shape": "PostHostCommandUpdateOutput"
+      },
+      "errors": [
+        {
+          "shape": "ClientException",
+          "exception": true,
+          "documentation": "This exception indicates that the request failed due to the fault of the customer (either an invalid request was provided, referred to a non-existant object, or another reason within the client's control)."
+        },
+        {
+          "shape": "ServerException",
+          "exception": true,
+          "fault": true,
+          "documentation": "This exception indicates that the request failed due to a problem on the server, or with the server's dependencies."
+        }
+      ],
+      "documentation": "This updates the central workflow engine with the current progress of the host command. This will also return the status of the host command centrally if possible, so agents can skip processing the command if it has been aborted / timed out. However, the status is optional, so if no status is returned the agent should treat it as if it was ok to continue. Throws ClientException for an invalid HostCommandIdentifier or Diagnostics. Throws ServerException for failures caused by the deployment system or its dependencies."
+    },
+    "PutHostCommandAcknowledgement": {
+      "name": "PutHostCommandAcknowledgement",
+      "http": {
+        "method": "POST",
+        "requestUri": "/"
+      },
+      "input": {
+        "shape": "PutHostCommandAcknowledgementInput"
+      },
+      "output": {
+        "shape": "PutHostCommandAcknowledgementOutput"
+      },
+      "errors": [
+        {
+          "shape": "ClientException",
+          "exception": true,
+          "documentation": "This exception indicates that the request failed due to the fault of the customer (either an invalid request was provided, referred to a non-existant object, or another reason within the client's control)."
+        },
+        {
+          "shape": "ServerException",
+          "exception": true,
+          "fault": true,
+          "documentation": "This exception indicates that the request failed due to a problem on the server, or with the server's dependencies."
+        }
+      ],
+      "documentation": "This notifies the central workflow engine that the agent has received the specified command and is ready to start execution. This will also return the status of the host command centrally if possible, so agents can skip processing the command if it has been aborted / timed out. However, the status is optional, so if no status is returned the agent should treat it as if it was ok to continue. Throws ClientException for an invalid HostCommandIdentifier or Diagnostics. Throws ServerException for failures caused by the deployment system or its dependencies."
+    },
+    "PutHostCommandComplete": {
+      "name": "PutHostCommandComplete",
+      "http": {
+        "method": "POST",
+        "requestUri": "/"
+      },
+      "input": {
+        "shape": "PutHostCommandCompleteInput"
+      },
+      "errors": [
+        {
+          "shape": "ClientException",
+          "exception": true,
+          "documentation": "This exception indicates that the request failed due to the fault of the customer (either an invalid request was provided, referred to a non-existant object, or another reason within the client's control)."
+        },
+        {
+          "shape": "ServerException",
+          "exception": true,
+          "fault": true,
+          "documentation": "This exception indicates that the request failed due to a problem on the server, or with the server's dependencies."
+        }
+      ],
+      "documentation": "This reports completion of the command back to the workflow engine. Throws ClientException for an invalid HostCommandIdentifier or Diagnostics. Throws ServerException for failures caused by the deployment system or its dependencies."
+    }
+  },
+  "shapes": {
+    "ClientException": {
+      "type": "structure",
+      "members": {
+      },
+      "exception": true,
+      "documentation": "This exception indicates that the request failed due to the fault of the customer (either an invalid request was provided, referred to a non-existant object, or another reason within the client's control)."
+    },
+    "DeploymentCommandName": {
+      "type": "string"
+    },
+    "DeploymentCommandStatus": {
+      "type": "string",
+      "enum": [
+        "Pending",
+        "InProgress",
+        "Succeeded",
+        "Failed"
+      ]
+    },
+    "DeploymentExecutionId": {
+      "type": "string",
+      "min": 1,
+      "max": 129,
+      "pattern": "^[^/]+/[^/]+/[^/]+/.+$"
+    },
+    "DeploymentSpecification": {
+      "type": "structure",
+      "members": {
+        "GenericEnvelope": {
+          "shape": "Envelope"
+        },
+        "VariantId": {
+          "shape": "VariantId"
+        },
+        "VariantEnvelope": {
+          "shape": "Envelope"
+        }
+      },
+      "documentation": "The DeploymentSpecification contains an envelope for the generic client metadata, and if there is variant-specific metadata, the ID of the variant for the host and the envelope containing that variant's metadata. All fields are optional, though an empty DeploymentSpecification is likely indicative of an error."
+    },
+    "Envelope": {
+      "type": "structure",
+      "members": {
+        "Format": {
+          "shape": "Format"
+        },
+        "Payload": {
+          "shape": "Payload"
+        }
+      },
+      "documentation": "For an Envelope used for host command diagnostics, Format is limited to 64 characters and Payload is limited to 8192 characters."
+    },
+    "Format": {
+      "type": "string",
+      "min": 1,
+      "max": 64
+    },
+    "GenericDateTimestamp": {
+      "type": "timestamp"
+    },
+    "GenericLong": {
+      "type": "long"
+    },
+    "GenericString": {
+      "type": "string"
+    },
+    "GetDeploymentSpecificationInput": {
+      "type": "structure",
+      "required": [
+        "DeploymentExecutionId",
+        "HostIdentifier"
+      ],
+      "members": {
+        "DeploymentExecutionId": {
+          "shape": "DeploymentExecutionId"
+        },
+        "HostIdentifier": {
+          "shape": "HostIdentifier"
+        }
+      }
+    },
+    "GetDeploymentSpecificationOutput": {
+      "type": "structure",
+      "members": {
+        "DeploymentSystem": {
+          "shape": "GenericString"
+        },
+        "DeploymentSpecification": {
+          "shape": "DeploymentSpecification"
+        }
+      }
+    },
+    "HostCommandIdentifier": {
+      "type": "string"
+    },
+    "HostCommandInstance": {
+      "type": "structure",
+      "members": {
+        "HostCommandIdentifier": {
+          "shape": "HostCommandIdentifier"
+        },
+        "HostIdentifier": {
+          "shape": "HostIdentifier"
+        },
+        "DeploymentExecutionId": {
+          "shape": "DeploymentExecutionId"
+        },
+        "CommandName": {
+          "shape": "DeploymentCommandName"
+        },
+        "Nonce": {
+          "shape": "GenericLong"
+        }
+      }
+    },
+    "HostIdentifier": {
+      "type": "string",
+      "documentation": "An identifier for referring to a unit of capacity."
+    },
+    "Payload": {
+      "type": "string",
+      "min": 0,
+      "max": 8192
+    },
+    "PollHostCommandInput": {
+      "type": "structure",
+      "required": [
+        "HostIdentifier"
+      ],
+      "members": {
+        "HostIdentifier": {
+          "shape": "HostIdentifier"
+        }
+      }
+    },
+    "PollHostCommandOutput": {
+      "type": "structure",
+      "members": {
+        "HostCommand": {
+          "shape": "HostCommandInstance"
+        }
+      }
+    },
+    "PostHostCommandUpdateInput": {
+      "type": "structure",
+      "required": [
+        "HostCommandIdentifier"
+      ],
+      "members": {
+        "HostCommandIdentifier": {
+          "shape": "HostCommandIdentifier"
+        },
+        "EstimatedCompletionTime": {
+          "shape": "GenericDateTimestamp"
+        },
+        "Diagnostics": {
+          "shape": "Envelope"
+        }
+      }
+    },
+    "PostHostCommandUpdateOutput": {
+      "type": "structure",
+      "members": {
+        "CommandStatus": {
+          "shape": "DeploymentCommandStatus"
+        }
+      }
+    },
+    "PutHostCommandAcknowledgementInput": {
+      "type": "structure",
+      "required": [
+        "HostCommandIdentifier"
+      ],
+      "members": {
+        "HostCommandIdentifier": {
+          "shape": "HostCommandIdentifier"
+        },
+        "Diagnostics": {
+          "shape": "Envelope"
+        }
+      }
+    },
+    "PutHostCommandAcknowledgementOutput": {
+      "type": "structure",
+      "members": {
+        "CommandStatus": {
+          "shape": "DeploymentCommandStatus"
+        }
+      }
+    },
+    "PutHostCommandCompleteInput": {
+      "type": "structure",
+      "required": [
+        "HostCommandIdentifier",
+        "CommandStatus"
+      ],
+      "members": {
+        "HostCommandIdentifier": {
+          "shape": "HostCommandIdentifier"
+        },
+        "CommandStatus": {
+          "shape": "DeploymentCommandStatus"
+        },
+        "Diagnostics": {
+          "shape": "Envelope"
+        }
+      }
+    },
+    "ServerException": {
+      "type": "structure",
+      "members": {
+      },
+      "exception": true,
+      "fault": true,
+      "documentation": "This exception indicates that the request failed due to a problem on the server, or with the server's dependencies."
+    },
+    "VariantId": {
+      "type": "string"
+    }
+  }
+}