easy-going-rails/test/formatters/test_wrapper.rb

require_relative '../../lib/formatters/wrapper'
require 'test/unit'

class TestWrapper < Test::Unit::TestCase
  def test_wrap_score
    assert_wrap("\
********************\n\
********************\n\
**********",
                ('*' * 50), 20)

    assert_wrap("\
> ******************\n\
> ******************\n\
> **************",
                ('*' * 50), 20, '> ')
  end

  def test_wrap_sql
    assert_wrap("\
\e[0;35mScore Load (0.6ms)\e[0m \e[0;36mS\e[0m\n\
\e[0;36mELECT \"scores\".* FRO\e[0m\n\
\e[0;36mM \"scores\" WHERE \"sc\e[0m\n\
\e[0;36mores\".\"id\" = $1 LIMI\e[0m\n\
\e[0;36mT $2\e[0m [[\"id\", 2], [\"L\n\
IMIT\", 1]]", "\
\e[0;35mScore Load (0.6ms)\e[0m \e[0;36mSELECT \"scores\".* FROM \"scores\" WHERE \"scores\".\"id\" = $1 LIMIT $2\e[0m [[\"id\", 2], [\"LIMIT\", 1]]",
                20)
  end

  private

  def assert_wrap(expectation, text, length, prefix = '')
    assert_equal(expectation, prefix + Wrapper.wrap(text, prefix, length - prefix.length))
    test_with_style(text, length, prefix)
  end

  def test_with_style(text, length, prefix = '')
    begint = "my text is:\n"
    endoft = "a final is:\n"
    puts "#{begint}#{prefix}#{text}\e[0m"
    puts "#{endoft}#{prefix}#{Wrapper.wrap(text, prefix, length - prefix.length)}\e[0m"
    puts
  end
end