Class: RSpec::Mocks::Matchers::Receive

Inherits:
Object
  • Object
show all
Defined in:
lib/rspec/armour/receive_patch.rb

Overview

This class is patched to prevent mocking ActiveRecord associations and finders.

Constant Summary collapse

NEGATIVE_EXPECTATION_METHODS = 
%i[
  setup_negative_expectation
  setup_any_instance_negative_expectation
  does_not_match?
].freeze

Class Method Summary collapse

Class Method Details

.patch_for_armour! ⇒ Object 



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/rspec/armour/receive_patch.rb', line 14

def self.patch_for_armour!
  # rubocop:disable ThreadSafety/ClassInstanceVariable
  @armour_patched ||= false
  return if @armour_patched

  @armour_patched = true
  # rubocop:enable ThreadSafety/ClassInstanceVariable

  %i[
    setup_expectation
    setup_negative_expectation
    setup_allowance
    setup_any_instance_expectation
    setup_any_instance_negative_expectation
    setup_any_instance_allowance
    matches?
    does_not_match?
  ].each do |method|
    next unless method_defined?(method)

    patch_method(method)
  end
end

.patch_method(method) ⇒ Object 



38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/rspec/armour/receive_patch.rb', line 38

def self.patch_method(method)
  original_method = instance_method(method)
  negative = NEGATIVE_EXPECTATION_METHODS.include?(method)
  define_method(method) do |subject, &block|
    if ENV["RSPEC_ARMOUR_DEBUG"] == "true"
      puts "Checking RSpec::Armour '#{@message}' on #{subject}"
    end
    if negative
      RSpec::Armour::Checker.as_negative_expectation! do
        original_method.bind_call(self, subject, &block)
      end
    else
      if RSpec::Armour::Checker.restricted?(subject, @message)
        message = "Mocking/stubbing ActiveRecord association or finder `#{@message}` " \
                  "on `#{subject.inspect}` is restricted by rspec-armour."
        raise RSpec::Armour::MockError, message
      end
      original_method.bind_call(self, subject, &block)
    end
  end
end