From: Ashley Penney <ashley.penney@puppetlabs.com>
Date: Wed, 19 Feb 2014 19:54:11 +0000 (+0000)
Subject: Add PE support.
X-Git-Tag: 1.0.1~10^2^2
X-Git-Url: https://review.fuel-infra.org/gitweb?a=commitdiff_plain;h=318f3fdae4ececfef62f5c8b27624922859664f6;p=puppet-modules%2Fpuppetlabs-firewall.git

Add PE support.
---

diff --git a/spec/spec_helper_acceptance.rb b/spec/spec_helper_acceptance.rb
index afa3f05..afd6c5c 100644
--- a/spec/spec_helper_acceptance.rb
+++ b/spec/spec_helper_acceptance.rb
@@ -12,11 +12,17 @@ def ip6tables_flush_all_tables
   end
 end
 
-hosts.each do |host|
-  # Install Puppet
-  install_package host, 'rubygems'
-  on host, 'gem install puppet --no-ri --no-rdoc'
-  on host, "mkdir -p #{host['distmoduledir']}"
+unless ENV['RS_PROVISION'] == 'no'
+  hosts.each do |host|
+    # Install Puppet
+    if host.is_pe?
+      install_pe
+    else
+      install_package host, 'rubygems'
+      on host, 'gem install puppet --no-ri --no-rdoc'
+      on host, "mkdir -p #{host['distmoduledir']}"
+    end
+  end
 end
 
 RSpec.configure do |c|