issue #45: new command line interface!

3 files changed, 173 insertions, 20 deletions

diff --git a/test/executable/source.py b/test/executable/source.py
new file mode 100644
index 0000000..1bb2c00
--- /dev/null
+++ b/test/executable/source.py
@@ -0,0 +1 @@
+class ClassName(): pass
\ No newline at end of file
diff --git a/test/executable/source.rb b/test/executable/source.rb
new file mode 100644
index 0000000..226f15f
--- /dev/null
+++ b/test/executable/source.rb
@@ -0,0 +1 @@
+class ClassName; end
\ No newline at end of file
diff --git a/test/executable/suite.rb b/test/executable/suite.rb
index ddb0527..fd40909 100644
--- a/test/executable/suite.rb
+++ b/test/executable/suite.rb
@@ -1,4 +1,6 @@
 require 'test/unit'
+require 'shoulda-context'
+
 require 'pathname'
 $:.unshift 'lib'
 require 'coderay'
@@ -8,38 +10,187 @@ puts "Running CodeRay #{CodeRay::VERSION} executable tests..."
 class TestCodeRayExecutable < Test::Unit::TestCase
   
   ruby = `ps -c #$$`[/\w+\Z/]
-  ruby = 'jruby' if ruby == 'java'
   
   ROOT_DIR = Pathname.new(File.dirname(__FILE__)) + '..' + '..'
   EXECUTABLE = ROOT_DIR + 'bin' + 'coderay'
   EXE_COMMAND = '%s -wI%s %s'% [
-    ruby,  # calling Ruby process command
+    ruby,              # calling Ruby process command
     ROOT_DIR + 'lib',  # library dir
-    EXECUTABLE
+    EXECUTABLE         # coderay
   ]
   
