lib/innate/helper/aspect.rb in manveru-innate-2009.04.18 vs lib/innate/helper/aspect.rb in manveru-innate-2009.05

@@ -92,31 +92,31 @@
 
         def before_all(&block)
           AOP[self][:before_all] = block
         end
 
-        def before(name, &block)
-          AOP[self][:before][name] = block
+        def before(*names, &block)
+          names.each{|name| AOP[self][:before][name] = block }
         end
 
         def after_all(&block)
           AOP[self][:after_all] = block
         end
 
-        def after(name, &block)
-          AOP[self][:after][name] = block
+        def after(*names, &block)
+          names.each{|name| AOP[self][:after][name] = block }
         end
 
-        def wrap(name, &block)
-          before(name, &block)
-          after(name, &block)
+        def wrap(*names, &block)
+          before(*names, &block)
+          after(*names, &block)
         end
 
         def add_action_wrapper(order, method_name)
           if wrap = trait[:wrap]
-            wrap.merge(SortedSet[[order, method_name]])
+            wrap.merge(SortedSet[[order, method_name.to_s]])
           else
-            trait :wrap => SortedSet[[order, method_name]]
+            trait :wrap => SortedSet[[order, method_name.to_s]]
           end
         end
       end
     end
   end