debian-mirror-gitlab/spec/rubocop/cop/database/rescue_statement_timeout_spec.rb

# frozen_string_literal: true

require 'rubocop_spec_helper'
require_relative '../../../../rubocop/cop/database/rescue_statement_timeout'

RSpec.describe RuboCop::Cop::Database::RescueStatementTimeout do
  it 'flags the use of ActiveRecord::StatementTimeout' do
    expect_offense(<<~CODE)
      begin
        do_something
      rescue ActiveRecord::StatementTimeout => e
      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Avoid rescuing the `ActiveRecord::StatementTimeout` [...]
        try_something_else
      end
    CODE
  end

  it 'does not flag a different exception' do
    expect_no_offenses(<<~CODE)
      begin
        do_something
      rescue ActiveRecord::RecordNotFound => e
        try_something_else
      end
    CODE
  end
end