lib/caploy/recipes/defaults.rb in caploy-0.1.9 vs lib/caploy/recipes/defaults.rb in caploy-0.1.10

@@ -40,19 +40,24 @@
 
     task :set_branch_info_file, :roles => :app do
       run "cd #{release_path} && echo \"#{branch}\" > CURRENT_BRANCH"
     end
 
+    task :set_staging_release_test_info_file, :roles => :app do
+      run "cd #{release_path} && echo \"#{branch}\" > RELEASE_TEST" if branch == 'master' and ENV['RELEASE_TEST']
+    end
+
     desc "build missing paperclip styles"
     task :build_missing_paperclip_styles, :roles => :app do
       run "cd #{current_path}; RAILS_ENV=#{rails_env} bundle exec rake paperclip:refresh:missing_styles"
     end
 
     desc 'Show deployed revision'
     task :revision, :roles => :app do
       run "cat #{current_path}/REVISION"
     end
 
+    after 'deploy:finalize_update', 'deploy:set_branch_info_file'
   end
 
   task :test_and_prepare_cap_env do
     abort "You must set :user before using defaults" unless fetch(:user, nil)
     abort "You must set :repository before using defaults" unless fetch(:repository, nil)