Watchr config for Rails app using Test::Unit.

.watchr

LINUX = RUBY_PLATFORM =~ /linux/i unless defined?(LINUX)

def escape(arg)
  arg.gsub(/\e\[..m?/, '')         # rid us of ansi escape sequences
     .gsub(/["`]/, "'")
     .gsub(/\r?\n/, "\\\\\\\\\\n")
end

def notify(pass, heading, body='')
  cmd = if LINUX
    %(notify-send --hint=int:transient:1 "#{escape heading}" "#{escape body[0..400]}")
  else
    %(growlnotify -m "#{escape body}" "#{escape heading}")
  end
  system(cmd)
end

def run_tests(test, force=false)
  if force || File.exist?(test)
    puts "-" * 80
    rspec_cmd = File.exists?('.zeus.sock') ? "zeus test" : "rake test"
    puts "#{rspec_cmd} #{test}"
    cmd = IO.popen("#{rspec_cmd} #{test} 2>&1")
    result = ''
    until cmd.eof?
      char = cmd.getc
      result << char
      $stdout.write(char)
    end
    if result =~/0 failures, 0 errors/
      summary = $~.to_s
      secs = result.match(/Finished tests in ([\d\.]+s)/)[1]
      notify(true, 'Test Success', summary + ' ' + secs)
    elsif result =~ /(\d+) failures?, (\d+) errors?/
      summary = $~.to_s
      notify(false, summary)
    else
      notify(false, 'Test Error', 'One or more tests could not run due to error.')
    end
  else
    puts "#{test} does not exist."
  end
end

def run_suite
  run_tests('test/*', :force)
end

watch('^test/unit/(.*)_test\.rb'       ) { |m| run_tests("test/unit/#{m[1]}_test.rb")       }
watch('^test/functional/(.*)_test\.rb' ) { |m| run_tests("test/functional/#{m[1]}_test.rb") }
watch('^test/factories/(.*)\.rb'       ) { |m| run_tests("test/unit/#{m[1]}_test.rb")       }
watch('^test/integration/(.*)_test\.rb') { |m| run_tests("test/integration/#{m[1]}_test.rb")}
watch('^app/models/(.*)\.rb'           ) { |m| run_tests("test/unit/#{m[1]}_test.rb")       }
watch('^app/controllers/(.*)\.rb'      ) { |m| run_tests("test/functional/#{m[1]}_test.rb") }
watch('^lib/(.*)\.rb'                  ) { |m| run_tests("test/lib/#{m[1]}_test.rb")        }

@interrupt_received = false

# ^C
Signal.trap 'INT' do
  if @interrupt_received
    exit 0
  else
    @interrupt_received = true
    puts "\nInterrupt a second time to quit"
    Kernel.sleep 1
    @interrupt_received = false
    run_suite
  end
end