fix specs

Author: aurelienbottazini

Gemfile.lock

@@ -17,6 +17,7 @@ GEM
     minitest (6.0.5)
       drb (~> 2.0)
       prism (~> 1.5)
+    minitest-mock (5.27.0)
     ostruct (0.6.1)
     path_expander (2.0.1)
     prism (1.9.0)
@@ -38,6 +39,7 @@ DEPENDENCIES
   flog
   markdown-run!
   minitest (= 6.0.5)
+  minitest-mock (~> 5.27)
   rake
   simplecov (~> 0.22)
 

exe/markdown-run

@@ -1,16 +1,30 @@
 #!/usr/bin/env ruby
 
+require "optparse"
 require_relative "../lib/markdown_run"
 
 # Script to process markdown files, execute code blocks based on language,
 # and insert their results back into the markdown.
 
+option_parser = OptionParser.new do |parser|
+  parser.banner = "Usage: markdown-run <file.md>"
+  parser.separator ""
+  parser.separator "Processes a markdown file and executes code blocks, inserting their results."
+  parser.separator ""
+  parser.separator "Options:"
+
+  parser.on("-h", "--help", "Show this help message") do
+    puts parser
+    exit 0 unless $0 != __FILE__  # Don't exit when loaded as a library
+  end
+end
+
+option_parser.parse!
+
 if ARGV.empty?
-  puts "Usage: markdown-run <file.md>"
-  puts "Processes a markdown file and executes code blocks, inserting their results."
+  puts option_parser
   exit 1 unless $0 != __FILE__  # Don't exit when loaded as a library
 else
   success = MarkdownRun.run_code_blocks(ARGV[0])
   exit success ? 0 : 1 unless $0 != __FILE__  # Don't exit when loaded as a library
 end
-

lib/code_executor.rb

@@ -1,3 +1,4 @@
+require "json"
 require "tempfile"
 require "open3"
 require_relative "language_configs"
