Creating test puppet facts

test_facts.rb

require 'facter'

hostname=Facter['hostname'].value

all_servers = ['denver', 'indy', 'sandiego', 'chicago', 'saltlake', 'nashville']

Facter.add('my_env') do

  dev = ['denver', 'indy', 'sandiego']
  prod = ['chicago', 'saltlake', 'nashville']

  setcode do
    case hostname
    when *dev
      'dev'
    when *prod
      'prod'
    else
      'unknown'
    end
  end

end