diff options
Diffstat (limited to 'lib/bacon.rb')
-rw-r--r-- | lib/bacon.rb | 4 |
1 files changed, 3 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 |