1 # Copyright (C) The Arvados Authors. All rights reserved.
3 # SPDX-License-Identifier: AGPL-3.0
7 class UserSessionsControllerTest < ActionController::TestCase
9 test "redirect to joshid" do
10 api_client_page = 'http://client.example.com/home'
11 get :login, params: {return_to: api_client_page}
12 assert_response :redirect
13 assert_equal("http://test.host/auth/joshid?return_to=%2Chttp%3A%2F%2Fclient.example.com%2Fhome", @response.redirect_url)
14 assert_nil assigns(:api_client)
17 test "send token when user is already logged in" do
18 authorize_with :inactive
19 api_client_page = 'http://client.example.com/home'
20 get :login, params: {return_to: api_client_page}
21 assert_response :redirect
22 assert_equal(0, @response.redirect_url.index(api_client_page + '?'),
23 'Redirect url ' + @response.redirect_url +
24 ' should start with ' + api_client_page + '?')
25 assert_not_nil assigns(:api_client)
28 test "login creates token without expiration by default" do
29 assert_equal Rails.configuration.Login.TokenLifetime, 0
30 authorize_with :inactive
31 api_client_page = 'http://client.example.com/home'
32 get :login, params: {return_to: api_client_page}
33 assert_not_nil assigns(:api_client)
34 assert_nil assigns(:api_client_auth).expires_at
37 test "login creates token with configured lifetime" do
38 token_lifetime = 1.hour
39 Rails.configuration.Login.TokenLifetime = token_lifetime
40 authorize_with :inactive
41 api_client_page = 'http://client.example.com/home'
42 get :login, params: {return_to: api_client_page}
43 assert_not_nil assigns(:api_client)
44 api_client_auth = assigns(:api_client_auth)
45 assert_in_delta(api_client_auth.expires_at,
46 api_client_auth.updated_at + token_lifetime,
51 [1.hour, 2.hour, 1.hour],
52 [2.hour, 1.hour, 1.hour],
53 [2.hour, nil, 2.hour],
54 ].each do |config_lifetime, request_lifetime, expect_lifetime|
55 test "login with TokenLifetime=#{config_lifetime} and request has expires_at=#{ request_lifetime.nil? ? "nil" : request_lifetime }" do
56 Rails.configuration.Login.TokenLifetime = config_lifetime
57 expected_expiration_time = Time.now() + expect_lifetime
58 authorize_with :inactive
59 @request.headers['Authorization'] = 'Bearer '+Rails.configuration.SystemRootToken
60 if request_lifetime.nil?
61 get :create, params: {provider: 'controller', auth_info: {email: "foo@bar.com"}, return_to: ',https://app.example'}
63 get :create, params: {provider: 'controller', auth_info: {email: "foo@bar.com", expires_at: Time.now() + request_lifetime}, return_to: ',https://app.example'}
65 assert_response :redirect
66 api_client_auth = assigns(:api_client_auth)
67 assert_not_nil api_client_auth
68 assert_not_nil assigns(:api_client)
69 assert_in_delta(api_client_auth.expires_at,
70 expected_expiration_time,
75 test "login with remote param returns a salted token" do
76 authorize_with :inactive
77 api_client_page = 'http://client.example.com/home'
78 remote_prefix = 'zbbbb'
79 get :login, params: {return_to: api_client_page, remote: remote_prefix}
80 assert_response :redirect
81 api_client_auth = assigns(:api_client_auth)
82 assert_not_nil api_client_auth
83 assert_includes(@response.redirect_url, 'api_token='+api_client_auth.salted_token(remote: remote_prefix))
86 test "login with malformed remote param returns an error" do
87 authorize_with :inactive
88 api_client_page = 'http://client.example.com/home'
89 remote_prefix = 'invalid_cluster_id'
90 get :login, params: {return_to: api_client_page, remote: remote_prefix}
94 test "login to LoginCluster" do
95 Rails.configuration.Login.LoginCluster = 'zbbbb'
96 Rails.configuration.RemoteClusters['zbbbb'] = ConfigLoader.to_OrderedOptions({'Host' => 'zbbbb.example.com'})
97 api_client_page = 'http://client.example.com/home'
98 get :login, params: {return_to: api_client_page}
99 assert_response :redirect
100 assert_equal("https://zbbbb.example.com/login?return_to=http%3A%2F%2Fclient.example.com%2Fhome", @response.redirect_url)
101 assert_nil assigns(:api_client)
104 test "don't go into redirect loop if LoginCluster is self" do
105 Rails.configuration.Login.LoginCluster = 'zzzzz'
106 api_client_page = 'http://client.example.com/home'
107 get :login, params: {return_to: api_client_page}
108 assert_response :redirect
109 assert_equal("http://test.host/auth/joshid?return_to=%2Chttp%3A%2F%2Fclient.example.com%2Fhome", @response.redirect_url)
110 assert_nil assigns(:api_client)
113 test "controller cannot create session without SystemRootToken" do
114 get :create, params: {provider: 'controller', auth_info: {email: "foo@bar.com"}, return_to: ',https://app.example'}
118 test "controller cannot create session with wrong SystemRootToken" do
119 @request.headers['Authorization'] = 'Bearer blah'
120 get :create, params: {provider: 'controller', auth_info: {email: "foo@bar.com"}, return_to: ',https://app.example'}
124 test "controller can create session using SystemRootToken" do
125 @request.headers['Authorization'] = 'Bearer '+Rails.configuration.SystemRootToken
126 get :create, params: {provider: 'controller', auth_info: {email: "foo@bar.com"}, return_to: ',https://app.example'}
127 assert_response :redirect
128 api_client_auth = assigns(:api_client_auth)
129 assert_not_nil api_client_auth
130 assert_includes(@response.redirect_url, 'api_token='+api_client_auth.token)