# frozen_string_literal: true require 'rails_helper' RSpec.describe Archivable, type: :model do let(:user) { create(:user) } let(:point) { create(:point, user: user, raw_data: { lon: 13.4, lat: 52.5 }) } describe 'associations and scopes' do it { expect(point).to belong_to(:raw_data_archive).optional } describe 'scopes' do let!(:archived_point) { create(:point, user: user, raw_data_archived: true) } let!(:not_archived_point) { create(:point, user: user, raw_data_archived: false) } it '.archived returns archived points' do expect(Point.archived).to include(archived_point) expect(Point.archived).not_to include(not_archived_point) end it '.not_archived returns non-archived points' do expect(Point.not_archived).to include(not_archived_point) expect(Point.not_archived).not_to include(archived_point) end end end describe '#raw_data_with_archive' do context 'when raw_data is present in database' do it 'returns raw_data from database' do expect(point.raw_data_with_archive).to eq({ 'lon' => 13.4, 'lat' => 52.5 }) end end context 'when raw_data is archived' do let(:archive) { create(:points_raw_data_archive, user: user) } let(:archived_point) do create(:point, user: user, raw_data: nil, raw_data_archived: true, raw_data_archive: archive) end before do # Mock archive file content with this specific point compressed_data = gzip_data([ { id: archived_point.id, raw_data: { lon: 14.0, lat: 53.0 } } ]) allow(archive.file.blob).to receive(:download).and_return(compressed_data) end it 'fetches raw_data from archive' do result = archived_point.raw_data_with_archive expect(result).to eq({ 'id' => archived_point.id, 'raw_data' => { 'lon' => 14.0, 'lat' => 53.0 } }['raw_data']) end end context 'when raw_data is archived but point not in archive' do let(:archive) { create(:points_raw_data_archive, user: user) } let(:archived_point) do create(:point, user: user, raw_data: nil, raw_data_archived: true, raw_data_archive: archive) end before do # Mock archive file with different point compressed_data = gzip_data([ { id: 999, raw_data: { lon: 14.0, lat: 53.0 } } ]) allow(archive.file.blob).to receive(:download).and_return(compressed_data) end it 'returns empty hash' do expect(archived_point.raw_data_with_archive).to eq({}) end end end describe '#restore_raw_data!' do let(:archive) { create(:points_raw_data_archive, user: user) } let(:archived_point) do create(:point, user: user, raw_data: nil, raw_data_archived: true, raw_data_archive: archive) end it 'restores raw_data to database and clears archive flags' do new_data = { lon: 15.0, lat: 54.0 } archived_point.restore_raw_data!(new_data) archived_point.reload expect(archived_point.raw_data).to eq(new_data.stringify_keys) expect(archived_point.raw_data_archived).to be false expect(archived_point.raw_data_archive_id).to be_nil end end describe 'temporary cache' do let(:june_point) { create(:point, user: user, timestamp: Time.new(2024, 6, 15).to_i) } it 'checks temporary restore cache with correct key format' do cache_key = "raw_data:temp:#{user.id}:2024:6:#{june_point.id}" cached_data = { lon: 16.0, lat: 55.0 } Rails.cache.write(cache_key, cached_data, expires_in: 1.hour) # Access through send since check_temporary_restore_cache is private result = june_point.send(:check_temporary_restore_cache) expect(result).to eq(cached_data) end end def gzip_data(points_array) io = StringIO.new gz = Zlib::GzipWriter.new(io) points_array.each do |point_data| gz.puts(point_data.to_json) end gz.close io.string end end