require "#{File.dirname(__FILE__)}/utils"
require 'capistrano/task_definition'

class TaskDefinitionTest < Test::Unit::TestCase
  def setup
    @namespace = namespace do |s|
      role(s, :app, "app1", :primary => true)
      role(s, :app, "app2", "app3")
      role(s, :web, "web1", "web2")
      role(s, :report, "app2", :no_deploy => true)
      role(s, :file, "file", :no_deploy => true)
    end
  end

  def test_task_without_roles_should_apply_to_all_defined_hosts
    task = new_task(:testing, @namespace)
    assert_equal %w(app1 app2 app3 web1 web2 file).sort, task.servers.map { |s| s.host }.sort
  end

  def test_task_with_explicit_role_list_should_apply_only_to_those_roles
    task = new_task(:testing, @namespace, :roles => %w(app web))
    assert_equal %w(app1 app2 app3 web1 web2).sort, task.servers.map { |s| s.host }.sort
  end

  def test_task_with_single_role_should_apply_only_to_that_role
    task = new_task(:testing, @namespace, :roles => :web)
    assert_equal %w(web1 web2).sort, task.servers.map { |s| s.host }.sort
  end

  def test_task_with_hosts_option_should_apply_only_to_those_hosts
    task = new_task(:testing, @namespace, :hosts => %w(foo bar))
    assert_equal %w(foo bar).sort, task.servers.map { |s| s.host }.sort
  end

  def test_task_with_single_hosts_option_should_apply_only_to_that_host
    task = new_task(:testing, @namespace, :hosts => "foo")
    assert_equal %w(foo).sort, task.servers.map { |s| s.host }.sort
  end

  def test_task_with_roles_as_environment_variable_should_apply_only_to_that_role
    ENV['ROLES'] = "app,file"
    task = new_task(:testing, @namespace)
    assert_equal %w(app1 app2 app3 file).sort, task.servers.map { |s| s.host }.sort
  ensure
    ENV['ROLES'] = nil
  end

  def test_task_with_hosts_as_environment_variable_should_apply_only_to_those_hosts
    ENV['HOSTS'] = "foo,bar"
    task = new_task(:testing, @namespace)
    assert_equal %w(foo bar).sort, task.servers.map { |s| s.host }.sort
  ensure
    ENV['HOSTS'] = nil
  end

  def test_task_with_only_should_apply_only_to_matching_tasks
    task = new_task(:testing, @namespace, :roles => :app, :only => { :primary => true })
    assert_equal %w(app1), task.servers.map { |s| s.host }
  end

  def test_task_with_except_should_apply_only_to_matching_tasks
    task = new_task(:testing, @namespace, :except => { :no_deploy => true })
    assert_equal %w(app1 app2 app3 web1 web2).sort, task.servers.map { |s| s.host }.sort
  end

  def test_fqn_at_top_level_should_be_task_name
    task = new_task(:testing, @namespace)
    assert_equal "testing", task.fully_qualified_name
  end

  def test_fqn_in_namespace_should_include_namespace_fqn
    ns = namespace("outer:inner")
    task = new_task(:testing, ns)
    assert_equal "outer:inner:testing", task.fully_qualified_name
  end

  def test_task_should_require_block
    assert_raises(ArgumentError) do
      Capistrano::TaskDefinition.new(:testing, @namespace)
    end
  end

  private

    def namespace(fqn=nil)
      space = stub(:roles => {}, :fully_qualified_name => fqn)
      yield(space) if block_given?
      space
    end

    def role(space, name, *args)
      opts = args.last.is_a?(Hash) ? args.pop : {}
      space.roles[name] ||= []
      space.roles[name].concat(args.map { |h| Capistrano::ServerDefinition.new(h, opts) })
    end

    def new_task(name, namespace, options={}, &block)
      block ||= Proc.new {}
      task = Capistrano::TaskDefinition.new(name, namespace, options, &block)
      assert_equal block, task.body
      return task
    end
end