gitlab-org--gitlab-foss/spec/models/ci/build_trace_chunk_spec.rb

563 lines
15 KiB
Ruby
Raw Normal View History

require 'spec_helper'
describe Ci::BuildTraceChunk, :clean_gitlab_redis_shared_state do
include ExclusiveLeaseHelpers
set(:build) { create(:ci_build, :running) }
let(:chunk_index) { 0 }
let(:data_store) { :redis }
let(:raw_data) { nil }
let(:build_trace_chunk) do
described_class.new(build: build, chunk_index: chunk_index, data_store: data_store, raw_data: raw_data)
end
before do
stub_feature_flags(ci_enable_live_trace: true)
2018-06-25 03:19:40 -04:00
stub_artifacts_object_storage
end
context 'FastDestroyAll' do
let(:parent) { create(:project) }
let(:pipeline) { create(:ci_pipeline, project: parent) }
let(:build) { create(:ci_build, :running, :trace_live, pipeline: pipeline, project: parent) }
let(:subjects) { build.trace_chunks }
it_behaves_like 'fast destroyable'
def external_data_counter
Gitlab::Redis::SharedState.with do |redis|
redis.scan_each(match: "gitlab:ci:trace:*:chunks:*").to_a.size
end
end
end
2018-04-05 07:39:35 -04:00
describe 'CHUNK_SIZE' do
it 'Chunk size can not be changed without special care' do
expect(described_class::CHUNK_SIZE).to eq(128.kilobytes)
end
end
describe '#data' do
subject { build_trace_chunk.data }
context 'when data_store is redis' do
let(:data_store) { :redis }
before do
2018-06-25 03:19:40 -04:00
build_trace_chunk.send(:unsafe_set_data!, 'Sample data in redis')
end
it { is_expected.to eq('Sample data in redis') }
end
context 'when data_store is database' do
2018-06-25 03:19:40 -04:00
let(:data_store) { :database }
let(:raw_data) { 'Sample data in database' }
2018-06-25 03:19:40 -04:00
it { is_expected.to eq('Sample data in database') }
end
2018-06-25 03:19:40 -04:00
context 'when data_store is fog' do
let(:data_store) { :fog }
2018-06-25 03:19:40 -04:00
before do
2018-06-25 04:59:46 -04:00
build_trace_chunk.send(:unsafe_set_data!, 'Sample data in fog')
2018-06-25 03:19:40 -04:00
end
2018-06-25 03:19:40 -04:00
it { is_expected.to eq('Sample data in fog') }
end
2018-06-25 03:19:40 -04:00
end
2018-06-25 03:19:40 -04:00
describe '#append' do
subject { build_trace_chunk.append(new_data, offset) }
2018-06-25 03:19:40 -04:00
let(:new_data) { 'Sample new data' }
let(:offset) { 0 }
let(:merged_data) { data + new_data.to_s }
2018-06-25 03:19:40 -04:00
shared_examples_for 'Appending correctly' do
context 'when offset is negative' do
let(:offset) { -1 }
2018-06-25 03:19:40 -04:00
it { expect { subject }.to raise_error('Offset is out of range') }
end
2018-06-25 03:19:40 -04:00
context 'when offset is bigger than data size' do
let(:offset) { data.bytesize + 1 }
2018-06-25 03:19:40 -04:00
it { expect { subject }.to raise_error('Offset is out of range') }
end
2018-06-25 03:19:40 -04:00
context 'when new data overflows chunk size' do
let(:new_data) { 'a' * (described_class::CHUNK_SIZE + 1) }
2018-06-25 03:19:40 -04:00
it { expect { subject }.to raise_error('Chunk size overflow') }
end
2018-06-25 03:19:40 -04:00
context 'when offset is EOF' do
let(:offset) { data.bytesize }
2018-06-25 03:19:40 -04:00
it 'appends' do
subject
2018-06-25 03:19:40 -04:00
expect(build_trace_chunk.data).to eq(merged_data)
end
2018-06-25 04:59:46 -04:00
context 'when the other process is appending' do
let(:lease_key) { "trace_write:#{build_trace_chunk.build.id}:chunks:#{build_trace_chunk.chunk_index}" }
before do
stub_exclusive_lease_taken(lease_key)
end
2018-06-25 04:59:46 -04:00
it 'raise an error' do
expect { subject }.to raise_error('Failed to obtain a lock')
2018-06-25 04:59:46 -04:00
end
end
2018-06-25 03:19:40 -04:00
context 'when new_data is nil' do
let(:new_data) { nil }
2018-06-25 04:59:46 -04:00
2018-06-25 03:19:40 -04:00
it 'raises an error' do
2018-07-04 02:19:22 -04:00
expect { subject }.to raise_error('New data is missing')
2018-06-25 03:19:40 -04:00
end
end
2018-06-25 03:19:40 -04:00
context 'when new_data is empty' do
let(:new_data) { '' }
2018-06-25 04:59:46 -04:00
2018-06-25 03:19:40 -04:00
it 'does not append' do
subject
2018-06-25 04:59:46 -04:00
2018-06-25 03:19:40 -04:00
expect(build_trace_chunk.data).to eq(data)
end
it 'does not execute UPDATE' do
ActiveRecord::QueryRecorder.new { subject }.log.map do |query|
expect(query).not_to include('UPDATE')
end
end
end
end
2018-06-25 03:19:40 -04:00
context 'when offset is middle of datasize' do
let(:offset) { data.bytesize / 2 }
2018-06-25 03:19:40 -04:00
it 'appends' do
subject
2018-06-25 03:19:40 -04:00
expect(build_trace_chunk.data).to eq(data.byteslice(0, offset) + new_data)
end
end
end
2018-06-25 03:19:40 -04:00
shared_examples_for 'Scheduling sidekiq worker to flush data to persist store' do
context 'when new data fullfilled chunk size' do
let(:new_data) { 'a' * described_class::CHUNK_SIZE }
2018-06-25 03:19:40 -04:00
it 'schedules trace chunk flush worker' do
expect(Ci::BuildTraceChunkFlushWorker).to receive(:perform_async).once
2018-06-25 03:19:40 -04:00
subject
end
2018-06-25 03:19:40 -04:00
it 'migrates data to object storage' do
Sidekiq::Testing.inline! do
subject
2018-06-25 03:19:40 -04:00
build_trace_chunk.reload
expect(build_trace_chunk.fog?).to be_truthy
expect(build_trace_chunk.data).to eq(new_data)
end
end
end
2018-06-25 03:19:40 -04:00
end
2018-06-25 03:19:40 -04:00
shared_examples_for 'Scheduling no sidekiq worker' do
context 'when new data fullfilled chunk size' do
let(:new_data) { 'a' * described_class::CHUNK_SIZE }
2018-06-25 03:19:40 -04:00
it 'does not schedule trace chunk flush worker' do
expect(Ci::BuildTraceChunkFlushWorker).not_to receive(:perform_async)
subject
2018-06-25 03:19:40 -04:00
end
2018-06-25 03:19:40 -04:00
it 'does not migrate data to object storage' do
Sidekiq::Testing.inline! do
data_store = build_trace_chunk.data_store
subject
build_trace_chunk.reload
expect(build_trace_chunk.data_store).to eq(data_store)
end
end
end
end
context 'when data_store is redis' do
let(:data_store) { :redis }
2018-06-25 03:19:40 -04:00
context 'when there are no data' do
let(:data) { '' }
it 'has no data' do
expect(build_trace_chunk.data).to be_empty
end
it_behaves_like 'Appending correctly'
it_behaves_like 'Scheduling sidekiq worker to flush data to persist store'
end
2018-06-25 03:19:40 -04:00
context 'when there are some data' do
let(:data) { 'Sample data in redis' }
before do
build_trace_chunk.send(:unsafe_set_data!, data)
end
it 'has data' do
expect(build_trace_chunk.data).to eq(data)
end
it_behaves_like 'Appending correctly'
it_behaves_like 'Scheduling sidekiq worker to flush data to persist store'
end
end
context 'when data_store is database' do
2018-06-25 03:19:40 -04:00
let(:data_store) { :database }
2018-06-25 03:19:40 -04:00
context 'when there are no data' do
let(:data) { '' }
it 'has no data' do
expect(build_trace_chunk.data).to be_empty
end
it_behaves_like 'Appending correctly'
it_behaves_like 'Scheduling no sidekiq worker'
end
context 'when there are some data' do
let(:raw_data) { 'Sample data in database' }
let(:data) { raw_data }
it 'has data' do
expect(build_trace_chunk.data).to eq(data)
end
it_behaves_like 'Appending correctly'
it_behaves_like 'Scheduling no sidekiq worker'
end
end
2018-06-25 03:19:40 -04:00
context 'when data_store is fog' do
let(:data_store) { :fog }
2018-06-25 06:59:28 -04:00
context 'when there are no data' do
let(:data) { '' }
it 'has no data' do
expect(build_trace_chunk.data).to be_empty
end
it_behaves_like 'Appending correctly'
it_behaves_like 'Scheduling no sidekiq worker'
end
context 'when there are some data' do
let(:data) { 'Sample data in fog' }
before do
build_trace_chunk.send(:unsafe_set_data!, data)
end
it 'has data' do
expect(build_trace_chunk.data).to eq(data)
end
it_behaves_like 'Appending correctly'
it_behaves_like 'Scheduling no sidekiq worker'
2018-06-25 03:19:40 -04:00
end
end
end
describe '#truncate' do
subject { build_trace_chunk.truncate(offset) }
2018-06-25 03:19:40 -04:00
shared_examples_for 'truncates' do
context 'when offset is negative' do
let(:offset) { -1 }
it { expect { subject }.to raise_error('Offset is out of range') }
end
context 'when offset is bigger than data size' do
let(:offset) { data.bytesize + 1 }
it { expect { subject }.to raise_error('Offset is out of range') }
end
context 'when offset is 10' do
let(:offset) { 10 }
2018-06-25 03:19:40 -04:00
it 'truncates' do
subject
2018-06-25 03:19:40 -04:00
expect(build_trace_chunk.data).to eq(data.byteslice(0, offset))
end
end
end
context 'when data_store is redis' do
let(:data_store) { :redis }
let(:data) { 'Sample data in redis' }
before do
2018-06-25 03:19:40 -04:00
build_trace_chunk.send(:unsafe_set_data!, data)
end
2018-06-25 03:19:40 -04:00
it_behaves_like 'truncates'
end
context 'when data_store is database' do
2018-06-25 03:19:40 -04:00
let(:data_store) { :database }
let(:raw_data) { 'Sample data in database' }
let(:data) { raw_data }
2018-06-25 03:19:40 -04:00
it_behaves_like 'truncates'
end
context 'when data_store is fog' do
let(:data_store) { :fog }
2018-06-25 06:59:28 -04:00
let(:data) { 'Sample data in fog' }
2018-06-25 03:19:40 -04:00
2018-06-25 06:59:28 -04:00
before do
build_trace_chunk.send(:unsafe_set_data!, data)
2018-06-25 03:19:40 -04:00
end
2018-06-25 06:59:28 -04:00
it_behaves_like 'truncates'
end
end
describe '#size' do
subject { build_trace_chunk.size }
context 'when data_store is redis' do
let(:data_store) { :redis }
context 'when data exists' do
let(:data) { 'Sample data in redis' }
before do
2018-06-25 03:19:40 -04:00
build_trace_chunk.send(:unsafe_set_data!, data)
end
it { is_expected.to eq(data.bytesize) }
end
context 'when data exists' do
it { is_expected.to eq(0) }
end
end
context 'when data_store is database' do
2018-06-25 03:19:40 -04:00
let(:data_store) { :database }
context 'when data exists' do
2018-06-25 03:19:40 -04:00
let(:raw_data) { 'Sample data in database' }
let(:data) { raw_data }
it { is_expected.to eq(data.bytesize) }
end
context 'when data does not exist' do
it { is_expected.to eq(0) }
end
end
2018-06-25 03:19:40 -04:00
context 'when data_store is fog' do
let(:data_store) { :fog }
context 'when data exists' do
let(:data) { 'Sample data in fog' }
let(:key) { "tmp/builds/#{build.id}/chunks/#{chunk_index}.log" }
before do
build_trace_chunk.send(:unsafe_set_data!, data)
end
it { is_expected.to eq(data.bytesize) }
end
context 'when data does not exist' do
2018-06-25 06:59:28 -04:00
it { is_expected.to eq(0) }
2018-06-25 03:19:40 -04:00
end
end
end
2018-06-18 04:56:16 -04:00
describe '#persist_data!' do
subject { build_trace_chunk.persist_data! }
2018-06-25 04:59:46 -04:00
shared_examples_for 'Atomic operation' do
context 'when the other process is persisting' do
let(:lease_key) { "trace_write:#{build_trace_chunk.build.id}:chunks:#{build_trace_chunk.chunk_index}" }
before do
stub_exclusive_lease_taken(lease_key)
end
2018-06-25 04:59:46 -04:00
it 'raise an error' do
expect { subject }.to raise_error('Failed to obtain a lock')
2018-06-25 04:59:46 -04:00
end
end
end
context 'when data_store is redis' do
let(:data_store) { :redis }
context 'when data exists' do
let(:data) { 'Sample data in redis' }
before do
2018-06-25 03:19:40 -04:00
build_trace_chunk.send(:unsafe_set_data!, data)
end
2018-06-25 03:19:40 -04:00
it 'persists the data' do
expect(build_trace_chunk.redis?).to be_truthy
expect(Ci::BuildTraceChunks::Redis.new.data(build_trace_chunk)).to eq(data)
expect(Ci::BuildTraceChunks::Database.new.data(build_trace_chunk)).to be_nil
expect { Ci::BuildTraceChunks::Fog.new.data(build_trace_chunk) }.to raise_error(Excon::Error::NotFound)
subject
2018-06-25 03:19:40 -04:00
expect(build_trace_chunk.fog?).to be_truthy
expect(Ci::BuildTraceChunks::Redis.new.data(build_trace_chunk)).to be_nil
expect(Ci::BuildTraceChunks::Database.new.data(build_trace_chunk)).to be_nil
expect(Ci::BuildTraceChunks::Fog.new.data(build_trace_chunk)).to eq(data)
end
2018-06-25 04:59:46 -04:00
it_behaves_like 'Atomic operation'
end
context 'when data does not exist' do
2018-06-25 03:19:40 -04:00
it 'does not persist' do
expect { subject }.to raise_error('Can not persist empty data')
end
end
end
context 'when data_store is database' do
2018-06-25 03:19:40 -04:00
let(:data_store) { :database }
2018-06-25 03:19:40 -04:00
context 'when data exists' do
let(:data) { 'Sample data in database' }
2018-06-25 03:19:40 -04:00
before do
build_trace_chunk.send(:unsafe_set_data!, data)
end
2018-06-25 03:19:40 -04:00
it 'persists the data' do
expect(build_trace_chunk.database?).to be_truthy
expect(Ci::BuildTraceChunks::Redis.new.data(build_trace_chunk)).to be_nil
expect(Ci::BuildTraceChunks::Database.new.data(build_trace_chunk)).to eq(data)
expect { Ci::BuildTraceChunks::Fog.new.data(build_trace_chunk) }.to raise_error(Excon::Error::NotFound)
2018-06-25 03:19:40 -04:00
subject
2018-06-25 03:19:40 -04:00
expect(build_trace_chunk.fog?).to be_truthy
expect(Ci::BuildTraceChunks::Redis.new.data(build_trace_chunk)).to be_nil
expect(Ci::BuildTraceChunks::Database.new.data(build_trace_chunk)).to be_nil
expect(Ci::BuildTraceChunks::Fog.new.data(build_trace_chunk)).to eq(data)
end
2018-06-25 04:59:46 -04:00
it_behaves_like 'Atomic operation'
2018-06-25 03:19:40 -04:00
end
2018-06-25 03:19:40 -04:00
context 'when data does not exist' do
it 'does not persist' do
expect { subject }.to raise_error('Can not persist empty data')
end
end
end
2018-06-25 03:19:40 -04:00
context 'when data_store is fog' do
let(:data_store) { :fog }
context 'when data exists' do
let(:data) { 'Sample data in fog' }
before do
build_trace_chunk.send(:unsafe_set_data!, data)
end
2018-06-25 03:19:40 -04:00
it 'does not change data store' do
expect(build_trace_chunk.fog?).to be_truthy
expect(Ci::BuildTraceChunks::Redis.new.data(build_trace_chunk)).to be_nil
expect(Ci::BuildTraceChunks::Database.new.data(build_trace_chunk)).to be_nil
expect(Ci::BuildTraceChunks::Fog.new.data(build_trace_chunk)).to eq(data)
2018-06-25 03:19:40 -04:00
subject
2018-06-25 03:19:40 -04:00
expect(build_trace_chunk.fog?).to be_truthy
expect(Ci::BuildTraceChunks::Redis.new.data(build_trace_chunk)).to be_nil
expect(Ci::BuildTraceChunks::Database.new.data(build_trace_chunk)).to be_nil
expect(Ci::BuildTraceChunks::Fog.new.data(build_trace_chunk)).to eq(data)
end
2018-06-25 04:59:46 -04:00
it_behaves_like 'Atomic operation'
2018-06-25 03:19:40 -04:00
end
end
end
2018-06-25 03:19:40 -04:00
2018-06-25 04:59:46 -04:00
describe 'deletes data in redis after a parent record destroyed' do
let(:project) { create(:project) }
before do
pipeline = create(:ci_pipeline, project: project)
create(:ci_build, :running, :trace_live, pipeline: pipeline, project: project)
create(:ci_build, :running, :trace_live, pipeline: pipeline, project: project)
create(:ci_build, :running, :trace_live, pipeline: pipeline, project: project)
end
shared_examples_for 'deletes all build_trace_chunk and data in redis' do
it do
Gitlab::Redis::SharedState.with do |redis|
expect(redis.scan_each(match: "gitlab:ci:trace:*:chunks:*").to_a.size).to eq(3)
end
expect(described_class.count).to eq(3)
subject
expect(described_class.count).to eq(0)
Gitlab::Redis::SharedState.with do |redis|
expect(redis.scan_each(match: "gitlab:ci:trace:*:chunks:*").to_a.size).to eq(0)
end
end
end
context 'when traces are archived' do
let(:subject) do
project.builds.each do |build|
build.success!
end
end
it_behaves_like 'deletes all build_trace_chunk and data in redis'
end
context 'when project is destroyed' do
let(:subject) do
project.destroy!
end
it_behaves_like 'deletes all build_trace_chunk and data in redis'
end
end
end