spec_helper.rb snippet for custom environments in rspec-puppet tests.

spec_helper_snippet.rb

require 'puppetlabs_spec_helper/module_spec_helper'

def set_environment(environment = :production)
  RSpec.configure { |config| config.default_facts['environment'] = environment.to_s }
end

RSpec.configure do |config|
  config.before(:each) do
    if defined?(environment)
      set_environment(environment)
    end
  end
end