qurd 0.0.1

data/test/configuration_test.rb

@@ -0,0 +1,333 @@
+require 'test_helper'
+describe Qurd::Configuration do
+  include WebMockStubs
+
+  subject { Qurd::Configuration.instance }
+  let(:mock) { Minitest::Mock.new }
+  let(:config) do
+    {
+      # dry_run:"true",
+      # daemonize:false,
+      pid_file: 'tmp/qurd.pid',
+      # wait_time:"0",
+      # visibility_timeout:"1",
+      log_file: 'tmp/qurd.log',
+      log_level: 'debug',
+      aws_credentials: [
+        {
+          name: 'credentials',
+          type: 'credentials',
+          options: {
+            access_key_id: 'access',
+            secret_access_key: 'secret'
+          }
+        }
+      ],
+      auto_scaling_queues: {
+        staging: {
+          credentials: 'credentials',
+          region: 'us-west-2',
+          queues: ['/ScalingNotificationsQueue/']
+        }
+      },
+      actions: {
+        launch: ['Qurd::Action::Dummy'],
+        launch_error: ['Qurd::Action::Dummy'],
+        terminate: ['Qurd::Action::Dummy'],
+        terminate_error: ['Qurd::Action::Dummy'],
+        test: ['Qurd::Action::Dummy']
+      }
+    }
+  end
+
+  def hashie_config(merge = {})
+    Hashie::Mash.new(config.merge!(merge))
+  end
+
+  def stub_yaml(config, &_block)
+    YAML.stub :load, config do
+      yield if block_given?
+    end
+  end
+
+  describe '#init' do
+    it 'sets default config values' do
+      config.merge!(log_file: nil, pid_file: nil)
+      stub_yaml config do
+        subject.init('/dev/null')
+        subject.config.daemonize.must_equal false
+        subject.config.dry_run.must_equal false
+        subject.config.listen_timeout.must_equal 300.0
+        subject.config.pid_file.must_equal '/var/run/qurd/qurd.pid'
+        subject.config.save_failures.must_equal true
+        subject.config.sqs_set_attributes_timeout.must_equal 10.0
+        subject.config.visibility_timeout.must_equal '300'
+        subject.config.wait_time.must_equal '20'
+      end
+    end
+
+    it 'keeps config values' do
+      config.merge!(
+        daemonize: true,
+        dry_run: true,
+        listen_timeout: 2,
+        pid_file: 'tmp/qurd.pid',
+        save_failures: false,
+        sqs_set_attributes_timeout: 3,
+        visibility_timeout: 0,
+        wait_time: 1
+      )
+      stub_yaml config do
+        subject.init('/dev/null')
+        subject.config.daemonize.must_equal true
+        subject.config.dry_run.must_equal true
+        subject.config.listen_timeout.must_equal 2.0
+        subject.config.pid_file.must_equal 'tmp/qurd.pid'
+        subject.config.save_failures.must_equal false
+        subject.config.sqs_set_attributes_timeout.must_equal 3.0
+        subject.config.visibility_timeout.must_equal '0'
+        subject.config.wait_time.must_equal '1'
+      end
+    end
+  end
+
+  describe '#debug?' do
+    it 'is true' do
+      hashie = hashie_config(log_level: 'debug')
+      subject.stub :config, hashie do
+        subject.debug?.must_equal true
+      end
+    end
+
+    it 'is false' do
+      hashie = hashie_config(log_level: 'info')
+      subject.stub :config, hashie do
+        subject.debug?.must_equal false
+      end
+    end
+  end
+
+  describe '#logger!' do
+    it 'logs and raises RuntimeError' do
+      mock.expect :error, nil, ['foo']
+      lambda do
+        subject.stub :logger, mock do
+          subject.logger!('foo')
+        end
+      end.must_raise RuntimeError, 'foo'
+      mock.verify
+    end
+
+    it 'logs and raises StandardError' do
+      mock.expect :error, nil, ['foo']
+      lambda do
+        subject.stub :logger, mock do
+          subject.logger!('foo', StandardError)
+        end
+      end.must_raise StandardError, 'foo'
+      mock.verify
+    end
+  end
+
+  describe '#configure' do
+    it 'configures logger, accounts, queues, and actions' do
+      aws_sqs_list_queues
+      aws_sqs_set_queue_attributes
+      stub_yaml config do
+        subject.configure('/dev/null')
+      end
+    end
+  end
+
+  describe '#configure_actions' do
+    it 'raises when objects are not found' do
+      config.merge!(actions: { launch: ['DoesNotExist'] })
+      stub_yaml config do
+        subject.init('/dev/null')
+        lambda do
+          subject.send :configure_actions
+        end.must_raise RuntimeError, 'Class undefined for actions: DoesNotExist'
+      end
+    end
+
+    it 'raises when action values are not arrays' do
+      config.merge!(actions: { launch: 'Qurd::Action::Dummy' })
+      stub_yaml config do
+        subject.init('/dev/null')
+        lambda do
+          subject.send :configure_actions
+        end.must_raise RuntimeError, 'Action types must be an array'
+      end
+    end
+  end
+
+  describe '#configure_listeners' do
+    it 'creates accounts data structure' do
+      aws_sqs_list_queues
+      aws_sqs_set_queue_attributes
+      stub_yaml config do
+        subject.init '/dev/null'
+        subject.send :configure_credentials
+        subject.send :configure_auto_scaling_queues
+        subject.config.listeners.count.must_equal 1
+      end
+    end
+  end
+
+  describe '#get_or_default' do
+    class Dummy
+      def nil
+        nil
+      end
+
+      def true
+        true
+      end
+    end
+    let(:obj) { Dummy.new }
+
+    it 'chooses a defined value' do
+      val = subject.send :get_or_default, obj, :true, 0
+      val.must_equal true
+    end
+
+    it 'chooses a default value' do
+      val = subject.send :get_or_default, obj, :nil, 0
+      val.must_equal 0
+    end
+
+    it 'casts a value' do
+      val = subject.send :get_or_default, obj, :nil, 0, :to_s
+      val.must_equal '0'
+    end
+  end
+
+  describe '#verify_account!' do
+    %w(credentials region queues).each do |key|
+      let(:monitor) do
+        {
+          'credentials' =>  'foo',
+          'region' =>  'bar',
+          'queues'  => 'baz'
+        }
+      end
+
+      it "raises if #{key} is nil" do
+        monitor.delete(key)
+        lambda do
+          subject.send :verify_account!, :bam, monitor
+        end.must_raise RuntimeError, "Account bam missing keys: #{key}"
+      end
+
+      it "raises if #{key} is empty" do
+        monitor[key] = ''
+        lambda do
+          subject.send :verify_account!, :bam, monitor
+        end.must_raise RuntimeError, "Account bam missing keys: #{key}"
+      end
+
+      it 'is ok' do
+        ret = subject.send :verify_account!, :bam, monitor
+        ret.must_equal nil
+      end
+    end
+  end
+
+  describe '#mkdir_p_file!' do
+    it 'makes directories' do
+      subject.send :mkdir_p_file!, '/tmp/qurd-test/foo/bar/bam/baz.txt'
+      File.new('/tmp/qurd-test/foo/bar/bam')
+      `rm -rf /tmp/qurd-test`
+    end
+
+    it 'raises RuntimeError' do
+      skip "can't be root" if Process.uid == 0
+      lambda do
+        subject.send :mkdir_p_file!, '/etc/foo'
+      end.must_raise RuntimeError, 'Directory not writable: /etc'
+    end
+  end
+
+  describe '#default_credentials' do
+    it 'creates a default' do
+      ec2metadata
+      ret = subject.send :default_credentials
+      ret[0][0].must_equal 'default'
+      ret[0][1].must_be_kind_of Aws::InstanceProfileCredentials
+      ret.count.must_equal 1
+    end
+  end
+
+  describe '#assume_role_credentials' do
+    let(:cred) do
+      Hashie::Mash.new(name: 'foo',
+                       options: {
+                         role_arn: 'arn:aws:iam::1:user/bob@example.com',
+                         role_session_name: 'foo'
+                       })
+    end
+    %w(policy duration_seconds external_id).each do |key|
+      it "sets option #{key}" do
+        Aws.config[:region] = 'us-west-2'
+        aws_sts_assume_role
+        cred.options[key] = '1'
+        ret = subject.send :assume_role_credentials, cred
+        ret[0].must_equal cred.name
+        ret[1].must_be_kind_of Aws::AssumeRoleCredentials
+        Aws.config[:region] = nil
+      end
+    end
+  end
+
+  describe '#credentials' do
+    let(:cred) do
+      Hashie::Mash.new(name: 'foo',
+                       options: {
+                         access_key_id: 'abc',
+                         secret_access_key: 'def'
+                       })
+    end
+    it 'sets access_key_id and secret_access_key' do
+      ret = subject.send :credentials, cred
+      ret[0].must_equal cred.name
+      ret[1].access_key_id.must_equal cred.options.access_key_id
+      ret[1].secret_access_key.must_equal cred.options.secret_access_key
+    end
+
+    it 'sets access_key_id, secret_access_key, and session_token' do
+      cred.options.session_token = 'foo'
+      ret = subject.send :credentials, cred
+      ret[0].must_equal cred.name
+      ret[1].access_key_id.must_equal cred.options.access_key_id
+      ret[1].secret_access_key.must_equal cred.options.secret_access_key
+      ret[1].session_token.must_equal cred.options.session_token
+    end
+  end
+
+  describe '#instance_profile_credentials' do
+    let(:cred) do
+      Hashie::Mash.new(name: 'foo',
+                       options: {
+                         retries: 0,
+                         http_open_timeout: 0,
+                         http_read_timeout: 0
+                       })
+    end
+    %w(port delay http_debug_output).each do |key|
+      it "sets option #{key}" do
+        ec2metadata
+        cred.options[key] = '80'
+        ret = subject.send :instance_profile_credentials, cred
+        ret[0].must_equal cred.name
+        ret[1].must_be_kind_of Aws::InstanceProfileCredentials
+      end
+    end
+  end
+
+  describe '#string2class' do
+    it 'returns a class' do
+      ret = subject.send :string2class, 'Qurd::Action::Dummy'
+      ret.must_equal Qurd::Action::Dummy
+    end
+  end
+end

