(maint) Add ability to switch between puppet4 and puppet5 repos

Author: Zak-Kent

spec/spec_helper_acceptance.rb

@@ -2,34 +2,67 @@
 require 'beaker/puppet_install_helper'
 require 'beaker/module_install_helper'
 
+def use_puppet5
+  env_var = 'PUPPET_INSTALL_VERSION'
+  ((ENV[env_var] =~ %r{2018}) || !ENV.include?(env_var)) ? true : false
+end
+
+def build_url(platform)
+  use_pup5 = use_puppet5
+  if use_pup5
+    url5 = 'http://%{mngr}.puppetlabs.com/%{dir}puppet5-release%{plat}'
+    case platform
+    when 'el' then url5 % { mngr: 'yum', dir: 'puppet5/', plat: '-el-' }
+    when 'fedora' then url5 % { mngr: 'yum', dir: 'puppet5/', plat: '-fedora-' }
+    when 'debian', 'ubuntu' then url5 % { mngr: 'apt', dir: '', plat: '-' }
+    else
+      raise "build_url() called with unsupported platform '#{platform}'"
+    end
+  else
+    url4 = 'http://%{mngr}.puppetlabs.com/puppetlabs-release-pc1%{plat}'
+    case platform
+    when 'el' then url4 % { mngr: 'yum', plat: '-el-' }
+    when 'fedora' then url4 % { mngr: 'yum', plat: '-fedora-' }
+    when 'debian', 'ubuntu' then url4 % { mngr: 'apt', plat: '-' }
+    else
+      raise "build_url() called with unsupported platform '#{platform}'"
+    end
+  end
+end
+
 hosts.each do |host|
   if host['platform'] =~ %r{debian}
     on host, 'echo \'export PATH=/var/lib/gems/1.8/bin/:${PATH}\' >> ~/.bashrc'
   end
   # install_puppet
   if host['platform'] =~ %r{el-(5|6|7)}
     relver = Regexp.last_match(1)
-    on host, "rpm -ivh http://yum.puppetlabs.com/puppet5/puppet5-release-el-#{relver}.noarch.rpm"
+    on host, "rpm -ivh #{build_url('el')}#{relver}.noarch.rpm"
     on host, 'yum install -y puppetserver'
   elsif host['platform'] =~ %r{fedora-(\d+)}
     relver = Regexp.last_match(1)
-    on host, "rpm -ivh http://yum.puppetlabs.com/puppet5/puppet5-release-fedora-#{relver}.noarch.rpm"
+    on host, "rpm -ivh #{build_url('fedora')}#{relver}.noarch.rpm"
     on host, 'yum install -y puppetserver'
   elsif host['platform'] =~ %r{(ubuntu|debian)}
     unless host.check_for_package 'curl'
       on host, 'apt-get install -y curl'
     end
     # For openjdk8
-    if host['platform'].version == '8'
+    use_pup5 = use_puppet5
+    if host['platform'].version == '8' && use_pup5
       create_remote_file(host,
                          '/etc/apt/sources.list.d/jessie-backports.list',
                          'deb http://httpredir.debian.org/debian jessie-backports main')
       on host, 'apt-get -y -m update'
       on host, 'apt-get install -y -t jessie-backports openjdk-8-jre-headless'
     end
     on host, 'apt-get install apt-transport-https --assume-yes'
-    on host, 'curl -O http://apt.puppetlabs.com/puppet5-release-$(lsb_release -c -s).deb'
-    on host, 'dpkg -i puppet5-release-$(lsb_release -c -s).deb'
+    on host, "curl -O #{build_url('debian')}$(lsb_release -c -s).deb"
+    if use_pup5
+      on host, 'dpkg -i puppet5-release-$(lsb_release -c -s).deb'
+    else
+      on host, 'dpkg -i puppetlabs-release-pc1-$(lsb_release -c -s).deb'
+    end
     on host, 'apt-get -y -m update'
     on host, 'apt-get install -y puppetserver'
   else