X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/3dfc071494cbb0386ecd7269c72a435fc7f9cc24..31d8469fa1adfa0c3a07a23725f26b26f2291df5:/services/api/test/test_helper.rb?ds=sidebyside diff --git a/services/api/test/test_helper.rb b/services/api/test/test_helper.rb index 3413f6486a..5747a85cf5 100644 --- a/services/api/test/test_helper.rb +++ b/services/api/test/test_helper.rb @@ -1,8 +1,18 @@ +# Copyright (C) The Arvados Authors. All rights reserved. +# +# SPDX-License-Identifier: AGPL-3.0 + ENV["RAILS_ENV"] = "test" unless ENV["NO_COVERAGE_TEST"] begin - require 'simplecov' - require 'simplecov-rcov' + verbose_orig = $VERBOSE + begin + $VERBOSE = nil + require 'simplecov' + require 'simplecov-rcov' + ensure + $VERBOSE = verbose_orig + end class SimpleCov::Formatter::MergedFormatter def format(result) SimpleCov::Formatter::HTMLFormatter.new.format(result) @@ -22,46 +32,83 @@ end require File.expand_path('../../config/environment', __FILE__) require 'rails/test_help' +require 'mocha' +require 'mocha/minitest' module ArvadosTestSupport def json_response - ActiveSupport::JSON.decode @response.body + Oj.strict_load response.body end def api_token(api_client_auth_name) - api_client_authorizations(api_client_auth_name).api_token + api_client_authorizations(api_client_auth_name).token end def auth(api_client_auth_name) - {'HTTP_AUTHORIZATION' => "OAuth2 #{api_token(api_client_auth_name)}"} + {'HTTP_AUTHORIZATION' => "Bearer #{api_token(api_client_auth_name)}"} + end + + def show_errors model + return lambda { model.errors.full_messages.inspect } end end class ActiveSupport::TestCase - include FactoryGirl::Syntax::Methods + include FactoryBot::Syntax::Methods fixtures :all include ArvadosTestSupport + include CurrentApiClient teardown do Thread.current[:api_client_ip_address] = nil Thread.current[:api_client_authorization] = nil Thread.current[:api_client_uuid] = nil Thread.current[:api_client] = nil + Thread.current[:token] = nil Thread.current[:user] = nil - # Restore configuration settings changed during tests - $application_config.each do |k,v| - if k.match /^[^.]*$/ - Rails.configuration.send (k + '='), v - end + restore_configuration + end + + def assert_equal(expect, *args) + if expect.nil? + assert_nil(*args) + else + super end end + def assert_not_allowed + # Provide a block that calls a Rails boolean "true or false" success value, + # like model.save or model.destroy. This method will test that it either + # returns false, or raises a Permission Denied exception. + begin + refute(yield) + rescue ArvadosModel::PermissionDeniedError + end + end + + def add_permission_link from_who, to_what, perm_type + act_as_system_user do + Link.create!(tail_uuid: from_who.uuid, + head_uuid: to_what.uuid, + link_class: 'permission', + name: perm_type) + end + end + + def restore_configuration + # Restore configuration settings changed during tests + ConfigLoader.copy_into_config $arvados_config, Rails.configuration + ConfigLoader.copy_into_config $remaining_config, Rails.configuration + end + def set_user_from_auth(auth_name) client_auth = api_client_authorizations(auth_name) Thread.current[:api_client_authorization] = client_auth Thread.current[:api_client] = client_auth.api_client Thread.current[:user] = client_auth.user + Thread.current[:token] = client_auth.token end def expect_json @@ -69,35 +116,83 @@ class ActiveSupport::TestCase end def authorize_with api_client_auth_name - authorize_with_token api_client_authorizations(api_client_auth_name).api_token + authorize_with_token api_client_authorizations(api_client_auth_name).token end def authorize_with_token token t = token - t = t.api_token if t.respond_to? :api_token + t = t.token if t.respond_to? :token ArvadosApiToken.new.call("rack.input" => "", - "HTTP_AUTHORIZATION" => "OAuth2 #{t}") + "HTTP_AUTHORIZATION" => "Bearer #{t}") + end + + def salt_token(fixture:, remote:) + auth = api_client_authorizations(fixture) + uuid = auth.uuid + token = auth.api_token + hmac = OpenSSL::HMAC.hexdigest('sha1', token, remote) + return "v2/#{uuid}/#{hmac}" + end + + def self.skip_slow_tests? + !(ENV['RAILS_TEST_SHORT'] || '').empty? + end + + def self.skip(*args, &block) + end + + def self.slow_test(name, &block) + test(name, &block) unless skip_slow_tests? end end class ActionController::TestCase setup do - @counter = 0 + @test_counter = 0 + self.request.headers['Accept'] = 'application/json' + self.request.headers['Content-Type'] = 'application/json' end def check_counter action - @counter += 1 - if @counter == 2 - # assert_equal 1, 2, "Multiple actions in functional test" + @test_counter += 1 + if @test_counter == 2 + assert_equal 1, 2, "Multiple actions in functional test" end end [:get, :post, :put, :patch, :delete].each do |method| define_method method do |action, *args| check_counter action + # After Rails 5.0 upgrade, some params don't get properly serialized. + # One case are filters: [['attr', 'op', 'val']] become [['attr'], ['op'], ['val']] + # if not passed upstream as a JSON string. + if args[0].is_a?(Hash) && args[0][:params].is_a?(Hash) + args[0][:params].each do |key, _| + next if key == :exclude_script_versions # Job Reuse tests + # Keys could be: :filters, :where, etc + if [Array, Hash].include?(args[0][:params][key].class) + args[0][:params][key] = SafeJSON.dump(args[0][:params][key]) + end + end + end super action, *args end end + + def self.suite + s = super + def s.run(*args) + @test_case.startup() + begin + super + ensure + @test_case.shutdown() + end + end + s + end + def self.startup; end + def self.shutdown; end end class ActionDispatch::IntegrationTest @@ -106,6 +201,7 @@ class ActionDispatch::IntegrationTest Thread.current[:api_client_authorization] = nil Thread.current[:api_client_uuid] = nil Thread.current[:api_client] = nil + Thread.current[:token] = nil Thread.current[:user] = nil end end