data/test/dummy_action_test.rb

@@ -0,0 +1,51 @@
+require 'test_helper'
+describe Qurd::Action::Dummy do
+  include WebMockStubs
+
+  describe 'class' do
+    def setup
+      aws_sqs_list_queues
+      aws_sqs_set_queue_attributes
+      Qurd::Configuration.instance.configure('test/inputs/qurd.yml')
+    end
+    let(:subject) { Qurd::Action::Dummy }
+
+    describe '#configure' do
+      %w(launch launch_error terminate terminate_error test).each do |action|
+        it "logs '#{action}'" do
+          aws_sqs_receive_message "test/responses/aws/sqs-receive-message-1-#{action}.xml"
+          mock = Minitest::Mock.new
+          mock.expect :debug, nil, [action]
+          subject.stub :qurd_logger, mock do
+            subject.configure(action)
+          end
+          mock.verify
+        end
+      end
+    end
+  end
+
+  describe 'instance' do
+    def setup
+      aws_sqs_list_queues
+      aws_sqs_set_queue_attributes
+      aws_ec2_describe_instances 'test/responses/aws/ec2-describe-instances-1.xml'
+      Qurd::Configuration.instance.configure('test/inputs/qurd.yml')
+    end
+    let(:sqs_client) { Aws::SQS::Client.new(region: 'us-west-2') }
+    let(:queue_url) { 'https://sqs.us-west-2.amazonaws.com/123456890/test2-ScalingNotificationsQueue-HPPYDAYSAGAI1' }
+    let(:sqs_message) { sqs_client.receive_message(queue_url: queue_url).messages.first }
+    let(:qurd_message) { Qurd::Message.new(message: sqs_message, region: 'us-west-2', aws_credentials: Aws::Credentials.new('a', 'b')) }
+    let(:subject) { Qurd::Action::Dummy.new(qurd_message) }
+
+    %w[launch launch_error terminate terminate_error test].each do |action|
+      describe "##{action}" do
+        it 'sets dummy context' do
+          aws_sqs_receive_message "test/responses/aws/sqs-receive-message-1-#{action}.xml"
+          subject.send action
+          subject.context[:dummy].must_equal true
+        end
+      end
+    end
+  end
+end

