# Copyright (C) The Arvados Authors. All rights reserved.
#
# SPDX-License-Identifier: AGPL-3.0

require 'test_helper'

class WorkUnitTest < ActiveSupport::TestCase

  reset_api_fixtures :after_each_test, false

  setup do
    Rails.configuration.Users.AnonymousUserToken = api_fixture('api_client_authorizations')['anonymous']['api_token']
  end

  [
    [Job, 'running_job_with_components', "jwu", 2, "Running", nil, 0.5],
    [PipelineInstance, 'pipeline_in_running_state', nil, 1, "Running", nil, 0.0],
    [PipelineInstance, 'has_component_with_completed_jobs', nil, 3, "Complete", true, 1.0],
    [PipelineInstance, 'pipeline_with_tagged_collection_input', "pwu", 1, "Ready", nil, 0.0],
    [PipelineInstance, 'failed_pipeline_with_two_jobs', nil, 2, "Cancelled", false, 0.0],
    [Container, 'requester', 'cwu', 1, "Complete", true, 1.0],
    [ContainerRequest, 'cr_for_requester', 'cwu', 1, "Complete", true, 1.0],
    [ContainerRequest, 'queued', 'cwu', 0, "Queued", nil, 0.0],   # priority 1
    [ContainerRequest, 'canceled_with_queued_container', 'cwu', 0, "Cancelled", false, 0.0],
    [ContainerRequest, 'canceled_with_locked_container', 'cwu', 0, "Cancelled", false, 0.0],
    [ContainerRequest, 'canceled_with_running_container', 'cwu', 1, "Running", nil, 0.0],
  ].each do |type, fixture, label, num_children, state, success, progress|
    test "children of #{fixture}" do
      use_token 'active'
      obj = find_fixture(type, fixture)
      wu = obj.work_unit(label)

      if label != nil
        assert_equal(label, wu.label)
      elsif obj.name.nil?
        assert_nil(wu.label)
      else
        assert_equal(obj.name, wu.label)
      end
      assert_equal(obj['uuid'], wu.uuid)
      assert_equal(state, wu.state_label)
      if success.nil?
        assert_nil(wu.success?)
      else
        assert_equal(success, wu.success?)
      end
      assert_equal(progress, wu.progress)

      assert_equal(num_children, wu.children.size)
      wu.children.each do |child|
        assert_equal(true, child.respond_to?(:script))
      end
    end
  end

  [
    ['cr_for_failed', 'Failed', 33],
    ['completed', 'Complete', 0],
  ].each do |cr_fixture, state, exit_code|
    test "Completed ContainerRequest state = #{state} with exit_code = #{exit_code}" do
      use_token 'active'
      obj = find_fixture(ContainerRequest, cr_fixture)
      wu = obj.work_unit
      assert_equal state, wu.state_label
      assert_equal exit_code, wu.exit_code
    end
  end

  [
    [Job, 'running_job_with_components', 1, 1, nil, true],
    [Job, 'queued', nil, 0, 1, false],
    [PipelineInstance, 'pipeline_in_running_state', 1, 1, nil, false],
    [PipelineInstance, 'has_component_with_completed_jobs', 60, 60, nil, true],
  ].each do |type, fixture, walltime, cputime, queuedtime, cputime_more_than_walltime|
    test "times for #{fixture}" do
      use_token 'active'
      obj = find_fixture(type, fixture)
      wu = obj.work_unit

      if walltime
        assert_equal true, (wu.walltime >= walltime)
      else
        if walltime.nil?
          assert_nil wu.walltime
        else
          assert_equal walltime, wu.walltime
        end
      end

      if cputime
        assert_equal true, (wu.cputime >= cputime)
      else
        assert_equal cputime, wu.cputime
      end

      if queuedtime
        assert_equal true, (wu.queuedtime >= queuedtime)
      elsif queuedtime.nil?
        assert_nil wu.queuedtime
      else
        assert_equal queuedtime, wu.queuedtime
      end

      assert_equal cputime_more_than_walltime, (wu.cputime > wu.walltime) if wu.cputime and wu.walltime
    end
  end

  test 'can_cancel?' do
    use_token 'active' do
      assert find_fixture(Job, 'running').work_unit.can_cancel?
      refute find_fixture(Container, 'running').work_unit.can_cancel?
      assert find_fixture(ContainerRequest, 'running').work_unit.can_cancel?
    end
    use_token 'spectator' do
      refute find_fixture(ContainerRequest, 'running_anonymous_accessible').work_unit.can_cancel?
    end
    use_token 'admin' do
      assert find_fixture(ContainerRequest, 'running_anonymous_accessible').work_unit.can_cancel?
    end
  end
end