services/api/test/unit/api_client_test.rb

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: AGPL-3.0
   4
   5 require 'test_helper'
   6
   7 class ApiClientTest < ActiveSupport::TestCase
   8   include CurrentApiClient
   9
  10   [true, false].each do |token_lifetime_enabled|
  11     test "configured workbench is trusted when token lifetime is#{token_lifetime_enabled ? '': ' not'} enabled" do
  12       Rails.configuration.Login.TokenLifetime = token_lifetime_enabled ? 8.hours : 0
  13       Rails.configuration.Services.Workbench1.ExternalURL = URI("http://wb1.example.com")
  14       Rails.configuration.Services.Workbench2.ExternalURL = URI("https://wb2.example.com:443")
  15
  16       act_as_system_user do
  17         [["http://wb0.example.com", false],
  18         ["http://wb1.example.com", true],
  19         ["http://wb2.example.com", false],
  20         ["https://wb2.example.com", true],
  21         ["https://wb2.example.com/", true],
  22         ].each do |pfx, result|
  23           a = ApiClient.create(url_prefix: pfx, is_trusted: false)
  24           if token_lifetime_enabled
  25             assert_equal false, a.is_trusted, "API client with url prefix '#{pfx}' shouldn't be trusted"
  26           else
  27             assert_equal result, a.is_trusted
  28           end
  29         end
  30
  31         a = ApiClient.create(url_prefix: "http://example.com", is_trusted: true)
  32         a.save!
  33         a.reload
  34         assert a.is_trusted
  35       end
  36     end
  37   end
  38 end