lib/heroku_release.rb in heroku_release-0.1.2 vs lib/heroku_release.rb in heroku_release-0.1.3

@@ -68,20 +68,24 @@
       previous = previous_release_version
       if previous
         output "Rolling back to '#{previous}' ..."
         execute "git push -f #{config.heroku_remote} #{previous}:master"
         output "Deleting rollbacked release '#{current}' ..."
-        execute "git tag -d #{current}"
-        execute "git push #{config.heroku_remote} :refs/tags/#{current}"
-        execute "git push origin :refs/tags/#{current}"
+        remove_tag(current)
         output 'Rollback completed'
       else
         output "No release tags found - cannot do rollback"
         output releases
       end
     end
 
     private
+
+    def remove_tag(tag_name)
+      execute "git tag -d #{tag_name}"
+      execute "git push #{config.heroku_remote} :refs/tags/#{tag_name}"
+      execute "git push origin :refs/tags/#{tag_name}"
+    end
 
     def tag_comment
       return ENV['COMMENT'] if ENV['COMMENT']      
       if config.prompt_for_comments
         print "Required - please enter a release comment: "