gems/activerecord-gitlab/spec/active_record/gitlab_patches/rescue_from_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

# frozen_string_literal: true

RSpec.describe ActiveRecord::GitlabPatches::RescueFrom do
  let(:model_with_rescue_from) do
    Class.new(Project) do
      rescue_from ActiveRecord::StatementInvalid, with: :handle_exception

      class << self
        def handle_exception(exception); end
      end
    end
  end

  let(:model_without_rescue_from) do
    Class.new(Project)
  end

  context 'for errors from ActiveRelation.load' do
    it 'triggers rescue_from' do
      expect(model_with_rescue_from).to receive(:handle_exception)

      expect { model_with_rescue_from.where('BADQUERY').load }.not_to raise_error
    end

    it 'does not trigger rescue_from' do
      expect { model_without_rescue_from.where('BADQUERY').load }.to raise_error(ActiveRecord::StatementInvalid)
    end
  end
end