mirror of
https://github.com/LaCasemate/fab-manager.git
synced 2024-11-29 10:24:20 +01:00
63 lines
2.1 KiB
Ruby
63 lines
2.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# Provides methods to help testing archives of accounting periods
|
|
module ArchiveHelper
|
|
# Force the generation of the archive now.
|
|
# Then extract it, then check its contents, then delete the archive, finally delete the extracted content
|
|
def assert_archive(accounting_period)
|
|
assert_not_nil accounting_period, 'AccountingPeriod was not created'
|
|
|
|
archive_worker = ArchiveWorker.new
|
|
archive_worker.perform(accounting_period.id)
|
|
|
|
assert FileTest.exist?(accounting_period.archive_file), 'ZIP archive was not generated'
|
|
|
|
dest = extract_archive(accounting_period)
|
|
|
|
# Check archive matches
|
|
file = check_integrity(dest)
|
|
|
|
archive = File.read("#{dest}/#{file}")
|
|
archive_json = JSON.parse(archive)
|
|
invoices = Invoice.where(
|
|
'created_at >= :start_date AND created_at <= :end_date',
|
|
start_date: accounting_period.start_at.to_time.in_time_zone, end_date: accounting_period.end_at.to_time.in_time_zone
|
|
)
|
|
|
|
assert_equal invoices.count, archive_json['invoices'].count
|
|
assert_equal accounting_period.footprint, archive_json['period_footprint']
|
|
|
|
require 'version'
|
|
assert_equal Version.current, archive_json['software']['version']
|
|
|
|
# we clean up the files before quitting
|
|
FileUtils.rm_rf(dest)
|
|
FileUtils.rm_rf(accounting_period.archive_folder)
|
|
end
|
|
|
|
private
|
|
|
|
# Extract the archive to the temporary folder
|
|
def extract_archive(accounting_period)
|
|
require 'tmpdir'
|
|
require 'fileutils'
|
|
dest = "#{Dir.tmpdir}/accounting/#{accounting_period.id}"
|
|
FileUtils.mkdir_p "#{dest}/accounting"
|
|
Zip::File.open(accounting_period.archive_file) do |zip_file|
|
|
# Handle entries one by one
|
|
zip_file.each do |entry|
|
|
# Extract to file/directory/symlink
|
|
entry.extract("#{dest}/#{entry.name}")
|
|
end
|
|
end
|
|
dest
|
|
end
|
|
|
|
def check_integrity(extracted_path)
|
|
require 'integrity/checksum'
|
|
sumfile = File.read("#{extracted_path}/checksum.sha256").split("\t")
|
|
assert_equal sumfile[0], Integrity::Checksum.file("#{extracted_path}/#{sumfile[1]}"), 'archive checksum does not match'
|
|
sumfile[1]
|
|
end
|
|
end
|