* lib/test/unit.rb (MiniTest::Unit#process_args): refactored.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@28668 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

ChangeLog

@@ -1,3 +1,7 @@
+Sat Jul 17 19:01:47 2010  Nobuyoshi Nakada  <nobu@ruby-lang.org>
+
+	* lib/test/unit.rb (MiniTest::Unit#process_args): refactored.
+
 Sat Jul 17 18:30:05 2010  Tanaka Akira  <akr@fsij.org>
 
 	* tool/file2lastrev.rb: don't depend on pathname.rb if File.realpath

bin/testrb

@@ -1,6 +1,6 @@
 #!/usr/bin/env ruby
 require 'test/unit'
-exit Test::Unit.start {|files|
+tests = Test::Unit.new {|files|
   if files.empty?
     puts "Usage: testrb [options] tests..."
     exit false
@@ -12,3 +12,4 @@ exit Test::Unit.start {|files|
   end
   files
 }
+exit tests.run(ARGV) || true

lib/test/unit.rb

@@ -20,93 +20,150 @@ module Test
         @@run_count += 1
         super
       end
+
+      def run_once
+        return if have_run?
+        return if $! # don't run if there was an exception
+        yield
+      end
+      module_function :run_once
     end
 
-    def self.setup_argv(original_argv=::ARGV)
+    module Options
-      minitest_argv = []
+      def initialize(&block)
-      files = []
+        @init_hook = block
-      reject = []
+        super(&nil)
-      original_argv = original_argv.dup
+      end
-      OptionParser.new do |parser|
-        parser.default_argv = original_argv
 
-        parser.on '-v', '--verbose' do |v|
+      def process_args(args = [])
-          minitest_argv << '-v' if v
+        options = {}
+        OptionParser.new do |opts|
+          setup_options(opts, options)
+          opts.parse!(args)
+        end
+        args = @init_hook.call(args, options) if @init_hook
+        non_options(args, options)
+        options
+      end
+
+      private
+      def setup_options(opts, options)
+        opts.banner  = 'minitest options:'
+        opts.version = MiniTest::Unit::VERSION
+
+        opts.on '-h', '--help', 'Display this help.' do
+          puts opts
+          exit
         end
 
-        parser.on '-n', '--name TESTNAME' do |name|
+        opts.on '-s', '--seed SEED', Integer, "Sets random seed" do |m|
-          minitest_argv << '-n'
+          options[:seed] = m.to_i
-          minitest_argv << name
         end
 
+        opts.on '-v', '--verbose', "Verbose. Show progress processing files." do
+          options[:verbose] = true
+        end
+
+        opts.on '-n', '--name PATTERN', "Filter test names on pattern." do |a|
+          options[:filter] = a
+        end
+      end
+
+      def non_options(files, options)
+        files.each {|f|
+          d = File.dirname(path = File.expand_path(f))
+          unless $:.include? d
+            $: << d
+          end
+          begin
+            require path
+          rescue LoadError
+            puts "#{f}: #{$!}"
+          end
+        }
+      end
+    end
+
+    module GlobOption
+      include Options
+
+      def non_options(files, options)
+        files.map! {|f|
+          f = f.tr(File::ALT_SEPARATOR, File::SEPARATOR) if File::ALT_SEPARATOR
+          if File.directory? f
+            Dir["#{f}/**/test_*.rb"]
+          elsif File.file? f
+            f
+          else
+            raise ArgumentError, "file not found: #{f}"
+          end
+        }
+        files.flatten!
+        super(files, options)
+      end
+    end
+
+    module RejectOption
+      include Options
+
+      def setup_options(parser, options)
+        super
         parser.on '-x', '--exclude PATTERN' do |pattern|
-          reject << pattern
+          (options[:reject] ||= []) << pattern
         end
+      end
 
+      def non_options(files, options)
+        if reject = options.delete(:reject)
+          reject_pat = Regexp.union(reject.map {|r| /#{r}/ })
+          files.reject! {|f| reject_pat =~ f }
+        end
+        super(files, options)
+      end
+    end
+
+    module LoadPathOption
+      include Options
+
+      def setup_options(parser, options)
+        super
         parser.on '-Idirectory' do |dirs|
           dirs.split(':').each { |d| $LOAD_PATH.unshift d }
         end
-      end.parse!
-      files = original_argv
-
-      if block_given?
-        files = yield files
       end
-
-      files.map! {|f|
-        f = f.tr(File::ALT_SEPARATOR, File::SEPARATOR) if File::ALT_SEPARATOR
-        if File.directory? f
-          Dir["#{f}/**/test_*.rb"]
-        elsif File.file? f
-          f
-        else
-          raise ArgumentError, "file not found: #{f}"
-        end
-      }
-      files.flatten!
-
-      reject_pat = Regexp.union(reject.map {|r| /#{r}/ })
-      files.reject! {|f| reject_pat =~ f }
-
-      MiniTest::Unit._install_at_exit {
-        next if RunCount.have_run?
-        next if $! # don't run if there was an exception
-        exit false unless run(minitest_argv)
-      }
-
-      files.each {|f|
-        d = File.dirname(path = File.expand_path(f))
-        unless $:.include? d
-          $: << d
-        end
-        begin
-          require path
-        rescue LoadError
-          puts "#{f}: #{$!}"
-        end
-      }
-
-      minitest_argv
     end
 
-    def self.run(args)
+    def self.new
-      exit_code = MiniTest::Unit.new.run(args)
+      Mini.new do |files, options|
-      !exit_code || exit_code == 0
+        if block_given?
+          files = yield files
+        end
+        files
+      end
     end
 
-    def self.start(argv=::ARGV, &block)
+    class Mini < MiniTest::Unit
-      run(setup_argv(argv, &block))
+      include Test::Unit::GlobOption
+      include Test::Unit::RejectOption
+      include Test::Unit::LoadPathOption
     end
   end
 end
 
 class MiniTest::Unit
   def self.new(*)
-    super.extend(Test::Unit::RunCount)
+    super
+      .extend(Test::Unit::RunCount)
+      .extend(Test::Unit::Options)
   end
 
-  def self._install_at_exit(&block)
+  class << self; undef autorun; end
-    at_exit(&block) unless @@installed_at_exit
+  def self.autorun
+    at_exit {
+      Test::Unit::RunCount.run_once {exit(new.run(ARGV) || true)}
+    } unless @@installed_at_exit
     @@installed_at_exit = true
   end
 end
+
+MiniTest::Unit.autorun

test/runner.rb

@@ -6,7 +6,7 @@ require 'test/unit'
 src_testdir = File.dirname(File.expand_path(__FILE__))
 srcdir = File.dirname(src_testdir)
 
-exit Test::Unit.start {|files|
+tests = Test::Unit.new {|files|
   if files.empty?
     [src_testdir]
   else
@@ -23,3 +23,4 @@ exit Test::Unit.start {|files|
     }
   end
 }
+exit tests.run(ARGV) || true