class Test::Unit::Collector::Dir

# File lib/test/unit/collector/dir.rb, line 13
def initialize(dir=::Dir, file=::File, object_space=::ObjectSpace, req=nil)
  super()
  @dir = dir
  @file = file
  @object_space = object_space
  @req = req
  @pattern = [/\btest_.*\.rb\Z/m]
  @exclude = []
  @base = nil
end

# File lib/test/unit/collector/dir.rb, line 24
def collect(*from)
  basedir = @base
  $:.push(basedir) if basedir
  if(from.empty?)
    recursive_collect('.', find_test_cases)
  elsif(from.size == 1)
    recursive_collect(from.first, find_test_cases)
  else
    suites = []
    from.each do |f|
      suite = recursive_collect(f, find_test_cases)
      suites << suite unless(suite.tests.empty?)
    end
    suite = TestSuite.new("[#{from.join(', ')}]")
    sort(suites).each{|s| suite << s}
    suite
  end
ensure
  $:.delete_at($:.rindex(basedir)) if basedir
end

# File lib/test/unit/collector/dir.rb, line 85
def collect_file(name, suites, already_gathered)
  dir = @file.dirname(@file.expand_path(name, @base))
  $:.unshift(dir)
  if(@req)
    @req.require(name)
  else
    require(name)
  end
  find_test_cases(already_gathered).each{|t| add_suite(suites, t.suite)}
ensure
  $:.delete_at($:.rindex(dir)) if(dir)
end

# File lib/test/unit/collector/dir.rb, line 45
def find_test_cases(ignore=[])
  cases = []
  @object_space.each_object(Class) do |c|
    cases << c if(c < TestCase && !ignore.include?(c))
  end
  ignore.concat(cases)
  cases
end

# File lib/test/unit/collector/dir.rb, line 98
def realdir(path)
  if @base
    @file.join(@base, path)
  else
    path
  end
end

# File lib/test/unit/collector/dir.rb, line 54
def recursive_collect(name, already_gathered)
  sub_suites = []
  path = realdir(name)
  if @file.directory?(path)
    dir_name = name unless name == '.'
    @dir.entries(path).each do |e|
      next if(e == '.' || e == '..')
      e_name = dir_name ? @file.join(dir_name, e) : e
      if @file.directory?(realdir(e_name))
        next if /\A(?:CVS|\.svn|\.git)\z/ =~ e
        sub_suite = recursive_collect(e_name, already_gathered)
        sub_suites << sub_suite unless(sub_suite.empty?)
      else
        next if /~\z/ =~ e_name or /\A\.\#/ =~ e
        if @pattern and !@pattern.empty?
          next unless @pattern.any? {|pat| pat =~ e_name}
        end
        if @exclude and !@exclude.empty?
          next if @exclude.any? {|pat| pat =~ e_name}
        end
        collect_file(e_name, sub_suites, already_gathered)
      end
    end
  else
    collect_file(name, sub_suites, already_gathered)
  end
  suite = TestSuite.new(@file.basename(name))
  sort(sub_suites).each{|s| suite << s}
  suite
end