File: monkey_patcher.rb

package info (click to toggle)
ruby-mocha 3.1.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,656 kB
  • sloc: ruby: 12,304; javascript: 499; makefile: 14
file content (26 lines) | stat: -rw-r--r-- 770 bytes parent folder | download
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
# frozen_string_literal: true

require 'mocha/api'

module Mocha
  module Integration
    module MonkeyPatcher
      def self.apply(mod, run_method_patch)
        if mod < Mocha::API
          warn "Mocha::API already included in #{mod}" if $DEBUG
        else
          mod.send(:include, Mocha::API)
        end
        if mod.method_defined?(:run_before_mocha)
          warn "#{mod}#run_before_mocha method already defined" if $DEBUG
        elsif mod.method_defined?(:run)
          mod.send(:alias_method, :run_before_mocha, :run)
          mod.send(:remove_method, :run)
          mod.send(:include, run_method_patch)
        else
          raise "Unable to monkey-patch #{mod}, because it does not define a `#run` method"
        end
      end
    end
  end
end