ext/packaging/tasks/pe_rpm.rake in puppet-2.7.20 vs ext/packaging/tasks/pe_rpm.rake in puppet-2.7.21

@@ -1,19 +1,19 @@
-if @build_pe
+if @build.build_pe
   namespace :pe do
     desc "Build a PE rpm using rpmbuild (requires all BuildRequires, rpmbuild, etc)"
     task :local_rpm => "package:rpm"
 
     desc "Build rpms using ALL final mocks in build_defaults yaml, keyed to PL infrastructure, pass MOCK to override"
-    task :local_mock_final => ["pl:fetch", "pl:load_extras", "pl:mock_final"] do
-      if @team == 'release'
+    task :local_mock_all => ["pl:fetch", "pl:mock_all"] do
+      if @build.team == 'release'
         Rake::Task["pe:sign_rpms"].invoke
       end
     end
 
     desc "Build a PE rpm using the default mock"
-    task :local_mock => ["pl:fetch", "pl:load_extras", "pl:mock" ] do
-      if @team == 'release'
+    task :local_mock => ["pl:fetch", "pl:mock"] do
+      if @build.team == 'release'
         Rake::Task["pe:sign_rpms"].invoke
       end
     end
   end
 end