spec/lib/gitlab/ci/trace/chunked_file/live_trace_spec.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

require 'spec_helper'

describe Gitlab::Ci::Trace::ChunkedFile::LiveTrace, :clean_gitlab_redis_cache do
  include ChunkedIOHelpers

  let(:chunked_io) { described_class.new(job_id, nil, mode) }
  let(:job) { create(:ci_build) }
  let(:job_id) { job.id }
  let(:mode) { 'rb' }

  let(:chunk_stores) do
    [Gitlab::Ci::Trace::ChunkedFile::ChunkStore::Redis,
      Gitlab::Ci::Trace::ChunkedFile::ChunkStore::Database]
  end

  describe 'ChunkStores are Redis and Database', :partial_support do
    it_behaves_like 'ChunkedIO shared tests'
  end

  describe '.exist?' do
    subject { described_class.exist?(job_id) }

    context 'when a chunk exists in a store' do
      before do
        fill_trace_to_chunks(sample_trace_raw)
      end

      it { is_expected.to be_truthy }
    end

    context 'when chunks do not exists in any store' do
      it { is_expected.to be_falsey }
    end
  end

  describe '#truncate' do
    subject { chunked_io.truncate(offset) }

    let(:mode) { 'a+b' }

    before do
      fill_trace_to_chunks(sample_trace_raw)
    end

    context 'when offset is 0' do
      let(:offset) { 0 }

      it 'deletes all chunks' do
        expect { subject }.to change { described_class.exist?(job_id) }.from(true).to(false)
      end
    end

    context 'when offset is size' do
      let(:offset) { sample_trace_raw.length }

      it 'does nothing' do
        expect { subject }.not_to change { described_class.exist?(job_id) }
      end
    end

    context 'when offset is else' do
      let(:offset) { 10 }

      it 'raises an error' do
        expect { subject }.to raise_error('Unexpected operation')
      end
    end
  end

  describe '#delete' do
    subject { chunked_io.delete }

    context 'when a chunk exists in a store' do
      before do
        fill_trace_to_chunks(sample_trace_raw)
      end

      it 'deletes' do
        expect { subject }.to change { described_class.exist?(job_id) }.from(true).to(false)
      end
    end

    context 'when chunks do not exists in any store' do
      it 'deletes' do
        expect { subject }.not_to change { described_class.exist?(job_id) }
      end
    end
  end
end