Autotest.send(:alias_method, :real_make_test_cmd, :make_test_cmd)
Autotest.send(:define_method, :make_test_cmd) do |*args|
  real_make_test_cmd(*args).sub('test/unit', %[rubygems -e "require 'redgreen'"])
end

module Autotest::Growl
  def self.growl title, msg, pri=0, stick=""
    system "growlnotify -n autotest --image /opt/local/lib/ruby/gems/1.8/gems/rails-1.1.6/html/images/rails.png -p #{pri} -m #{msg.inspect} #{title} #{stick}"
  end

  Autotest.add_hook :ran_command do |at|
    output = at.results.slice(/(\d+).*errors/)
    if output =~ /ns.*[1-9]/
      growl "Test Results", "#{output}"[3..-1], 2, "-s"
    else
      growl "Test Results", "#{output}"[3..-1]
    end
  end
end