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