File: ignoring_warning.rb

package info (click to toggle)
ruby-mocha 3.0.2-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,652 kB
  • sloc: ruby: 12,324; javascript: 499; makefile: 14
file content (20 lines) | stat: -rw-r--r-- 503 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
module Mocha
  # @private
  module IgnoringWarning
    def ignoring_warning(pattern, if_: true)
      return yield unless if_

      begin
        original_warn = Warning.method(:warn)
        Warning.singleton_class.define_method(:warn) do |message|
          original_warn.call(message) unless message =~ pattern
        end

        yield
      ensure
        Warning.singleton_class.undef_method(:warn)
        Warning.singleton_class.define_method(:warn, original_warn)
      end
    end
  end
end