module Mocha::Integration::MonkeyPatcher

def self.apply(mod, run_method_patch)
  if mod < Mocha::API
    Debug.puts "Mocha::API already included in #{mod}"
  else
    mod.send(:include, Mocha::API)
  end
  if mod.method_defined?(:run_before_mocha)
    Debug.puts "#{mod}#run_before_mocha method already defined"
  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