X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/4dd3d867e590ff54b9e2c1abeea879bcdf87c529..ecdc08f29a97a6127cd36c71cc56683f09907289:/services/api/test/unit/container_request_test.rb diff --git a/services/api/test/unit/container_request_test.rb b/services/api/test/unit/container_request_test.rb index 70bcc74cc3..69e277cc2c 100644 --- a/services/api/test/unit/container_request_test.rb +++ b/services/api/test/unit/container_request_test.rb @@ -5,12 +5,30 @@ require 'test_helper' require 'helpers/container_test_helper' require 'helpers/docker_migration_helper' +require 'arvados/collection' class ContainerRequestTest < ActiveSupport::TestCase include DockerMigrationHelper include DbCurrentTime include ContainerTestHelper + def with_container_auth(ctr) + auth_was = Thread.current[:api_client_authorization] + Thread.current[:api_client_authorization] = ApiClientAuthorization.find_by_uuid(ctr.auth_uuid) + begin + yield + ensure + Thread.current[:api_client_authorization] = auth_was + end + end + + def lock_and_run(ctr) + act_as_system_user do + ctr.update_attributes!(state: Container::Locked) + ctr.update_attributes!(state: Container::Running) + end + end + def create_minimal_req! attrs={} defaults = { command: ["echo", "foo"], @@ -52,7 +70,7 @@ class ContainerRequestTest < ActiveSupport::TestCase cr.container_image = "img3" cr.cwd = "/tmp3" cr.environment = {"BUP" => "BOP"} - cr.mounts = {"BAR" => "BAZ"} + cr.mounts = {"BAR" => {"kind" => "BAZ"}} cr.output_path = "/tmp4" cr.priority = 2 cr.runtime_constraints = {"vcpus" => 4} @@ -64,29 +82,33 @@ class ContainerRequestTest < ActiveSupport::TestCase end [ - {"vcpus" => 1}, - {"vcpus" => 1, "ram" => nil}, - {"vcpus" => 0, "ram" => 123}, - {"vcpus" => "1", "ram" => "123"} - ].each do |invalid_constraints| - test "Create with #{invalid_constraints}" do + {"runtime_constraints" => {"vcpus" => 1}}, + {"runtime_constraints" => {"vcpus" => 1, "ram" => nil}}, + {"runtime_constraints" => {"vcpus" => 0, "ram" => 123}}, + {"runtime_constraints" => {"vcpus" => "1", "ram" => "123"}}, + {"mounts" => {"FOO" => "BAR"}}, + {"mounts" => {"FOO" => {}}}, + {"mounts" => {"FOO" => {"kind" => "tmp", "capacity" => 42.222}}}, + {"command" => ["echo", 55]}, + {"environment" => {"FOO" => 55}} + ].each do |value| + test "Create with invalid #{value}" do set_user_from_auth :active assert_raises(ActiveRecord::RecordInvalid) do - cr = create_minimal_req!(state: "Committed", - priority: 1, - runtime_constraints: invalid_constraints) + cr = create_minimal_req!({state: "Committed", + priority: 1}.merge(value)) cr.save! end end - test "Update with #{invalid_constraints}" do + test "Update with invalid #{value}" do set_user_from_auth :active cr = create_minimal_req!(state: "Uncommitted", priority: 1) cr.save! assert_raises(ActiveRecord::RecordInvalid) do cr = ContainerRequest.find_by_uuid cr.uuid - cr.update_attributes!(state: "Committed", - runtime_constraints: invalid_constraints) + cr.update_attributes!({state: "Committed", + priority: 1}.merge(value)) end end end @@ -143,7 +165,7 @@ class ContainerRequestTest < ActiveSupport::TestCase assert_equal({"/out" => {"kind"=>"tmp", "capacity"=>1000000}}, c.mounts) assert_equal "/out", c.output_path assert_equal({"keep_cache_ram"=>268435456, "vcpus" => 2, "ram" => 30}, c.runtime_constraints) - assert_equal 1, c.priority + assert_operator 0, :<, c.priority assert_raises(ActiveRecord::RecordInvalid) do cr.priority = nil @@ -159,50 +181,17 @@ class ContainerRequestTest < ActiveSupport::TestCase assert_equal 0, c.priority end - - test "Container request max priority" do - set_user_from_auth :active - cr = create_minimal_req!(priority: 5, state: "Committed") - - c = Container.find_by_uuid cr.container_uuid - assert_equal 5, c.priority - - cr2 = create_minimal_req! - cr2.priority = 10 - cr2.state = "Committed" - cr2.container_uuid = cr.container_uuid - act_as_system_user do - cr2.save! - end - - # cr and cr2 have priority 5 and 10, and are being satisfied by - # the same container c, so c's priority should be - # max(priority)=10. - c.reload - assert_equal 10, c.priority - - cr2.update_attributes!(priority: 0) - - c.reload - assert_equal 5, c.priority - - cr.update_attributes!(priority: 0) - - c.reload - assert_equal 0, c.priority - end - - test "Independent container requests" do set_user_from_auth :active cr1 = create_minimal_req!(command: ["foo", "1"], priority: 5, state: "Committed") cr2 = create_minimal_req!(command: ["foo", "2"], priority: 10, state: "Committed") c1 = Container.find_by_uuid cr1.container_uuid - assert_equal 5, c1.priority + assert_operator 0, :<, c1.priority c2 = Container.find_by_uuid cr2.container_uuid - assert_equal 10, c2.priority + assert_operator c1.priority, :<, c2.priority + c2priority_was = c2.priority cr1.update_attributes!(priority: 0) @@ -210,12 +199,13 @@ class ContainerRequestTest < ActiveSupport::TestCase assert_equal 0, c1.priority c2.reload - assert_equal 10, c2.priority + assert_equal c2priority_was, c2.priority end test "Request is finalized when its container is cancelled" do set_user_from_auth :active cr = create_minimal_req!(priority: 1, state: "Committed", container_count_max: 1) + assert_equal users(:active).uuid, cr.modified_by_user_uuid act_as_system_user do Container.find_by_uuid(cr.container_uuid). @@ -224,6 +214,7 @@ class ContainerRequestTest < ActiveSupport::TestCase cr.reload assert_equal "Final", cr.state + assert_equal users(:active).uuid, cr.modified_by_user_uuid end test "Request is finalized when its container is completed" do @@ -232,6 +223,7 @@ class ContainerRequestTest < ActiveSupport::TestCase cr = create_minimal_req!(owner_uuid: project.uuid, priority: 1, state: "Committed") + assert_equal users(:active).uuid, cr.modified_by_user_uuid c = act_as_system_user do c = Container.find_by_uuid(cr.container_uuid) @@ -253,18 +245,20 @@ class ContainerRequestTest < ActiveSupport::TestCase cr.reload assert_equal "Final", cr.state - ['output', 'log'].each do |out_type| - pdh = Container.find_by_uuid(cr.container_uuid).send(out_type) - assert_equal(1, Collection.where(portable_data_hash: pdh, - owner_uuid: project.uuid).count, - "Container #{out_type} should be copied to #{project.uuid}") - end + assert_equal users(:active).uuid, cr.modified_by_user_uuid + assert_not_nil cr.output_uuid assert_not_nil cr.log_uuid output = Collection.find_by_uuid cr.output_uuid assert_equal output_pdh, output.portable_data_hash + assert_equal output.owner_uuid, project.uuid, "Container output should be copied to #{project.uuid}" + assert_not_nil output.modified_at + log = Collection.find_by_uuid cr.log_uuid - assert_equal log_pdh, log.portable_data_hash + assert_equal log.manifest_text, ". 37b51d194a7513e45b56f6524f2d51f2+3 0:3:bar +./log\\040for\\040container\\040#{cr.container_uuid} 37b51d194a7513e45b56f6524f2d51f2+3 0:3:bar\n" + + assert_equal log.owner_uuid, project.uuid, "Container log should be copied to #{project.uuid}" end test "Container makes container request, then is cancelled" do @@ -272,14 +266,17 @@ class ContainerRequestTest < ActiveSupport::TestCase cr = create_minimal_req!(priority: 5, state: "Committed", container_count_max: 1) c = Container.find_by_uuid cr.container_uuid - assert_equal 5, c.priority + assert_operator 0, :<, c.priority + lock_and_run(c) - cr2 = create_minimal_req! - cr2.update_attributes!(priority: 10, state: "Committed", requesting_container_uuid: c.uuid, command: ["echo", "foo2"], container_count_max: 1) - cr2.reload + cr2 = with_container_auth(c) do + create_minimal_req!(priority: 10, state: "Committed", container_count_max: 1, command: ["echo", "foo2"]) + end + assert_not_nil cr2.requesting_container_uuid + assert_equal users(:active).uuid, cr2.modified_by_user_uuid c2 = Container.find_by_uuid cr2.container_uuid - assert_equal 10, c2.priority + assert_operator 0, :<, c2.priority act_as_system_user do c.state = "Cancelled" @@ -291,43 +288,101 @@ class ContainerRequestTest < ActiveSupport::TestCase cr2.reload assert_equal 0, cr2.priority + assert_equal users(:active).uuid, cr2.modified_by_user_uuid c2.reload assert_equal 0, c2.priority end + test "child container priority follows same ordering as corresponding top-level ancestors" do + findctr = lambda { |cr| Container.find_by_uuid(cr.container_uuid) } - test "Container makes container request, then changes priority" do set_user_from_auth :active - cr = create_minimal_req!(priority: 5, state: "Committed", container_count_max: 1) - - c = Container.find_by_uuid cr.container_uuid - assert_equal 5, c.priority - - cr2 = create_minimal_req! - cr2.update_attributes!(priority: 5, state: "Committed", requesting_container_uuid: c.uuid, command: ["echo", "foo2"], container_count_max: 1) - cr2.reload - c2 = Container.find_by_uuid cr2.container_uuid - assert_equal 5, c2.priority + toplevel_crs = [ + create_minimal_req!(priority: 5, state: "Committed", environment: {"workflow" => "0"}), + create_minimal_req!(priority: 5, state: "Committed", environment: {"workflow" => "1"}), + create_minimal_req!(priority: 5, state: "Committed", environment: {"workflow" => "2"}), + ] + parents = toplevel_crs.map(&findctr) + + children = parents.map do |parent| + lock_and_run(parent) + with_container_auth(parent) do + create_minimal_req!(state: "Committed", + priority: 1, + environment: {"child" => parent.environment["workflow"]}) + end + end.map(&findctr) + + grandchildren = children.reverse.map do |child| + lock_and_run(child) + with_container_auth(child) do + create_minimal_req!(state: "Committed", + priority: 1, + environment: {"grandchild" => child.environment["child"]}) + end + end.reverse.map(&findctr) - act_as_system_user do - c.priority = 10 - c.save! - end + shared_grandchildren = children.map do |child| + with_container_auth(child) do + create_minimal_req!(state: "Committed", + priority: 1, + environment: {"grandchild" => "shared"}) + end + end.map(&findctr) - cr.reload + assert_equal shared_grandchildren[0].uuid, shared_grandchildren[1].uuid + assert_equal shared_grandchildren[0].uuid, shared_grandchildren[2].uuid + shared_grandchild = shared_grandchildren[0] - cr2.reload - assert_equal 10, cr2.priority + set_user_from_auth :active - c2.reload - assert_equal 10, c2.priority + # parents should be prioritized by submit time. + assert_operator parents[0].priority, :>, parents[1].priority + assert_operator parents[1].priority, :>, parents[2].priority + + # children should be prioritized in same order as their respective + # parents. + assert_operator children[0].priority, :>, children[1].priority + assert_operator children[1].priority, :>, children[2].priority + + # grandchildren should also be prioritized in the same order, + # despite having been submitted in the opposite order. + assert_operator grandchildren[0].priority, :>, grandchildren[1].priority + assert_operator grandchildren[1].priority, :>, grandchildren[2].priority + + # shared grandchild container should be prioritized above + # everything that isn't needed by parents[0], but not above + # earlier-submitted descendants of parents[0] + assert_operator shared_grandchild.priority, :>, grandchildren[1].priority + assert_operator shared_grandchild.priority, :>, children[1].priority + assert_operator shared_grandchild.priority, :>, parents[1].priority + assert_operator shared_grandchild.priority, :<=, grandchildren[0].priority + assert_operator shared_grandchild.priority, :<=, children[0].priority + assert_operator shared_grandchild.priority, :<=, parents[0].priority + + # increasing priority of the most recent toplevel container should + # reprioritize all of its descendants (including the shared + # grandchild) above everything else. + toplevel_crs[2].update_attributes!(priority: 72) + (parents + children + grandchildren + [shared_grandchild]).map(&:reload) + assert_operator shared_grandchild.priority, :>, grandchildren[0].priority + assert_operator shared_grandchild.priority, :>, children[0].priority + assert_operator shared_grandchild.priority, :>, parents[0].priority + assert_operator shared_grandchild.priority, :>, grandchildren[1].priority + assert_operator shared_grandchild.priority, :>, children[1].priority + assert_operator shared_grandchild.priority, :>, parents[1].priority + # ...but the shared container should not have higher priority than + # the earlier-submitted descendants of the high-priority workflow. + assert_operator shared_grandchild.priority, :<=, grandchildren[2].priority + assert_operator shared_grandchild.priority, :<=, children[2].priority + assert_operator shared_grandchild.priority, :<=, parents[2].priority end [ - ['running_container_auth', 'zzzzz-dz642-runningcontainr', 12], - ['active_no_prefs', nil, 0], + ['running_container_auth', 'zzzzz-dz642-runningcontainr', 501], + ['active_no_prefs', nil, 0] ].each do |token, expected, expected_priority| test "create as #{token} and expect requesting_container_uuid to be #{expected}" do set_user_from_auth token @@ -338,6 +393,15 @@ class ContainerRequestTest < ActiveSupport::TestCase end end + test "create as container_runtime_token and expect requesting_container_uuid to be zzzzz-dz642-20isqbkl8xwnsao" do + set_user_from_auth :container_runtime_token + Thread.current[:token] = "#{Thread.current[:token]}/zzzzz-dz642-20isqbkl8xwnsao" + cr = ContainerRequest.create(container_image: "img", output_path: "/tmp", command: ["echo", "foo"]) + assert_not_nil cr.uuid, 'uuid should be set for newly created container_request' + assert_equal 'zzzzz-dz642-20isqbkl8xwnsao', cr.requesting_container_uuid + assert_equal 1, cr.priority + end + [[{"vcpus" => [2, nil]}, lambda { |resolved| resolved["vcpus"] == 2 }], [{"vcpus" => [3, 7]}, @@ -379,6 +443,27 @@ class ContainerRequestTest < ActiveSupport::TestCase "path" => "/foo", } end], + [{"/out" => { + "kind" => "collection", + "portable_data_hash" => "1f4b0bc7583c2a7f9102c395f4ffc5e3+45", + "path" => "/foo"}}, + lambda do |resolved| + resolved["/out"] == { + "portable_data_hash" => "1f4b0bc7583c2a7f9102c395f4ffc5e3+45", + "kind" => "collection", + "path" => "/foo", + } + end], + # Empty collection + [{"/out" => { + "kind" => "collection", + "path" => "/foo"}}, + lambda do |resolved| + resolved["/out"] == { + "kind" => "collection", + "path" => "/foo", + } + end], ].each do |mounts, okfunc| test "resolve mounts #{mounts.inspect} to values" do set_user_from_auth :active @@ -412,9 +497,8 @@ class ContainerRequestTest < ActiveSupport::TestCase "path" => "/foo", }, } - assert_raises(ArgumentError) do - Container.resolve_mounts(m) - end + resolved_mounts = Container.resolve_mounts(m) + assert_equal m['portable_data_hash'], resolved_mounts['portable_data_hash'] end ['arvados/apitestfixture:latest', @@ -431,7 +515,7 @@ class ContainerRequestTest < ActiveSupport::TestCase test "Container.resolve_container_image(pdh)" do set_user_from_auth :active [[:docker_image, 'v1'], [:docker_image_1_12, 'v2']].each do |coll, ver| - Rails.configuration.docker_image_formats = [ver] + Rails.configuration.Containers.SupportedDockerImageFormats = [ver] pdh = collections(coll).portable_data_hash resolved = Container.resolve_container_image(pdh) assert_equal resolved, pdh @@ -450,8 +534,14 @@ class ContainerRequestTest < ActiveSupport::TestCase end end + test "allow unrecognized container when there are remote_hosts" do + set_user_from_auth :active + Rails.configuration.RemoteClusters = Rails.configuration.RemoteClusters.merge({foooo: ActiveSupport::InheritableOptions.new({Host: "bar.com"})}) + Container.resolve_container_image('acbd18db4cc2f85cedef654fccc4a4d8+3') + end + test "migrated docker image" do - Rails.configuration.docker_image_formats = ['v2'] + Rails.configuration.Containers.SupportedDockerImageFormats = ['v2'] add_docker19_migration_link # Test that it returns only v2 images even though request is for v1 image. @@ -469,7 +559,7 @@ class ContainerRequestTest < ActiveSupport::TestCase end test "use unmigrated docker image" do - Rails.configuration.docker_image_formats = ['v1'] + Rails.configuration.Containers.SupportedDockerImageFormats = ['v1'] add_docker19_migration_link # Test that it returns only supported v1 images even though there is a @@ -488,7 +578,7 @@ class ContainerRequestTest < ActiveSupport::TestCase end test "incompatible docker image v1" do - Rails.configuration.docker_image_formats = ['v1'] + Rails.configuration.Containers.SupportedDockerImageFormats = ['v1'] add_docker19_migration_link # Don't return unsupported v2 image even if we ask for it directly. @@ -501,7 +591,7 @@ class ContainerRequestTest < ActiveSupport::TestCase end test "incompatible docker image v2" do - Rails.configuration.docker_image_formats = ['v2'] + Rails.configuration.Containers.SupportedDockerImageFormats = ['v2'] # No migration link, don't return unsupported v1 image, set_user_from_auth :active @@ -566,7 +656,7 @@ class ContainerRequestTest < ActiveSupport::TestCase test "requesting_container_uuid at create is not allowed" do set_user_from_auth :active - assert_raises(ActiveRecord::RecordNotSaved) do + assert_raises(ActiveRecord::RecordInvalid) do create_minimal_req!(state: "Uncommitted", priority: 1, requesting_container_uuid: 'youcantdothat') end end @@ -615,6 +705,89 @@ class ContainerRequestTest < ActiveSupport::TestCase assert_not_equal cr2.container_uuid, cr.container_uuid end + test "Retry on container cancelled with runtime_token" do + set_user_from_auth :spectator + spec = api_client_authorizations(:active) + cr = create_minimal_req!(priority: 1, state: "Committed", + runtime_token: spec.token, + container_count_max: 2) + prev_container_uuid = cr.container_uuid + + c = act_as_system_user do + c = Container.find_by_uuid(cr.container_uuid) + assert_equal spec.token, c.runtime_token + c.update_attributes!(state: Container::Locked) + c.update_attributes!(state: Container::Running) + c + end + + cr.reload + assert_equal "Committed", cr.state + assert_equal prev_container_uuid, cr.container_uuid + prev_container_uuid = cr.container_uuid + + act_as_system_user do + c.update_attributes!(state: Container::Cancelled) + end + + cr.reload + assert_equal "Committed", cr.state + assert_not_equal prev_container_uuid, cr.container_uuid + prev_container_uuid = cr.container_uuid + + c = act_as_system_user do + c = Container.find_by_uuid(cr.container_uuid) + assert_equal spec.token, c.runtime_token + c.update_attributes!(state: Container::Cancelled) + c + end + + cr.reload + assert_equal "Final", cr.state + assert_equal prev_container_uuid, cr.container_uuid + end + + + test "Retry saves logs from previous attempts" do + set_user_from_auth :active + cr = create_minimal_req!(priority: 1, state: "Committed", container_count_max: 3) + + c = act_as_system_user do + c = Container.find_by_uuid(cr.container_uuid) + c.update_attributes!(state: Container::Locked) + c.update_attributes!(state: Container::Running) + c + end + + container_uuids = [] + + [0, 1, 2].each do + cr.reload + assert_equal "Committed", cr.state + container_uuids << cr.container_uuid + + c = act_as_system_user do + logc = Collection.new(manifest_text: ". 37b51d194a7513e45b56f6524f2d51f2+3 0:3:bar\n") + logc.save! + c = Container.find_by_uuid(cr.container_uuid) + c.update_attributes!(state: Container::Cancelled, log: logc.portable_data_hash) + c + end + end + + container_uuids.sort! + + cr.reload + assert_equal "Final", cr.state + assert_equal 3, cr.container_count + assert_equal ". 37b51d194a7513e45b56f6524f2d51f2+3 0:3:bar +./log\\040for\\040container\\040#{container_uuids[0]} 37b51d194a7513e45b56f6524f2d51f2+3 0:3:bar +./log\\040for\\040container\\040#{container_uuids[1]} 37b51d194a7513e45b56f6524f2d51f2+3 0:3:bar +./log\\040for\\040container\\040#{container_uuids[2]} 37b51d194a7513e45b56f6524f2d51f2+3 0:3:bar +" , Collection.find_by_uuid(cr.log_uuid).manifest_text + + end + test "Output collection name setting using output_name with name collision resolution" do set_user_from_auth :active output_name = 'unimaginative name' @@ -664,7 +837,7 @@ class ContainerRequestTest < ActiveSupport::TestCase assert_not_nil(trash) assert_not_nil(delete) assert_in_delta(trash, now + 1.second, 10) - assert_in_delta(delete, now + Rails.configuration.blob_signature_ttl.second, 10) + assert_in_delta(delete, now + Rails.configuration.Collections.BlobSigningTTL, 10) end def check_output_ttl_1y(now, trash, delete) @@ -708,12 +881,109 @@ class ContainerRequestTest < ActiveSupport::TestCase assert_equal ContainerRequest::Final, cr3.state end + [ + [false, ActiveRecord::RecordInvalid], + [true, nil], + ].each do |preemptible_conf, expected| + test "having Rails.configuration.Containers.UsePreemptibleInstances=#{preemptible_conf}, create preemptible container request and verify #{expected}" do + sp = {"preemptible" => true} + common_attrs = {cwd: "test", + priority: 1, + command: ["echo", "hello"], + output_path: "test", + scheduling_parameters: sp, + mounts: {"test" => {"kind" => "json"}}} + Rails.configuration.Containers.UsePreemptibleInstances = preemptible_conf + set_user_from_auth :active + + cr = create_minimal_req!(common_attrs) + cr.state = ContainerRequest::Committed + + if !expected.nil? + assert_raises(expected) do + cr.save! + end + else + cr.save! + assert_equal sp, cr.scheduling_parameters + end + end + end + + [ + 'zzzzz-dz642-runningcontainr', + nil, + ].each do |requesting_c| + test "having preemptible instances active on the API server, a committed #{requesting_c.nil? ? 'non-':''}child CR should not ask for preemptible instance if parameter already set to false" do + common_attrs = {cwd: "test", + priority: 1, + command: ["echo", "hello"], + output_path: "test", + scheduling_parameters: {"preemptible" => false}, + mounts: {"test" => {"kind" => "json"}}} + + Rails.configuration.Containers.UsePreemptibleInstances = true + set_user_from_auth :active + + if requesting_c + cr = with_container_auth(Container.find_by_uuid requesting_c) do + create_minimal_req!(common_attrs) + end + assert_not_nil cr.requesting_container_uuid + else + cr = create_minimal_req!(common_attrs) + end + + cr.state = ContainerRequest::Committed + cr.save! + + assert_equal false, cr.scheduling_parameters['preemptible'] + end + end + + [ + [true, 'zzzzz-dz642-runningcontainr', true], + [true, nil, nil], + [false, 'zzzzz-dz642-runningcontainr', nil], + [false, nil, nil], + ].each do |preemptible_conf, requesting_c, schedule_preemptible| + test "having Rails.configuration.Containers.UsePreemptibleInstances=#{preemptible_conf}, #{requesting_c.nil? ? 'non-':''}child CR should #{schedule_preemptible ? '':'not'} ask for preemptible instance by default" do + common_attrs = {cwd: "test", + priority: 1, + command: ["echo", "hello"], + output_path: "test", + mounts: {"test" => {"kind" => "json"}}} + + Rails.configuration.Containers.UsePreemptibleInstances = preemptible_conf + set_user_from_auth :active + + if requesting_c + cr = with_container_auth(Container.find_by_uuid requesting_c) do + create_minimal_req!(common_attrs) + end + assert_not_nil cr.requesting_container_uuid + else + cr = create_minimal_req!(common_attrs) + end + + cr.state = ContainerRequest::Committed + cr.save! + + assert_equal schedule_preemptible, cr.scheduling_parameters['preemptible'] + end + end + [ [{"partitions" => ["fastcpu","vfastcpu", 100]}, ContainerRequest::Committed, ActiveRecord::RecordInvalid], [{"partitions" => ["fastcpu","vfastcpu", 100]}, ContainerRequest::Uncommitted], [{"partitions" => "fastcpu"}, ContainerRequest::Committed, ActiveRecord::RecordInvalid], [{"partitions" => "fastcpu"}, ContainerRequest::Uncommitted], [{"partitions" => ["fastcpu","vfastcpu"]}, ContainerRequest::Committed], + [{"max_run_time" => "one day"}, ContainerRequest::Committed, ActiveRecord::RecordInvalid], + [{"max_run_time" => "one day"}, ContainerRequest::Uncommitted], + [{"max_run_time" => -1}, ContainerRequest::Committed, ActiveRecord::RecordInvalid], + [{"max_run_time" => -1}, ContainerRequest::Uncommitted], + [{"max_run_time" => 86400}, ContainerRequest::Committed], ].each do |sp, state, expected| test "create container request with scheduling_parameters #{sp} in state #{state} and verify #{expected}" do common_attrs = {cwd: "test", @@ -740,6 +1010,26 @@ class ContainerRequestTest < ActiveSupport::TestCase end end + test "Having preemptible_instances=true create a committed child container request and verify the scheduling parameter of its container" do + common_attrs = {cwd: "test", + priority: 1, + command: ["echo", "hello"], + output_path: "test", + state: ContainerRequest::Committed, + mounts: {"test" => {"kind" => "json"}}} + set_user_from_auth :active + Rails.configuration.Containers.UsePreemptibleInstances = true + + cr = with_container_auth(Container.find_by_uuid 'zzzzz-dz642-runningcontainr') do + create_minimal_req!(common_attrs) + end + assert_equal 'zzzzz-dz642-runningcontainr', cr.requesting_container_uuid + assert_equal true, cr.scheduling_parameters["preemptible"] + + c = Container.find_by_uuid(cr.container_uuid) + assert_equal true, c.scheduling_parameters["preemptible"] + end + [['Committed', true, {name: "foobar", priority: 123}], ['Committed', false, {container_count: 2}], ['Committed', false, {container_count: 0}], @@ -792,8 +1082,7 @@ class ContainerRequestTest < ActiveSupport::TestCase c = Container.find_by_uuid(cr.container_uuid) assert_equal 1, c.priority - # destroy the cr - assert_nothing_raised {cr.destroy} + cr.destroy # the cr's container now has priority of 0 c = Container.find_by_uuid(cr.container_uuid) @@ -838,24 +1127,25 @@ class ContainerRequestTest < ActiveSupport::TestCase end end - test "reuse container with same secret_mounts" do - set_user_from_auth :active - cr1 = create_minimal_req!(state: "Committed", priority: 1) - cr1.save! - cr2 = create_minimal_req!(state: "Committed", priority: 1, secret_mounts: {}) - cr2.save! - assert_not_nil cr1.container_uuid - assert_equal cr1.container_uuid, cr2.container_uuid - end - - secrets = {"/foo" => {"kind" => "binary"}} + # Note: some of these tests might look redundant because they test + # that out-of-order spellings of hashes are still considered equal + # regardless of whether the existing (container) or new (container + # request) hash needs to be re-ordered. + secrets = {"/foo" => {"kind" => "text", "content" => "xyzzy"}} + same_secrets = {"/foo" => {"content" => "xyzzy", "kind" => "text"}} + different_secrets = {"/foo" => {"kind" => "text", "content" => "something completely different"}} [ [true, nil, nil], [true, nil, {}], + [true, {}, nil], [true, {}, {}], - [true, secrets, secrets], + [true, secrets, same_secrets], + [true, same_secrets, secrets], [false, nil, secrets], [false, {}, secrets], + [false, secrets, {}], + [false, secrets, nil], + [false, secrets, different_secrets], ].each do |expect_reuse, sm1, sm2| test "container reuse secret_mounts #{sm1.inspect}, #{sm2.inspect}" do set_user_from_auth :active @@ -894,14 +1184,48 @@ class ContainerRequestTest < ActiveSupport::TestCase assert_no_secrets_logged end - test "not reuse container with different secret_mounts" do - secrets = {"/foo" => {"kind" => "binary"}} + test "conflicting key in mounts and secret_mounts" do + sm = {'/secret/foo' => {'kind' => 'text', 'content' => secret_string}} set_user_from_auth :active - cr1 = create_minimal_req!(state: "Committed", priority: 1, secret_mounts: secrets.dup) - cr1.save! - cr2 = create_minimal_req!(state: "Committed", priority: 1, secret_mounts: secrets.dup) - cr2.save! - assert_not_nil cr1.container_uuid - assert_equal cr1.container_uuid, cr2.container_uuid + cr = create_minimal_req! + assert_equal false, cr.update_attributes(state: "Committed", + priority: 1, + mounts: cr.mounts.merge(sm), + secret_mounts: sm) + assert_equal [:secret_mounts], cr.errors.messages.keys + end + + test "using runtime_token" do + set_user_from_auth :spectator + spec = api_client_authorizations(:active) + cr = create_minimal_req!(state: "Committed", runtime_token: spec.token, priority: 1) + cr.save! + c = Container.find_by_uuid cr.container_uuid + lock_and_run c + assert_nil c.auth_uuid + assert_equal c.runtime_token, spec.token + + assert_not_nil ApiClientAuthorization.find_by_uuid(spec.uuid) + + act_as_system_user do + c.update_attributes!(state: Container::Complete, + exit_code: 0, + output: '1f4b0bc7583c2a7f9102c395f4ffc5e3+45', + log: 'fa7aeb5140e2848d39b416daeef4ffc5+45') + end + + cr.reload + c.reload + assert_nil cr.runtime_token + assert_nil c.runtime_token + end + + test "invalid runtime_token" do + set_user_from_auth :active + spec = api_client_authorizations(:spectator) + assert_raises(ArgumentError) do + cr = create_minimal_req!(state: "Committed", runtime_token: "#{spec.token}xx") + cr.save! + end end end