lib/lucid/ast/specs.rb in lucid-0.0.6 vs lib/lucid/ast/specs.rb in lucid-0.0.7

@@ -21,14 +21,19 @@
         @features << feature
       end
 
       def accept(visitor)
         return if Lucid.wants_to_quit
-        start = Time.now
-        self.each do |feature|
-          visitor.visit_feature(feature)
+
+        visitor.visit_features(self) do
+          start = Time.now
+
+          self.each do |feature|
+            feature.accept(visitor)
+          end
+
+          @duration = Time.now - start
         end
-        @duration = Time.now - start
       end
 
       def step_count
         @features.inject(0) { |total, feature| total += feature.step_count }
       end