skaar-daemon-kit 0.1.7.10

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/rspec.rake

@@ -0,0 +1,21 @@
+begin
+  require 'spec'
+rescue LoadError
+  require 'rubygems'
+  require 'spec'
+end
+begin
+  require 'spec/rake/spectask'
+rescue LoadError
+  puts <<-EOS
+To use rspec for testing you must install rspec gem:
+    gem install rspec
+EOS
+  exit(0)
+end
+
+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

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 %> <%= DaemonKit.root %>/bin/<%= DaemonKit.configuration.daemon_name %> start"
+  w.stop = "/usr/bin/env DAEMON_ENV=<%= DaemonKit.env %> <%= DaemonKit.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

data/test/test_amqp_generator.rb

@@ -0,0 +1,48 @@
+require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
+
+
+class TestAmqpGenerator < Test::Unit::TestCase
+  include RubiGen::GeneratorTestHelper
+
+  def setup
+    bare_setup
+  end
+
+  def teardown
+    bare_teardown
+  end
+
+  # Some generator-related assertions:
+  #   assert_generated_file(name, &block) # block passed the file contents
+  #   assert_directory_exists(name)
+  #   assert_generated_class(name, &block)
+  #   assert_generated_module(name, &block)
+  #   assert_generated_test_for(name, &block)
+  # The assert_generated_(class|module|test_for) &block is passed the body of the class/module within the file
+  #   assert_has_method(body, *methods) # check that the body has a list of methods (methods with parentheses not supported yet)
+  #
+  # Other helper methods are:
+  #   app_root_files - put this in teardown to show files generated by the test method (e.g. p app_root_files)
+  #   bare_setup - place this in setup method to create the APP_ROOT folder for each test
+  #   bare_teardown - place this in teardown method to destroy the TMP_ROOT or APP_ROOT folder after each test
+
+  def test_generator_without_options
+    name = "myapp"
+    run_generator('amqp', [name], sources)
+    assert_directory_exists "config"
+    assert_directory_exists "config/initializers"
+    assert_generated_file   "config/amqp.yml"
+    assert_generated_file   "config/initializers/myapp.rb"
+    assert_generated_file   "libexec/myapp-daemon.rb"
+  end
+
+  private
+  def sources
+    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path))
+    ]
+  end
+
+  def generator_path
+    "daemon_generators"
+  end
+end

data/test/test_cron_generator.rb

@@ -0,0 +1,45 @@
+require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
+
+
+class TestCronGenerator < Test::Unit::TestCase
+  include RubiGen::GeneratorTestHelper
+
+  def setup
+    bare_setup
+  end
+
+  def teardown
+    bare_teardown
+  end
+
+  # Some generator-related assertions:
+  #   assert_generated_file(name, &block) # block passed the file contents
+  #   assert_directory_exists(name)
+  #   assert_generated_class(name, &block)
+  #   assert_generated_module(name, &block)
+  #   assert_generated_test_for(name, &block)
+  # The assert_generated_(class|module|test_for) &block is passed the body of the class/module within the file
+  #   assert_has_method(body, *methods) # check that the body has a list of methods (methods with parentheses not supported yet)
+  #
+  # Other helper methods are:
+  #   app_root_files - put this in teardown to show files generated by the test method (e.g. p app_root_files)
+  #   bare_setup - place this in setup method to create the APP_ROOT folder for each test
+  #   bare_teardown - place this in teardown method to destroy the TMP_ROOT or APP_ROOT folder after each test
+
+  def test_generator_without_options
+    name = "myapp"
+    run_generator('cron', [name], sources)
+    assert_directory_exists "config/initializers"
+    assert_generated_file   "config/initializers/myapp.rb"
+  end
+
+  private
+  def sources
+    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path))
+    ]
+  end
+
+  def generator_path
+    "daemon_generators"
+  end
+end

data/test/test_daemon-kit_generator.rb

