diff options
| -rw-r--r-- | lib/bacon.rb | 4 | ||||
| -rw-r--r-- | test/spec_bacon.rb | 1 |
2 files changed, 4 insertions, 1 deletions
diff --git a/lib/bacon.rb b/lib/bacon.rb index 256cd0c..232b2df 100644 --- 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 index b3981ee..a0eeffd 100644 --- a/test/spec_bacon.rb +++ b/test/spec_bacon.rb @@ -299,6 +299,7 @@ describe "before/after" do it "should run in the right order" do @a.should.equal 5 + @a = 2 end end |