require 'rake_helper' describe 'gitlab:traces rake tasks' do before do Rake.application.rake_require 'tasks/gitlab/traces' end shared_examples 'passes the job id to worker' do it do expect(ArchiveTraceWorker).to receive(:bulk_perform_async).with([[job.id]]) run_rake_task('gitlab:traces:archive') end end shared_examples 'does not pass the job id to worker' do it do expect(ArchiveTraceWorker).not_to receive(:bulk_perform_async) run_rake_task('gitlab:traces:archive') end end context 'when trace file stored in default path' do let!(:job) { create(:ci_build, :success, :trace_live) } it_behaves_like 'passes the job id to worker' end context 'when trace is stored in database' do let!(:job) { create(:ci_build, :success) } before do job.update_column(:trace, 'trace in db') end it_behaves_like 'passes the job id to worker' end context 'when job has trace artifact' do let!(:job) { create(:ci_build, :success) } before do create(:ci_job_artifact, :trace, job: job) end it_behaves_like 'does not pass the job id to worker' end context 'when job is not finished yet' do let!(:build) { create(:ci_build, :running, :trace_live) } it_behaves_like 'does not pass the job id to worker' end end