2019-04-29 20:27:42 -04:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2018-09-13 01:24:08 -04:00
|
|
|
require 'rails_helper'
|
|
|
|
|
|
|
|
RSpec.describe UploadRecovery do
|
2019-05-06 23:12:20 -04:00
|
|
|
fab!(:user) { Fabricate(:user) }
|
2018-09-13 01:24:08 -04:00
|
|
|
|
|
|
|
let(:upload) do
|
|
|
|
UploadCreator.new(
|
2018-09-18 22:44:36 -04:00
|
|
|
file_from_fixtures("smallest.png"),
|
2018-09-13 01:24:08 -04:00
|
|
|
"logo.png"
|
|
|
|
).create_for(user.id)
|
|
|
|
end
|
|
|
|
|
2018-09-18 22:44:36 -04:00
|
|
|
let(:upload2) do
|
|
|
|
UploadCreator.new(
|
|
|
|
file_from_fixtures("small.pdf", "pdf"),
|
|
|
|
"some.pdf"
|
|
|
|
).create_for(user.id)
|
|
|
|
end
|
|
|
|
|
2018-09-13 01:24:08 -04:00
|
|
|
let(:post) do
|
|
|
|
Fabricate(:post,
|
2018-09-18 22:44:36 -04:00
|
|
|
raw: <<~SQL,
|
|
|
|
![logo.png](#{upload.short_url})
|
|
|
|
SQL
|
2018-09-13 01:24:08 -04:00
|
|
|
user: user
|
2018-09-13 01:41:38 -04:00
|
|
|
).tap(&:link_post_uploads)
|
2018-09-13 01:24:08 -04:00
|
|
|
end
|
|
|
|
|
2018-09-13 01:59:17 -04:00
|
|
|
let(:upload_recovery) { UploadRecovery.new }
|
|
|
|
|
2018-09-13 01:24:08 -04:00
|
|
|
before do
|
2018-09-18 22:44:36 -04:00
|
|
|
SiteSetting.authorized_extensions = 'png|pdf'
|
2019-03-14 10:47:38 -04:00
|
|
|
Jobs.run_immediately!
|
2018-09-13 01:24:08 -04:00
|
|
|
end
|
|
|
|
|
2018-09-30 22:51:25 -04:00
|
|
|
after do
|
|
|
|
[upload, upload2].each do |u|
|
2018-10-01 02:20:50 -04:00
|
|
|
next if u
|
2018-09-30 22:51:25 -04:00
|
|
|
public_path = "#{Discourse.store.public_dir}#{u.url}"
|
|
|
|
|
|
|
|
[
|
|
|
|
public_path,
|
|
|
|
public_path.sub("uploads", "uploads/tombstone")
|
|
|
|
].each { |path| File.delete(path) if File.exists?(path) }
|
2018-09-13 01:59:17 -04:00
|
|
|
end
|
2018-09-30 22:51:25 -04:00
|
|
|
end
|
2018-09-13 01:59:17 -04:00
|
|
|
|
2018-09-30 22:51:25 -04:00
|
|
|
describe '#recover' do
|
2018-09-13 01:59:17 -04:00
|
|
|
describe 'when given an invalid sha1' do
|
|
|
|
it 'should not do anything' do
|
|
|
|
upload_recovery.expects(:recover_from_local).never
|
|
|
|
|
|
|
|
post.update!(
|
|
|
|
raw: "![logo.png](upload://#{'a' * 28}.png)"
|
|
|
|
)
|
|
|
|
|
|
|
|
upload_recovery.recover
|
2018-09-20 02:21:57 -04:00
|
|
|
|
|
|
|
post.update!(
|
|
|
|
raw: "<a href=#{"/uploads/test/original/3X/a/6%0A/#{upload.sha1}.png"}>test</a>"
|
|
|
|
)
|
|
|
|
|
|
|
|
upload_recovery.recover
|
2018-09-13 01:24:08 -04:00
|
|
|
end
|
|
|
|
end
|
2018-09-13 01:59:17 -04:00
|
|
|
|
2018-09-13 04:32:35 -04:00
|
|
|
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
|
|
|
|
|
2018-09-18 23:52:57 -04:00
|
|
|
describe 'for a missing attachment' do
|
|
|
|
let(:post) do
|
|
|
|
Fabricate(:post,
|
|
|
|
raw: <<~SQL,
|
|
|
|
<a class="attachment" href="#{upload2.url}">some.pdf</a>
|
|
|
|
<a>blank</a>
|
|
|
|
SQL
|
|
|
|
user: user
|
|
|
|
).tap(&:link_post_uploads)
|
|
|
|
end
|
|
|
|
|
|
|
|
it 'should recover the attachment' do
|
|
|
|
expect do
|
|
|
|
upload2.destroy!
|
|
|
|
end.to change { post.reload.uploads.count }.from(1).to(0)
|
|
|
|
|
|
|
|
expect do
|
|
|
|
upload_recovery.recover
|
|
|
|
end.to change { post.reload.uploads.count }.from(0).to(1)
|
2018-09-19 04:03:52 -04:00
|
|
|
|
|
|
|
expect(File.read(Discourse.store.path_for(post.uploads.first)))
|
|
|
|
.to eq(File.read(file_from_fixtures("small.pdf", "pdf")))
|
2018-09-18 23:52:57 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2018-09-18 22:44:36 -04:00
|
|
|
it 'should recover uploads and attachments' do
|
2018-09-13 01:59:17 -04:00
|
|
|
stub_request(:get, "http://test.localhost#{upload.url}")
|
|
|
|
.to_return(status: 200)
|
|
|
|
|
|
|
|
expect do
|
|
|
|
upload.destroy!
|
2018-09-18 23:52:57 -04:00
|
|
|
end.to change { post.reload.uploads.count }.from(1).to(0)
|
2018-09-13 01:59:17 -04:00
|
|
|
|
|
|
|
expect do
|
|
|
|
upload_recovery.recover
|
2018-09-18 23:52:57 -04:00
|
|
|
end.to change { post.reload.uploads.count }.from(0).to(1)
|
2018-09-19 04:03:52 -04:00
|
|
|
|
|
|
|
expect(File.read(Discourse.store.path_for(post.uploads.first)))
|
|
|
|
.to eq(File.read(file_from_fixtures("smallest.png")))
|
2018-09-13 01:59:17 -04:00
|
|
|
end
|
2019-04-01 23:41:00 -04:00
|
|
|
|
|
|
|
describe 'image tag' do
|
|
|
|
let(:post) do
|
|
|
|
Fabricate(:post,
|
|
|
|
raw: <<~SQL,
|
|
|
|
<img src='#{upload.url}'>
|
|
|
|
SQL
|
|
|
|
user: user
|
|
|
|
).tap(&:link_post_uploads)
|
|
|
|
end
|
|
|
|
|
|
|
|
it 'should recover the upload' do
|
|
|
|
stub_request(:get, "http://test.localhost#{upload.url}")
|
|
|
|
.to_return(status: 200)
|
|
|
|
|
|
|
|
expect do
|
|
|
|
upload.destroy!
|
|
|
|
end.to change { post.reload.uploads.count }.from(1).to(0)
|
|
|
|
|
|
|
|
expect do
|
|
|
|
upload_recovery.recover
|
|
|
|
end.to change { post.reload.uploads.count }.from(0).to(1)
|
|
|
|
|
|
|
|
expect(File.read(Discourse.store.path_for(post.uploads.first)))
|
|
|
|
.to eq(File.read(file_from_fixtures("smallest.png")))
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2019-04-10 04:22:35 -04:00
|
|
|
describe 'image markdown' do
|
|
|
|
let(:post) do
|
|
|
|
Fabricate(:post,
|
|
|
|
raw: <<~SQL,
|
|
|
|
![image](#{upload.url})
|
|
|
|
SQL
|
|
|
|
user: user
|
|
|
|
).tap(&:link_post_uploads)
|
|
|
|
end
|
|
|
|
|
|
|
|
it 'should recover the upload' do
|
|
|
|
stub_request(:get, "http://test.localhost#{upload.url}")
|
|
|
|
.to_return(status: 200)
|
|
|
|
|
|
|
|
expect do
|
|
|
|
upload.destroy!
|
|
|
|
end.to change { post.reload.uploads.count }.from(1).to(0)
|
|
|
|
|
|
|
|
expect do
|
|
|
|
upload_recovery.recover
|
|
|
|
end.to change { post.reload.uploads.count }.from(0).to(1)
|
|
|
|
|
|
|
|
expect(File.read(Discourse.store.path_for(post.uploads.first)))
|
|
|
|
.to eq(File.read(file_from_fixtures("smallest.png")))
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2019-04-01 23:41:00 -04:00
|
|
|
describe 'bbcode' do
|
|
|
|
let(:post) do
|
|
|
|
Fabricate(:post,
|
|
|
|
raw: <<~SQL,
|
|
|
|
[img]#{upload.url}[/img]
|
|
|
|
SQL
|
|
|
|
user: user
|
|
|
|
).tap(&:link_post_uploads)
|
|
|
|
end
|
|
|
|
|
|
|
|
it 'should recover the upload' do
|
|
|
|
stub_request(:get, "http://test.localhost#{upload.url}")
|
|
|
|
.to_return(status: 200)
|
|
|
|
|
|
|
|
expect do
|
|
|
|
upload.destroy!
|
|
|
|
end.to change { post.reload.uploads.count }.from(1).to(0)
|
|
|
|
|
|
|
|
expect do
|
|
|
|
upload_recovery.recover
|
|
|
|
end.to change { post.reload.uploads.count }.from(0).to(1)
|
|
|
|
|
|
|
|
expect(File.read(Discourse.store.path_for(post.uploads.first)))
|
|
|
|
.to eq(File.read(file_from_fixtures("smallest.png")))
|
|
|
|
end
|
|
|
|
end
|
2018-09-13 01:24:08 -04:00
|
|
|
end
|
|
|
|
end
|