Rakefile

   1 require 'puppet_blacksmith/rake_tasks'
   2 require 'puppet-lint/tasks/puppet-lint'
   3 require 'puppetlabs_spec_helper/rake_tasks'
   4
   5 PuppetLint.configuration.send('relative')
   6 PuppetLint.configuration.send('disable_documentation')
   7 PuppetLint.configuration.send('disable_single_quote_string_with_variables')
   8
   9 desc 'Generate pooler nodesets'
  10 task :gen_nodeset do
  11   require 'beaker-hostgenerator'
  12   require 'securerandom'
  13   require 'fileutils'
  14
  15   agent_target = ENV['TEST_TARGET']
  16   if ! agent_target
  17     STDERR.puts 'TEST_TARGET environment variable is not set'
  18     STDERR.puts 'setting to default value of "redhat-64default."'
  19     agent_target = 'redhat-64default.'
  20   end
  21
  22   master_target = ENV['MASTER_TEST_TARGET']
  23   if ! master_target
  24     STDERR.puts 'MASTER_TEST_TARGET environment variable is not set'
  25     STDERR.puts 'setting to default value of "redhat7-64mdcl"'
  26     master_target = 'redhat7-64mdcl'
  27   end
  28
  29   targets = "#{master_target}-#{agent_target}"
  30   cli = BeakerHostGenerator::CLI.new([targets])
  31   nodeset_dir = "tmp/nodesets"
  32   nodeset = "#{nodeset_dir}/#{targets}-#{SecureRandom.uuid}.yaml"
  33   FileUtils.mkdir_p(nodeset_dir)
  34   File.open(nodeset, 'w') do |fh|
  35     fh.print(cli.execute)
  36   end
  37   puts nodeset
  38 end