-require 'puppet_blacksmith/rake_tasks'
-require 'puppet-lint/tasks/puppet-lint'
require 'puppetlabs_spec_helper/rake_tasks'
+require 'puppet-syntax/tasks/puppet-syntax'
+require 'puppet_blacksmith/rake_tasks' if Bundler.rubygems.find_name('puppet-blacksmith').any?
-PuppetLint.configuration.send('relative')
-PuppetLint.configuration.send('disable_documentation')
-PuppetLint.configuration.send('disable_single_quote_string_with_variables')
+PuppetLint.configuration.send('disable_relative')
-desc 'Generate pooler nodesets'
-task :gen_nodeset do
- require 'beaker-hostgenerator'
- require 'securerandom'
- require 'fileutils'
-
- agent_target = ENV['TEST_TARGET']
- if ! agent_target
- STDERR.puts 'TEST_TARGET environment variable is not set'
- STDERR.puts 'setting to default value of "redhat-64default."'
- agent_target = 'redhat-64default.'
- end
-
- master_target = ENV['MASTER_TEST_TARGET']
- if ! master_target
- STDERR.puts 'MASTER_TEST_TARGET environment variable is not set'
- STDERR.puts 'setting to default value of "redhat7-64mdcl"'
- master_target = 'redhat7-64mdcl'
- end
-
- targets = "#{master_target}-#{agent_target}"
- cli = BeakerHostGenerator::CLI.new([targets])
- nodeset_dir = "tmp/nodesets"
- nodeset = "#{nodeset_dir}/#{targets}-#{SecureRandom.uuid}.yaml"
- FileUtils.mkdir_p(nodeset_dir)
- File.open(nodeset, 'w') do |fh|
- fh.print(cli.execute)
- end
- puts nodeset
-end