amqp-daemon-kit 0.1.8.1

data/spec/argument_spec.rb

@@ -0,0 +1,70 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe DaemonKit::Arguments do
+
+  describe "parsing ARGV" do
+
+    it "should extract the given command" do
+      argv = [ 'start', '-f', 'foo' ]
+      res = DaemonKit::Arguments.parse( argv )
+
+      res.first.should == :start
+    end
+
+    it "should have a default command if missing" do
+      argv = [ '-h' ]
+      res = DaemonKit::Arguments.parse( argv )
+
+      res.first.should == :run
+    end
+
+    it "should extract explicit configuration options" do
+      argv = [ 'start', '--config', 'environment=development' ]
+      res = DaemonKit::Arguments.parse( argv )
+
+      res.shift
+      res.first.should == [ 'environment=development' ]
+
+      res.last.should == []
+    end
+
+    it "should extract implicit configuration options" do
+      argv = [ '-e', 'production' ]
+      res = DaemonKit::Arguments.parse( argv )
+
+      res.shift
+      res.first.should == ['environment=production']
+
+      res.last.should == []
+    end
+
+    it "should extract daemon options" do
+      argv = [ 'start', '-h' ]
+      res = DaemonKit::Arguments.parse( argv )
+
+      res.shift
+      res.first.should == []
+
+      res.last.should == [ '-h' ]
+    end
+
+    it "should handle different ordered configurations easily" do
+      argv = [ '--pidfile', '/tmp/piddy', '--log', '/tmp/loggy' ]
+      res = DaemonKit::Arguments.configuration( argv )
+
+      # No additional args
+      res.last.should be_empty
+
+      res.first[0].should == "pid_file=/tmp/piddy"
+      res.first[1].should == "log_path=/tmp/loggy"
+    end
+
+    it "should handle mixed configurations easily" do
+      argv = [ '--rest', 'yes', '-l', '/tmp/loggy', '-f', 'bar' ]
+      res = DaemonKit::Arguments.configuration( argv )
+
+      res.first.should == [ 'log_path=/tmp/loggy' ]
+      res.last.should  == [ '--rest', 'yes', '-f', 'bar' ]
+    end
+  end
+end

data/spec/config_spec.rb

@@ -0,0 +1,83 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe DaemonKit::Config do
+
+  describe "working with config data" do
+    before(:each) do
+      @config = DaemonKit::Config.new('foo' => 'bar', 'nes' => { 'ted' => 'value' }, 'dash-ed' => 'keys')
+    end
+
+    it "should have string key access to values" do
+      @config['foo'].should == 'bar'
+    end
+
+    it "should have symbol key access to values" do
+      @config[:foo].should == 'bar'
+    end
+
+    it "should have instance accessors to values" do
+      @config.foo.should == 'bar'
+    end
+
+    it "should have nested instance accessors to values" do
+      @config.nes.ted.should == 'value'
+    end
+
+    it "should return the config as a hash" do
+      @config.to_h.should == { 'foo' => 'bar', 'nes' => { 'ted' => 'value' }, 'dash-ed' => 'keys' }
+    end
+
+    it "should be able to symbolize keys in returned hash" do
+      @config.to_h(true).should == { :foo => 'bar', :nes => { :ted => 'value' }, :'dash-ed' => 'keys' }
+    end
+
+    it "should be able to handle dashed keys via accessors" do
+      @config.dash_ed.should == 'keys'
+    end
+  end
+
+  describe "parsing files" do
+    before(:all) do
+      FileUtils.mkdir_p( DAEMON_ROOT + "/config" )
+      FileUtils.cp( File.dirname(__FILE__) + '/fixtures/env.yml', DAEMON_ROOT + '/config/' )
+      FileUtils.cp( File.dirname(__FILE__) + '/fixtures/noenv.yml', DAEMON_ROOT + '/config/' )
+    end
+
+    it "should parse env keys correctly" do
+      config = DaemonKit::Config.load('env')
+
+      config.test.should == 'yes!'
+      config.array.should_not be_empty
+    end
+
+    it "should not be worried about missing env keys" do
+      config = DaemonKit::Config.load('noenv')
+
+      config.string.should == 'value'
+    end
+
+    it "should accept symbol file names" do
+      config = DaemonKit::Config.load(:env)
+      config.test.should == 'yes!'
+    end
+
+    it "should bail on missing files" do
+      lambda {
+        DaemonKit::Config.load('missing')
+      }.should raise_error(ArgumentError)
+    end
+
+    it "should give direct hash access to a config" do
+      config = DaemonKit::Config.hash(:env)
+
+      config.should be_a_kind_of(Hash)
+      config.keys.should include('test')
+    end
+
+    it "should give direct symbolized hash access to a config" do
+      config = DaemonKit::Config.hash(:env, true)
+
+      config.keys.should include(:test)
+    end
+  end
+end

