Project

General

Profile

Download (7.44 KB) Statistics
| Branch: | Tag: | Revision:
require 'rubygems'

ENV["RAILS_ENV"] = "test"
require 'minitest/mock'
require File.expand_path('../config/environment', __dir__)
require 'rails/test_help'
require 'mocha/minitest'
require 'factory_bot_rails'
require 'controllers/shared/basic_rest_response_test'
require 'facet_test_helper'
require 'active_support_test_case_helper'
require 'fact_importer_test_helper'
require 'rfauxfactory'
require 'webmock/minitest'
require 'webmock'
require 'robottelo/reporter/attributes'
require 'test_report_helper'

# FactoryBot 5 changed the default to 'true'
FactoryBot.use_parent_strategy = false

# Do not allow network connections and external processes
WebMock.disable_net_connect!(allow_localhost: true)

# Configure shoulda
Shoulda::Matchers.configure do |config|
config.integrate do |with|
with.test_framework :minitest
with.library :rails
end
end

Mocha.configure do |config|
config.strict_keyword_argument_matching = true
end

# Use our custom test runner, and register a fake plugin to skip a specific test
Foreman::Plugin.register :skip_test do
tests_to_skip "CustomRunnerTest" => ["custom runner is working"]
end

# Turn of Apipie validation for tests
Apipie.configuration.validate = false

# To prevent Postgres' errors "permission denied: "RI_ConstraintTrigger"
if ActiveRecord::Base.connection.adapter_name == 'PostgreSQL'
ActiveRecord::Migration.execute "SET CONSTRAINTS ALL DEFERRED;"
end

# List of valid record name field.
def valid_name_list
[
RFauxFactory.gen_alpha(1),
RFauxFactory.gen_alpha(255),
*RFauxFactory.gen_strings(1..255, exclude: [:html]).values.map { |x| x.truncate_bytes(255, omission: '') },
RFauxFactory.gen_html(rand((1..230))),
]
end

# List of invalid record name field .
def invalid_name_list
[
'',
' ',
' ',
"\t",
]
end

module ReactjsHelper
def read_webpack_manifest
manifest = Rails.root.join('public/webpack/manifest.json')
if manifest.exist?
JSON.parse(manifest.read)
else
# Stubbed method to deal with a missing manifest for controller tests
{"assetsByChunkName" => {"foreman-vendor" => ["foreman-vendor.js", "foreman-vendor.css"]}}
end
end
end

module TestCaseRailsLoggerExtensions
def before_setup
super
ensure
@_ext_current_buffer = StringIO.new
@_ext_old_logger = Rails.logger
@_ext_old_ar_logger = ActiveRecord::Base.logger
Rails.logger = Foreman::SilencedLogger.new(ActiveSupport::TaggedLogging.new(Logger.new(@_ext_current_buffer)))
ActiveRecord::Base.logger = Rails.logger if ENV['PRINT_TEST_LOGS_SQL']
end

def after_teardown
Rails.logger = @_ext_old_logger if @_ext_old_logger
ActiveRecord::Base.logger = @_ext_old_ar_logger if @_ext_old_ar_logger
if (ENV['PRINT_TEST_LOGS_ON_ERROR'] && error?) || (ENV['PRINT_TEST_LOGS_ON_FAILURE'] && !passed?)
@_ext_current_buffer.close_write
$stdout << "\n\nRails logs for #{name} FAILURE:\n"
$stdout << @_ext_current_buffer.string
end
super
ensure
@_ext_current_buffer&.close
@_ext_current_buffer = nil
end
end

class ActiveSupport::TestCase
extend Robottelo::Reporter::TestAttributes
prepend TestCaseRailsLoggerExtensions
setup :setup_dns_stubs

class << self
alias_method :test, :it
end

def setup_dns_stubs
Resolv::DNS.any_instance.stubs(:getname).raises(Resolv::ResolvError, "DNS must be stub: Resolv::DNS.any_instance.stubs(:getname).returns('example.com')")
Resolv::DNS.any_instance.stubs(:getnames).raises(Resolv::ResolvError, "DNS must be stub: Resolv::DNS.any_instance.stubs(:getnames).returns(['example.com'])")
Resolv::DNS.any_instance.stubs(:getaddress).raises(Resolv::ResolvError, "DNS must be stub: Resolv::DNS.any_instance.stubs(:getaddress).returns('127.0.0.15')")
Resolv::DNS.any_instance.stubs(:getaddresses).raises(Resolv::ResolvError, "DNS must be stub: Resolv::DNS.any_instance.stubs(:getaddresses).returns(['127.0.0.15'])")
end

