hercules 0.1.1

data/tests/fixtures/Gemfile.with_git_gem.lock

@@ -0,0 +1,10 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    git (1.2.5)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  git

data/tests/fixtures/bogus_config.yml

@@ -0,0 +1,9 @@
+ $%#$% 
+  token-> abc
+  repository: file:///tmp/hercules_test_repository
+  target_directory: /tmp/hercules_test_checkout
+  master:
+    checkouts_to_keep: 1
+  test:
+    checkouts_to_keep: 3
+

data/tests/fixtures/bogus_deployer.rb

@@ -0,0 +1,2 @@
+class Deployer
+end

data/tests/fixtures/config.yml

@@ -0,0 +1,12 @@
+---
+test_project:
+  token: abc
+  repository: file:///tmp/hercules_test_repository
+  target_directory: /tmp/hercules_test_checkout
+  master:
+    checkout_on_startup: false
+    checkouts_to_keep: 1
+  test:
+    checkout_on_startup: false
+    checkouts_to_keep: 3
+

data/tests/fixtures/config_empty.yml

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

data/tests/fixtures/config_empty_branches.yml

@@ -0,0 +1,8 @@
+---
+test_project:
+  token: abc
+  repository: file:///tmp/hercules_test_repository
+  target_directory: /tmp/hercules_test_checkout
+  master:
+  test:
+

data/tests/fixtures/config_empty_projects.yml

@@ -0,0 +1,2 @@
+---
+test_project:

data/tests/fixtures/config_global.yml

@@ -0,0 +1,14 @@
+---
+host: "127.0.0.1"
+port: 8081
+test_project:
+  token: abc
+  repository: file:///tmp/hercules_test_repository
+  target_directory: /tmp/hercules_test_checkout
+  master:
+    checkout_on_startup: false
+    checkouts_to_keep: 1
+  test:
+    checkout_on_startup: false
+    checkouts_to_keep: 3
+

data/tests/fixtures/config_partial_1.yml

@@ -0,0 +1,7 @@
+---
+test_project:
+  repository: file:///tmp/hercules_test_repository
+  target_directory: /tmp/hercules_test_checkout
+  master:
+  test:
+

data/tests/fixtures/config_partial_2.yml

@@ -0,0 +1,7 @@
+---
+test_project:
+  token: abc
+  target_directory: /tmp/hercules_test_checkout
+  master:
+  test:
+

data/tests/fixtures/config_partial_3.yml

@@ -0,0 +1,7 @@
+---
+test_project:
+  token: abc
+  repository: file:///tmp/hercules_test_repository
+  master:
+  test:
+

data/tests/fixtures/deployer_branch.rb

@@ -0,0 +1,11 @@
+module Hercules
+  class Triggers
+    def self.before_deploy(options)
+      true
+    end
+
+    def self.after_deploy(options)
+      FileUtils.touch "./branch_name_#{options[:branch]}"
+    end
+  end
+end

data/tests/fixtures/deployer_exception.rb

@@ -0,0 +1,11 @@
+module Hercules
+  class Triggers
+    def self.before_deploy(options)
+      raise "test exception"
+    end
+
+    def self.after_deploy(options)
+      FileUtils.touch "./after_deploy"
+    end
+  end
+end

data/tests/fixtures/deployer_false.rb

@@ -0,0 +1,11 @@
+module Hercules
+  class Triggers
+    def self.before_deploy(options)
+      false
+    end
+
+    def self.after_deploy(options)
+      FileUtils.touch "#{options[:path]}/after_deploy"
+    end
+  end
+end

data/tests/fixtures/deployer_path.rb

@@ -0,0 +1,11 @@
+module Hercules
+  class Triggers
+    def self.before_deploy(options)
+      true
+    end
+
+    def self.after_deploy(options)
+      FileUtils.touch "./after_deploy"
+    end
+  end
+end

data/tests/fixtures/deployer_true.rb

@@ -0,0 +1,11 @@
+module Hercules
+  class Triggers
+    def self.before_deploy(options)
+      true
+    end
+
+    def self.after_deploy(options)
+      FileUtils.touch "#{options[:path]}/after_deploy"
+    end
+  end
+end

data/tests/fixtures/deployer_undefined_variable.rb

@@ -0,0 +1,11 @@
+module Hercules
+  class Triggers
+    def self.before_deploy(options)
+      puts "test with #{undefined_variable}"
+    end
+
+    def self.after_deploy(options)
+      FileUtils.touch "./after_deploy"
+    end
+  end
+end

