Having higher-level after blocks also be run for nested contexts.

jeffkreeftmeijer · Oct 24, 2008 · 072975b · 072975b
1 parent 6c8e263
commit 072975b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lib/bacon.rb b/lib/bacon.rb
@@ -258,7 +258,9 @@ module Kernel
   private
   def describe(*args, &block)
     context = Bacon::Context.new(args.join(' '), &block)
-    (instance_variable_get('@before') || []).each { |b|  context.before &b }
+    %w[before after].each do |block_type|
+      (instance_variable_get("@#{block_type}") || []).each { |b|  context.send(block_type, &b) }
+    end
     context.run
     context
   end

diff --git a/test/spec_bacon.rb b/test/spec_bacon.rb
@@ -299,6 +299,7 @@ def equal_string(x)
 
     it "should run in the right order" do
       @a.should.equal 5
+      @a = 2
     end
   end