spec/rubocop/cop/usage_data/large_table_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
90

# frozen_string_literal: true

require 'fast_spec_helper'

require 'rubocop'
require 'rubocop/rspec/support'

require_relative '../../../../rubocop/cop/usage_data/large_table'

RSpec.describe RuboCop::Cop::UsageData::LargeTable, type: :rubocop do
  include CopHelper

  let(:large_tables) { %i[Rails Time] }
  let(:count_methods) { %i[count distinct_count] }
  let(:allowed_methods) { %i[minimum maximum] }

  let(:config) do
    RuboCop::Config.new('UsageData/LargeTable' => {
                          'NonRelatedClasses' => large_tables,
                          'CountMethods' => count_methods,
                          'AllowedMethods' => allowed_methods
                        })
  end

  subject(:cop) { described_class.new(config) }

  context 'when in usage_data files' do
    before do
      allow(cop).to receive(:usage_data_files?).and_return(true)
    end

    context 'with large tables' do
      context 'when calling Issue.count' do
        it 'register an offence' do
          inspect_source('Issue.count')

          expect(cop.offenses.size).to eq(1)
        end
      end

      context 'when calling Issue.active.count' do
        it 'register an offence' do
          inspect_source('Issue.active.count')

          expect(cop.offenses.size).to eq(1)
        end
      end

      context 'when calling count(Issue)' do
        it 'does not register an offence' do
          inspect_source('count(Issue)')

          expect(cop.offenses).to be_empty
        end
      end

      context 'when calling count(Ci::Build.active)' do
        it 'does not register an offence' do
          inspect_source('count(Ci::Build.active)')

          expect(cop.offenses).to be_empty
        end
      end

      context 'when calling Ci::Build.active.count' do
        it 'register an offence' do
          inspect_source('Ci::Build.active.count')

          expect(cop.offenses.size).to eq(1)
        end
      end

      context 'when using allowed methods' do
        it 'does not register an offence' do
          inspect_source('Issue.minimum')

          expect(cop.offenses).to be_empty
        end
      end
    end

    context 'with non related class' do
      it 'does not register an offence' do
        inspect_source('Rails.count')

        expect(cop.offenses).to be_empty
      end
    end
  end
end