diff --git a/lib/irb/color_printer.rb b/lib/irb/color_printer.rb index 187c33718..3667ef16f 100644 --- a/lib/irb/color_printer.rb +++ b/lib/irb/color_printer.rb @@ -10,7 +10,12 @@ def self.pp(obj, out = $>, width = 79) out end - def text(str, width = str.length) + def text(str, width = nil) + unless str.is_a?(String) + str = str.inspect + end + width ||= str.length + case str when /\A#' super(Color.colorize(str, [:GREEN]), width) diff --git a/test/irb/test_color.rb b/test/irb/test_color.rb index c0682b1bb..6ce652104 100644 --- a/test/irb/test_color.rb +++ b/test/irb/test_color.rb @@ -1,6 +1,7 @@ # frozen_string_literal: false require 'test/unit' require 'irb/color' +require 'irb/color_printer' require 'rubygems' require 'stringio' @@ -152,6 +153,20 @@ def test_colorize_code_complete_false end end + def test_color_printer + unless ripper_lexer_scan_supported? + skip 'Ripper::Lexer#scan is supported in Ruby 2.7+' + end + { + 1 => "#{BLUE}#{BOLD}1#{CLEAR}", + Struct.new('IRBTestColorPrinter', :a).new('test') => "#{GREEN}##{CLEAR}", + Ripper::Lexer.new('1').scan => "[#{GREEN}##{CLEAR}]", + }.each do |object, result| + actual = with_term { IRB::ColorPrinter.pp(object, '') } + assert_equal(result, actual, "Case: IRB::ColorPrinter.pp(#{object.inspect}, '')") + end + end + def test_inspect_colorable { 1 => true, @@ -184,6 +199,10 @@ def complete_option_supported? Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('2.7.0') end + def ripper_lexer_scan_supported? + Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('2.7.0') + end + def with_term stdout = $stdout io = StringIO.new