Merge pull request #62 from syncrou/expose_playbooks_off_of_projects

Expose playbooks off of projects

lib/ansible_tower_client/base_models/project.rb

@@ -3,5 +3,9 @@ class Project < BaseModel
     def self.endpoint
       "projects".freeze
     end
+
+    def playbooks
+      Collection.new(api).find_all_by_url(related['playbooks'])
+    end
   end
 end

lib/ansible_tower_client/collection.rb

@@ -47,17 +47,22 @@ def class_from_type(type)
     def fetch_more_results(next_page, get_options)
       return if next_page.nil?
       body = parse_response(api.get(next_page, get_options))
-      parse_result_set(body["results"])
-
-      body["next"]
+      parse_result_set(body)
     end
 
     def parse_response(response)
       JSON.parse(response.body)
     end
 
-    def parse_result_set(results)
-      results.each { |result| @collection << build_object(result) }
+    def parse_result_set(body)
+      case body.class.name
+      when "Array" then
+        @collection = body
+        nil
+      when "Hash" then
+        body["results"].each { |result| @collection << build_object(result) }
+        body["next"]
+      end
     end
 
     def build_object(result)

spec/support/shared_examples/collection_methods.rb

@@ -18,6 +18,20 @@
     expect(obj_collection_array.first.url).to eq(url)
   end
 
+  it ".find_all_by_url returns a collection of items via a url" do
+    raw_collection_array = raw_url_collection.to_a.flatten
+    expect(api).to receive(:get).and_return(instance_double("Faraday::Result", :body => raw_collection_array.to_json))
+    url = raw_url_collection['url']
+
+    obj_collection = collection.find_all_by_url(url)
+    expect(obj_collection).to be_a Enumerator
+
+    obj_collection_array = obj_collection.to_a
+    expect(obj_collection_array.length).to    eq(8)
+    expect(obj_collection_array.first).to_not eq described_class
+    expect(obj_collection_array.first).to be_a String
+  end
+
   it ".find returns an instance" do
     expect(api).to receive(:get).and_return(instance_double("Faraday::Result", :body => raw_instance.to_json))