# frozen_string_literal: true RSpec.describe UploadReference do describe 'badge uploads' do fab!(:upload) { Fabricate(:upload) } it 'creates upload references' do badge = nil expect { badge = Fabricate(:badge, image_upload_id: upload.id) } .to change { UploadReference.count }.by(1) upload_reference = UploadReference.last expect(upload_reference.upload).to eq(upload) expect(upload_reference.target).to eq(badge) expect { badge.destroy! } .to change { UploadReference.count }.by(-1) end end describe 'category uploads' do fab!(:upload1) { Fabricate(:upload) } fab!(:upload2) { Fabricate(:upload) } it 'creates upload references' do category = nil expect { category = Fabricate(:category, uploaded_logo_id: upload1.id, uploaded_background_id: upload2.id) } .to change { UploadReference.count }.by(2) upload_reference = UploadReference.last expect(upload_reference.target).to eq(category) expect { category.destroy! } .to change { UploadReference.count }.by(-2) end end describe 'custom emoji uploads' do fab!(:upload) { Fabricate(:upload) } it 'creates upload references' do custom_emoji = nil expect { custom_emoji = CustomEmoji.create!(name: 'emoji', upload_id: upload.id) } .to change { UploadReference.count }.by(1) upload_reference = UploadReference.last expect(upload_reference.target).to eq(custom_emoji) expect { custom_emoji.destroy! } .to change { UploadReference.count }.by(-1) end end describe 'group uploads' do fab!(:upload) { Fabricate(:upload) } it 'creates upload references' do group = nil expect { group = Fabricate(:group, flair_upload_id: upload.id) } .to change { UploadReference.count }.by(1) upload_reference = UploadReference.last expect(upload_reference.upload).to eq(upload) expect(upload_reference.target).to eq(group) expect { group.destroy! } .to change { UploadReference.count }.by(-1) end end describe 'post uploads' do fab!(:upload) { Fabricate(:upload) } fab!(:post) { Fabricate(:post, raw: "[](#{upload.short_url})") } it 'creates upload references' do expect { post.link_post_uploads } .to change { UploadReference.count }.by(1) upload_reference = UploadReference.last expect(upload_reference.upload).to eq(upload) expect(upload_reference.target).to eq(post) expect { post.destroy! } .to change { UploadReference.count }.by(-1) end end describe 'site setting uploads' do let(:provider) { SiteSettings::DbProvider.new(SiteSetting) } fab!(:upload) { Fabricate(:upload) } fab!(:upload2) { Fabricate(:upload) } it 'creates upload references for uploads' do expect { provider.save('logo', upload.id, SiteSettings::TypeSupervisor.types[:upload]) } .to change { UploadReference.count }.by(1) upload_reference = UploadReference.last expect(upload_reference.upload).to eq(upload) expect(upload_reference.target).to eq(SiteSetting.find_by(name: 'logo')) expect { provider.destroy('logo') } .to change { UploadReference.count }.by(-1) end it 'creates upload references for uploaded_image_lists' do expect { provider.save('selectable_avatars', "#{upload.id}|#{upload2.id}", SiteSettings::TypeSupervisor.types[:uploaded_image_list]) } .to change { UploadReference.count }.by(2) upload_references = UploadReference.all.where(target: SiteSetting.find_by(name: 'selectable_avatars')) expect(upload_references.pluck(:upload_id)).to contain_exactly(upload.id, upload2.id) expect { provider.destroy('selectable_avatars') } .to change { UploadReference.count }.by(-2) end end describe 'theme field uploads' do fab!(:upload) { Fabricate(:upload) } it 'creates upload refererences' do theme_field = nil expect do theme_field = ThemeField.create!( theme_id: Fabricate(:theme).id, target_id: 0, name: 'field', value: '', upload: upload, type_id: ThemeField.types[:theme_upload_var], ) end.to change { UploadReference.count }.by(1) upload_reference = UploadReference.last expect(upload_reference.upload).to eq(upload) expect(upload_reference.target).to eq(theme_field) expect { theme_field.destroy! } .to change { UploadReference.count }.by(-1) end end describe 'theme setting uploads' do fab!(:upload) { Fabricate(:upload) } it 'creates upload refererences' do theme_setting = nil expect do theme_setting = ThemeSetting.create!( name: 'field', data_type: ThemeSetting.types[:upload], value: upload.id, theme_id: Fabricate(:theme).id, ) end.to change { UploadReference.count }.by(1) upload_reference = UploadReference.last expect(upload_reference.upload).to eq(upload) expect(upload_reference.target).to eq(theme_setting) expect { theme_setting.destroy! } .to change { UploadReference.count }.by(-1) end end describe 'user uploads' do fab!(:upload) { Fabricate(:upload) } it 'creates upload references' do user = nil expect { user = Fabricate(:user, uploaded_avatar_id: upload.id) } .to change { UploadReference.count }.by(1) upload_reference = UploadReference.last expect(upload_reference.upload).to eq(upload) expect(upload_reference.target).to eq(user) expect { user.destroy! } .to change { UploadReference.count }.by(-1) end end describe 'user avatar uploads' do fab!(:upload1) { Fabricate(:upload) } fab!(:upload2) { Fabricate(:upload) } it 'creates upload references' do user_avatar = nil expect { user_avatar = Fabricate(:user_avatar, custom_upload_id: upload1.id, gravatar_upload_id: upload2.id) } .to change { UploadReference.count }.by(2) upload_reference = UploadReference.last expect(upload_reference.target).to eq(user_avatar) expect { user_avatar.destroy! } .to change { UploadReference.count }.by(-2) end end describe 'user export uploads' do fab!(:upload) { Fabricate(:upload) } it 'creates upload references' do user_export = nil expect do user_export = UserExport.create!( file_name: 'export', user: Fabricate(:user), upload: upload, topic: Fabricate(:topic), ) end.to change { UploadReference.count }.by(1) upload_reference = UploadReference.last expect(upload_reference.upload).to eq(upload) expect(upload_reference.target).to eq(user_export) expect { user_export.destroy! } .to change { UploadReference.count }.by(-1) end end describe 'user profile uploads' do fab!(:user) { Fabricate(:user) } fab!(:upload1) { Fabricate(:upload) } fab!(:upload2) { Fabricate(:upload) } it 'creates upload references' do user_profile = user.user_profile expect { user_profile.update!(profile_background_upload_id: upload1.id, card_background_upload_id: upload2.id) } .to change { UploadReference.count }.by(2) upload_reference = UploadReference.last expect(upload_reference.target).to eq(user_profile) expect { user_profile.destroy! } .to change { UploadReference.count }.by(-2) end end end