data/test/inputs/foo.pem

@@ -0,0 +1,27 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIIEowIBAAKCAQEA2vEv2wptCoW4z0mrsGQ/t9gHGai0u7O2NiNXHURHNDnr722o
+QaFkedwnOfPRAW3YzV0t5mbjudN1XiT7L9EAV1Yek9fxSPTkPttRkTyr9ya4CAer
+W4J6uIpz2VPOo57DX8bo/sv2STSiMlaTYgKizZlPWLpNKYbDt34+lYJerffhC7in
+eHPLJifhH6sYQ0XCG5fEo+QA8VecLS3OdNJ/dGeWWrOgP9MdklaquSsVrGzZoTDR
+OzuX81L4ml2bdJJY55tfpbMyACkNxcmwIAcvXLsHa5+yL/wcVHQVvNTl/mgOmUws
+hYa7lcjsjZUSy2ky4syKvsiDgdJFU5L++n8suQIDAQABAoIBAQDLwHUmxZcgmb2Z
+uxe6ejEKKFLDQEuh+ubxjwX9SJNzOQwmr8hL+6SD+6vNtOenCtEenT6gEtSPFQk+
+6e+RWlGU9aJTOD1mBfF7xaMtJGdtBG1zm5+O8+i9YKvbbrD5eECeG4CtzqaPDhp7
++jICni1gZPKAwjAZGz98vlID1WXDQxlTd5z7S+HcHx4bWDKFVb1b7fjhPMljTkvJ
+AUBGiyhfrPohsieaxvbn/whzQyIdKLnV9ve7IDZgAvPW1BUn9qI+ya/4ZP7XV5/T
+OHIzxJ+Q60CA4RPyy2gkdFKJLzsJXjSEkCmky5iySBuUlvXqpc6VoOyqIKDJHMlp
+Qb6SH+thAoGBAO7hnga6beEcbAC2JsbS2Ill8RM23V/aKB8ZdDpRpdsA39dDc7JU
+nEHS7oEwmjhwEBt/T0enlcYuzU50C9/YVgXCAIibTJ9NCzJAt3Wf1CsDcxTANEwO
+/vMjMF5k9QYX+hSEE6gtwjiPGREtD7hJTMfR5dJ6cxDMdBvofCWMGK4NAoGBAOqh
+xs/osLolQ8mrLguN5DCimQO3GEOfTOhoYd9jsOHSgxAdYpgyBa7QiFZz8QmxisJI
+7iocSDPVwm90O15+qV09aa9mYxWBnmtT8Y7tM9nSkOJxHohSTYMeSk9zeCFpKCqC
+ZqMhilALfml7Sir+xnVgkfTBbY3d5llG6T/ZPTpdAoGAdugbbyHxe/JX2GrTFnjx
+jEMQIw5WV1YSTpivADCQLFldEpvixUvUk/dY/TmiKqGtguJ2JibHKOU3EOw21O+s
+siUFevpCw7Dn99W6/HPYeqi26gdagNmeEozbkMEULjpVI00zM63DVb/1LFWP0524
++GYEziZhjGUZDqNE2LPJvSkCgYBbLdsQ2KA09l5NdJREMTPIIqfL5c8sGC/O8SF3
+qkoGt8Pu8zxpYKcMtEUHM68r0S7XKXhELRFx81PH9CdOJXKzpTX4z5UZunFRzCFH
+Ja7MqyNanBL9bt5oYdEG5xx7/114h4xTQKl7SbwqSZsc8j01R/3wW2qzgtQNRZ1H
+DqjYkQKBgEruMhyLVO0b/V5iVGKIxYnoA3oCcfYEa5oKHGD5nEHIM67J947MnxU/
+CIpVmMGZ6w9ah5kMhIVroZvEjlk6YMuWvaoUjwB/EN505ylq0Wacb8kscbjqdaLu
+dZO8vHOGxI92WtXrINU/97nnRAeMdLm4RqW86yWoKSlm72RWmcol
+-----END RSA PRIVATE KEY-----

