Create Rakefile
This commit is contained in:
Abir Majumdar
GitHub
parent
e8bea34ed5
commit
9a88a3f890
209
Rakefile
Normal file
209
Rakefile
Normal file
@ -0,0 +1,209 @@
|
||||
require 'fileutils'
|
||||
require 'puppet-lint/tasks/puppet-lint'
|
||||
require 'puppetlabs_spec_helper/rake_tasks'
|
||||
require 'ra10ke'
|
||||
require 'r10k/puppetfile'
|
||||
require 'erb'
|
||||
require 'json'
|
||||
require 'rest-client'
|
||||
require 'onceover/rake_tasks'
|
||||
|
||||
PuppetSyntax.app_management = true
|
||||
PuppetSyntax.exclude_paths = ["site/**/plans/*"]
|
||||
PuppetLint.configuration.fail_on_warnings = true
|
||||
PuppetLint.configuration.send('relative')
|
||||
PuppetLint.configuration.send('disable_140chars')
|
||||
PuppetLint.configuration.send('disable_class_inherits_from_params_class')
|
||||
PuppetLint.configuration.send('disable_documentation')
|
||||
PuppetLint.configuration.ignore_paths = ["spec/**/*.pp", "pkg/**/*.pp", "bundle/**/*", "vendor/**/*"]
|
||||
|
||||
|
||||
Rake::Task[:spec_prep].enhance [:generate_fixtures]
|
||||
|
||||
desc "Run tests"
|
||||
task :run_tests do
|
||||
print "Executing Lint Test...\n"
|
||||
Rake::Task[:lint].execute
|
||||
print " -> Success!\n\n"
|
||||
|
||||
print "Executing Syntax Test...\n"
|
||||
Rake::Task[:syntax].execute
|
||||
print " -> Success!\n\n"
|
||||
|
||||
print "Executing r10k(Puppetfile) Syntax Test...\n -> "
|
||||
Rake::Task['r10k:syntax'].execute
|
||||
print "\n"
|
||||
|
||||
print "Checking for missing spec tests...\n"
|
||||
Rake::Task[:check_for_spec_tests].execute
|
||||
print " -> No missing tests!\n\n"
|
||||
|
||||
print "Launching rspec tests...\n"
|
||||
Rake::Task[:spec].execute
|
||||
end
|
||||
|
||||
desc "Generate Fixtures files for role/profile"
|
||||
task :generate_fixtures do
|
||||
print "Generating Fixtures..."
|
||||
build_fixtures(File.dirname(__FILE__))
|
||||
print "Done!\n"
|
||||
end
|
||||
|
||||
desc "Generate spec tests for missing classes"
|
||||
task :generate_spec_tests do
|
||||
spec_gen(true)
|
||||
end
|
||||
|
||||
desc "Get spec test status"
|
||||
task :check_for_spec_tests do
|
||||
spec_gen
|
||||
end
|
||||
|
||||
desc "Show PE Only Modules"
|
||||
task :pe_only_mods do
|
||||
puts get_pe_modules
|
||||
end
|
||||
|
||||
def get_pe_modules
|
||||
# Query Puppet Forge for the latest list of PE-only modules
|
||||
# Thanks to dan-wittenberg for the original logic on this!
|
||||
modules = {}
|
||||
|
||||
url="https://forgeapi.puppetlabs.com/v3/modules?module_groups=pe_only"
|
||||
r = RestClient.get url, { :accept => 'application/json', :charset => 'utf-8' }
|
||||
|
||||
JSON.parse(r.force_encoding("UTF-8"))['results'].each do |x|
|
||||
name = x['current_release']['metadata']['name'].gsub('/','-')
|
||||
modules[name] = "git@github.com:puppetlabs/#{name}.git"
|
||||
end
|
||||
|
||||
modules
|
||||
end
|
||||
|
||||
def spec_gen(create=false)
|
||||
exit_code = 0
|
||||
['role','profile'].each do |m|
|
||||
# For role or profile, find all the classes
|
||||
classes = Array.new
|
||||
|
||||
pattern = 'site/profile/manifests/*/*.pp' if m == 'profile'
|
||||
pattern = 'site/role/manifests/*/*.pp' if m == 'role'
|
||||
Dir.glob("#{pattern}").each do |f|
|
||||
File.open(f).read.each_line do |l|
|
||||
c = l.scan(/(\s+)?class\s+([a-zA-Z:_]+)\s+[\{,\(]/)
|
||||
# Add this class to the classes array
|
||||
classes.push(c[0][1]) if !c.empty?
|
||||
end
|
||||
end
|
||||
|
||||
# For each class, see if a spec file exists - using naming convention
|
||||
# <class>_<subclass>[_<subclass>_]_spec.rb
|
||||
classes.each do |c|
|
||||
spec_file = "#{File.dirname(__FILE__)}/spec/classes/#{m}/#{c.split('::').join('_')}_spec.rb"
|
||||
|
||||
# If no spec file exists, create a blank should compile test file
|
||||
if File.exists?(spec_file)
|
||||
puts "Class #{c} - Spec file already exists at #{spec_file}!" if create == true
|
||||
else
|
||||
if create == true
|
||||
puts "Class #{c} - Creating... #{spec_file}!"
|
||||
File.open(spec_file, 'w') do |f|
|
||||
f.write evaluate_template('spec_template.rb.erb',binding)
|
||||
end
|
||||
else
|
||||
puts "Class #{c} - Spec file missing!"
|
||||
exit_code = 1
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
if exit_code != 0
|
||||
raise(exit_code)
|
||||
end
|
||||
end
|
||||
|
||||
# Most of this logic was lifted from onceover (comments and all) - thank you!
|
||||
# https://github.com/dylanratcliffe/onceover/blob/98811bee7bf373e1a22706d98f9ccc1360aff482/lib/onceover/controlrepo.rb
|
||||
def evaluate_template(template_name,bind)
|
||||
template_dir = File.expand_path('./scripts',File.dirname(__FILE__))
|
||||
template = File.read(File.expand_path("./#{template_name}",template_dir))
|
||||
ERB.new(template, nil, '-').result(bind)
|
||||
end
|
||||
|
||||
def build_fixtures(controlrepo)
|
||||
# Load up the Puppetfile using R10k
|
||||
puppetfile = R10K::Puppetfile.new(controlrepo)
|
||||
fail 'Could not load Puppetfile' unless puppetfile.load
|
||||
modules = puppetfile.modules
|
||||
|
||||
# Store PE Only Mods list
|
||||
pe_only = get_pe_modules
|
||||
|
||||
# Iterate over everything and seperate it out for the sake of readability
|
||||
symlinks = []
|
||||
forge_modules = []
|
||||
repositories = []
|
||||
|
||||
modules.each do |mod|
|
||||
# This logic could probably be cleaned up. A lot.
|
||||
if mod.is_a? R10K::Module::Forge
|
||||
if mod.expected_version.is_a?(Hash)
|
||||
# Set it up as a symlink, because we are using local files in the Puppetfile
|
||||
symlinks << {
|
||||
'name' => mod.name,
|
||||
'dir' => mod.expected_version[:path]
|
||||
}
|
||||
elsif mod.expected_version.is_a?(String)
|
||||
|
||||
# Verify if this is a PE mod or not
|
||||
# if it is a PE only module; we need to set it up as a git repo for fixtures b/c of license issues
|
||||
if pe_only.keys.include?(mod.title.gsub('/','-'))
|
||||
# Its PE Only
|
||||
repositories << {
|
||||
'name' => mod.name,
|
||||
'repo' => mod.instance_variable_get(:@remote) =~ /\.git/ ? mod.instance_variable_get(:@remote) : pe_only[mod.title],
|
||||
# ^^ This isn't perfect, as some of the repo names don't match - but its a start
|
||||
'ref' => mod.expected_version
|
||||
}
|
||||
else
|
||||
# Set it up as a normal forge module
|
||||
forge_modules << {
|
||||
'name' => mod.name,
|
||||
'repo' => mod.title,
|
||||
'ref' => mod.expected_version
|
||||
}
|
||||
end
|
||||
|
||||
end
|
||||
elsif mod.is_a? R10K::Module::Git
|
||||
# Set it up as a git repo
|
||||
repositories << {
|
||||
'name' => mod.name,
|
||||
'repo' => mod.instance_variable_get(:@remote),
|
||||
'ref' => mod.version
|
||||
}
|
||||
end
|
||||
end
|
||||
|
||||
symlinks << {
|
||||
'name' => "profile",
|
||||
'dir' => '"#{source_dir}/site/profile"',
|
||||
}
|
||||
|
||||
symlinks << {
|
||||
'name' => "role",
|
||||
'dir' => '"#{source_dir}/site/role"',
|
||||
}
|
||||
|
||||
symlinks << {
|
||||
'name' => "manifests",
|
||||
'dir' => '"#{source_dir}/manifests"',
|
||||
}
|
||||
|
||||
File.open("#{File.dirname(__FILE__)}/.fixtures.yml",'w') do |f|
|
||||
f.write evaluate_template('fixtures.yml.erb',binding)
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user