|
desc 'Snapshots tasks'
|
|
namespace :snapshots do
|
|
desc "Generate snapshots"
|
|
task :generate => :environment do
|
|
unless Rails.env.test?
|
|
puts "This task can only be run in test environment"
|
|
exit
|
|
end
|
|
|
|
require 'database_cleaner'
|
|
require 'factory_bot_rails'
|
|
|
|
# don't advertise any plugins to prevent different results
|
|
module PluginSnapshotStub
|
|
def find(id)
|
|
nil
|
|
end
|
|
end
|
|
|
|
module BaseMacrosStub
|
|
def dns_lookup(_name_or_ip)
|
|
'foreman.example.com'
|
|
end
|
|
end
|
|
|
|
::Foreman::Plugin.singleton_class.send :prepend, PluginSnapshotStub
|
|
::Foreman::Renderer::Scope::Base.prepend BaseMacrosStub
|
|
|
|
# clean the snapshot directory in order to delete renamed ones and keep it clean
|
|
FileUtils.rm_rf(Dir.glob(File.join(
|
|
::Foreman::Renderer::Source::Snapshot::SNAPSHOTS_DIRECTORY, '*')))
|
|
|
|
DatabaseCleaner.cleaning do
|
|
Foreman.settings.load
|
|
Setting[:unattended_url] = "http://foreman.example.com"
|
|
Setting[:foreman_url] = "http://foreman.example.com"
|
|
Setting[:ct_location] = "/usr/bin/cat"
|
|
Setting[:ct_arguments] = []
|
|
|
|
User.current = FactoryBot.build(:user, :admin)
|
|
admin = FactoryBot.create(:user, :admin, password: 'password123', auth_source: FactoryBot.create(:auth_source_ldap))
|
|
|
|
User.as(admin.login) do
|
|
Foreman::TemplateSnapshotService.templates.each do |template|
|
|
Foreman::Renderer::Source::Snapshot.hosts(template).each do |host|
|
|
snapshot_path = Foreman::Renderer::Source::Snapshot.snapshot_path(template, host)
|
|
dir = File.dirname(snapshot_path)
|
|
FileUtils.mkdir_p(dir) unless File.directory?(dir)
|
|
|
|
snapshot = Foreman::TemplateSnapshotService.render_template(template, host)
|
|
if snapshot =~ /^#cloud-config/
|
|
puts "Validating YAML #{snapshot_path}"
|
|
YAML.safe_load(snapshot)
|
|
end
|
|
puts "Writing #{snapshot_path}"
|
|
File.write(snapshot_path, snapshot)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
