ruby_test.rb

#!/usr/bin/env ruby

require 'fileutils'

class TestFinder

  attr_reader :prefix

  def initialize(prefix)
    @prefix  = prefix
  end


  def name_to_file(name)
    "#{prefix}/#{name}_test.rb"
  end

  def file_to_name(file)
    file.sub(%r{^#{Regexp.escape(prefix)}/}, '').sub(%r{_test\.rb$}, '')
  end

  def test_files
    @test_files ||= Dir["#{prefix}/**/*_test.rb"]
  end

  def test_names
    @test_names ||= test_files.collect {|file| file_to_name(file) }
  end

  def run(tests, pattern)
    if tests.empty?
      if test_files.empty?
        puts "Nothing to do."
      else
        run(test_files)
      end
    else
      tests = tests.collect {|test| force_test_file(test) }
      command = command_to_run(tests, pattern)
      puts command
      exec command
    end
  end

  def list
    puts test_names.sort.join("\n")
  end

  private

  def force_test_file(test)
    if test_files.include?(test)
      test
    elsif test_names.include?(test)
      name_to_file(test)
    else
      raise "No such test: #{test.inspect}"
    end
  end

  def command_to_run(files, pattern)
    files_string = files.inspect
    require_string = 'require "#{f}"'
    command_arguments = %w(-Itest -e)
    command_arguments << "#{files_string}.each {|f| #{require_string} }"
    if pattern != ''
      command_arguments << '--' << '-n' << "/#{pattern}/"
    end
    exec('ruby', *command_arguments)
  end
end

list = ARGV.delete('-l') || ARGV.delete('--list')
prefix = ARGV.shift
prefix = prefix ? "test/#{prefix}" : 'test'
finder = TestFinder.new(prefix)
args = ARGV.dup

if list
  finder.list
else
  pattern = []
  tests   = []

  ARGV.each do |arg|
    if pattern.size > 0 || arg =~ /^\//
      pattern << arg.sub(/^\//, '')
    else
      tests << arg
    end
  end
  pattern = pattern.join(' ')

  finder.run(tests, pattern)
end