Add colorize

test/test_blob.rb

@@ -170,4 +170,13 @@ class TestBlob < Test::Unit::TestCase
     assert_equal nil, blob("script.foo").shebang_language
     assert_equal nil, blob("foo.rb").shebang_language
   end
+
+  def test_colorize
+    assert_equal <<-HTML, blob("foo.rb").colorize
+<div class="highlight"><pre><span class="k">module</span> <span class="nn">Foo</span>
+<span class="k">end</span>
+</pre>
+</div>
+    HTML
+  end
 end

test/test_language.rb

@@ -95,4 +95,20 @@ class TestLanguage < Test::Unit::TestCase
     assert Language['Assembly'].unpopular?
     assert Language['Brainfuck'].unpopular?
   end
+
+  def test_colorize
+    assert_equal <<-HTML, Language['Text'].colorize("Hello")
+<div class="highlight"><pre>Hello
+</pre>
+</div>
+    HTML
+
+    assert_equal <<-HTML, Language['Ruby'].colorize("def foo\n  'foo'\nend\n")
+<div class="highlight"><pre><span class="k">def</span> <span class="nf">foo</span>
+  <span class="s1">&#39;foo&#39;</span>
+<span class="k">end</span>
+</pre>
+</div>
+    HTML
+  end
 end

test/test_lexer.rb

@@ -34,4 +34,20 @@ class TestLexer < Test::Unit::TestCase
     assert !Lexer['Ruby'].eql?(Lexer['Python'])
     assert !Lexer['Ruby'].eql?(Lexer.new('Ruby'))
   end
+
+  def test_colorize
+    assert_equal <<-HTML, Lexer['Text only'].colorize("Hello")
+<div class="highlight"><pre>Hello
+</pre>
+</div>
+    HTML
+
+    assert_equal <<-HTML, Lexer['Ruby'].colorize("def foo\n  'foo'\nend\n")
+<div class="highlight"><pre><span class="k">def</span> <span class="nf">foo</span>
+  <span class="s1">&#39;foo&#39;</span>
+<span class="k">end</span>
+</pre>
+</div>
+    HTML
+  end
 end