spec/lib/gitlab/utils/deep_size_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

# frozen_string_literal: true

require 'spec_helper'

describe Gitlab::Utils::DeepSize do
  let(:data) do
    {
      a: [1, 2, 3],
      b: {
        c: [4, 5],
        d: [
          { e: [[6], [7]] }
        ]
      }
    }
  end

  let(:max_size) { 1.kilobyte }
  let(:max_depth) { 10 }
  let(:deep_size) { described_class.new(data, max_size: max_size, max_depth: max_depth) }

  describe '#evaluate' do
    context 'when data within size and depth limits' do
      it 'returns true' do
        expect(deep_size).to be_valid
      end
    end

    context 'when data not within size limit' do
      let(:max_size) { 200.bytes }

      it 'returns false' do
        expect(deep_size).not_to be_valid
      end
    end

    context 'when data not within depth limit' do
      let(:max_depth) { 2 }

      it 'returns false' do
        expect(deep_size).not_to be_valid
      end
    end
  end
end