Contents

require File.join(File.expand_path(File.dirname(__FILE__)), '../../..', 'test_helper.rb')
require 'rbbt/workflow'
require 'rbbt/hpc/orchestrate/rules'

module TestWFA
  extend Workflow

  task :a1 => :string do self.task_name.to_s end

  dep :a1
  task :a2 => :string do self.task_name.to_s end

  dep :a2
  task :a3 => :string do self.task_name.to_s end
end

module TestWFB
  extend Workflow

  dep TestWFA, :a2
  task :b1 => :string do self.task_name.to_s end

  dep :b1
  task :b2 => :string do self.task_name.to_s end
end

module TestWFC
  extend Workflow

  dep TestWFA, :a1
  dep_task :c1, TestWFB, :b2

  task :c2 => :string do self.task_name.to_s end

  dep :c1
  dep :c2
  task :c3 => :string do self.task_name.to_s end
end

class TestOrchestrate < Test::Unit::TestCase

  RULES = IndiferentHash.setup(YAML.load(<<-EOF))
---
defaults:
  queue: first_queue
  time: 1h
  log: 2
  config_keys: key1 value1 token1
chains:
  chain_a:
    workflow: TestWFA
    tasks: a1, a2, a3
    config_keys: key2 value2 token2, key3 value3 token3.1 token3.2
  chain_b:
    workflow: TestWFB
    tasks: b1, b2
  chain_b2:
    comment: This chain is not valid, it is missing a2
    tasks: TestWFA#a1, TestWFA#a3, TestWFB#b1, TestWFB#b2
TestWFA:
  defaults:
    log: 4
    config_keys: key4 value4 token4
  a1:
    cpus: 10
    config_keys: key5 value5 token5
TestWFC:
  defaults:
    skip: true
    log: 4
  EOF

  def test_defaults
    rules = HPC::Orchestration.task_specific_rules RULES, "TestWFA", :a1
    assert_equal "first_queue" , rules[:queue]
  end


  def test_task_options
    rules = HPC::Orchestration.task_specific_rules RULES, "TestWFA", :a1
    assert_equal 10, rules[:cpus]
    assert_equal 4, rules[:log]
  end

  def test_skip
    rules = HPC::Orchestration.task_specific_rules RULES, "TestWFC", :c1
    assert rules[:skip]
  end


end

Version data entries

78 entries across 78 versions & 1 rubygems

Version	Path
rbbt-util-5.37.4	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.37.3	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.37.1	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.37.0	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.36.0	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.35.4	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.35.3	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.35.2	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.35.1	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.27	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.26	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.25	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.24	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.23	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.22	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.21	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.20	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.18	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.17	test/rbbt/hpc/orchestrate/test_rules.rb
rbbt-util-5.34.16	test/rbbt/hpc/orchestrate/test_rules.rb