def clear_plugins
@clear_plugins = true
@plugins_backup = Foreman::Plugin.registered_plugins
@registries_backup = Foreman::Plugin.registries
Foreman::Plugin.send(:clear)
end

def after_teardown
super

return unless @clear_plugins
Foreman::Plugin.send(:clear, @plugins_backup, @registries_backup)
@clear_plugins = nil
end

def assert_sql_queries(num_of_queries, match = /SELECT/)
queries = []
ActiveSupport::Notifications.subscribe("sql.active_record") do |_name, _start, _finish, _id, payload|
queries << payload[:sql] if payload[:sql] =~ match
end
yield
ActiveSupport::Notifications.unsubscribe("sql.active_record")
assert_equal num_of_queries, queries.size, "Expected #{num_of_queries} queries, but got #{queries.size}"
end

def assert_equal_arrays(array1, array2)
assert_equal array1.sort, array2.sort
end
end

class ActionView::TestCase
helper Rails.application.routes.url_helpers
end

::Rails::Engine.subclasses.map(&:instance).each do |engine|
support_file = "#{engine.root}/test/support/foreman_test_helper_additions.rb"
require support_file if File.exist?(support_file)
end

class ActionController::TestCase
extend Robottelo::Reporter::TestAttributes
include ::BasicRestResponseTest
setup :setup_set_script_name, :set_api_user, :turn_off_login, :set_admin

class << self
alias_method :test, :it
end

def set_admin
User.current = users(:admin)
end

def turn_off_login
SETTINGS[:require_ssl] = false
end

def setup_set_script_name
@request.env["SCRIPT_NAME"] = @controller.config.relative_url_root
end

def set_api_user
return unless self.class.to_s[/api/i]
set_basic_auth(users(:apiadmin), "secret")
end

def reset_api_credentials
@request.env.delete('HTTP_AUTHORIZATION')
end

def set_basic_auth(user, password)
login = user.is_a?(User) ? user.login : user
@request.env['HTTP_AUTHORIZATION'] = ActionController::HttpAuthentication::Basic.encode_credentials(login, password)
@request.env['CONTENT_TYPE'] = 'application/json'
@request.env['HTTP_ACCEPT'] = 'application/json'
end

def with_temporary_settings(**kwargs)
old_settings = SETTINGS.slice(*kwargs.keys)
begin
SETTINGS.update(kwargs)

yield
ensure
SETTINGS.update(old_settings)
end
end
end

class GraphQLQueryTestCase < ActiveSupport::TestCase
let(:variables) { {} }
let(:context_user) { FactoryBot.create(:user, :admin) }
let(:context) { { current_user: context_user } }
let(:result) { ForemanGraphqlSchema.execute(query, context: context, variables: variables) }

def assert_record(expected, actual, type_name: nil)
assert_not_nil expected
assert_equal Foreman::GlobalId.for(expected), actual['id']
end

def assert_collection(expected, actual, type_name: nil)
assert expected.any?, 'The expected records array can not be empty to assert_collection'
assert_equal expected.count, actual['totalCount']

expected_global_ids = expected.map { |r| Foreman::GlobalId.for(r) }
actual_global_ids = actual['edges'].map { |e| e['node']['id'] }

assert_same_elements expected_global_ids, actual_global_ids
end
end

def with_auditing(klass)
auditing_was_enabled = klass.auditing_enabled
klass.enable_auditing
yield
ensure
klass.disable_auditing unless auditing_was_enabled
end

def json_response
ActiveSupport::JSON.decode(response.body)
end

def json_data(key)
data = json_response.fetch('data', {})
data.fetch(key, {})
end

def json_errors
json_response.fetch('errors', [])
end

def json_error_messages
json_errors.map { |e| e.fetch('message') }
end
(8-8/9)