Closes issue #20

thoughtbot · Sep 9, 2011 · ba8a264 · ba8a264
1 parent 62e9202
commit ba8a264
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 5 deletions.
diff --git a/lib/kumade/deployer.rb b/lib/kumade/deployer.rb
@@ -14,11 +14,15 @@ def initialize(environment = 'staging', pretending = false)
     end
 
     def deploy
-      ensure_heroku_remote_exists
-      pre_deploy
-      sync_heroku
-      heroku_migrate
-      post_deploy
+      begin
+        ensure_heroku_remote_exists
+        pre_deploy
+        sync_heroku
+        heroku_migrate
+        post_deploy
+      rescue
+        post_deploy
+      end
     end
 
     def pre_deploy

diff --git a/spec/kumade/deployer_spec.rb b/spec/kumade/deployer_spec.rb
@@ -57,6 +57,17 @@
 
     subject.deploy
   end
+
+  it "should call post_deploy if deploy fails" do
+    subject.stub(:run => true)
+
+    subject.should_receive(:ensure_heroku_remote_exists).and_raise(RuntimeError)
+
+    subject.should_receive(:post_deploy)
+
+    subject.deploy
+  end
+
 end
 
 describe Kumade::Deployer, "#sync_github" do