require 'puppetlabs_spec_helper/rake_tasks' require 'rake' task 'beaker:test',[:host,:type] => :set_beaker_variables do |t,args| Rake::Task['beaker-rspec:test'].invoke(args) if File.exists?('./acceptance') Dir.chdir('./acceptance') exec(build_beaker_command args) Dir.chdir('../') else puts "No acceptance directory found, not running beaker tests" end end desc "Run beaker rspec tasks against pe" RSpec::Core::RakeTask.new('beaker-rspec:test',[:host,:type]=>:set_beaker_variables) do |t,args| t.pattern = 'spec/acceptance' t.rspec_opts = '--color' t.verbose = true end desc "Run beaker and beaker-rspec tasks" task 'beaker:test:pe',:host do |t,args| args.with_defaults(:type=> 'pe') Rake::Task['beaker:test'].invoke(args[:host],args[:type]) end task 'beaker:test:git',:host do |t,args| args.with_defaults({:type=> 'git'}) Rake::Task['beaker:test'].invoke(args[:host],args[:type]) end task :set_beaker_variables do |t,args| puts 'Setting environment variables for testing' if args[:host] ENV['BEAKER_set'] = args[:host] puts "Host to test #{ENV['BEAKER_set']}" end ENV['BEAKER_IS_PE'] = args[:type] == 'pe'? "true": "false" end def build_beaker_command(args) cmd = ["beaker"] cmd << "--type #{args[:type]}" unless !args[:type] if File.exists?("./.beaker-#{args[:type]}.cfg") cmd << "--options-file ./.beaker-#{args[:type]}.cfg" end if File.exists?("config/#{args[:host]}.cfg") cmd << "--hosts config/#{args[:host]}.cfg" end if File.exists?("./tests") cmd << "--tests ./tests" end cmd.join(" ") end