spike of a `OptionParser` backed test runner.

1 files changed, 83 insertions, 0 deletions

diff --git a/railties/lib/rails/test_unit/runner.rb b/railties/lib/rails/test_unit/runner.rb
new file mode 100644
index 0000000000..25f30f6b80
--- /dev/null
+++ b/railties/lib/rails/test_unit/runner.rb
@@ -0,0 +1,83 @@
+require "ostruct"
+require "optparse"
+require "rake/file_list"
+
+module Rails
+  class TestRunner
+    class Options
+      def self.parse(args)
+        options = { backtrace: false, name: nil }
+
+        opt_parser = ::OptionParser.new do |opts|
+          opts.banner = "Usage: bin/rails test [options]"
+
+          opts.separator ""
+          opts.separator "Filter options:"
+
+          opts.on("-n", "--name [NAME]",
+                  "Only run tests matching NAME") do |name|
+            options[:name] = name
+          end
+
+          opts.separator ""
+          opts.separator "Output options:"
+
+          opts.on("-b", "--backtrace",
+                  "show the complte backtrace") do
+            options[:backtrace] = true
+          end
+
+          opts.separator ""
+          opts.separator "Common options:"
+
+          opts.on_tail("-h", "--help", "Show this message") do
+            puts opts
+            exit
+          end
+        end
+
+        opt_parser.order!(args)
+        options[:pattern] = ARGV.shift
+        options
+      end
+    end
+
+    def initialize(options = {})
+      @options = options
+    end
+
+    def run
+      enable_backtrace if @options[:backtrace]
+      filter_tests_by_name if @options[:name]
+
+      run_tests
+    end
+
+    private
+    def run_tests
+      test_files.to_a.each do |file|
+        require File.expand_path file
+      end
+    end
+
+    NAMED_PATTERNS = {
+      "models" => "test/models/**/*_test.rb"
+    }
+    def test_files
+      if @options[:pattern]
+        pattern = NAMED_PATTERNS[@options[:pattern]]
+      else
+        pattern = "test/**/*_test.rb"
+      end
+      Rake::FileList[pattern]
+    end
+
+    def enable_backtrace
+      ENV["BACKTRACE"] = "1"
+    end
+
+    def filter_tests_by_name
+      ARGV.push("-n", @options[:name])
+    end
+  end
+end