debian-mirror-gitlab/spec/lib/gitlab/ci/trace_spec.rb

135 lines
3.6 KiB
Ruby
Raw Normal View History

2019-10-12 21:52:04 +05:30
# frozen_string_literal: true
2017-08-17 22:00:37 +05:30
require 'spec_helper'
2021-01-03 14:25:43 +05:30
RSpec.describe Gitlab::Ci::Trace, :clean_gitlab_redis_shared_state, factory_default: :keep do
let_it_be(:project) { create_default(:project) }
let_it_be_with_reload(:build) { create(:ci_build) }
2017-08-17 22:00:37 +05:30
let(:trace) { described_class.new(build) }
describe "associations" do
it { expect(trace).to respond_to(:job) }
it { expect(trace).to delegate_method(:old_trace).to(:job) }
end
2020-11-24 15:15:51 +05:30
context 'when trace is migrated to object storage' do
let!(:job) { create(:ci_build, :trace_artifact) }
let!(:artifact1) { job.job_artifacts_trace }
let!(:artifact2) { job.reload.job_artifacts_trace }
let(:test_data) { "hello world" }
before do
stub_artifacts_object_storage
artifact1.file.migrate!(ObjectStorage::Store::REMOTE)
end
it 'reloads the trace after is it migrated' do
stub_const('Gitlab::HttpIO::BUFFER_SIZE', test_data.length)
expect_next_instance_of(Gitlab::HttpIO) do |http_io|
expect(http_io).to receive(:get_chunk).and_return(test_data, "")
end
expect(artifact2.job.trace.raw).to eq(test_data)
end
2021-01-03 14:25:43 +05:30
it 'reloads the trace in case of a chunk error' do
chunk_error = described_class::ChunkedIO::FailedToGetChunkError
allow_any_instance_of(described_class::Stream)
.to receive(:raw).and_raise(chunk_error)
expect(build).to receive(:reset).and_return(build)
expect { trace.raw }.to raise_error(chunk_error)
end
2020-11-24 15:15:51 +05:30
end
2018-10-15 14:42:47 +05:30
context 'when live trace feature is disabled' do
2017-08-17 22:00:37 +05:30
before do
2018-10-15 14:42:47 +05:30
stub_feature_flags(ci_enable_live_trace: false)
2017-08-17 22:00:37 +05:30
end
2018-10-15 14:42:47 +05:30
it_behaves_like 'trace with disabled live trace feature'
2017-08-17 22:00:37 +05:30
end
2018-10-15 14:42:47 +05:30
context 'when live trace feature is enabled' do
2017-08-17 22:00:37 +05:30
before do
2018-10-15 14:42:47 +05:30
stub_feature_flags(ci_enable_live_trace: true)
2018-03-27 19:54:05 +05:30
end
2018-10-15 14:42:47 +05:30
it_behaves_like 'trace with enabled live trace feature'
2018-03-27 19:54:05 +05:30
end
2020-03-13 15:44:24 +05:30
describe '#update_interval' do
context 'it is not being watched' do
it 'returns 30 seconds' do
expect(trace.update_interval).to eq(30.seconds)
end
end
context 'it is being watched' do
before do
trace.being_watched!
end
it 'returns 3 seconds' do
expect(trace.update_interval).to eq(3.seconds)
end
end
end
describe '#being_watched!' do
let(:cache_key) { "gitlab:ci:trace:#{build.id}:watched" }
it 'sets gitlab:ci:trace:<job.id>:watched in redis' do
trace.being_watched!
result = Gitlab::Redis::SharedState.with do |redis|
redis.exists(cache_key)
end
expect(result).to eq(true)
end
it 'updates the expiry of gitlab:ci:trace:<job.id>:watched in redis', :clean_gitlab_redis_shared_state do
Gitlab::Redis::SharedState.with do |redis|
redis.set(cache_key, true, ex: 4.seconds)
end
expect do
trace.being_watched!
end.to change { Gitlab::Redis::SharedState.with { |redis| redis.pttl(cache_key) } }
end
end
describe '#being_watched?' do
context 'gitlab:ci:trace:<job.id>:watched in redis is set', :clean_gitlab_redis_shared_state do
before do
Gitlab::Redis::SharedState.with do |redis|
redis.set("gitlab:ci:trace:#{build.id}:watched", true)
end
end
it 'returns true' do
expect(trace.being_watched?).to be(true)
end
end
context 'gitlab:ci:trace:<job.id>:watched in redis is not set' do
it 'returns false' do
expect(trace.being_watched?).to be(false)
end
end
end
2021-01-03 14:25:43 +05:30
describe '#lock' do
it 'acquires an exclusive lease on the trace' do
trace.lock do
expect { trace.lock }
.to raise_error described_class::LockedError
end
end
end
2017-08-17 22:00:37 +05:30
end