Accept custom AR relation for `UploadRecovery`.

lib/upload_recovery.rb

@@ -3,8 +3,8 @@ class UploadRecovery
     @dry_run = dry_run
   end
 
-  def recover
-    Post.where("raw LIKE '%upload:\/\/%'").find_each do |post|
+  def recover(posts = Post)
+    posts.where("raw LIKE '%upload:\/\/%'").find_each do |post|
       begin
         analyzer = PostAnalyzer.new(post.raw, post.topic_id)
         cooked_stripped = analyzer.send(:cooked_stripped)

spec/lib/upload_recovery_spec.rb

@@ -46,6 +46,14 @@ RSpec.describe UploadRecovery do
       end
     end
 
+    it 'accepts a custom ActiveRecord relation' do
+      post.update!(updated_at: 2.days.ago)
+      upload.destroy!
+
+      upload_recovery.expects(:recover_from_local).never
+      upload_recovery.recover(Post.where("updated_at >= ?", 1.day.ago))
+    end
+
     it 'should recover the upload' do
       stub_request(:get, "http://test.localhost#{upload.url}")
         .to_return(status: 200)