data/tests/fixtures/startup_checkout_config.yml

@@ -0,0 +1,12 @@
+---
+test_project:
+  token: abc
+  repository: file:///tmp/hercules_test_repository
+  target_directory: /tmp/hercules_test_checkout
+  master:
+    checkout_on_startup: true
+    checkouts_to_keep: 1
+  test:
+    checkout_on_startup: false
+    checkouts_to_keep: 3
+

data/tests/fixtures/startup_checkout_error_config.yml

@@ -0,0 +1,12 @@
+---
+test_project:
+  token: abc
+  repository: file:///tmp/hercules_test_repository
+  target_directory: /tmp/hercules_test_checkout
+  master:
+    checkout_on_startup: true
+    checkouts_to_keep: 1
+  test:
+    checkout_on_startup: true
+    checkouts_to_keep: 3
+

data/tests/git_handler_test.rb

@@ -0,0 +1,95 @@
+# coding: utf-8
+require 'tests/git_setup'
+require 'lib/git_handler'
+require 'test/unit'
+
+class GitHandlerTest < Test::Unit::TestCase
+  include GitSetup
+  def setup
+    git_setup
+  end
+
+  def test_get_last_commit
+    g = Hercules::GitHandler.new(@config['test_project'])
+    g.export_branch('master')
+    assert_equal @head_sha, g.last_commit
+  end
+
+  def test_export_branch
+    g = Hercules::GitHandler.new(@config['test_project'])
+    g.export_branch('master')
+    assert_is_checkout @config['test_project']['target_directory'] + '/checkouts/master/' + @head_sha
+  end
+
+  def test_export_non_existent_branch
+    g = Hercules::GitHandler.new(@config['test_project'])
+    g.export_branch('branch_that_will_not_exist') rescue nil
+    assert !File.exists?(@config['test_project']['target_directory'] + '/checkouts/branch_that_will_not_exist/' + @head_sha)
+    assert_equal 2, Dir.glob(@config['test_project']['target_directory'] + '/checkouts/branch_that_will_not_exist/.*').size
+    assert_equal 0, Dir.glob(@config['test_project']['target_directory'] + '/checkouts/branch_that_will_not_exist/*').size
+  end
+
+  def test_export_non_existent_or_invalid_repository
+    @config['test_project']['repository'] = 'repository_that_does_not_exist_or_is_invalid'
+    g = Hercules::GitHandler.new(@config['test_project'])
+    begin
+      g.export_branch('branch_that_will_not_exist')
+      assert false
+    rescue Exception => e
+      assert_match /Error while cloning/, e.message
+    end
+
+  end
+
+  def test_create_branches_dir
+    g = Hercules::GitHandler.new(@config['test_project'])
+    g.create_branches_dir
+    assert (File.exists?(@config['test_project']['target_directory'] + '/branches'))
+  end
+
+  def test_branches_dir
+    g = Hercules::GitHandler.new(@config['test_project'])
+    assert_equal @config['test_project']['target_directory'] + '/branches', g.branches_path
+  end
+
+  def test_deploy_branch
+    g = Hercules::GitHandler.new(@config['test_project'])
+    g.deploy_branch('master')
+    assert_is_checkout @config['test_project']['target_directory'] + '/branches/master'
+  end
+
+  def test_should_deploy_test_branch
+    @g.branch('test').checkout
+    g = Hercules::GitHandler.new(@config['test_project'])
+    g.deploy_branch('test')
+    assert_is_checkout @config['test_project']['target_directory'] + '/branches/test'
+  end
+
+  def test_should_maintain_at_most_three_test_checkouts
+    @g.branch('test').checkout
+    g = Hercules::GitHandler.new(@config['test_project'])
+    FileUtils.mkdir_p(@config['test_project']['target_directory'] + '/logs')
+    4.times do |t|
+      sha1 = generate_commit "new_commit#{t}"
+      g.deploy_branch('test')
+      # Here we must simulate the log creating which will be done by the deployer
+      FileUtils.touch("#{@config['test_project']['target_directory']}/logs/#{sha1}.log")
+      assert File.exists?(@config['test_project']['target_directory'] + "/logs")
+      assert File.exists?(@config['test_project']['target_directory'] + "/branches/test/new_commit#{t}")
+      # The checkouts_to_keep in config.yml is set to 3 (so 5 is the maximum: 3 + '.' + '..')
+      assert_equal (3+t > 5 ? 5 : 3+t), Dir.entries(@config['test_project']['target_directory'] + '/checkouts/test').size
+      assert_equal (3+t > 5 ? 5 : 3+t), Dir.entries(@config['test_project']['target_directory'] + '/logs').size
+    end
+  end
+
+  def test_should_maintain_only_one_master_checkout
+    g = Hercules::GitHandler.new(@config['test_project'])
+    g.deploy_branch('master')
+    generate_commit 'new_commit'
+    g.deploy_branch('master')
+    assert File.exists?(@config['test_project']['target_directory'] + '/branches/master/new_commit')
+    assert_equal 3, Dir.entries(@config['test_project']['target_directory'] + '/checkouts/master').size
+  end
+
+end
+

