X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/cca78a112d6a2aa4f76c1956cfe8ec2d43a68759..4139c5dfde60cdd20d39f385ca36107b0c44906f:/services/api/test/unit/fail_jobs_test.rb diff --git a/services/api/test/unit/fail_jobs_test.rb b/services/api/test/unit/fail_jobs_test.rb index 8c6539e8dc..304335c6f0 100644 --- a/services/api/test/unit/fail_jobs_test.rb +++ b/services/api/test/unit/fail_jobs_test.rb @@ -1,3 +1,7 @@ +# Copyright (C) The Arvados Authors. All rights reserved. +# +# SPDX-License-Identifier: AGPL-3.0 + require 'test_helper' require 'crunch_dispatch' @@ -36,14 +40,14 @@ class FailJobsTest < ActiveSupport::TestCase end test 'cancel slurm jobs' do - Rails.configuration.crunch_job_wrapper = :slurm_immediate - Rails.configuration.crunch_job_user = 'foobar' - fake_squeue = File.popen("echo #{@job[:before_reboot].uuid}") - fake_scancel = File.popen("true") - File.expects(:popen). + Rails.configuration.Containers.JobsAPI.CrunchJobWrapper = "slurm_immediate" + Rails.configuration.Containers.JobsAPI.CrunchJobUser = 'foobar' + fake_squeue = IO.popen("echo #{@job[:before_reboot].uuid}") + fake_scancel = IO.popen("true") + IO.expects(:popen). with(['squeue', '-a', '-h', '-o', '%j']). returns(fake_squeue) - File.expects(:popen). + IO.expects(:popen). with(includes('sudo', '-u', 'foobar', 'scancel', '-n', @job[:before_reboot].uuid)). returns(fake_scancel) @dispatch.fail_jobs(before: Time.at(BOOT_TIME).to_s) @@ -51,7 +55,7 @@ class FailJobsTest < ActiveSupport::TestCase end test 'use reboot time' do - Rails.configuration.crunch_job_wrapper = nil + Rails.configuration.Containers.JobsAPI.CrunchJobWrapper = nil @dispatch.expects(:open).once.with('/proc/stat'). returns open(Rails.root.join('test/fixtures/files/proc_stat')) @dispatch.fail_jobs(before: 'reboot')