before_all and after_all accept parameter or proc

1 files changed, 4 insertions, 3 deletions

diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
index 9369519..262e966 100644
--- a/spec/spec_helper.rb
+++ b/spec/spec_helper.rb
@@ -103,13 +103,14 @@ class Node3 < Zorglub::Node
     @after=0
     class << self
         attr_accessor :before, :after
+        def post obj
+            @after +=1
+        end
     end
     before_all do |node|
         Node3.before +=1
     end
-    after_all do |node|
-        Node3.after +=1
-    end
+    after_all Node3.method(:post)
     layout! 'layout-2'
     engine! 'engine-2'
     def index