require 'rake'
require 'rake/testtask'
require 'rake/rdoctask'
require 'rake/gempackagetask'
require 'rake/contrib/sshpublisher'
require 'yaml'
$LOAD_PATH << File.dirname(__FILE__) + "/lib"
require "deep_test/rake_tasks"

task :default => %w[
  test 
  spec 
  failing_test
  deep_test
  deep_spec
  run_distributed
  run_distributed_with_worker_down
  run_distributed_with_failover
  test_rails_project
]

task :pc => :default

Rake::TestTask.new do |t|
  t.pattern = "test/**/*_test.rb"
  t.libs += ['test', 'lib']
end

DeepTest::TestTask.new :deep_test do |t|
  t.number_of_workers = 2
  t.pattern = "test/**/*_test.rb"
  t.metrics_file = "deep_test.metrics"
end

DeepTest::TestTask.new(:distributed_test) do |t|
  t.number_of_workers = 2
  t.pattern = "test/**/*_test.rb"
  t.distributed_server = "druby://localhost:8000"
  t.sync_options = {:source => File.dirname(__FILE__), 
                    :local => true,
                    :rsync_options => "--exclude=.svn"}
end

def rspec_present?
  defined?(Spec)
end

if rspec_present?
  Spec::Rake::SpecTask.new(:spec) do |t|
    t.spec_files = FileList['spec/**/*_spec.rb']
  end
  
  Spec::Rake::SpecTask.new(:deep_spec) do |t|
    t.spec_files = FileList['spec/**/*_spec.rb']
    t.deep_test :number_of_workers => 2
  end

  Spec::Rake::SpecTask.new(:distributed_spec) do |t|
    t.spec_files = FileList['spec/**/*_spec.rb']
    t.deep_test :number_of_workers => 2, 
                :distributed_server => "druby://localhost:8000",
                :sync_options => {:source => File.dirname(__FILE__), 
                                  :local => true,
                                  :rsync_options => "--exclude=.svn"}
  end
end


DeepTest::TestTask.new(:distribute_tests_to_minis) do |t|
  t.number_of_workers = 2
  t.pattern = "test/**/*_test.rb"
  t.distributed_server = "druby://alpha.local:8000"
  t.sync_options = {:source => File.dirname(__FILE__), 
                    :username => "tworker",
                    :password => "thought",
                    :rsync_options => "--exclude=.svn"}
end

task :run_distributed do |t|
  begin
    FileUtils.mkdir('/tmp/test_1') unless File.exist?('/tmp/test_1')
    FileUtils.mkdir('/tmp/test_2') unless File.exist?('/tmp/test_2')

    test_1_pid = fork do
      exec "ruby bin/deep_test test_server --uri drubyall://localhost:8001 --work_dir /tmp/test_1"
    end

    test_2_pid = fork do
      exec "ruby bin/deep_test test_server --uri drubyall://localhost:8002 --work_dir /tmp/test_2"
    end

    master_pid = fork do
      exec "ruby bin/deep_test master_test_server --uri drubyall://localhost:8000 druby://localhost:8001 druby://localhost:8002"
    end

    sleep 1

    Rake::Task[:distributed_test].invoke
    Rake::Task[:distributed_spec].invoke

    sh "ruby bin/deep_test test_throughput druby://localhost:8000 20"
  ensure
    Process.kill('TERM', master_pid) if master_pid rescue nil
    Process.kill('TERM', test_1_pid) if test_1_pid rescue nil
    Process.kill('TERM', test_2_pid) if test_2_pid rescue nil
    Process.waitall
  end

  sleep 1
end

task :run_distributed_with_worker_down do |t|
  begin
    FileUtils.mkdir('/tmp/test_1') unless File.exist?('/tmp/test_1')

    test_1_pid = fork do
      exec "ruby bin/deep_test test_server --uri drubyall://localhost:8001 --work_dir /tmp/test_1"
    end

    # don't start worker 2

    master_pid = fork do
      exec "ruby bin/deep_test master_test_server --uri drubyall://localhost:8000 druby://localhost:8001 druby://localhost:8002"
    end

    sleep 1

    # Use throughput to make sure we can run tests
    sh "ruby bin/deep_test test_throughput druby://localhost:8000 20"
  ensure
    Process.kill('TERM', master_pid) if master_pid rescue nil
    Process.kill('TERM', test_1_pid) if test_1_pid rescue nil
    Process.waitall
  end
end

task :run_distributed_with_failover do |t|
  puts
  puts "*** Running distributed with no server - expect a failover message ***"
  puts
  Rake::Task[:distributed_spec].execute "dummy arg"
end

task :failing_test do
  command = "rake --rakefile test/failing.rake 2>&1"
  puts command
  `#{command}`
  if $?.success?
    puts "F"
    fail "****\ntest/failing.rake should have failed\n****"
  else
    puts "."
  end
end

Rake::TestTask.new(:test_rails_project) do |t|
  t.pattern = "sample_rails_project/deep_test.rb"
end

desc "Generate documentation"
Rake::RDocTask.new(:rdoc) do |rdoc|
  rdoc.rdoc_dir = "doc"
  rdoc.title    = "DeepTest"
  rdoc.options << '--line-numbers'
  rdoc.rdoc_files.include(
    'README.rdoc', 
    'CHANGELOG', 
    'lib/deep_test/null_worker_listener.rb',
    'lib/deep_test/database/*.rb'
  )
end

desc "Upload RDoc to RubyForge"
task :publish_rdoc => [:rerdoc] do
  rubyforge_config = "#{ENV['HOME']}/.rubyforge/user-config.yml"
  username = YAML.load_file(rubyforge_config)["username"]
  sh "chmod -R 775 doc"
  sh "scp -rqp doc/* #{username}@rubyforge.org:/var/www/gforge-projects/deep-test"
end

Gem::manage_gems

specification = Gem::Specification.new do |s|
  s.platform = Gem::Platform::RUBY
	s.name   = "deep_test"
  s.summary = "DeepTest runs tests in multiple processes."
	s.version = "1.2.2"
	s.author = "anonymous z, Dan Manges, David Vollbracht"
	s.description = s.summary
	s.email = "daniel.manges@gmail.com"
  s.homepage = "http://deep-test.rubyforge.org"
  s.rubyforge_project = "deep-test"
  s.executables << "deep_test"

  s.has_rdoc = true
  s.extra_rdoc_files = ['README.rdoc', 'CHANGELOG']
  s.rdoc_options << '--title' << "DeepTest" << '--main' << 'README.rdoc' << '--line-numbers'

  s.files = FileList['{lib,script,test,bin}/**/*.{rb,rake,rhtml}', 'README.rdoc', 'CHANGELOG', 'Rakefile'].to_a
end

Rake::GemPackageTask.new(specification) do |package|
  package.need_zip = true
  package.need_tar = true
end

Rake::Task[:gem].prerequisites.unshift :default

task :tar do
  system "tar zcf pkg/deep_test.tar.gz --exclude=.svn --exclude='*.tar.gz' --exclude='*.gem' --directory=.. deep_test"
end