apps/workbench/test/unit/work_unit_test.rb

   1 require 'test_helper'
   2
   3 class WorkUnitTest < ActiveSupport::TestCase
   4
   5   reset_api_fixtures :after_each_test, false
   6
   7   setup do
   8     Rails.configuration.anonymous_user_token = api_fixture('api_client_authorizations')['anonymous']['api_token']
   9   end
  10
  11   [
  12     [Job, 'running_job_with_components', "jwu", 2, "Running", nil, 0.5],
  13     [PipelineInstance, 'pipeline_in_running_state', nil, 1, "Running", nil, 0.0],
  14     [PipelineInstance, 'has_component_with_completed_jobs', nil, 3, "Complete", true, 1.0],
  15     [PipelineInstance, 'pipeline_with_tagged_collection_input', "pwu", 1, "Ready", nil, 0.0],
  16     [Container, 'requester', 'cwu', 1, "Complete", true, 1.0],
  17     [ContainerRequest, 'cr_for_requester', 'cwu', 1, "Complete", true, 1.0],
  18     [ContainerRequest, 'queued', 'cwu', 0, "Queued", nil, 0.0],   # priority 1
  19     [ContainerRequest, 'canceled_with_queued_container', 'cwu', 0, "Cancelled", false, 0.0],
  20     [ContainerRequest, 'canceled_with_locked_container', 'cwu', 0, "Queued", nil, 0.0],
  21     [ContainerRequest, 'canceled_with_running_container', 'cwu', 1, "Running", nil, 0.0],
  22   ].each do |type, fixture, label, num_children, state, success, progress|
  23     test "children of #{fixture}" do
  24       use_token 'active'
  25       obj = find_fixture(type, fixture)
  26       wu = obj.work_unit(label)
  27
  28       if label != nil
  29         assert_equal(label, wu.label)
  30       else
  31         assert_equal(obj.name, wu.label)
  32       end
  33       assert_equal(obj['uuid'], wu.uuid)
  34       assert_equal(state, wu.state_label)
  35       assert_equal(success, wu.success?)
  36       assert_equal(progress, wu.progress)
  37
  38       assert_equal(num_children, wu.children.size)
  39       wu.children.each do |child|
  40         assert_equal(true, child.respond_to?(:script))
  41       end
  42     end
  43   end
  44
  45   [
  46     ['cr_for_failed', 'Failed', 33],
  47     ['completed', 'Complete', 0],
  48   ].each do |cr_fixture, state, exit_code|
  49     test "Completed ContainerRequest state = #{state} with exit_code = #{exit_code}" do
  50       use_token 'active'
  51       obj = find_fixture(ContainerRequest, cr_fixture)
  52       wu = obj.work_unit
  53       assert_equal state, wu.state_label
  54       assert_equal exit_code, wu.exit_code
  55     end
  56   end
  57
  58   [
  59     [Job, 'running_job_with_components', 1, 1, nil, true],
  60     [Job, 'queued', nil, 0, 1, false],
  61     [PipelineInstance, 'pipeline_in_running_state', 1, 1, nil, false],
  62     [PipelineInstance, 'has_component_with_completed_jobs', 60, 60, nil, true],
  63   ].each do |type, fixture, walltime, cputime, queuedtime, cputime_more_than_walltime|
  64     test "times for #{fixture}" do
  65       use_token 'active'
  66       obj = find_fixture(type, fixture)
  67       wu = obj.work_unit
  68
  69       if walltime
  70         assert_equal true, (wu.walltime >= walltime)
  71       else
  72         assert_equal walltime, wu.walltime
  73       end
  74
  75       if cputime
  76         assert_equal true, (wu.cputime >= cputime)
  77       else
  78         assert_equal cputime, wu.cputime
  79       end
  80
  81       if queuedtime
  82         assert_equal true, (wu.queuedtime >= queuedtime)
  83       else
  84         assert_equal queuedtime, wu.queuedtime
  85       end
  86
  87       assert_equal cputime_more_than_walltime, (wu.cputime > wu.walltime) if wu.cputime and wu.walltime
  88     end
  89   end
  90
  91   test 'can_cancel?' do
  92     use_token 'active' do
  93       assert find_fixture(Job, 'running').work_unit.can_cancel?
  94       refute find_fixture(Container, 'running').work_unit.can_cancel?
  95       assert find_fixture(ContainerRequest, 'running').work_unit.can_cancel?
  96     end
  97     use_token 'spectator' do
  98       refute find_fixture(ContainerRequest, 'running_anonymous_accessible').work_unit.can_cancel?
  99     end
 100     use_token 'admin' do
 101       assert find_fixture(ContainerRequest, 'running_anonymous_accessible').work_unit.can_cancel?
 102     end
 103   end
 104 end