minitest-colorize 0.0.1 → 0.0.2

data/lib/minitest/colorize.rb

@@ -1,4 +1,5 @@
 require "minitest/colorize/version"
+require "minitest/unit"
 
 module MiniTest
   class Colorize
@@ -8,25 +9,49 @@ module MiniTest
       self.io = io
     end
 
-    def print(out)
-      if color = colors[out]
-        io.print "#{color}#{out}#{clear}"
+    def print(string = nil)
+      return io.print if string.nil?
+
+      if color = colors[string]
+        io.print tint(color, string)
       else
-        io.print out
+        io.print string
       end
     end
 
-    def colors
-      { "F" => "\e[31m", "E" => "\e[31m", "S" => "\e[33m", "." => "\e[32m" }
-    end
+    def puts(string = nil)
+      return io.puts if string.nil?
 
-    def clear
-      "\e[0m"
+      if string =~ /(\d+) tests, (\d+) assertions, (\d+) failures, (\d+) errors, (\d+) skips/
+        color = if $3 != '0'
+                  colors['F']
+                elsif $4 != '0'
+                  colors['E']
+                elsif $5 != '0'
+                  colors['S']
+                else
+                  colors['.']
+                end
+
+        io.puts tint(color, string)
+      else
+        io.puts string
+      end
     end
 
     def method_missing(method, *args, &block)
       io.send(method, *args, &block)
     end
+
+    protected
+
+    def tint(color, string)
+      "\e[#{color}m#{string}\e[0m"
+    end
+
+    def colors
+      { "F" => 31, "E" => 31, "S" => 33, "." => 32 }
+    end
   end
 end
 

data/lib/minitest/colorize/version.rb

@@ -1,5 +1,5 @@
 module MiniTest
   class Colorize
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

data/test/minitest_colorize_test.rb

@@ -1,38 +1,87 @@
 require 'test_helper'
 
 class MiniTest::ColorizeTest < MiniTest::Unit::TestCase
-  def test_fail
+  def test_print_fail
     assert_output "\e[31mF\e[0m" do
       colorize = MiniTest::Colorize.new($stdout)
       colorize.print 'F'
     end
   end
 
-  def test_error
+  def test_print_error
     assert_output "\e[31mE\e[0m" do
       colorize = MiniTest::Colorize.new($stdout)
       colorize.print 'E'
     end
   end
 
-  def test_skip
+  def test_print_skip
     assert_output "\e[33mS\e[0m" do
       colorize = MiniTest::Colorize.new($stdout)
       colorize.print 'S'
     end
   end
 
-  def test_success
+  def test_print_success
     assert_output "\e[32m.\e[0m" do
       colorize = MiniTest::Colorize.new($stdout)
       colorize.print '.'
     end
   end
 
-  def test_others
+  def test_print
     assert_output "colorize" do
       colorize = MiniTest::Colorize.new($stdout)
       colorize.print 'colorize'
     end
   end
+
+  def test_print_without_arguments
+    assert_output '' do
+      colorize = MiniTest::Colorize.new($stdout)
+      colorize.print
+    end
+  end
+
+  def test_puts_fail
+    assert_output "\e[31m12 tests, 12 assertions, 4 failures, 0 errors, 0 skips\e[0m\n" do
+      colorize = MiniTest::Colorize.new($stdout)
+      colorize.puts '12 tests, 12 assertions, 4 failures, 0 errors, 0 skips'
+    end
+  end
+
+  def test_puts_error
+    assert_output "\e[31m12 tests, 12 assertions, 0 failures, 4 errors, 0 skips\e[0m\n" do
+      colorize = MiniTest::Colorize.new($stdout)
+      colorize.puts '12 tests, 12 assertions, 0 failures, 4 errors, 0 skips'
+    end
+  end
+
+  def test_puts_skip
+    assert_output "\e[33m12 tests, 12 assertions, 0 failures, 0 errors, 4 skips\e[0m\n" do
+      colorize = MiniTest::Colorize.new($stdout)
+      colorize.puts '12 tests, 12 assertions, 0 failures, 0 errors, 4 skips'
+    end
+  end
+
+  def test_puts_success
+    assert_output "\e[32m12 tests, 12 assertions, 0 failures, 0 errors, 0 skips\e[0m\n" do
+      colorize = MiniTest::Colorize.new($stdout)
+      colorize.puts '12 tests, 12 assertions, 0 failures, 0 errors, 0 skips'
+    end
+  end
+
+  def test_puts
+    assert_output "colorize\n" do
+      colorize = MiniTest::Colorize.new($stdout)
+      colorize.puts 'colorize'
+    end
+  end
+
+  def test_puts_without_arguments
+    assert_output "\n" do
+      colorize = MiniTest::Colorize.new($stdout)
+      colorize.puts
+    end
+  end
 end

metadata

@@ -2,7 +2,7 @@
 name: minitest-colorize
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors: 
 - Gabriel Sobrinho
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-06-24 00:00:00 -03:00
+date: 2011-06-25 00:00:00 -03:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency