Assertions on non-tty

nobu · nobu · commit ede12890d2c9 · 2021-04-08T21:46:07.000+09:00
diff --git a/test/irb/test_color.rb b/test/irb/test_color.rb
@@ -31,6 +31,8 @@ def test_colorize
         [:CYAN]      => "#{CYAN}#{text}#{CLEAR}",
       }.each do |seq, result|
         assert_equal_with_term(result, text, seq: seq)
+
+        assert_equal_with_term(text, text, seq: seq, tty: false)
       end
     end
 
@@ -124,6 +126,9 @@ def test_colorize_code
         if colorize_code_supported?
           assert_equal_with_term(result, code, complete: true)
           assert_equal_with_term(result, code, complete: false)
+
+          assert_equal_with_term(code, code, complete: true, tty: false)
+          assert_equal_with_term(code, code, complete: false, tty: false)
         else
           assert_equal_with_term(code, code)
         end
@@ -141,6 +146,8 @@ def test_colorize_code_complete_true
         "('foo" => "(#{RED}#{BOLD}'#{CLEAR}#{RED}#{REVERSE}foo#{CLEAR}",
       }.each do |code, result|
         assert_equal_with_term(result, code, complete: true)
+
+        assert_equal_with_term(code, code, complete: true, tty: false)
       end
     end
 
@@ -153,8 +160,12 @@ def test_colorize_code_complete_false
         if colorize_code_supported?
           assert_equal_with_term(result, code, complete: false)
 
+          assert_equal_with_term(code, code, complete: false, tty: false)
+
           unless complete_option_supported?
             assert_equal_with_term(result, code, complete: true)
+
+            assert_equal_with_term(code, code, complete: true, tty: false)
           end
         else
           assert_equal_with_term(code, code)
@@ -194,10 +205,10 @@ def complete_option_supported?
       Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('2.7.0')
     end
 
-    def with_term
+    def with_term(tty: true)
       stdout = $stdout
       io = StringIO.new
-      def io.tty?; true; end
+      def io.tty?; true; end if tty
       $stdout = io
 
       env = ENV.to_h.dup
@@ -209,8 +220,8 @@ def io.tty?; true; end
       ENV.replace(env) if env
     end
 
-    def assert_equal_with_term(result, code, seq: nil, **opts)
-      actual = with_term do
+    def assert_equal_with_term(result, code, seq: nil, tty: true, **opts)
+      actual = with_term(tty: tty) do
         if seq
           IRB::Color.colorize(code, seq, **opts)
         else
