pe_utils.rb in beaker-pe-2.1.4

- old
+ new

@@ -4,10 +4,11 @@
 require 'beaker-pe/install/feature_flags'
 require "beaker-answers"
 require "timeout"
 require "json"
 require "deep_merge"
+
 module Beaker
   module DSL
     module InstallUtils
       #
       # This module contains methods to help installing/upgrading PE builds - including Higgs installs
@@ -1552,9 +1553,13 @@
 
           step "Finish install/upgrade on infrastructure" do
               [master, database, dashboard].uniq.each do |host|
                 execute_installer_cmd(host, opts)
               end
+          end
+
+          step "Stop agent service on infrastructure nodes" do
+            stop_agent_on(pe_infrastructure, :run_in_parallel => true)
           end
 
           step "First puppet run on infrastructure + postgres node" do
             [master, database, dashboard, pe_postgres].uniq.each do |host|
               on host, 'puppet agent -t', :acceptable_exit_codes => [0,2]