module ParallelTests::Tasks

def check_for_pending_migrations
  ["db:abort_if_pending_migrations", "app:db:abort_if_pending_migrations"].each do |abort_migrations|
    if Rake::Task.task_defined?(abort_migrations)
      Rake::Task[abort_migrations].invoke
      break
    end
  end
end

def load_lib
  $LOAD_PATH << File.expand_path(File.join(File.dirname(__FILE__), '..'))
  require "parallel_tests"
end

def parse_args(args)
  # order as given by user
  args = [args[:count], args[:pattern], args[:options], args[:pass_through]]
  # count given or empty ?
  # parallel:spec[2,models,options]
  # parallel:spec[,models,options]
  count = args.shift if args.first.to_s =~ /^\d*$/
  num_processes = count.to_i unless count.to_s.empty?
  pattern = args.shift
  options = args.shift
  pass_through = args.shift
  [num_processes, pattern.to_s, options.to_s, pass_through.to_s]
end

def purge_before_load
  if Gem::Version.new(Rails.version) > Gem::Version.new('4.2.0')
    Rake::Task.task_defined?('db:purge') ? 'db:purge' : 'app:db:purge'
  end
end

def rails_env
  ENV['RAILS_ENV'] || 'test'
end

def run_in_parallel(cmd, options={})
  load_lib
  count = " -n #{options[:count]}" unless options[:count].to_s.empty?
  # Using the relative path to find the binary allow to run a specific version of it
  executable = File.expand_path("../../../bin/parallel_test", __FILE__)
  command = "#{ParallelTests.with_ruby_binary(Shellwords.escape(executable))} --exec '#{cmd}'#{count}#{' --non-parallel' if options[:non_parallel]}"
  abort unless system(command)
end

def suppress_output(command, ignore_regex)
  activate_pipefail = "set -o pipefail"
  remove_ignored_lines = %Q{(grep -v "#{ignore_regex}" || test 1)}
  if File.executable?('/bin/bash') && system('/bin/bash', '-c', "#{activate_pipefail} 2>/dev/null && test 1")
    # We need to shell escape single quotes (' becomes '"'"') because
    # run_in_parallel wraps command in single quotes
    %Q{/bin/bash -c '"'"'#{activate_pipefail} && (#{command}) | #{remove_ignored_lines}'"'"'}
  else
    command
  end
end

def suppress_schema_load_output(command)
  ParallelTests::Tasks.suppress_output(command, "^   ->\\|^-- ")
end