Allow zjit/test_runner.rb to take argument and options

Author: st0012

zjit/test_runner.rb

@@ -1,4 +1,23 @@
 require "pathname"
+require "optparse"
+
+# Parse command-line arguments
+options = {}
+parser = OptionParser.new do |opts|
+  opts.banner = "Usage: ruby zjit/test_runner.rb [options] [test_file]"
+  
+  opts.on("--miniruby PATH", "Path to miniruby executable") do |path|
+    options[:miniruby] = path
+  end
+  
+  opts.on("-h", "--help", "Show this message") do
+    puts opts
+    exit
+  end
+end
+
+args = parser.parse!(ARGV)
+specific_test = args.first
 
 # Tests that are known to fail or cause issues with ZJIT
 excluded = [
@@ -61,28 +80,55 @@
   "test/ruby/test_yjit.rb",
 ]
 
-repo_root = ENV["CI"] ? "../src" : "."
-miniruby = Dir.exist?("#{repo_root}/build") ? "#{repo_root}/build/miniruby" : "#{repo_root}/miniruby"
+# Determine repo root based on the context
+if specific_test && specific_test.start_with?("test/")
+  # When a specific test file is provided, use current directory as repo root
+  repo_root = "."
+else
+  repo_root = ENV["CI"] ? "../src" : "."
+end
+
+# Use provided miniruby path or default
+if options[:miniruby]
+  miniruby = options[:miniruby]
+else
+  miniruby = Dir.exist?("#{repo_root}/build") ? "#{repo_root}/build/miniruby" : "#{repo_root}/miniruby"
+end
+
 test_runner = "#{repo_root}/test/runner.rb"
 miniruby_opts = "-I#{repo_root}/lib -I. -I.ext/common --zjit-call-threshold=1"
 test_dir = "#{repo_root}/test/ruby"
 
-total_test_files = Dir.glob("#{test_dir}/**/*.rb")
-
-# Filter out excluded tests
-test_files = total_test_files.reject do |file|
-  excluded.any? { |excluded_file| file.end_with?(excluded_file) }
+# Determine which tests to run
+if specific_test
+  # Run specific test file
+  test_files = [specific_test]
+  total_test_files = test_files
+else
+  # Run all tests except excluded ones
+  total_test_files = Dir.glob("#{test_dir}/**/*.rb")
+  test_files = total_test_files.reject do |file|
+    excluded.any? { |excluded_file| file.end_with?(excluded_file) }
+  end
 end
 
 command = "RUST_BACKTRACE=1 ruby --disable=gems #{test_runner} --ruby=\"#{miniruby} #{miniruby_opts}\""
 
-puts "Running #{test_files.length} (#{total_test_files.length} total) test files with ZJIT..."
+if specific_test
+  puts "Running specific test file: #{specific_test}"
+else
+  puts "Running #{test_files.length} (#{total_test_files.length} total) test files with ZJIT..."
+end
 puts "Command: #{command}"
 
 # Run each test
 test_files.each_with_index do |test_file, idx|
   relative_path = Pathname.new(test_file).relative_path_from(Pathname.new("."))
-  print "[#{idx + 1}/#{test_files.length}] #{relative_path}... "
+  if test_files.length == 1
+    print "Running #{relative_path}... "
+  else
+    print "[#{idx + 1}/#{test_files.length}] #{relative_path}... "
+  end
 
   # Build command using the test runner framework
   cmd = %Q{#{command} #{test_file} 2>&1}