Rakefile

require 'rubygems'
require 'puppetlabs_spec_helper/rake_tasks'

def io_popen(command)
  IO.popen(command) do |io|
    io.each do |line|
      print line
      yield line if block_given?
    end
  end
end

# Customize lint option
task :lint do
  PuppetLint.configuration.send("disable_80chars")
  PuppetLint.configuration.send("disable_class_parameter_defaults")
end

# Initialize vagrant instance for testing
desc "Powers on Vagrant VMs with specific manifests"
task :vagrant, :manifest do |t, args|
  Rake::Task["spec_prep"].execute

  prefix = "VAGRANT_MANIFEST='#{args[:manifest]||'init.pp'}'"

  puts args[:manifest]
  provision = false
  io_popen("export #{prefix}; vagrant up --provider=vmware_fusion") do |line|
    provision = true if line =~ /Machine is already running./
  end
  io_popen("export #{prefix}; vagrant provision") if provision
end

# Cleanup vagrant environment
desc "Destroys Vagrant VMs and cleanup spec directory"
task :vagrant_clean do
  `vagrant destroy -f`
  Rake::Task["spec_clean"].execute
end