@@ -36,16 +37,17 @@ def execute_with_config(code_content, lang_config, temp_dir, lang_key, input_fil
     temp_file_suffix = lang_config[:temp_file_suffix]
 
     if temp_file_suffix
-      execute_with_temp_file(code_content, cmd_lambda, temp_file_suffix, temp_dir, lang_key, input_file_path, explain, flamegraph)
+      write_code_to_temp_file = lang_config.fetch(:write_code_to_temp_file, true)
+      execute_with_temp_file(code_content, cmd_lambda, temp_file_suffix, temp_dir, lang_key, input_file_path, explain, flamegraph, write_code_to_temp_file)
     else
       execute_direct_command(code_content, cmd_lambda, input_file_path, explain, flamegraph)
     end
   end
 
-  def execute_with_temp_file(code_content, cmd_lambda, temp_file_suffix, temp_dir, lang_key, input_file_path = nil, explain = false, flamegraph = false)
+  def execute_with_temp_file(code_content, cmd_lambda, temp_file_suffix, temp_dir, lang_key, input_file_path = nil, explain = false, flamegraph = false, write_code_to_temp_file = true)
     result = nil
     Tempfile.create([lang_key, temp_file_suffix], temp_dir) do |temp_file|
-      temp_file.write(code_content)
+      temp_file.write(code_content) if write_code_to_temp_file
       temp_file.close
       command_to_run, exec_options = cmd_lambda.call(**{
         code_content: code_content,

lib/language_configs.rb

@@ -19,7 +19,8 @@
 
 SQLITE_CONFIG = {
   command: proc { |code_content: nil, temp_file_path: nil, **| [ "sqlite3 #{temp_file_path}", { stdin_data: code_content } ] },
-  temp_file_suffix: ".db" # Temp file is the database
+  temp_file_suffix: ".db", # Temp file is the database
+  write_code_to_temp_file: false
 }.freeze
 
 SUPPORTED_LANGUAGES = {
@@ -46,7 +47,8 @@
       else
         [ "#{psql_cmd} -A -t -X", { stdin_data: code_content } ]
       end
-    }
+    },
+    result_block_type: "plain"
   },
   "ruby" => {
     command: proc { |temp_file_path: nil, **|

lib/result_helper.rb

@@ -16,8 +16,16 @@ def mermaid_style_result?(lang)
     lang_config && lang_config[:result_handling] == :mermaid_svg
   end
 
+  def plain_style_result?(lang)
+    lang_config = SUPPORTED_LANGUAGES[lang]
+    lang_config && lang_config[:result_block_type] == "plain"
+  end
+
   def result_block_header(lang)
-    ruby_style_result?(lang) ? "```ruby\n" : "``` {result}\n"
+    return "```ruby\n" if ruby_style_result?(lang)
+    return "```RESULT\n" if plain_style_result?(lang)
+
+    "``` {result}\n"
   end
 
   def result_block_regex(lang)
@@ -27,6 +35,8 @@ def result_block_regex(lang)
     elsif ruby_style_result?(lang)
       # For ruby, check for old-style ```ruby RESULT blocks (for backward compatibility during migration)
       /^```ruby\s+RESULT$/i
+    elsif plain_style_result?(lang)
+      /^```\s*RESULT$/i
     else
       /^```\s*\{result\}$/i
     end

lib/tasks/test_profiler.rake

@@ -0,0 +1,53 @@
+desc "Run tests with detailed timing information"
+task :test_profile do
+  require 'benchmark'
+
+  puts "Running tests with detailed profiling..."
+
+  # Run tests with verbose output and capture timing
+  output = `bundle exec rake test TESTOPTS="-v" 2>&1`
+
+  # Extract test timings
+  timings = []
+  output.scan(/^(.+) = ([0-9]+\.[0-9]+) s = \.$/) do |test_name, time|
+    timings << [test_name, time.to_f]
+  end
+
+  # Sort by time (descending)
+  timings.sort_by! { |_, time| -time }
+
+  puts "\n" + "="*80
+  puts "TOP 15 SLOWEST TESTS"
+  puts "="*80
+
+  timings.first(15).each_with_index do |(test_name, time), index|
+    printf "%2d. %-60s %6.2f s\n", index + 1, test_name.truncate(60), time
+  end
+
+  puts "\n" + "="*80
+  puts "SUMMARY"
+  puts "="*80
+  total_time = timings.sum { |_, time| time }
+  slow_tests = timings.select { |_, time| time > 0.1 }
+
+  puts "Total test time: #{total_time.round(2)} seconds"
+  puts "Tests slower than 0.1s: #{slow_tests.count}"
+  puts "Time spent in slow tests: #{slow_tests.sum { |_, time| time }.round(2)} seconds (#{((slow_tests.sum { |_, time| time } / total_time) * 100).round(1)}%)"
+
+  # Show original test output
+  puts "\n" + "="*80
+  puts "FULL TEST OUTPUT"
+  puts "="*80
+  puts output
+end
+
+# String extension for truncation
+class String
+  def truncate(max_length, omission = "...")
+    if length > max_length
+      self[0, max_length - omission.length] + omission
+    else
+      self
+    end
+  end
+end

markdown-run-sample/markdown-run-sample-flamegraph-20250905-120101-9c8e2d8be98f.svg

@@ -39,6 +39,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'ostruct', '0.6.1'
 
   spec.add_development_dependency 'minitest', "6.0.5"
+  spec.add_development_dependency 'minitest-mock', '~> 5.27'
   spec.add_development_dependency 'rake'
   spec.add_development_dependency 'flog'
   spec.add_development_dependency 'simplecov', '~> 0.22'

markdown-run.gemspec

@@ -0,0 +1,28 @@
+require 'minitest/autorun'
+require 'open3'
+require 'rbconfig'
+
+class TestCli < Minitest::Test
+  def test_help_matches_empty_invocation
+    empty_stdout, empty_stderr, empty_status = run_cli
+    help_stdout, help_stderr, help_status = run_cli("--help")
+
+    assert_equal help_stdout, empty_stdout
+    assert_equal "", empty_stderr
+    assert_equal "", help_stderr
+    assert_equal 1, empty_status.exitstatus
+    assert_equal 0, help_status.exitstatus
+    assert_includes help_stdout, "Usage: markdown-run <file.md>"
+    assert_includes help_stdout, "-h, --help"
+  end
+
+  private
+
+  def run_cli(*args)
+    bundler_env = ENV.each_key.each_with_object({ "RUBYOPT" => nil }) do |key, env|
+      env[key] = nil if key.start_with?("BUNDLE")
+    end
+
+    Open3.capture3(bundler_env, RbConfig.ruby, File.expand_path("../exe/markdown-run", __dir__), *args)
+  end
+end

test/cli_test.rb

@@ -84,6 +84,10 @@ def test_psql_config_when_available
     end
   end
 
+  def test_psql_config_uses_plain_result_blocks
+    assert_equal "plain", SUPPORTED_LANGUAGES["psql"][:result_block_type]
+  end
+
   def test_psql_config_when_unavailable
     # Stub PostgresHelper to indicate psql is not available
     PostgresHelper.stub :available?, false do
@@ -259,9 +263,10 @@ def test_all_supported_languages_have_command
 
   def test_sqlite_config_properties
     assert_equal ".db", SQLITE_CONFIG[:temp_file_suffix]
+    assert_equal false, SQLITE_CONFIG[:write_code_to_temp_file]
 
     command, options = SQLITE_CONFIG[:command].call(**{ code_content: "SELECT 1;", temp_file_path: "/tmp/test.db" })
     assert_equal "sqlite3 /tmp/test.db", command
     assert_equal({ stdin_data: "SELECT 1;" }, options)
   end
-end
+end