1 # Copyright (C) The Arvados Authors. All rights reserved.
3 # SPDX-License-Identifier: AGPL-3.0
6 require 'webrick/https'
8 require 'helpers/users_test_helper'
10 class RemoteUsersTest < ActionDispatch::IntegrationTest
13 def salted_active_token(remote:)
14 salt_token(fixture: :active, remote: remote).sub('/zzzzz-', '/'+remote+'-')
18 token = salted_active_token(remote: remote)
19 {"HTTP_AUTHORIZATION" => "Bearer #{token}"}
22 # For remote authentication tests, we bring up a simple stub server
23 # (on a port chosen by webrick) and configure the SUT so the stub is
24 # responsible for clusters "zbbbb" (a well-behaved cluster) and
25 # "zbork" (a misbehaving cluster).
27 # Test cases can override the stub's default response to
28 # .../users/current by changing @stub_status and @stub_content.
31 clnt.ssl_config.verify_mode = OpenSSL::SSL::VERIFY_NONE
32 HTTPClient.stubs(:new).returns clnt
34 @controller = Arvados::V1::UsersController.new
35 ready = Thread::Queue.new
36 srv = WEBrick::HTTPServer.new(
38 Logger: WEBrick::Log.new(
39 Rails.root.join("log", "webrick.log").to_s,
41 AccessLog: [[File.open(Rails.root.join(
42 "log", "webrick_access.log").to_s, 'a+'),
43 WEBrick::AccessLog::COMBINED_LOG_FORMAT]],
45 SSLVerifyClient: OpenSSL::SSL::VERIFY_NONE,
46 SSLPrivateKey: OpenSSL::PKey::RSA.new(
47 File.open(Rails.root.join("tmp", "self-signed.key")).read),
48 SSLCertificate: OpenSSL::X509::Certificate.new(
49 File.open(Rails.root.join("tmp", "self-signed.pem")).read),
50 SSLCertName: [["CN", WEBrick::Utils::getservername]],
51 StartCallback: lambda { ready.push(true) })
52 srv.mount_proc '/discovery/v1/apis/arvados/v1/rest' do |req, res|
53 Rails.cache.delete 'arvados_v1_rest_discovery'
54 res.body = Arvados::V1::SchemaController.new.send(:discovery_doc).to_json
56 srv.mount_proc '/arvados/v1/users/current' do |req, res|
57 res.status = @stub_status
58 res.body = @stub_content.is_a?(String) ? @stub_content : @stub_content.to_json
65 @remote_host = "127.0.0.1:#{srv.config[:Port]}"
66 Rails.configuration.remote_hosts['zbbbb'] = @remote_host
67 Rails.configuration.remote_hosts['zbork'] = @remote_host
68 Arvados::V1::SchemaController.any_instance.stubs(:root_url).returns "https://#{@remote_host}"
71 uuid: 'zbbbb-tpzed-000000000000000',
78 @remote_server.andand.stop
81 test 'authenticate with remote token' do
82 get '/arvados/v1/users/current', {format: 'json'}, auth(remote: 'zbbbb')
83 assert_response :success
84 assert_equal 'zbbbb-tpzed-000000000000000', json_response['uuid']
85 assert_equal false, json_response['is_admin']
87 # revoke original token
89 @stub_content = {error: 'not authorized'}
91 # re-authorize before cache expires
92 get '/arvados/v1/users/current', {format: 'json'}, auth(remote: 'zbbbb')
93 assert_response :success
95 # simulate cache expiry
96 ApiClientAuthorization.where(
97 uuid: salted_active_token(remote: 'zbbbb').split('/')[1]).
98 update_all(expires_at: db_current_time - 1.minute)
100 # re-authorize after cache expires
101 get '/arvados/v1/users/current', {format: 'json'}, auth(remote: 'zbbbb')
105 test 'authenticate with remote token from misbhehaving remote cluster' do
106 get '/arvados/v1/users/current', {format: 'json'}, auth(remote: 'zbork')
110 test 'authenticate with remote token that fails validate' do
113 error: 'not authorized',
115 get '/arvados/v1/users/current', {format: 'json'}, auth(remote: 'zbbbb')
119 test 'remote api server is not an api server' do
121 @stub_content = '<html>bad</html>'
122 get '/arvados/v1/users/current', {format: 'json'}, auth(remote: 'zbbbb')
126 ['zbbbb', 'z0000'].each do |token_valid_for|
127 test "validate #{token_valid_for}-salted token for remote cluster zbbbb" do
128 salted_token = salt_token(fixture: :active, remote: token_valid_for)
129 get '/arvados/v1/users/current', {format: 'json', remote: 'zbbbb'}, {
130 "HTTP_AUTHORIZATION" => "Bearer #{salted_token}"
132 if token_valid_for == 'zbbbb'
134 assert_equal(users(:active).uuid, json_response['uuid'])
141 test "list readable groups with salted token" do
142 salted_token = salt_token(fixture: :active, remote: 'zbbbb')
143 get '/arvados/v1/groups', {
148 "HTTP_AUTHORIZATION" => "Bearer #{salted_token}"
151 group_uuids = json_response['items'].collect { |i| i['uuid'] }
152 assert_includes(group_uuids, 'zzzzz-j7d0g-fffffffffffffff')
153 refute_includes(group_uuids, 'zzzzz-j7d0g-000000000000000')