mirror of
https://github.com/Freika/dawarich.git
synced 2026-01-11 09:41:40 -05:00
171 lines
5.6 KiB
Ruby
171 lines
5.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'rails_helper'
|
|
|
|
RSpec.describe Users::ExportData::Imports, type: :service do
|
|
let(:user) { create(:user) }
|
|
let(:files_directory) { Pathname.new(Dir.mktmpdir('test_exports')) }
|
|
let(:service) { described_class.new(user, files_directory) }
|
|
|
|
subject { service.call }
|
|
|
|
after do
|
|
FileUtils.rm_rf(files_directory) if files_directory.exist?
|
|
end
|
|
|
|
describe '#call' do
|
|
context 'when user has no imports' do
|
|
it 'returns an empty array' do
|
|
expect(subject).to eq([])
|
|
end
|
|
end
|
|
|
|
context 'when user has imports without files' do
|
|
let!(:import1) { create(:import, user: user, name: 'Import 1') }
|
|
let!(:import2) { create(:import, user: user, name: 'Import 2') }
|
|
|
|
it 'returns import data without file information' do
|
|
expect(service.call.size).to eq(2)
|
|
|
|
first_import = service.call.find { |i| i['name'] == 'Import 1' }
|
|
expect(first_import['file_name']).to be_nil
|
|
expect(first_import['original_filename']).to be_nil
|
|
expect(first_import).not_to have_key('user_id')
|
|
expect(first_import).not_to have_key('raw_data')
|
|
expect(first_import).not_to have_key('id')
|
|
end
|
|
|
|
it 'logs processing information' do
|
|
expect(Rails.logger).to receive(:info).at_least(:once)
|
|
service.call
|
|
end
|
|
end
|
|
|
|
context 'when user has imports with attached files' do
|
|
let(:file_content) { 'test file content' }
|
|
let(:blob) { create_blob(filename: 'test_file.json', content_type: 'application/json') }
|
|
let!(:import_with_file) do
|
|
import = create(:import, user: user, name: 'Import with File')
|
|
import.file.attach(blob)
|
|
import
|
|
end
|
|
|
|
before do
|
|
allow(Imports::SecureFileDownloader).to receive(:new).and_return(
|
|
double(download_with_verification: file_content)
|
|
)
|
|
end
|
|
|
|
it 'returns import data with file information' do
|
|
import_data = subject.first
|
|
|
|
expect(import_data['name']).to eq('Import with File')
|
|
expect(import_data['file_name']).to eq("import_#{import_with_file.id}_test_file.json")
|
|
expect(import_data['original_filename']).to eq('test_file.json')
|
|
expect(import_data['file_size']).to be_present
|
|
expect(import_data['content_type']).to eq('application/json')
|
|
end
|
|
|
|
it 'downloads and saves the file to the files directory' do
|
|
import_data = subject.first
|
|
|
|
file_path = files_directory.join(import_data['file_name'])
|
|
expect(File.exist?(file_path)).to be true
|
|
expect(File.read(file_path)).to eq(file_content)
|
|
end
|
|
|
|
it 'sanitizes the filename' do
|
|
blob = create_blob(filename: 'test file with spaces & symbols!.json')
|
|
import_with_file.file.attach(blob)
|
|
|
|
import_data = subject.first
|
|
|
|
expect(import_data['file_name']).to match(/import_\d+_test_file_with_spaces___symbols_.json/)
|
|
end
|
|
end
|
|
|
|
context 'when file download fails' do
|
|
let!(:import_with_file) do
|
|
import = create(:import, user: user, name: 'Import with error file')
|
|
import.file.attach(create_blob)
|
|
import
|
|
end
|
|
|
|
before do
|
|
allow(Imports::SecureFileDownloader).to receive(:new).and_raise(StandardError, 'Download failed')
|
|
end
|
|
|
|
it 'handles download errors gracefully' do
|
|
import_data = subject.find { |i| i['name'] == 'Import with error file' }
|
|
|
|
expect(import_data['file_error']).to eq('Failed to download: Download failed')
|
|
end
|
|
end
|
|
|
|
context 'with single import (no parallel processing)' do
|
|
let!(:import) { create(:import, user: user, name: 'Single import') }
|
|
|
|
it 'processes without using parallel threads' do
|
|
expect(Parallel).not_to receive(:map)
|
|
service.call
|
|
end
|
|
end
|
|
|
|
context 'with multiple imports (parallel processing)' do
|
|
let!(:import1) { create(:import, user: user, name: 'Multiple Import 1') }
|
|
let!(:import2) { create(:import, user: user, name: 'Multiple Import 2') }
|
|
let!(:import3) { create(:import, user: user, name: 'Multiple Import 3') }
|
|
|
|
let!(:imports) { [import1, import2, import3] }
|
|
|
|
it 'uses parallel processing with limited threads' do
|
|
expect(Parallel).to receive(:map).with(anything, in_threads: 2).and_call_original
|
|
service.call
|
|
end
|
|
|
|
it 'returns all imports' do
|
|
expect(subject.size).to eq(3)
|
|
end
|
|
end
|
|
|
|
context 'with multiple users' do
|
|
let(:other_user) { create(:user) }
|
|
let!(:user_import) { create(:import, user: user, name: 'User Import') }
|
|
let!(:other_user_import) { create(:import, user: other_user, name: 'Other User Import') }
|
|
|
|
it 'only returns imports for the specified user' do
|
|
expect(subject.size).to eq(1)
|
|
expect(subject.first['name']).to eq('User Import')
|
|
end
|
|
end
|
|
|
|
context 'performance considerations' do
|
|
let!(:import1) { create(:import, user: user, name: 'Perf Import 1') }
|
|
let!(:import2) { create(:import, user: user, name: 'Perf Import 2') }
|
|
|
|
let!(:imports_with_files) { [import1, import2] }
|
|
|
|
before do
|
|
imports_with_files.each do |import|
|
|
import.file.attach(create_blob)
|
|
end
|
|
end
|
|
|
|
it 'includes file_attachment to avoid N+1 queries' do
|
|
# This test verifies that we're using .includes(:file_attachment)
|
|
expect(user.imports).to receive(:includes).with(:file_attachment).and_call_original
|
|
service.call
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def create_blob(filename: 'test.txt', content_type: 'text/plain')
|
|
ActiveStorage::Blob.create_and_upload!(
|
|
io: StringIO.new('test content'),
|
|
filename: filename,
|
|
content_type: content_type
|
|
)
|
|
end
|
|
end
|