Make sure the task is actually ran before running the after callback

judofyr · Oct 17, 2009 · 8a8d6ab · 8a8d6ab
1 parent fa88025
commit 8a8d6ab
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/extras/rdoctask.rb b/extras/rdoctask.rb
@@ -8,6 +8,10 @@ def after_running_rdoc(&block)
     end
 
     def define
+      pre = @before_running_rdoc
+      ran = false
+      @before_running_rdoc = proc { ran = true; pre.call if pre }
+
       super
       return unless after = @after_running_rdoc
 
@@ -18,7 +22,7 @@ def define
         begin
           target.invoke
         ensure
-          after.call
+          after.call if ran
         end
       end
     end