feat: return included response in collection

lib/prizepicks/collection.rb

@@ -1,20 +1,35 @@
 # frozen_string_literal: true
 module PrizePicks
   class Collection
-    attr_reader :data, :total_pages, :current_page
+    attr_reader :data, :included, :total_pages, :current_page
 
     def self.from_response(response, type:)
       new(
         data: response['data'].map { |attrs| type.new(attrs) },
+        included: parse_included_data(response['included']),
         current_page: response.dig('meta', 'current_page'),
         total_pages: response.dig('meta', 'total_pages')
       )
     end
 
-    def initialize(data:, current_page:, total_pages:)
+    def initialize(data:, included:, current_page:, total_pages:)
       @data = data
+      @included = included
       @current_page = current_page
       @total_pages = total_pages
     end
+
+    def self.parse_included_data(included)
+      included.map do |attrs|
+        type = attrs['type'].capitalize
+        class_name = "PrizePicks::#{type}"
+        klass = if Object.const_defined?(class_name)
+                  Object.const_get(class_name)
+                else
+                  PrizePicks::Base
+                end
+        klass.new(attrs)
+      end
+    end
   end
 end

test/prizepicks/api/endpoints/test_entries.rb

@@ -20,14 +20,24 @@ def test_entries
           stub = stub_request(PrizePicks::Api::Endpoints::Entries::ENDPOINT,
                               response: stub_response(fixture: 'entries'))
           client = init_client(email: 'fake@fake.com', password: 'fake123!',
-                               stub: stub)
+                               stub:)
           resp = client.entries
           assert_equal PrizePicks::Collection, resp.class
           first_entry = resp.data.first
           assert_equal PrizePicks::Entry, first_entry.class
           assert_equal 2000, first_entry.amount_bet_cents
           assert_equal '134928677', first_entry.data['id']
         end
+
+        def test_entries_included
+          stub = stub_request(PrizePicks::Api::Endpoints::Entries::ENDPOINT,
+                              response: stub_response(fixture: 'entries'))
+          client = init_client(email: 'fake@fake.com', password: 'fake123!',
+                               stub:)
+          resp = client.entries
+          included = resp.included
+          assert_equal 2, included.count
+        end
       end
     end
   end

test/prizepicks/api/endpoints/test_leagues.rb

@@ -12,7 +12,7 @@ def test_leagues_endpoint
         def test_entries
           stub = stub_request(PrizePicks::Api::Endpoints::Leagues::ENDPOINT,
                               response: stub_response(fixture: 'leagues'))
-          client = init_client(stub: stub)
+          client = init_client(stub:)
           resp = client.leagues
           assert_equal PrizePicks::Collection, resp.class
           assert_equal PrizePicks::League, resp.data.first.class

test/prizepicks/api/endpoints/test_projections.rb

@@ -12,7 +12,7 @@ def test_projections_endpoint
         def test_projections
           stub = stub_request(PrizePicks::Api::Endpoints::Projections::ENDPOINT,
                               response: stub_response(fixture: 'projections'))
-          client = init_client(stub: stub)
+          client = init_client(stub:)
           resp = client.projections
           assert_equal PrizePicks::Collection, resp.class
           assert_equal PrizePicks::Projection, resp.data.first.class

test/prizepicks/api/endpoints/test_sign_in.rb

@@ -16,7 +16,7 @@ def test_sign_in
                                 },
                               })
           client = init_client(email: 'fake@fake.com', password: 'fake123!',
-                               stub: stub)
+                               stub:)
           resp = client.sign_in
           assert_equal PrizePicks::User, resp.class
           assert_equal 'example@example.com', resp.email