data/spec/configurable_spec.rb

@@ -0,0 +1,56 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+class FooConfig
+  include DaemonKit::Configurable
+
+  configurable :has_default, true
+  configurable :no_default
+  configurable :has_lock, :locked => true
+end
+
+describe DaemonKit::Configurable do
+
+  before(:each) do
+    @foo = FooConfig.new
+  end
+
+  it "should support default values" do
+    lambda {
+      @foo.has_default.should be_true
+    }.should_not raise_error( NoMethodError )
+  end
+
+  it "should support overwriting unlocked defaults" do
+    lambda {
+      @foo.has_default = false
+      @foo.has_default.should be_false
+    }.should_not raise_error
+  end
+
+  it "should support no default values" do
+    lambda {
+      @foo.no_default.should be_nil
+    }.should_not raise_error( NoMethodError )
+  end
+
+  it "should allow setting locked values once" do
+    lambda {
+      @foo.has_lock = 1
+      @foo.has_lock.should == 1
+
+      @foo.has_lock = 2
+      @foo.has_lock.should == 1
+    }.should_not raise_error
+  end
+
+  it "should allow bypassing the lock explicitly" do
+    lambda {
+      @foo.has_lock = 1
+      @foo.has_lock.should == 1
+
+      @foo.set(:has_lock, 2)
+      @foo.has_lock.should == 2
+    }.should_not raise_error
+  end
+
+end

data/spec/daemon_kit_spec.rb

@@ -0,0 +1,7 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+describe DaemonKit do
+
+  it "should work"
+
+end

data/spec/error_handlers_spec.rb

@@ -0,0 +1,23 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe DaemonKit::Safety do
+end
+
+describe DaemonKit::ErrorHandlers::Mail do
+  it "should send an email report" do
+    conf = Object.new
+    conf.stubs(:daemon_name).returns('test')
+    DaemonKit.stubs(:configuration).returns(conf)
+
+    fake_smtp = Object.new
+    fake_smtp.expects(:start).with('localhost.localdomain', nil, nil, nil)
+    Net::SMTP.expects(:new).with('localhost', 25).returns(fake_smtp)
+
+    begin
+      raise RuntimeError, "specs don't fail :)"
+    rescue => e
+      handler = DaemonKit::ErrorHandlers::Mail.instance
+      handler.handle_exception( e )
+    end
+  end
+end

data/spec/fixtures/env.yml

@@ -0,0 +1,15 @@
+# Test ENV keys
+
+defaults: &defaults
+  string: value
+  array:
+    - entry
+    - entry
+
+development:
+  test: no!
+  <<: *defaults
+
+test:
+  test: yes!
+  <<: *defaults

data/spec/fixtures/noenv.yml

@@ -0,0 +1,4 @@
+string: value
+array:
+  - entry
+  - entry

data/spec/initializer_spec.rb

@@ -0,0 +1,26 @@
+require File.dirname(__FILE__) + '/spec_helper'
+require 'daemon_kit/initializer'
+
+describe DaemonKit::Configuration do
+  before(:each) do
+    @configuration = DaemonKit::Configuration.new
+  end
+
+  it "should know our environment" do
+    @configuration.environment.should_not be_nil
+  end
+
+  it "should have a default log path" do
+    @configuration.log_path.should_not be_nil
+  end
+
+  it "should have a default log level" do
+    @configuration.log_level.should_not be_nil
+  end
+
+  it "should set a default umask" do
+    File.umask.should_not be(0)
+    File.umask.should be(18)
+  end
+
+end

data/spec/spec.opts

@@ -0,0 +1 @@
+--colour

data/spec/spec_helper.rb

@@ -0,0 +1,27 @@
+require 'rubygems'
+gem 'rspec'
+require 'spec'
+
+require 'mocha'
+require 'fileutils'
+
+DAEMON_ENV  = "test"
+DAEMON_ROOT = "#{File.dirname(__FILE__)}/../tmp"
+
+$:.unshift( File.dirname(__FILE__) + '/../lib' )
+require 'daemon_kit'
+
+Spec::Runner.configure do |config|
+  # == Mock Framework
+  #
+  # RSpec uses it's own mocking framework by default. If you prefer to
+  # use mocha, flexmock or RR, uncomment the appropriate line:
+  #
+  config.mock_with :mocha
+  # config.mock_with :flexmock
+  # config.mock_with :rr
+
+  # setup a fake daemon_root
+  config.before(:all) { File.directory?( DAEMON_ROOT ) ? FileUtils.rm_rf("#{DAEMON_ROOT}/*") : FileUtils.mkdir_p( DAEMON_ROOT ) }
+  config.after(:all) { FileUtils.rm_rf("#{DAEMON_ROOT}/*") }
+end

data/tasks/cucumber.rake

