1 # Copyright (C) The Arvados Authors. All rights reserved.
3 # SPDX-License-Identifier: AGPL-3.0
5 require 'update_permissions'
7 ENV["RAILS_ENV"] = "test"
8 unless ENV["NO_COVERAGE_TEST"]
10 verbose_orig = $VERBOSE
14 require 'simplecov-rcov'
16 $VERBOSE = verbose_orig
18 class SimpleCov::Formatter::MergedFormatter
20 SimpleCov::Formatter::HTMLFormatter.new.format(result)
21 SimpleCov::Formatter::RcovFormatter.new.format(result)
24 SimpleCov.formatter = SimpleCov::Formatter::MergedFormatter
27 add_filter 'initializers/secret_token'
28 add_filter 'initializers/omniauth'
31 $stderr.puts "SimpleCov unavailable (#{e}). Proceeding without."
35 require File.expand_path('../../config/environment', __FILE__)
36 require 'rails/test_help'
38 require 'mocha/minitest'
40 module ArvadosTestSupport
42 Oj.strict_load response.body
45 def api_token(api_client_auth_name)
46 api_client_authorizations(api_client_auth_name).token
49 def auth(api_client_auth_name)
50 {'HTTP_AUTHORIZATION' => "Bearer #{api_token(api_client_auth_name)}"}
54 return lambda { model.errors.full_messages.inspect }
58 class ActiveSupport::TestCase
59 include FactoryBot::Syntax::Methods
62 include ArvadosTestSupport
63 include CurrentApiClient
66 Thread.current[:api_client_ip_address] = nil
67 Thread.current[:api_client_authorization] = nil
68 Thread.current[:api_client_uuid] = nil
69 Thread.current[:api_client] = nil
70 Thread.current[:token] = nil
71 Thread.current[:user] = nil
75 def assert_equal(expect, *args)
83 def assert_not_allowed
84 # Provide a block that calls a Rails boolean "true or false" success value,
85 # like model.save or model.destroy. This method will test that it either
86 # returns false, or raises a Permission Denied exception.
89 rescue ArvadosModel::PermissionDeniedError
93 def add_permission_link from_who, to_what, perm_type
95 Link.create!(tail_uuid: from_who.uuid,
96 head_uuid: to_what.uuid,
97 link_class: 'permission',
102 def confirm_keys_as_symbols(a_hash, section_name)
103 a_hash.keys.each do |k|
104 assert(k.is_a?(Symbol), "Key '#{k}' on section '#{section_name}' should be a Symbol")
105 confirm_keys_as_symbols(a_hash[k], "#{section_name}.#{k}") if a_hash[k].is_a?(Hash)
109 def restore_configuration
110 # Confirm that any changed configuration doesn't include non-symbol keys
111 $arvados_config.keys.each do |config_section_name|
112 config_section = Rails.configuration.send("#{config_section_name}")
113 if config_section.is_a?(Hash)
114 confirm_keys_as_symbols(config_section, config_section_name)
117 # Restore configuration settings changed during tests
118 ConfigLoader.copy_into_config $arvados_config, Rails.configuration
119 ConfigLoader.copy_into_config $remaining_config, Rails.configuration
122 def set_user_from_auth(auth_name)
123 client_auth = api_client_authorizations(auth_name)
124 Thread.current[:api_client_authorization] = client_auth
125 Thread.current[:api_client] = client_auth.api_client
126 Thread.current[:user] = client_auth.user
127 Thread.current[:token] = client_auth.token
131 self.request.headers["Accept"] = "text/json"
134 def authorize_with api_client_auth_name
135 authorize_with_token api_client_authorizations(api_client_auth_name).token
138 def authorize_with_token token
140 t = t.token if t.respond_to? :token
141 ArvadosApiToken.new.call("rack.input" => "",
142 "HTTP_AUTHORIZATION" => "Bearer #{t}")
145 def salt_token(fixture:, remote:)
146 auth = api_client_authorizations(fixture)
148 token = auth.api_token
149 hmac = OpenSSL::HMAC.hexdigest('sha1', token, remote)
150 return "v2/#{uuid}/#{hmac}"
153 def self.skip_slow_tests?
154 !(ENV['RAILS_TEST_SHORT'] || '').empty?
157 def self.skip(*args, &block)
160 def self.slow_test(name, &block)
161 test(name, &block) unless skip_slow_tests?
165 class ActionController::TestCase
168 self.request.headers['Accept'] = 'application/json'
169 self.request.headers['Content-Type'] = 'application/json'
172 def check_counter action
174 if @test_counter == 2
175 assert_equal 1, 2, "Multiple actions in functional test"
179 [:get, :post, :put, :patch, :delete].each do |method|
180 define_method method do |action, *args|
182 # After Rails 5.0 upgrade, some params don't get properly serialized.
183 # One case are filters: [['attr', 'op', 'val']] become [['attr'], ['op'], ['val']]
184 # if not passed upstream as a JSON string.
185 if args[0].is_a?(Hash) && args[0][:params].is_a?(Hash)
186 args[0][:params].each do |key, _|
187 next if key == :exclude_script_versions # Job Reuse tests
188 # Keys could be: :filters, :where, etc
189 if [Array, Hash].include?(args[0][:params][key].class)
190 args[0][:params][key] = SafeJSON.dump(args[0][:params][key])
205 @test_case.shutdown()
210 def self.startup; end
211 def self.shutdown; end
214 class ActionDispatch::IntegrationTest
216 Thread.current[:api_client_ip_address] = nil
217 Thread.current[:api_client_authorization] = nil
218 Thread.current[:api_client_uuid] = nil
219 Thread.current[:api_client] = nil
220 Thread.current[:token] = nil
221 Thread.current[:user] = nil
225 # Ensure permissions are computed from the test fixtures.