[ruby/optparse] DidYouMean::PlainFormatter is deprecated

https://github.com/ruby/optparse/commit/0ac9957696

1 files changed, 11 insertions, 3 deletions

diff --git a/test/optparse/test_did_you_mean.rb b/test/optparse/test_did_you_mean.rb
index a9ad051f03..14f6ebee17 100644
--- a/test/optparse/test_did_you_mean.rb
+++ b/test/optparse/test_did_you_mean.rb
@@ -12,12 +12,20 @@ class TestOptionParser::DidYouMean < TestOptionParser
     @opt.def_option("--foo", Integer) { |v| @foo = v }
     @opt.def_option("--bar", Integer) { |v| @bar = v }
     @opt.def_option("--baz", Integer) { |v| @baz = v }
-    @formatter = ::DidYouMean.formatter
-    ::DidYouMean.formatter = ::DidYouMean::Formatter
+    unless ::DidYouMean::Formatter.respond_to?(:message_for)
+      @formatter = ::DidYouMean.formatter
+      case @formatter
+      when ::DidYouMean::PlainFormatter
+      else
+        ::DidYouMean.formatter = ::DidYouMean::PlainFormatter.new
+      end
+    end
   end
 
   def teardown
-    ::DidYouMean.formatter = @formatter
+    unless ::DidYouMean::Formatter.respond_to?(:message_for)
+      ::DidYouMean.formatter = @formatter
+    end
   end
 
   def test_no_suggestion