[ruby/error_highlight] Redefine in the module context

https://github.com/ruby/error_highlight/commit/0048bd0285

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