diff options
| -rw-r--r-- | test/error_highlight/test_error_highlight.rb | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/test/error_highlight/test_error_highlight.rb b/test/error_highlight/test_error_highlight.rb index 9a447328c1..6136bcdcc7 100644 --- a/test/error_highlight/test_error_highlight.rb +++ b/test/error_highlight/test_error_highlight.rb @@ -1362,12 +1362,17 @@ undefined method `time' for #{ ONE_RECV_MESSAGE } warning = nil original_warn = Warning.method(:warn) - Warning.define_method(:warn) {|s| warning = s} + Warning.class_eval do + remove_method(:warn) + define_method(:warn) {|str| warning = str} + end begin ErrorHighlight::DefaultFormatter.max_snippet_width = custom_max_width ensure - Warning.remove_method(:warn) - Warning.define_method(:warn, original_warn) + Warning.class_eval do + remove_method(:warn) + define_method(:warn, original_warn) + end end assert_match "'max_snippet_width' adjusted to minimum value of #{min_snippet_width}", warning |