result_set.rb in reportable-1.4.0

- old
+ new

@@ -2,19 +2,20 @@
 
   module Reportable
 
     # A result set as it is returned by the report methods.
     # This is basically a subclass of +Array+ that adds two
-    # attributes, +model_name+ and +report_name+ that store
+    # attributes, +model_class_name+ and +report_name+ that store
     # the name of the model and the report the result set
     # was generated from.
     #
     class ResultSet
+      include Enumerable
 
       # the name of the model the result set is based on
       #
-      attr_reader :model_name
+      attr_reader :model_class_name
 
       # the name of the report the result is based on
       #
       attr_reader :report_name
 
@@ -26,18 +27,24 @@
 
       # Initializes a new result set.
       #
       # @param [Array] array
       #   the array that is the actual result
-      # @param [String] model_name
+      # @param [String] model_class_name
       #   the name of the model the result set is based on
       # @param [String] report_name
       #   the name of the report the result is based on
       #
-      def initialize(array, model_name, report_name)
+      def initialize(array, model_class_name, report_name)
         @results = array
-        @model_name  = model_name
+        @model_class_name  = model_class_name
         @report_name = report_name.to_s
+      end
+
+      # quack like an Enumerable
+      #
+      def each(&block)
+        @results.each(&block)
       end
 
     end
 
   end