-  def coderay args
-    command = "#{EXE_COMMAND} #{args}"
-    # puts command
-    `#{command}`
+  def coderay args, fake_tty = false
+    if fake_tty
+      command = "#{EXE_COMMAND} #{args} --tty"
+    else
+      command = "#{EXE_COMMAND} #{args}"
+    end
+    puts command if $DEBUG
+    output = `#{command} 2>&1`
+    if output[EXECUTABLE.to_s]
+      raise output
+    else
+      output
+    end
+  end
+  
+  context 'a simple call with no arguments' do
+    should 'work' do
+      assert_nothing_raised { coderay('') }
+    end
+    should 'print version and help' do
+      assert_match(/CodeRay #{CodeRay::VERSION}/, coderay(''))
+      assert_match(/Usage:/, coderay(''))
+    end
+  end
+  
+  context 'version' do
+    should 'be printed with -v' do
+      assert_match(/\ACodeRay #{CodeRay::VERSION}\Z/, coderay('-v'))
+    end
+    should 'be printed with --version' do
+      assert_match(/\ACodeRay #{CodeRay::VERSION}\Z/, coderay('--version'))
+    end
+  end
+  
+  context 'help' do
+    should 'be printed with -h' do
+      assert_match(/^Usage:/, coderay('-h'))
+    end
+    should 'be printed with --help' do
+      assert_match(/^Usage:/, coderay('--help'))
+    end
+    should 'be printed with subcommand help' do
+      assert_match(/^Usage:/, coderay('help'))
+    end
+  end
+  
+  context 'commands' do
+    should 'be printed with subcommand commands' do
+      assert_match(/^ +help/, coderay('commands'))
+      assert_match(/^ +version/, coderay('commands'))
+    end
+  end
+  
+  context 'highlighting a file to the terminal' do
+    source_file = 'test/executable/source.py'
+    
+    source = File.read source_file
+    
+    ansi_seq = /\e\[[0-9;]+m/
+    
+    should 'not throw an error' do
+      assert_nothing_raised { coderay(source_file, :tty) }
+    end
+    should 'output its contents to stdout' do
+      target = coderay(source_file, :tty)
+      assert_equal source, target.chomp.gsub(ansi_seq, '')
+    end
+    should 'output ANSI-colored text' do
+      target = coderay(source_file, :tty)
+      assert_not_equal source, target.chomp
+      assert_equal 6, target.scan(ansi_seq).size
+    end
+  end
+  
+  context 'highlighting a file into a pipe (source.rb > source.rb.html)' do
+    source_file = 'test/executable/source.rb'
+    target_file = "#{source_file}.html"
+    command = "#{source_file} > #{target_file}"
+    
+    source = File.read source_file
+    
+    pre = %r{<pre [^>]*>(.*?)</pre>}m
+    tag = /<[^>]*>/
+    
+    should 'not throw an error' do
+      assert_nothing_raised { coderay(command) }
+    end
+    should 'output its contents to the pipe' do
+      coderay(command)
+      target = File.read(target_file)
+      if target = target[pre, 1]
+        assert_equal source, target.gsub(tag, '').strip
+      else
+        flunk "target code has no <pre> tag: #{target}"
+      end
+    end
+    should 'output valid HTML' do
+      coderay(command)
+      target = File.read(target_file)
+      assert_not_equal source, target[pre, 1]
+      assert_equal 6, target[pre, 1].scan(tag).size
+      assert_match %r{\A<!DOCTYPE html>\n<html>\n<head>}, target
+    end
+  end
+  
+  context 'highlighting a file into another file (source.rb source.rb.json)' do
+    source_file = 'test/executable/source.rb'
+    target_file = "#{source_file}.json"
+    command = "#{source_file} #{target_file}"
+    
+    source = File.read source_file
+    
+    text = /"text":"([^"]*)"/
+    
+    should 'not throw an error' do
+      assert_nothing_raised { coderay(command) }
+    end
+    should 'output its contents to the file' do
+      coderay(command)
+      target = File.read(target_file)
+      assert_equal source, target.scan(text).join
+    end
+    should 'output JSON' do
+      coderay(command)
+      target = File.read(target_file)
+      assert_not_equal source, target
+      assert_equal 6, target.scan(text).size
+    end
   end
   
-  def test_simple
-    assert_nothing_raised { coderay('') }
+  context 'highlighting a file without explicit input type (source.py)' do
+    source_file = 'test/executable/source.py'
+    command = source_file
+    
+    source = File.read source_file
+    
+    pre = %r{<pre [^>]*>(.*?)</pre>}m
+    tag_class = /<span class="([^>"]*)"?[^>]*>/
+    
+    should 'respect the file extension and highlight the input as Python' do
+      target = coderay(command)
+      assert_equal %w(kw cl kw), target[pre, 1].scan(tag_class).flatten
+    end
   end
   
-  VERSION_PATTERN = /CodeRay \d\.\d\.\d/
-  def test_version
-    assert_match(VERSION_PATTERN, coderay(''))
-    # assert_match(VERSION_PATTERN, coderay('--version'))
-    # assert_match(VERSION_PATTERN, coderay('-v'))
+  context 'highlighting a file with explicit input type (-ruby source.py)' do
+    source_file = 'test/executable/source.py'
+    command = "-ruby #{source_file}"
+    
+    source = File.read source_file
+    
+    pre = %r{<pre [^>]*>(.*?)</pre>}m
+    tag_class = /<span class="([^>"]*)"?[^>]*>/
+    
+    should 'ignore the file extension and highlight the input as Ruby' do
+      target = coderay(command)
+      assert_equal %w(kw cl), target[pre, 1].scan(tag_class).flatten
+    end
   end
   
-  HELP_PATTERN = /Usage:/
-  def test_help
-    assert_match(HELP_PATTERN, coderay(''))
-    # assert_match(HELP_PATTERN, coderay('--help'))
-    # assert_match(HELP_PATTERN, coderay('-h'))
+  context 'highlighting a file with explicit input and output type (-ruby source.py -span)' do
+    source_file = 'test/executable/source.py'
+    command = "-ruby #{source_file} -span"
+    
+    source = File.read source_file
+    
+    span_tags = /<\/?span[^>]*>/
+    
+    should 'just respect the output type and include span tags' do
+      target = coderay(command)
+      assert_equal source, target.chomp.gsub(span_tags, '')
+    end
   end
   
-end
\ No newline at end of file
+end