@@ -0,0 +1,13 @@
+begin
+  require 'cucumber/rake/task'
+
+  Cucumber::Rake::Task.new(:features) do |t|
+    t.fork = true
+    t.cucumber_opts = ['--format', (ENV['CUCUMBER_FORMAT'] || 'pretty')]
+  end
+rescue LoadError
+  desc 'Cucumber rake task not available'
+  task :features do
+    abort 'Cucumber rake task is not available. Be sure to install cucumber as a gem or plugin'
+  end
+end

data/tasks/rspec.rake

@@ -0,0 +1,20 @@
+begin
+  require 'spec'
+rescue LoadError
+  require 'rubygems'
+  require 'spec'
+end
+begin
+  require 'spec/rake/spectask'
+
+  desc "Run the specs under spec/models"
+  Spec::Rake::SpecTask.new do |t|
+    t.spec_opts = ['--options', "spec/spec.opts"]
+    t.spec_files = FileList['spec/**/*_spec.rb']
+  end
+rescue LoadError
+  puts <<-EOS
+To use rspec for testing you must install rspec gem:
+    gem install rspec
+EOS
+end

data/tasks/tests.rake

@@ -0,0 +1,6 @@
+require 'rake/testtask'
+Rake::TestTask.new do |t|
+  t.libs << "test"
+  t.test_files = FileList['test/test_*.rb']
+  t.verbose = true
+end

data/templates/god/god.erb

@@ -0,0 +1,69 @@
+#
+# This is a configuration template for 'god' process monitoring.
+#
+# More information can be found at http://god.rubyforge.org/
+#
+
+DAEMON_ROOT = "<%= DaemonKit.root %>"
+
+God.watch do |w|
+  w.name = '<%= DaemonKit.configuration.daemon_name %>'
+  w.interval = 30.seconds
+  w.start = "/usr/bin/env DAEMON_ENV=<%= DaemonKit.env %> #{DAEMON_ROOT}/bin/<%= DaemonKit.configuration.daemon_name %> start"
+  w.stop = "/usr/bin/env DAEMON_ENV=<%= DaemonKit.env %> #{DAEMON_ROOT}/bin/<%= DaemonKit.configuration.daemon_name %> stop"
+  w.start_grace = 10.seconds
+  w.stop_grace = 10.seconds
+  w.pid_file = "<%= DaemonKit.configuration.pid_file %>"
+  w.behavior(:clean_pid_file)
+  #w.uid = 'nobody'
+  #w.gid = 'nobody'
+
+  w.start_if do |start|
+    start.condition(:process_running) do |c|
+      c.interval = 30.seconds
+      c.running = false
+      c.notify = 'sysadmin'
+    end
+  end
+
+  w.restart_if do |restart|
+    restart.condition(:memory_usage) do |c|
+      c.above = 250.megabytes
+      c.times = [3, 5] # 3 out of 5 intervals
+      c.notify = 'sysadmin'
+    end
+
+    restart.condition(:cpu_usage) do |c|
+      c.above = 50.percent
+      c.times = 5
+      c.notify = 'sysadmin'
+    end
+  end
+
+  w.lifecycle do |on|
+    on.condition(:flapping) do |c|
+      c.to_state = [:start, :restart]
+      c.times = 5
+      c.within = 5.minute
+      c.transition = :unmonitored
+      c.retry_in = 10.minutes
+      c.retry_times = 5
+      c.retry_within = 2.hours
+      c.notify = 'sysadmin'
+    end
+  end
+end
+
+God::Contacts::Email.server_settings = {
+  :address => "smtp.gmail.com",
+  :port => 587,
+  :domain => "localhost",
+  :authentication => :plain,
+  :user_name => "you@gmail.com",
+  :password => "secret"
+}
+
+God.contact(:email) do |c|
+  c.name = 'sysadmin'
+  c.email = 'you@gmail.com'
+end

data/templates/monit/monit.erb

@@ -0,0 +1,14 @@
+#
+# This monit configuration file is built from the monit wiki with some
+# very basic checks on memory and CPU usage.
+#
+# See the man page at http://mmonit.com/monit/documentation/monit.html
+# for more information.
+#
+check process <%= DaemonKit.configuration.daemon_name %> with pidfile <%= DaemonKit.configuration.pid_file %>
+  group daemons
+  start program = "/usr/bin/env DAEMON_ENV=<%= DaemonKit.env %> <%= DaemonKit.root %>/bin/<%= DaemonKit.configuration.daemon_name %> start"
+  stop program  = "/usr/bin/env DAEMON_ENV=<%= DaemonKit.env %> <%= DaemonKit.root %>/bin/<%= DaemonKit.configuration.daemon_name %> stop"
+  if 5 restarts within 5 cycles then timeout
+  if cpu usage > 99% for 5 cycles then alert
+  if mem usage > 99% for 5 cycles then alert