1 # Copyright (C) The Arvados Authors. All rights reserved.
3 # SPDX-License-Identifier: AGPL-3.0
7 class Arvados::V1::ManagementControllerTest < ActionController::TestCase
9 [false, nil, 404, 'disabled'],
10 [true, nil, 401, 'authorization required'],
11 [true, 'badformatwithnoBearer', 403, 'authorization error'],
12 [true, 'Bearer wrongtoken', 403, 'authorization error'],
13 [true, 'Bearer configuredmanagementtoken', 200, '{"health":"OK"}'],
14 ].each do |enabled, header, error_code, error_msg|
15 test "_health/ping when #{if enabled then 'enabled' else 'disabled' end} with header '#{header}'" do
17 Rails.configuration.ManagementToken = 'configuredmanagementtoken'
19 Rails.configuration.ManagementToken = ""
22 @request.headers['Authorization'] = header
23 get :health, params: {check: 'ping'}
24 assert_response error_code
26 resp = JSON.parse(@response.body)
28 assert_equal(JSON.load('{"health":"OK"}'), resp)
30 assert_equal(error_msg, resp['errors'])
36 mtime = File.mtime(ENV["ARVADOS_CONFIG"])
37 hash = Digest::SHA256.hexdigest(File.read(ENV["ARVADOS_CONFIG"]))
38 Rails.configuration.ManagementToken = "configuredmanagementtoken"
39 @request.headers['Authorization'] = "Bearer configuredmanagementtoken"
41 assert_response :success
42 assert_equal 'text/plain', @response.content_type
44 assert_match /\narvados_config_source_timestamp_seconds{sha256="#{hash}"} #{Regexp.escape mtime.utc.to_f.to_s}\n/, @response.body
46 # Expect mtime < loadtime < now
47 m = @response.body.match(/\narvados_config_load_timestamp_seconds{sha256="#{hash}"} (.*?)\n/)
48 assert_operator m[1].to_f, :>, mtime.utc.to_f
49 assert_operator m[1].to_f, :<, Time.now.utc.to_f
51 assert_match /\narvados_version_running{version="#{Regexp.escape AppVersion.package_version}"} 1\n/, @response.body
54 test "metrics disabled" do
55 Rails.configuration.ManagementToken = ""
56 @request.headers['Authorization'] = "Bearer configuredmanagementtoken"
61 test "metrics bad token" do
62 Rails.configuration.ManagementToken = "configuredmanagementtoken"
63 @request.headers['Authorization'] = "Bearer asdf"
68 test "metrics unauthorized" do
69 Rails.configuration.ManagementToken = "configuredmanagementtoken"