# 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