Default DidYouMean.formatter to PlainFormatter

1 files changed, 5 insertions, 1 deletions

diff --git a/test/optparse/test_did_you_mean.rb b/test/optparse/test_did_you_mean.rb
index 2d0a9a66a2..82a086acae 100644
--- a/test/optparse/test_did_you_mean.rb
+++ b/test/optparse/test_did_you_mean.rb
@@ -9,6 +9,11 @@ class TestOptionParser::DidYouMean < TestOptionParser
     @opt.def_option("--bar", Integer) { |v| @bar = v }
     @opt.def_option("--baz", Integer) { |v| @baz = v }
     @formatter = ::DidYouMean.formatter
+    case @formatter
+    when ::DidYouMean::PlainFormatter
+    else
+      ::DidYouMean.formatter = ::DidYouMean::PlainFormatter.new
+    end
   end
 
   def teardown
@@ -22,7 +27,6 @@ class TestOptionParser::DidYouMean < TestOptionParser
   end
 
   def test_plain
-    ::DidYouMean.formatter = ::DidYouMean::PlainFormatter.new
     assert_raise_with_message(OptionParser::InvalidOption, /invalid option: --baa\nDid you mean\?\s+baz\s+bar\Z/) do
       @opt.permute!(%w"--baa")
     end