upgrade_test_cluster: Download custom PE archive before calling peadm

Author: bastelfreak

spec/acceptance/peadm_spec/plans/upgrade_test_cluster.pp

@@ -16,12 +16,10 @@
   $common_params = {
     download_mode          => $download_mode,
     version                => $version,
-    pe_installer_source    => $pe_installer_source,
     permit_unsafe_versions => $permit_unsafe_versions,
   }
 
-  $arch_params =
-    case $architecture {
+  $arch_params = case $architecture {
     'standard': {{
         primary_host => $t.filter |$n| { $n.vars['role'] == 'primary' },
     } }
@@ -53,6 +51,25 @@
     default: { fail('Invalid architecture!') }
   }
 
+  if $pe_installer_source {
+    $platform            = run_task('peadm::precheck', $arch_params['primary_host']).first['platform']
+    $pe_tarball_name     = "puppet-enterprise-${version}-${platform}.tar.gz"
+    $upload_tarball_path = "/tmp/${pe_tarball_name}"
+
+    if $download_mode == 'bolthost' {
+      run_plan('peadm::util::retrieve_and_upload', $arch_params['primary_host'],
+        source      => $pe_installer_source,
+        local_path  => "/tmp/${pe_tarball_name}",
+        upload_path => $upload_tarball_path,
+      )
+    } else {
+      run_task('peadm::download', $arch_params['primary_host'],
+        source => $pe_installer_source,
+        path   => $upload_tarball_path,
+      )
+    }
+  }
+
   $params = $arch_params + $common_params
   run_plan('peadm::upgrade', $params)
 }