Refactoring collection wrapper

2 files changed, 5 insertions, 9 deletions

diff --git a/lib/github/collection.rb b/lib/github/collection.rb
index bbca12a1c84..1b0c00928c5 100644
--- a/lib/github/collection.rb
+++ b/lib/github/collection.rb
@@ -1,18 +1,14 @@
 module Github
   class Collection
-    def initialize(url)
-      @url = url
-    end
-
-    def fetch(query = {})
-      return [] if @url.blank?
+    def fetch(url, query = {})
+      return [] if url.blank?
 
       Enumerator.new do |yielder|
         loop do
-          response = client.get(@url, query)
+          response = client.get(url, query)
           response.body.each { |item| yielder << item }
           raise StopIteration unless response.rels.key?(:next)
-          @url = response.rels[:next]
+          url = response.rels[:next]
         end
       end.lazy
     end
diff --git a/lib/github/repositories.rb b/lib/github/repositories.rb
index 42342471102..b6a03173673 100644
--- a/lib/github/repositories.rb
+++ b/lib/github/repositories.rb
@@ -5,7 +5,7 @@ module Github
     end
 
     def fetch
-      Collection.new(repos_url).fetch
+      Collection.new.fetch(repos_url)
     end
 
     private