require 'rbconfig' require 'rake/testtask' require 'rails/test_unit/sub_test_task'

TEST_CHANGES_SINCE = Time.now - 600

# Look up tests for recently modified sources. def recent_tests(source_pattern, test_path, touched_since = 10.minutes.ago)

FileList[source_pattern].map do |path|
  if File.mtime(path) > touched_since
    tests = []
    source_dir = File.dirname(path).split("/")
    source_file = File.basename(path, '.rb')

    # Support subdirs in app/models and app/controllers
    modified_test_path = source_dir.length > 2 ? "#{test_path}/" << source_dir[1..source_dir.length].join('/') : test_path

    # For modified files in app/ run the tests for it. ex. /test/functional/account_controller.rb
    test = "#{modified_test_path}/#{source_file}_test.rb"
    tests.push test if File.exist?(test)

    # For modified files in app, run tests in subdirs too. ex. /test/functional/account  _test.rb
    test = "#{modified_test_path}/#{File.basename(path, '.rb').sub("_controller","")}"
    FileList["#{test}  _test.rb"].each { |f| tests.push f } if File.exist?(test)

    return tests

  end
end.flatten.compact

end

# Recreated here from Active Support because :uncommitted needs it before Rails is available module Kernel

remove_method :silence_stderr # Removing old method to prevent method redefined warning
def silence_stderr
  old_stderr = STDERR.dup
  STDERR.reopen(RbConfig::CONFIG['host_os'] =~ /mswin|mingw/ ? 'NUL:' : '/dev/null')
  STDERR.sync = true
  yield
ensure
  STDERR.reopen(old_stderr)
end

end

task :default => :test

desc 'Runs test:units, test:functionals, test:integration together (also available: test:benchmark, test:profile, test:plugins)' task :test do

Rake::Task[ENV['TEST'] ? 'test:single' : 'test:run'].invoke

end

namespace :test do

task :prepare do
  # Placeholder task for other Railtie and plugins to enhance. See Active Record for an example.
end

task :run do
  errors = %w(test:units test:functionals test:integration).collect do |task|
    begin
      Rake::Task[task].invoke
      nil
    rescue => e
      { :task => task, :exception => e }
    end
  end.compact

  if errors.any?
    puts errors.map { |e| "Errors running #{e[:task]}! #{e[:exception].inspect}" }.join("\n")
    abort
  end
end

Rake::TestTask.new(:recent => "test:prepare") do |t|
  since = TEST_CHANGES_SINCE
  touched = FileList['test/   *_test.rb'].select { |path| File.mtime(path) > since } +
    recent_tests('app/models/   *.rb', 'test/unit', since) +
    recent_tests('app/controllers/**/*.rb', 'test/functional', since)

  t.libs << 'test'
  t.test_files = touched.uniq
end
Rake::Task['test:recent'].comment = "Test recent changes"

Rake::TestTask.new(:uncommitted => "test:prepare") do |t|
  def t.file_list
    if File.directory?(".svn")
      changed_since_checkin = silence_stderr { %xsvn status` }.split.map { |path| path.chomp[7 .. -1] }
    elsif File.directory?(".git")
      changed_since_checkin = silence_stderr { %xgit ls-files --modified --others` }.split.map { |path| path.chomp }
    else
      abort "Not a Subversion or Git checkout."
    end

    models      = changed_since_checkin.select { |path| path =~ /app[\\/]models[\\/].*\.rb$/ }
    controllers = changed_since_checkin.select { |path| path =~ /app[\\/]controllers[\\/].*\.rb$/ }

    unit_tests       = models.map { |model| "test/unit/#{File.basename(model, '.rb')}_test.rb" }
    functional_tests = controllers.map { |controller| "test/functional/#{File.basename(controller, '.rb')}_test.rb" }
    (unit_tests + functional_tests).uniq.select { |file| File.exist?(file) }
  end

  t.libs << 'test'
end
Rake::Task['test:uncommitted'].comment = "Test changes since last checkin (only Subversion and Git)"

Rake::TestTask.new(:single => "test:prepare") do |t|
  t.libs << "test"
end

Rails::SubTestTask.new(:units => "test:prepare") do |t|
  t.libs << "test"
  t.pattern = 'test/unit/**/*_test.rb'
end

Rails::SubTestTask.new(:functionals => "test:prepare") do |t|
  t.libs << "test"
  t.pattern = 'test/functional/**/*_test.rb'
end

Rails::SubTestTask.new(:integration => "test:prepare") do |t|
  t.libs << "test"
  t.pattern = 'test/integration/**/*_test.rb'
end

task 'test:benchmark_mode' do
  ENV["BENCHMARK_TESTS"] = '1'
end

Rails::SubTestTask.new(:benchmark => ['test:prepare', 'test:benchmark_mode']) do |t|
  t.libs << 'test'
  t.pattern = 'test/performance/**/*_test.rb'
end

Rails::SubTestTask.new(:profile => 'test:prepare') do |t|
  t.libs << 'test'
  t.pattern = 'test/performance/**/*_test.rb'
end

Rails::SubTestTask.new(:plugins => :environment) do |t|
  t.libs << "test"

  if ENV['PLUGIN']
    t.pattern = "vendor/plugins/#{ENV['PLUGIN']}/test/**/*_test.rb"
  else
    t.pattern = 'vendor/plugins/*/**/test/**/*_test.rb'
  end
end

end