require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\1')

class TestTask < Test::Unit::TestCase
  def test_basic_task
    task = Task.setup do |s=""|
      (self + s).length
    end

    assert_equal 4, task.exec_on("1234")
    assert_equal 6, task.exec_on("1234","56")
  end

  def test_step
    task = Task.setup do |s=""|
      s.length
    end

    s = task.job('test', ['12'])
    s.clean
    assert_equal 2, s.run
  end

  def test_override_inputs
    wf = Workflow.annonymous_workflow "TaskInputs" do
      input :input1, :string
      task :step1 => :string do |i| i end

      dep :step1, :input1 => 1
      input :input2, :string
      task :step2 => :string do |i| i end
    end

    job = wf.job(:step2, :input1 => 2)
    assert_equal Task::DEFAULT_NAME, job.name
    assert_not_equal Task::DEFAULT_NAME, job.step(:step1).name
  end

  def test_override_inputs_block
    Log.severity = 0
    wf = Workflow.annonymous_workflow "TaskInputs" do
      input :input1, :string
      task :step1 => :string do |i| i end

      dep :step1 do |id,options|
        {:inputs => {:input1 => 1}}
      end
      input :input2, :string
      task :step2 => :string do |i| i end
    end

    job = wf.job(:step2, :input1 => 2)
    assert_equal Task::DEFAULT_NAME, job.name
    assert_not_equal Task::DEFAULT_NAME, job.step(:step1).name
  end

  def test_task_override_dep
    Log.severity = 0
    wf = Workflow.annonymous_workflow "TaskInputs" do
      input :input1, :integer
      task :step1 => :integer do |i| i end

      dep :step1
      input :input2, :integer, "Integer", 3
      task :step2 => :integer do |i| i * step(:step1).load end
    end

    assert_equal 6, wf.job(:step2, :input1 => 2, :input2 => 3).run

    step1_job = wf.job(:step1, :input1 => 6)
    assert_equal 18, wf.job(:step2, :input1 => 2, "TaskInputs#step1" => step1_job).run

    assert_equal 18, wf.job(:step2, :input1 => 2, "TaskInputs#step1" => step1_job).run

    assert_equal 18, wf.job(:step2, "TaskInputs#step1" => step1_job).run
  end

  def test_input_dep
    wf = Workflow.annonymous_workflow "TaskInputs" do
      input :v1, :integer
      input :v2, :integer
      task :sum => :integer do |v1,v2|
        v1 + v2
      end

      input :input1, :integer
      task :step1 => :integer do |i| i end

      input :input2, :integer
      task :step2 => :integer do |i| i end

      dep :step1, :input1 => 2
      dep :step2, :input2 => 3
      dep :sum, :v1 => :step1, :v2 => :step2
      task :my_sum => :integer do
        dependencies.last.load
      end

      dep :my_sum
      task :double => :integer do
        step(:my_sum).load * 2 
      end
    end

    job = wf.job(:my_sum)
    assert_equal 5, job.run
    assert_equal Task::DEFAULT_NAME, job.name

    job = wf.job(:double)
    assert_equal Task::DEFAULT_NAME, job.name

    step1 = wf.job(:step1, :input1 => 3)
    job = wf.job(:double, "TaskInputs#step1" => step1)
    assert_equal 12, job.run
    assert_not_equal Task::DEFAULT_NAME, job.name

    step1 = wf.job(:step1, :input1 => 4)
    job = wf.job(:double, "TaskInputs#step1" => step1)
    assert_equal 14, job.run
    assert_not_equal Task::DEFAULT_NAME, job.name
  end

  def test_input_dep_override
    wf = Workflow.annonymous_workflow "TaskInputs" do
      input :v1, :integer
      input :v2, :integer
      task :sum => :integer do |v1,v2|
        v1 + v2
      end

      input :input1, :integer
      task :step1 => :integer do |i| i end

      input :input2, :integer
      task :step2 => :integer do |i| i end

      dep :step1
      dep :step2
      task :my_sum => :integer do
        dependencies.inject(0){|acc,d| acc += d.load }
      end
    end

    step2 = wf.job(:step2, :input2 => 4)
    job = wf.job(:my_sum, :input1 => 2, :input2 => 3, "TaskInputs#step2"=> step2)
    assert_equal 6, job.exec

    job = wf.job(:my_sum, :input1 => 2, :input2 => 3)
    assert_equal 5, job.run

    TmpFile.with_file(4) do |file|
      job = wf.job(:my_sum, :input1 => 2, :input2 => 3, "TaskInputs#step2"=> file)
      assert_equal 6, job.run
      assert_not_equal Task::DEFAULT_NAME, job.name
    end
  end

  def test_input_rename
    wf = Workflow.annonymous_workflow "TaskInputs" do
      input :v1, :integer
      input :v2, :integer
      task :sum => :integer do |v1,v2|
        v1 + v2
      end

      input :vv1, :integer
      input :vv2, :integer
      dep :sum, :v1 => :vv1, :v2 => :vv2
      task :my_sum => :integer do
        dependencies.last.load
      end
    end

    Log.severity = 0
    job = wf.job(:my_sum, :vv1 => 2, :vv2 => 3)
    assert_equal 5, job.run
  end
end