rails-profiler 0.25.0 → 0.26.0

data/lib/profiler/test_runner/run_store.rb

@@ -0,0 +1,120 @@
+# frozen_string_literal: true
+
+require "concurrent"
+require "securerandom"
+
+module Profiler
+  module TestRunner
+    class RunStore
+      TTL = 3600 # 1 hour
+
+      Run = Struct.new(:id, :status, :pid, :started_at, :finished_at, :output_lines, :exit_code, :files, :framework, keyword_init: true) do
+        def to_h
+          super.merge(
+            started_at: started_at&.iso8601,
+            finished_at: finished_at&.iso8601,
+            output: output_lines.join,
+            duration: finished_at && started_at ? ((finished_at - started_at) * 1000).round(2) : nil
+          ).except(:output_lines)
+        end
+      end
+
+      TERMINAL_STATUSES = %w[passed failed killed error].freeze
+
+      def initialize
+        @runs  = Concurrent::Hash.new
+        @locks = Concurrent::Hash.new
+      end
+
+      def create(files:, framework:)
+        id  = SecureRandom.hex(8)
+        run = Run.new(
+          id: id,
+          status: "pending",
+          pid: nil,
+          started_at: Time.now,
+          finished_at: nil,
+          output_lines: Concurrent::Array.new,
+          exit_code: nil,
+          files: files,
+          framework: framework.to_s
+        )
+        @runs[id]  = run
+        @locks[id] = { mutex: Mutex.new, cond: ConditionVariable.new }
+        cleanup_old_runs
+        run
+      end
+
+      def find(id)
+        @runs[id]
+      end
+
+      def update(id, **attrs)
+        run = @runs[id]
+        return unless run
+
+        attrs.each { |k, v| run.send(:"#{k}=", v) }
+        signal(id)
+        run
+      end
+
+      def append_output(id, chunk)
+        run = @runs[id]
+        return unless run
+
+        run.output_lines.push(chunk)
+        signal(id)
+      end
+
+      # Block until new output is available at +position+ or the run terminates.
+      # Returns { chunks: [...], status: "...", position: N, finished: bool }.
+      def wait_for_output(id, position:, timeout: 10)
+        run = @runs[id]
+        return { chunks: [], status: "not_found", position: 0, finished: true } unless run
+
+        lock = @locks[id]
+        return snapshot(run, position) unless lock
+
+        lock[:mutex].synchronize do
+          current = run.output_lines.size
+          if current <= position && !TERMINAL_STATUSES.include?(run.status)
+            lock[:cond].wait(lock[:mutex], timeout)
+          end
+        end
+
+        snapshot(run, position)
+      end
+
+      def all
+        @runs.values.sort_by { |r| r.started_at || Time.at(0) }.reverse
+      end
+
+      private
+
+      def signal(id)
+        lock = @locks[id]
+        return unless lock
+        lock[:mutex].synchronize { lock[:cond].broadcast }
+      end
+
+      def snapshot(run, position)
+        lines = run.output_lines.dup
+        {
+          chunks: lines[position..] || [],
+          status: run.status,
+          position: lines.size,
+          finished: TERMINAL_STATUSES.include?(run.status)
+        }
+      end
+
+      def cleanup_old_runs
+        cutoff = Time.now - TTL
+        @runs.delete_if { |id, r| r.finished_at && r.finished_at < cutoff && @locks.delete(id) }
+      end
+    end
+
+    def self.run_store
+      @run_store ||= RunStore.new
+    end
+  end
+end

data/lib/profiler/test_runner/runner.rb

@@ -0,0 +1,106 @@
+# frozen_string_literal: true
+
+require_relative "run_store"
+require_relative "discovery"
+
+module Profiler
+  module TestRunner
+    class Runner
+      def self.start(files:, framework:)
+        run = Profiler::TestRunner.run_store.create(files: files, framework: framework)
+        spawn_async(run)
+        run
+      end
+
+      def self.kill(run_id)
+        run = Profiler::TestRunner.run_store.find(run_id)
+        return false unless run && run.pid && run.status == "running"
+
+        begin
+          Process.kill("TERM", run.pid)
+          Profiler::TestRunner.run_store.update(run_id, status: "killed", finished_at: Time.now)
+          true
+        rescue Errno::ESRCH
+          # Process already exited
+          false
+        end
+      end
+
+      private
+
+      def self.spawn_async(run)
+        Thread.new do
+          run_process(run)
+        rescue => e
+          Profiler::TestRunner.run_store.update(
+            run.id,
+            status: "error",
+            finished_at: Time.now
+          )
+          Profiler::TestRunner.run_store.append_output(run.id, "\n[Profiler] Error: #{e.message}\n")
+        end
+      end
+
+      def self.run_process(run)
+        cmd = build_command(run.files, run.framework)
+        env = build_env
+
+        Profiler::TestRunner.run_store.update(run.id, status: "running", started_at: Time.now)
+
+        IO.popen([env, *cmd, err: [:child, :out]], "r") do |io|
+          Profiler::TestRunner.run_store.update(run.id, pid: io.pid)
+
+          while (chunk = io.read(256))
+            break if chunk.empty?
+
+            Profiler::TestRunner.run_store.append_output(run.id, chunk)
+          end
+        end
+
+        exit_code = $?.exitstatus || 0
+        status = exit_code == 0 ? "passed" : "failed"
+
+        Profiler::TestRunner.run_store.update(
+          run.id,
+          status: status,
+          finished_at: Time.now,
+          exit_code: exit_code
+        )
+      end
+
+      def self.build_command(files, framework)
+        root = defined?(Rails) ? Rails.root.to_s : Dir.pwd
+        absolute_files = files.map { |f| File.join(root, f) }
+
+        case framework.to_sym
+        when :rspec
+          ["bundle", "exec", "rspec", "--format", "progress", "--color", *absolute_files]
+        when :minitest
+          ["bundle", "exec", "rails", "test", *absolute_files]
+        else
+          ["bundle", "exec", "rspec", "--format", "progress", "--color", *absolute_files]
+        end
+      end
+
+      BLOCKED_ENV_KEYS = %w[RAILS_ENV RACK_ENV DATABASE_URL SECRET_KEY_BASE].freeze
+
+      def self.build_env
+        base = ENV.to_h
+
+        # Inject env var overrides configured in the profiler — skip blocked keys
+        overrides = Profiler.env_override_store.all_overrides
+        overrides.each do |key, entry|
+          next if BLOCKED_ENV_KEYS.include?(key.upcase)
+          value = entry.is_a?(Hash) ? entry["value"] : entry
+          base[key] = value
+        end
+
+        # Ensure test environment regardless of overrides
+        base["RAILS_ENV"] = "test"
+        base["RACK_ENV"]  = "test"
+
+        base
+      end
+    end
+  end
+end