data/test/inputs/knife.rb

@@ -0,0 +1,9 @@
+current_dir = File.dirname(__FILE__)
+log_level                :info
+log_location             STDOUT
+node_name                "foo"
+client_name              "foo"
+client_key               "#{current_dir}/foo.pem"
+validation_client_name   "foo-validator"
+validation_key           "#{current_dir}/validator.pem"
+chef_server_url          "https://api.opscode.com/organizations/foo"

data/test/inputs/qurd.yml

@@ -0,0 +1,32 @@
+---
+daemonize: false
+pid_file: tmp/qurd.pid
+wait_time: 20
+visibility_timeout: 600
+log_file: tmp/qurd.log
+log_level: debug
+aws_credentials:
+  - name: test
+    type: credentials
+    options:
+      access_key_id: bar
+      secret_access_key: foo
+auto_scaling_queues:
+  staging:
+    credentials: test
+    region: us-west-2
+    queues:
+      - "/ScalingNotificationsQueue/"
+
+actions:
+  launch:
+    - "Qurd::Action::Dummy"
+  launch_error:
+    - "Qurd::Action::Dummy"
+  terminate:
+    - "Qurd::Action::Dummy"
+  terminate_error:
+    - "Qurd::Action::Dummy"
+  test:
+    - "Qurd::Action::Dummy"
+