feat: SerpParser::Collection

Author: kjellberg

lib/serp_parser/collection.rb

@@ -0,0 +1,18 @@
+require "forwardable"
+
+module SerpParser
+  class Collection
+    include Enumerable
+    extend Forwardable
+
+    def_delegators :@items, :size, :length, :[], :empty?, :last, :index
+
+    def initialize(items)
+      @items = items
+    end
+
+    def each(&block)
+      @items.each(&block)
+    end
+  end
+end

lib/serp_parser/parser.rb

@@ -47,7 +47,7 @@ def run_schema
     # @return [Array] The collection of data.
     def build_collection(parsers)
       results = parse_children(parsers)
-      results
+      SerpParser::Collection.new(results)
     end
 
     # Extracts data from the document and merges it into a single hash.

spec/serp_parser/google/search_spec.rb

@@ -25,7 +25,11 @@
   end
 
   describe "#organic_results" do
-    it "returns an array of organic results" do
+    it "returns a collection object" do
+      expect(parser.organic_results).to be_an_instance_of(SerpParser::Collection)
+    end
+
+    it "returns organic results" do
       expect(parser.organic_results).to all(be_an_instance_of(SerpParser::Models::OrganicResult))
     end