@@ -0,0 +1,84 @@
+require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
+
+class TestDaemonKitGenerator < Test::Unit::TestCase
+  include RubiGen::GeneratorTestHelper
+
+  attr_reader :daemon_name
+  def setup
+    bare_setup
+    @daemon_name = File.basename(File.expand_path(APP_ROOT))
+  end
+
+  def teardown
+    bare_teardown
+  end
+
+  # Some generator-related assertions:
+  #   assert_generated_file(name, &block) # block passed the file contents
+  #   assert_directory_exists(name)
+  #   assert_generated_class(name, &block)
+  #   assert_generated_module(name, &block)
+  #   assert_generated_test_for(name, &block)
+  # The assert_generated_(class|module|test_for) &block is passed the body of the class/module within the file
+  #   assert_has_method(body, *methods) # check that the body has a list of methods (methods with parentheses not supported yet)
+  #
+  # Other helper methods are:
+  #   app_root_files - put this in teardown to show files generated by the test method (e.g. p app_root_files)
+  #   bare_setup - place this in setup method to create the APP_ROOT folder for each test
+  #   bare_teardown - place this in teardown method to destroy the TMP_ROOT or APP_ROOT folder after each test
+
+  def test_generator_without_options
+    run_generator('daemon_kit', [APP_ROOT], sources)
+
+    assert_generated_file   "README"
+    assert_generated_file   "Rakefile"
+    assert_directory_exists "bin"
+    assert_generated_file   "bin/#{daemon_name}"
+    assert_directory_exists "config"
+    assert_generated_file   "config/arguments.rb"
+    assert_generated_file   "config/boot.rb"
+    assert_generated_file   "config/environment.rb"
+    assert_generated_file   "config/environments/development.rb"
+    assert_generated_file   "config/environments/test.rb"
+    assert_generated_file   "config/environments/production.rb"
+    assert_directory_exists "config/pre-daemonize"
+    assert_generated_file   "config/pre-daemonize/readme"
+    assert_directory_exists "config/post-daemonize"
+    assert_generated_file   "config/post-daemonize/readme"
+    assert_directory_exists "lib"
+    assert_generated_file   "lib/#{daemon_name}.rb"
+    assert_directory_exists "libexec"
+    assert_generated_file   "libexec/#{daemon_name}-daemon.rb"
+    assert_directory_exists "log"
+    assert_directory_exists "script"
+    assert_generated_file   "script/console"
+    assert_generated_file   "script/destroy"
+    assert_generated_file   "script/generate"
+    assert_directory_exists "spec"
+    assert_directory_exists "tasks"
+    assert_directory_exists "vendor"
+    assert_directory_exists "tmp"
+
+    assert !File.exists?("#{APP_ROOT}/Capfile"), 'Capfile generated, but should not be'
+  end
+
+  def test_generator_with_capistrano_deployer
+    run_generator('daemon_kit', [APP_ROOT, '-d capistrano'], sources)
+
+    assert_generated_file "Capfile"
+    assert_generated_file "config/deploy.rb"
+  end
+
+  private
+  def sources
+    [
+     RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path)),
+     RubiGen::PathSource.new(:app, File.join(File.dirname(__FILE__), "..", "daemon_generators")),
+     RubiGen::PathSource.new(:app, File.join(File.dirname(__FILE__), "..", "rubygems_generators"))
+    ]
+  end
+
+  def generator_path
+    "app_generators"
+  end
+end

data/test/test_daemon_kit_config.rb

@@ -0,0 +1,28 @@
+require File.join(File.dirname(__FILE__), "test_helper")
+
+class TestDaemonKitConfig < Test::Unit::TestCase
+
+  def test_initialize_with_env_hash
+    data = {
+      'test' => {
+        'foo' => 'bar'
+      }
+    }
+
+    config = DaemonKit::Config.new( data )
+
+    assert_equal config['foo'], 'bar'
+    assert_equal config.foo, 'bar'
+  end
+
+  def test_initialize_without_env_hash
+    data = {
+      'foo' => 'bar'
+    }
+
+    config = DaemonKit::Config.new( data )
+
+    assert_equal config['foo'], 'bar'
+    assert_equal config.foo, 'bar'
+  end
+end

data/test/test_deploy_capistrano_generator.rb

@@ -0,0 +1,48 @@
+require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
+
+
+class TestDeployCapistranoGenerator < Test::Unit::TestCase
+  include RubiGen::GeneratorTestHelper
+
+  def setup
+    bare_setup
+  end
+
+  def teardown
+    bare_teardown
+  end
+
+  # Some generator-related assertions:
+  #   assert_generated_file(name, &block) # block passed the file contents
+  #   assert_directory_exists(name)
+  #   assert_generated_class(name, &block)
+  #   assert_generated_module(name, &block)
+  #   assert_generated_test_for(name, &block)
+  # The assert_generated_(class|module|test_for) &block is passed the body of the class/module within the file
+  #   assert_has_method(body, *methods) # check that the body has a list of methods (methods with parentheses not supported yet)
+  #
+  # Other helper methods are:
+  #   app_root_files - put this in teardown to show files generated by the test method (e.g. p app_root_files)
+  #   bare_setup - place this in setup method to create the APP_ROOT folder for each test
+  #   bare_teardown - place this in teardown method to destroy the TMP_ROOT or APP_ROOT folder after each test
+
+  def test_generator_without_options
+    run_generator('deploy_capistrano', ['myapp'], sources)
+    assert_generated_file   "Capfile"
+    assert_directory_exists "config"
+    assert_generated_file   "config/deploy.rb"
+    assert_directory_exists "config/deploy"
+    assert_generated_file   "config/deploy/staging.rb"
+    assert_generated_file   "config/deploy/production.rb"
+  end
+
+  private
+  def sources
+    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path))
+    ]
+  end
+
+  def generator_path
+    "daemon_generators"
+  end
+end