data/tests/git_setup.rb

@@ -0,0 +1,70 @@
+# coding: utf-8
+require 'rubygems'
+require 'git'
+require 'lib/git_handler'
+require 'yaml'
+
+module GitSetup
+  def git_setup
+    @config = YAML.load_file(File.dirname(__FILE__) + '/fixtures/config.yml')
+    dir = @config['test_project']['repository'].gsub(/file:\/\//, '')
+    FileUtils.rm_rf(dir)
+    FileUtils.rm_rf(@config['test_project']['target_directory'])
+    FileUtils.mkdir_p(dir)
+    FileUtils.cp(File.dirname(__FILE__) + '/fixtures/config.yml', "#{dir}/config.yml")
+    FileUtils.cp(File.dirname(__FILE__) + '/fixtures/Gemfile', "#{dir}/Gemfile")
+    FileUtils.cp(File.dirname(__FILE__) + '/fixtures/Gemfile.lock', "#{dir}/Gemfile.lock")
+    @g = Git.init(dir)
+    @g.chdir do
+      @g.config('user.name', 'Test User')
+      @g.config('user.email', 'email@email.com')
+      @g.add("./config.yml")
+      @g.add("./Gemfile")
+      @g.add("./Gemfile.lock")
+      @g.commit_all('message')
+      @head_sha = @g.gcommit('HEAD').sha
+    end
+  end
+
+  def generate_bogus_gemfile
+    change_repository do
+      File.open('./Gemfile', 'a'){|f| f.write('gem syntax error "Gem_That_Does_Not_Exist", "0.0.0", - % $ ') }
+    end
+  end
+
+  def generate_gemfile_with_gem
+    path = File.expand_path( File.dirname(__FILE__) )
+    change_repository do
+      FileUtils.cp("#{path}/fixtures/Gemfile.with_git_gem", "./Gemfile")
+      FileUtils.cp("#{path}/fixtures/Gemfile.with_git_gem.lock", "./Gemfile.lock")
+    end
+  end
+
+  def generate_deployer(deployer_name)
+    path = File.expand_path( File.dirname(__FILE__) )
+    change_repository do
+      FileUtils.mkdir_p("./lib")
+      FileUtils.cp("#{path}/fixtures/#{deployer_name}.rb", "./lib/hercules_triggers.rb")
+    end
+  end
+
+  def generate_commit file_name
+    change_repository{ FileUtils.touch file_name }
+  end
+
+  def change_repository
+    head_sha = @head_sha
+    @g.chdir do
+      yield
+      @g.add("./*")
+      @g.commit_all("added files")
+      head_sha = @g.gcommit('HEAD').sha
+    end
+    return head_sha
+  end
+
+  def assert_is_checkout(dir)
+    assert_equal File.open(File.dirname(__FILE__) + '/fixtures/config.yml').read, File.open(dir + '/config.yml').read
+    assert !File.exists?(dir + '/.git')
+  end
+end

data/tests/hercules_test.rb

@@ -0,0 +1,128 @@
+# coding: utf-8
+require 'tests/startup'
+require 'tests/git_setup'
+require 'test/unit'
+
+class HerculesTest < Test::Unit::TestCase
+  include Startup
+  include GitSetup
+
+  def setup
+    prepare_startup
+  end
+
+  def test_startup_validations
+    verbose(false) do
+      # Here we test for config file require
+      sh "src/hercules.rb -l /dev/null > /dev/null 2>&1" rescue nil
+      sleep 1
+      assert !File.exist?(@pidfile)
+      
+      # Test with an invalid  yaml
+      sh "src/hercules.rb -c src/hercules.rb -l /dev/null > /dev/null 2>&1" rescue nil
+      sleep 1
+      assert !File.exist?(@pidfile)
+    end
+  end
+
+  def test_logfile_and_piddir
+    start_hercules do |pid,log|
+      assert File.exist?(@logfile)
+      assert_match /Start/, log.read()
+      assert File.exist?(@pidfile)
+      assert_match /\d+/, pid
+    end
+  end
+
+  def test_send_term
+    start_hercules do |pid,log|
+      Process.kill("TERM", pid.to_i)
+      sleep 1
+      assert !File.exist?(@pidfile)
+      assert_match /Terminating hercules/, log.read
+    end
+  end
+
+  def test_send_hup
+    start_hercules do |pid,log|
+      Process.kill("HUP", pid.to_i)
+      sleep 1
+      assert File.exist?(@pidfile)
+      log_content = log.read
+      assert_match /Reloading config/, log_content
+      assert_no_match /Error reading/, log_content
+    end
+  end
+
+  def test_send_hup_with_bad_config_file
+    start_hercules do |pid,log|
+      FileUtils.mv "tests/fixtures/config.yml", "tests/fixtures/config.old.yml"
+      FileUtils.mv "tests/fixtures/bogus_config.yml", "tests/fixtures/config.yml"
+      Process.kill("HUP", pid.to_i)
+      sleep 1
+      assert File.exist?(@pidfile)
+      log_content = log.read
+      FileUtils.mv "tests/fixtures/config.yml", "tests/fixtures/bogus_config.yml"
+      FileUtils.mv "tests/fixtures/config.old.yml", "tests/fixtures/config.yml"
+      assert_match /Reloading config/, log_content
+      assert_match /Error reading/, log_content
+    end
+  end
+
+  def test_checkouts_on_startup_with_errors
+    git_setup
+    FileUtils.mv "tests/fixtures/config.yml", "tests/fixtures/config.old.yml"
+    FileUtils.mv "tests/fixtures/startup_checkout_error_config.yml", "tests/fixtures/config.yml"
+    begin
+      log_content = ""
+      start_hercules do |pid,log|
+        sleep 5
+        log_content = log.read
+        assert_match /Branch master deployed/, log_content
+        assert File.exist?(@pidfile)
+      end
+    ensure
+      FileUtils.mv "tests/fixtures/config.yml", "tests/fixtures/startup_checkout_error_config.yml"
+      FileUtils.mv "tests/fixtures/config.old.yml", "tests/fixtures/config.yml"
+    end
+  end
+
+  def test_checkouts_on_startup
+    git_setup
+    FileUtils.mv "tests/fixtures/config.yml", "tests/fixtures/config.old.yml"
+    FileUtils.mv "tests/fixtures/startup_checkout_config.yml", "tests/fixtures/config.yml"
+    start_hercules do |pid,log|
+      FileUtils.mv "tests/fixtures/config.yml", "tests/fixtures/startup_checkout_config.yml"
+      FileUtils.mv "tests/fixtures/config.old.yml", "tests/fixtures/config.yml"
+      sleep 5
+      log_content = log.read
+      assert_match /Branch master deployed/, log_content
+      assert_is_checkout @config['test_project']['target_directory'] + '/branches/master'
+    end
+  end
+
+  def test_no_checkouts_on_startups_with_default_config
+    git_setup
+    start_hercules do |pid,log|
+      sleep 5
+      log_content = log.read
+      assert_no_match /Branch master deployed/, log_content
+      assert !File.exist?(@config['test_project']['target_directory'] + '/branches/master')
+    end
+  end
+
+  def test_checkouts_on_startup_with_hidden_tmp
+    git_setup
+    FileUtils.mv "tests/fixtures/config.yml", "tests/fixtures/config.old.yml"
+    FileUtils.mv "tests/fixtures/startup_checkout_config.yml", "tests/fixtures/config.yml"
+    FileUtils.mkdir_p @config['test_project']['target_directory'] + '/checkouts/master/.tmp'
+    start_hercules do |pid,log|
+      FileUtils.mv "tests/fixtures/config.yml", "tests/fixtures/startup_checkout_config.yml"
+      FileUtils.mv "tests/fixtures/config.old.yml", "tests/fixtures/config.yml"
+      sleep 20
+      log_content = log.read
+      assert_match /Branch master deployed/, log_content
+      assert_is_checkout @config['test_project']['target_directory'] + '/branches/master'
+    end
+  end
+end