github-mirror/discourse
2
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2024-11-22 14:38:17 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Accept custom AR relation for UploadRecovery.

Browse Source
This commit is contained in:
Guo Xiang Tan 2018-09-13 16:32:35 +08:00
parent 6c65718301
commit 0a06b3d977
2 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/upload_recovery.rb
View File

@ -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)

8
spec/lib/upload_recovery_spec.rb
View File

@ -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)