data/lib/profiler/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Profiler
-  VERSION = "0.25.0"
+  VERSION = "0.26.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails-profiler
 version: !ruby/object:Gem::Version
-  version: 0.25.0
+  version: 0.26.0
 platform: ruby
 authors:
 - Sébastien Duplessy
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2026-06-01 00:00:00.000000000 Z
+date: 2026-06-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -114,14 +114,18 @@ files:
 - app/controllers/profiler/api/jobs_controller.rb
 - app/controllers/profiler/api/outbound_http_controller.rb
 - app/controllers/profiler/api/profiles_controller.rb
+- app/controllers/profiler/api/test_runner_controller.rb
+- app/controllers/profiler/api/tests_controller.rb
 - app/controllers/profiler/api/toolbar_controller.rb
 - app/controllers/profiler/application_controller.rb
 - app/controllers/profiler/assets_controller.rb
 - app/controllers/profiler/profiles_controller.rb
+- app/controllers/profiler/test_runner_controller.rb
 - app/views/layouts/profiler/application.html.erb
 - app/views/layouts/profiler/embedded.html.erb
 - app/views/profiler/profiles/index.html.erb
 - app/views/profiler/profiles/show.html.erb
+- app/views/profiler/test_runner/index.html.erb
 - config/routes.rb
 - exe/profiler-mcp
 - lib/profiler.rb
@@ -142,6 +146,7 @@ files:
 - lib/profiler/collectors/mailer_collector.rb
 - lib/profiler/collectors/request_collector.rb
 - lib/profiler/collectors/routes_collector.rb
+- lib/profiler/collectors/test_collector.rb
 - lib/profiler/collectors/view_collector.rb
 - lib/profiler/configuration.rb
 - lib/profiler/console_profiler.rb
@@ -158,10 +163,12 @@ files:
 - lib/profiler/mcp/body_formatter.rb
 - lib/profiler/mcp/file_cache.rb
 - lib/profiler/mcp/path_extractor.rb
+- lib/profiler/mcp/resources/failing_tests.rb
 - lib/profiler/mcp/resources/n1_patterns.rb
 - lib/profiler/mcp/resources/recent_jobs.rb
 - lib/profiler/mcp/resources/recent_requests.rb
 - lib/profiler/mcp/resources/slow_queries.rb
+- lib/profiler/mcp/resources/slow_tests.rb
 - lib/profiler/mcp/server.rb
 - lib/profiler/mcp/tools/analyze_queries.rb
 - lib/profiler/mcp/tools/clear_profiles.rb
@@ -171,12 +178,15 @@ files:
 - lib/profiler/mcp/tools/get_profile_detail.rb
 - lib/profiler/mcp/tools/get_profile_dumps.rb
 - lib/profiler/mcp/tools/get_profile_http.rb
+- lib/profiler/mcp/tools/get_test_profile_detail.rb
 - lib/profiler/mcp/tools/list_env_vars.rb
 - lib/profiler/mcp/tools/query_jobs.rb
 - lib/profiler/mcp/tools/query_mailers.rb
 - lib/profiler/mcp/tools/query_profiles.rb
+- lib/profiler/mcp/tools/query_test_profiles.rb
 - lib/profiler/mcp/tools/reset_all_env_vars.rb
 - lib/profiler/mcp/tools/reset_env_var.rb
+- lib/profiler/mcp/tools/run_tests.rb
 - lib/profiler/mcp/tools/set_env_var.rb
 - lib/profiler/middleware/cors_middleware.rb
 - lib/profiler/middleware/profiler_middleware.rb
@@ -192,6 +202,13 @@ files:
 - lib/profiler/storage/redis_store.rb
 - lib/profiler/storage/sqlite_store.rb
 - lib/profiler/tasks/profiler.rake
+- lib/profiler/test_helpers/minitest_support.rb
+- lib/profiler/test_helpers/reporter.rb
+- lib/profiler/test_helpers/rspec_support.rb
+- lib/profiler/test_profiler.rb
+- lib/profiler/test_runner/discovery.rb
+- lib/profiler/test_runner/run_store.rb
+- lib/profiler/test_runner/runner.rb
 - lib/profiler/version.rb
 homepage: https://git.duplessy.eu/sebastien/rails-profiler-gem
 licenses: