bitclust-core 1.2.6 → 1.6.0

data/test/test_searcher.rb

@@ -0,0 +1,153 @@
+require "test/unit"
+require "bitclust"
+require "bitclust/searcher"
+require "fileutils"
+require "tmpdir"
+
+class TestTerminalView < Test::Unit::TestCase
+  include BitClust
+
+  def test_show_class
+    view = TerminalView.new(Plain.new, {})
+    db = Database.dummy
+    foo = ClassEntry.new(db, "Foo")
+    bar = ClassEntry.new(db, "Bar")
+    out, err = capture_output do
+      assert_nothing_raised do
+        view.show_class([foo, bar])
+      end
+    end
+    assert_equal %w[Bar Foo], out.split
+    assert_empty err
+  end
+end
+
+class TestSearcherWindowsDrivePath < Test::Unit::TestCase
+  include BitClust
+
+  ENV_KEYS = %w[REFE2_SERVER BITCLUST_SERVER REFE2_DATADIR BITCLUST_DATADIR]
+
+  def setup
+    @saved_env = {}
+    ENV_KEYS.each {|key| @saved_env[key] = ENV.delete(key) }
+    @saved_dir = Dir.pwd
+    @searcher = Searcher.new
+  end
+
+  def teardown
+    Dir.chdir(@saved_dir)
+    ENV_KEYS.each do |key|
+      if @saved_env[key]
+        ENV[key] = @saved_env[key]
+      else
+        ENV.delete(key)
+      end
+    end
+  end
+
+  def test_windows_drive_path_predicate
+    assert_true @searcher.send(:windows_drive_path?, "G:/Users/foo/.bitclust/db-2.2.0")
+    assert_true @searcher.send(:windows_drive_path?, 'C:\Users\foo\.bitclust\db-2.2.0')
+    assert_false @searcher.send(:windows_drive_path?, "/home/foo/.bitclust/db-2.2.0")
+    assert_false @searcher.send(:windows_drive_path?, "druby://localhost:10001")
+  end
+
+  def test_drive_path_uri_keeps_drive_letter
+    uri = @searcher.send(:drive_path_uri, "G:/Users/foo/.bitclust/db-2.2.0")
+    assert_equal "file", uri.scheme
+    assert_equal "G:/Users/foo/.bitclust/db-2.2.0", uri.path
+  end
+
+  def test_find_dblocation_keeps_windows_drive_letter
+    with_fake_datadir("G:/Users/foo/.bitclust/db-2.2.0") do |relative_path|
+      ENV["BITCLUST_DATADIR"] = relative_path
+      location = @searcher.send(:find_dblocation)
+      assert_equal "file", location.scheme
+      assert_equal relative_path, location.path
+    end
+  end
+
+  def test_find_dblocation_keeps_unix_absolute_path
+    Dir.mktmpdir do |dir|
+      datadir = File.join(dir, ".bitclust", "db-2.2.0")
+      FileUtils.mkdir_p(datadir)
+      FileUtils.touch(File.join(datadir, "properties"))
+      ENV["BITCLUST_DATADIR"] = datadir
+      location = @searcher.send(:find_dblocation)
+      assert_equal "file", location.scheme
+      assert_equal datadir, location.path
+    end
+  end
+
+  def test_database_option_keeps_windows_drive_letter
+    @searcher.parser.parse!(["--database=G:/Users/foo/.bitclust/db-2.2.0"])
+    dblocation = @searcher.instance_variable_get(:@dblocation)
+    assert_equal "file", dblocation.scheme
+    assert_equal "G:/Users/foo/.bitclust/db-2.2.0", dblocation.path
+  end
+
+  private
+
+  # datadir/properties を tmpdir 内に作り、tmpdir へ chdir したうえで
+  # datadir までの相対パス(ドライブレターを含む)を渡す。":" は Unix の
+  # ファイル名として有効な文字なので、Linux 上でも "G:" というディレク
+  # トリを作ってドライブレター付きパスの再現ができる
+  def with_fake_datadir(relative_path)
+    Dir.mktmpdir do |dir|
+      datadir = File.join(dir, *relative_path.split("/"))
+      FileUtils.mkdir_p(datadir)
+      FileUtils.touch(File.join(datadir, "properties"))
+      Dir.chdir(dir) do
+        yield relative_path
+      end
+    end
+  end
+end
+
+# refe の module_function 表示(bitclust#297)。
+# HTML 側(#282)と同じく、DB のバージョンが 4.0 以降なら
+# "Foo.#baz" ではなく "Foo?.baz" と表示する。
+class TestSearcherModuleFunctionDisplay < Test::Unit::TestCase
+  include BitClust
+
+  def record_for(version)
+    db = MethodDatabase.dummy("version" => version)
+    spec = MethodSpec.parse("Foo.#baz")
+    SearchResult::Record.new(db, spec, spec)
+  end
+
+  def test_names_folds_module_function_typemark_for_40
+    assert_equal ["Foo?.baz"], record_for("4.0").names
+  end
+
+  def test_names_keeps_typemark_before_40
+    assert_equal ["Foo.#baz"], record_for("3.4").names
+  end
+
+  def test_names_without_db_version_keeps_typemark
+    spec = MethodSpec.parse("Foo.#baz")
+    assert_equal ["Foo.#baz"], SearchResult::Record.new(nil, spec, spec).names
+  end
+end
+
+# refe のクエリ側も 4.0 以降の表示形式 "Foo?.baz" を受け付ける(bitclust#297)
+class TestSearcherModuleFunctionQuery < Test::Unit::TestCase
+  include BitClust
+
+  def parse(pat)
+    Searcher.new.send(:parse_method_spec_pattern, pat)
+  end
+
+  def test_question_dot_pattern
+    assert_equal ["Foo", ".#", "baz"], parse("Foo?.baz")
+  end
+
+  def test_dot_hash_pattern_still_works
+    assert_equal ["Foo", ".#", "baz"], parse("Foo.#baz")
+  end
+
+  def test_predicate_method_names_unaffected
+    assert_equal ["Foo", "#", "empty?"], parse("Foo#empty?")
+    assert_equal ["Foo", ".", "b?"], parse("Foo.b?")
+  end
+end

data/test/test_searchpage_command.rb

@@ -0,0 +1,167 @@
+# frozen_string_literal: true
+require 'test/unit'
+require 'json'
+require 'fileutils'
+require 'bitclust'
+require 'bitclust/methoddatabase'
+require 'bitclust/subcommands/searchpage_command'
+
+# searchpage サブコマンド: 複数バージョンの DB から /ja/search/ 置換用の
+# 静的検索ページ一式（index.html + 統合 search_data.js + JS/CSS アセット）を
+# 生成する。rurema-search（サーバ常駐の全文検索）のリタイア先。
+#
+# テストリスト:
+# [x] 統合 index: 全版共通エントリは versions に全版、片方だけのものはその版のみ
+# [x] index.html に search_versions（昇順）と検索 UI の要素が埋まる
+# [x] vendored JS + NOTICE + search_page.js + search.css がコピーされる
+# [x] ページ専用 UI なので search_init.js（ページ内ボックス用）は含めない
+# [x] DB を持たない引数なし呼び出しはエラー
+# [x] グローバル --database 不要（needs_database? が false）
+class TestSearchpageCommand < Test::Unit::TestCase
+  def setup
+    @tmpdir = "searchpage_test_tmp"
+    FileUtils.rm_rf(@tmpdir)
+    @db34 = build_db('3.4', <<~'RD')
+      description
+
+      = class Foo < Object
+      == Instance Methods
+      --- foo
+    RD
+    @db41 = build_db('4.1', <<~'RD')
+      description
+
+      = class Foo < Object
+      == Instance Methods
+      --- foo
+      --- bar
+    RD
+    @out = "#{@tmpdir}/out"
+  end
+
+  def teardown
+    FileUtils.rm_rf(@tmpdir)
+  end
+
+  def build_db(version, builtin_rd)
+    root = "#{@tmpdir}/tree-#{version}/refm/api/src"
+    FileUtils.mkdir_p("#{root}/_builtin")
+    File.write("#{root}/LIBRARIES", "_builtin\n")
+    File.write("#{root}/_builtin.rd", builtin_rd)
+    prefix = "#{@tmpdir}/db-#{version}"
+    db = BitClust::MethodDatabase.new(prefix)
+    db.init
+    db.transaction do
+      db.propset('version', version)
+      db.propset('encoding', 'utf-8')
+    end
+    db.transaction do
+      db.update_by_stdlibtree(root)
+    end
+    prefix
+  end
+
+  def run_command(argv)
+    cmd = BitClust::Subcommands::SearchpageCommand.new
+    cmd.parse(argv)
+    cmd.exec(argv, { prefix: nil, capi: false })
+    cmd
+  end
+
+  def test_needs_no_global_database_option
+    cmd = BitClust::Subcommands::SearchpageCommand.new
+    assert_false cmd.needs_database?
+  end
+
+  def test_merged_index_versions
+    run_command(["--outputdir=#{@out}", @db41, @db34])
+    js = File.read("#{@out}/js/search_data.js")
+    assert_match(/\Avar search_data = \{/, js)
+    data = JSON.parse(js.sub(/\Avar search_data = /, '').sub(/;\z/, ''))
+    foo = data['index'].find { |e| e['full_name'] == 'Foo#foo' }
+    bar = data['index'].find { |e| e['full_name'] == 'Foo#bar' }
+    klass = data['index'].find { |e| e['full_name'] == 'Foo' }
+    assert_equal %w[3.4 4.1], foo['versions']
+    assert_equal %w[4.1], bar['versions']
+    assert_equal %w[3.4 4.1], klass['versions']
+  end
+
+  def test_merged_index_shows_module_functions_as_question_dot_only
+    # bitclust#279 コメント対応: 版によって表示が ".#"(4.0 より前)と
+    # "?."(4.0 以降)に割れる module function は、統合ページでは "?."
+    # 表記の1エントリに合流させる(両表記が併記されるとわかりにくい)
+    db33 = build_db('3.3', <<~'RD')
+      description
+
+      = module Kernel
+      == Module Functions
+      --- at_exit{ ... } -> Proc
+      aaa
+    RD
+    db40 = build_db('4.0', <<~'RD')
+      description
+
+      = module Kernel
+      == Module Functions
+      --- at_exit{ ... } -> Proc
+      aaa
+    RD
+    run_command(["--outputdir=#{@out}", db40, db33])
+    js = File.read("#{@out}/js/search_data.js")
+    data = JSON.parse(js.sub(/\Avar search_data = /, '').sub(/;\z/, ''))
+    entries = data['index'].select { |e| e['name'] == 'at_exit' }
+    assert_equal ['Kernel?.at_exit'], entries.map { |e| e['full_name'] }
+    assert_equal %w[3.3 4.0], entries[0]['versions']
+    assert_not_match(/Kernel\.#/, js)
+  end
+
+  def test_index_html_embeds_versions_and_ui
+    run_command(["--outputdir=#{@out}", @db41, @db34])
+    html = File.read("#{@out}/index.html")
+    assert_match(/var search_versions = \["3\.4","4\.1"\];/, html)
+    assert_match(/var search_version_base = "\.\.\/";/, html)
+    assert_match(/id="search-field"/, html)
+    assert_match(/id="search-results"/, html)
+    assert_match(%r{js/search_data\.js}, html)
+    assert_match(%r{js/search_page\.js}, html)
+  end
+
+  def test_assets_are_copied
+    run_command(["--outputdir=#{@out}", @db41, @db34])
+    %w[search_navigation.js search_ranker.js search_controller.js
+       search_page.js NOTICE].each do |f|
+      assert File.exist?("#{@out}/js/#{f}"), "js/#{f} not copied"
+    end
+    assert File.exist?("#{@out}/search.css")
+    # ページ内ドロップダウン用の search_init.js はこのページでは使わない
+    assert_false File.exist?("#{@out}/js/search_init.js")
+  end
+
+  def test_no_database_arguments_is_an_error
+    cmd = BitClust::Subcommands::SearchpageCommand.new
+    argv = ["--outputdir=#{@out}"]
+    cmd.parse(argv)
+    assert_raise(SystemExit) do
+      capture_stderr { cmd.exec(argv, { prefix: nil, capi: false }) }
+    end
+  end
+
+  def test_missing_outputdir_is_an_error
+    cmd = BitClust::Subcommands::SearchpageCommand.new
+    argv = [@db34]
+    cmd.parse(argv)
+    assert_raise(SystemExit) do
+      capture_stderr { cmd.exec(argv, { prefix: nil, capi: false }) }
+    end
+  end
+
+  private
+
+  def capture_stderr
+    orig = $stderr
+    $stderr = StringIO.new
+    yield
+  ensure
+    $stderr = orig
+  end
+end

data/test/test_simplesearcher.rb

@@ -15,6 +15,12 @@ class TestSearcher <  Test::Unit::TestCase
 == Class Methods
 --- bar
 = reopen Kernel
+== Module Functions
+--- open
+
+== Instance Methods
+--- puts
+
 == Special Variables
 --- $spespe
 
@@ -45,4 +51,26 @@ HERE
     assert_equal([], search_pattern(@db, " "), 'space')
     assert_equal([], search_pattern(@db, ""), 'blank')
   end
+
+  # bitclust#250 follow-up: the dynamic server's /search accepts free-text
+  # queries, and a module function may be typed in either notation --
+  # "Kernel.#open" (bitclust's own internal spec-string form, still what
+  # docs for Ruby < 4.0 display) or "Kernel?.open" (what docs for Ruby >=
+  # 4.0 display since #277). Both must resolve to the same method,
+  # regardless of which notation the *query* uses -- the query parser has
+  # no idea (and shouldn't need to know) which doc version the user is
+  # thinking of.
+  def test_module_function_query_notation
+    [['Kernel?.open', 'open'],  # new (>=4.0 display) notation
+     ['Kernel.#open', 'open'],  # existing (bitclust-internal) notation: regression check
+     ['?.open',       'open'],  # bare module-function marker, no class name
+     ['Kernel#puts',  'puts'],  # instance method: unaffected by the "?." change
+    ].each{|q, expected|
+      ret = search_pattern(@db, q)
+      assert_not_equal([], ret, q)
+      assert_equal(expected, ret[0].name, q)
+    }
+    # Singleton-method dot notation ('Hoge.h' => 'hoge' in test_simple_search
+    # above) is also unaffected; kept there rather than duplicated here.
+  end
 end

data/test/test_statichtml_command.rb

@@ -0,0 +1,204 @@
+require 'test/unit'
+require 'tmpdir'
+require 'stringio'
+require 'bitclust'
+require 'bitclust/subcommands/statichtml_command'
+
+class TestStatichtmlURLMapperEx < Test::Unit::TestCase
+  def setup
+    @urlmapper = BitClust::Subcommands::StatichtmlCommand::URLMapperEx.new(
+      :suffix => 'html',
+      :edit_base_url => 'https://github.com/rurema/doctree/edit/master',
+    )
+  end
+
+  # The edit URL must not contain a line number, so that shifting source lines
+  # does not churn the generated HTML diff.
+  def test_edit_url_has_no_line_number
+    location = BitClust::Location.new('refm/api/src/_builtin/Array', 42)
+    assert_equal('https://github.com/rurema/doctree/edit/master/refm/api/src/_builtin/Array',
+                 @urlmapper.edit_url(location))
+  end
+
+  # A location restored from the database has no line number (nil); the edit URL
+  # must still be a clean file link without a trailing "#L".
+  def test_edit_url_without_line_in_location
+    location = BitClust::Location.new('refm/api/src/_builtin/Array', nil)
+    assert_equal('https://github.com/rurema/doctree/edit/master/refm/api/src/_builtin/Array',
+                 @urlmapper.edit_url(location))
+  end
+end
+
+class TestStatichtmlRunRubyWasm < Test::Unit::TestCase
+  def build_command(themedir, outputdir)
+    cmd = BitClust::Subcommands::StatichtmlCommand.new
+    cmd.instance_variable_set(:@manager_config, { :themedir => Pathname.new(themedir) })
+    cmd.instance_variable_set(:@outputdir, Pathname.new(outputdir))
+    cmd.instance_variable_set(:@verbose, false)
+    cmd
+  end
+
+  def test_option_defaults_to_nil
+    cmd = BitClust::Subcommands::StatichtmlCommand.new
+    assert_nil(cmd.instance_variable_get(:@run_ruby_wasm))
+  end
+
+  def test_option_is_parsed
+    url = 'https://cdn.jsdelivr.net/npm/@ruby/3.4-wasm-wasi@2.9.3-2.9.4/dist/ruby+stdlib.wasm'
+    cmd = BitClust::Subcommands::StatichtmlCommand.new
+    cmd.parse(["--run-ruby-wasm=#{url}"])
+    assert_equal(url, cmd.instance_variable_get(:@run_ruby_wasm))
+  end
+
+  def test_run_js_is_copied
+    Dir.mktmpdir do |dir|
+      themedir = File.join(dir, 'theme')
+      outputdir = File.join(dir, 'out')
+      FileUtils.mkdir_p(File.join(themedir, 'js'))
+      FileUtils.mkdir_p(outputdir)
+      File.write(File.join(themedir, 'js', 'run.js'), "// run\n")
+      File.write(File.join(themedir, 'js', 'run-worker.js'), "// worker\n")
+      cmd = build_command(themedir, outputdir)
+      cmd.send(:copy_run_ruby_wasm_script)
+      assert_true(File.file?(File.join(outputdir, 'js', 'run.js')))
+      assert_true(File.file?(File.join(outputdir, 'js', 'run-worker.js')))
+    end
+  end
+
+  def test_theme_without_run_js_is_tolerated
+    Dir.mktmpdir do |dir|
+      themedir = File.join(dir, 'theme')
+      outputdir = File.join(dir, 'out')
+      FileUtils.mkdir_p(themedir)
+      FileUtils.mkdir_p(outputdir)
+      cmd = build_command(themedir, outputdir)
+      orig_stderr, $stderr = $stderr, StringIO.new
+      begin
+        assert_nothing_raised { cmd.send(:copy_run_ruby_wasm_script) }
+        assert_match(/run\.js not found/, $stderr.string)
+        assert_match(/run-worker\.js not found/, $stderr.string)
+      ensure
+        $stderr = orig_stderr
+      end
+      assert_false(File.exist?(File.join(outputdir, 'js', 'run.js')))
+      assert_false(File.exist?(File.join(outputdir, 'js', 'run-worker.js')))
+    end
+  end
+
+  # A themedir with run.js but not yet upgraded with run-worker.js (or vice
+  # versa) should still copy whichever file it does have, rather than
+  # aborting or silently skipping both.
+  def test_partial_theme_copies_what_it_has
+    Dir.mktmpdir do |dir|
+      themedir = File.join(dir, 'theme')
+      outputdir = File.join(dir, 'out')
+      FileUtils.mkdir_p(File.join(themedir, 'js'))
+      FileUtils.mkdir_p(outputdir)
+      File.write(File.join(themedir, 'js', 'run.js'), "// run\n")
+      cmd = build_command(themedir, outputdir)
+      orig_stderr, $stderr = $stderr, StringIO.new
+      begin
+        cmd.send(:copy_run_ruby_wasm_script)
+        assert_match(/run-worker\.js not found/, $stderr.string)
+      ensure
+        $stderr = orig_stderr
+      end
+      assert_true(File.file?(File.join(outputdir, 'js', 'run.js')))
+      assert_false(File.exist?(File.join(outputdir, 'js', 'run-worker.js')))
+    end
+  end
+end
+
+class TestStatichtmlSitemap < Test::Unit::TestCase
+  def build_command(outputdir)
+    cmd = BitClust::Subcommands::StatichtmlCommand.new
+    cmd.instance_variable_set(:@outputdir, Pathname.new(outputdir))
+    cmd.instance_variable_set(:@verbose, false)
+    cmd
+  end
+
+  def test_option_defaults_to_nil
+    cmd = BitClust::Subcommands::StatichtmlCommand.new
+    assert_nil(cmd.instance_variable_get(:@sitemap_baseurl))
+  end
+
+  def test_option_is_parsed
+    cmd = BitClust::Subcommands::StatichtmlCommand.new
+    cmd.parse(['--sitemap-baseurl=https://docs.ruby-lang.org/ja/3.4/'])
+    assert_equal('https://docs.ruby-lang.org/ja/3.4/',
+                 cmd.instance_variable_get(:@sitemap_baseurl))
+  end
+
+  def test_record_sitemap_path_is_noop_without_baseurl
+    Dir.mktmpdir do |dir|
+      cmd = build_command(dir)
+      cmd.send(:record_sitemap_path, Pathname.new(dir) + 'class' + 'String.html')
+      assert_empty(cmd.instance_variable_get(:@sitemap_paths))
+    end
+  end
+
+  def test_record_sitemap_path_keeps_relative_path
+    Dir.mktmpdir do |dir|
+      cmd = build_command(dir)
+      cmd.instance_variable_set(:@sitemap_baseurl, 'https://docs.ruby-lang.org/ja/3.4/')
+      cmd.send(:record_sitemap_path, Pathname.new(dir) + 'class' + 'String.html')
+      assert_equal(%w[class/String.html], cmd.instance_variable_get(:@sitemap_paths))
+    end
+  end
+
+  def test_create_sitemap_writes_loc_urls
+    Dir.mktmpdir do |dir|
+      cmd = build_command(dir)
+      cmd.instance_variable_set(:@sitemap_paths,
+                                 ['class/String.html', 'method/String/i/upcase.html'])
+      cmd.send(:create_sitemap, Pathname.new(dir), 'https://docs.ruby-lang.org/ja/3.4/')
+      xml = File.read(File.join(dir, 'sitemap.xml'))
+      assert_match(%r{\A<\?xml version="1\.0" encoding="UTF-8"\?>\n}, xml)
+      assert_match(%r{<urlset xmlns="http://www\.sitemaps\.org/schemas/sitemap/0\.9">}, xml)
+      assert_match(%r{<url><loc>https://docs\.ruby-lang\.org/ja/3\.4/class/String\.html</loc></url>}, xml)
+      assert_match(%r{<url><loc>https://docs\.ruby-lang\.org/ja/3\.4/method/String/i/upcase\.html</loc></url>}, xml)
+      assert_match(%r{</urlset>\n\z}, xml)
+    end
+  end
+
+  def test_create_sitemap_adds_trailing_slash_to_baseurl
+    Dir.mktmpdir do |dir|
+      cmd = build_command(dir)
+      cmd.instance_variable_set(:@sitemap_paths, ['class/String.html'])
+      cmd.send(:create_sitemap, Pathname.new(dir), 'https://docs.ruby-lang.org/ja/3.4')
+      xml = File.read(File.join(dir, 'sitemap.xml'))
+      assert_match(%r{<loc>https://docs\.ruby-lang\.org/ja/3\.4/class/String\.html</loc>}, xml)
+    end
+  end
+
+  def test_create_sitemap_escapes_special_characters
+    Dir.mktmpdir do |dir|
+      cmd = build_command(dir)
+      cmd.instance_variable_set(:@sitemap_paths, ['method/String/i/=3d=3d.html'])
+      cmd.send(:create_sitemap, Pathname.new(dir), 'https://example.com/a&b/')
+      xml = File.read(File.join(dir, 'sitemap.xml'))
+      assert_match(%r{<loc>https://example\.com/a&amp;b/method/String/i/=3d=3d\.html</loc>}, xml)
+      assert_not_match(/a&b/, xml)
+    end
+  end
+
+  def test_create_sitemap_warns_and_truncates_beyond_limit
+    Dir.mktmpdir do |dir|
+      cmd = build_command(dir)
+      limit = BitClust::Subcommands::StatichtmlCommand::MAX_SITEMAP_URLS
+      paths = (1..(limit + 1)).map {|i| "class/C#{i}.html" }
+      cmd.instance_variable_set(:@sitemap_paths, paths)
+      orig_stderr, $stderr = $stderr, StringIO.new
+      begin
+        cmd.send(:create_sitemap, Pathname.new(dir), 'https://example.com/')
+        assert_match(/warning:.*#{limit}/, $stderr.string)
+      ensure
+        $stderr = orig_stderr
+      end
+      xml = File.read(File.join(dir, 'sitemap.xml'))
+      assert_equal(limit, xml.scan('<url>').size)
+      assert_match(%r{<loc>https://example\.com/class/C1\.html</loc>}, xml)
+      assert_not_match(/C#{limit + 1}\.html/, xml)
+    end
+  end
+end

data/test/test_syntax_highlighter.rb

@@ -82,4 +82,64 @@ class TestSyntaxHighlighter < Test::Unit::TestCase
     END
     assert_equal(expected, highlight(source))
   end
+
+  test 'symbol const' do
+    source = <<~END
+      class Foo
+        def to_hash
+          {:Bar => 'a',
+           :Baz => 'b'
+          }
+        end
+      end
+    END
+    expected = <<~END
+      <span class="k">class</span> <span class="nn"></span><span class="o"></span><span class="nc">Foo</span>
+        <span class="k">def</span> <span class="nf">to_hash</span>
+          <span class="p">{</span><span class="ss">:Bar</span> <span class="o">=&gt;</span> <span class="s1">'a'</span>,
+           <span class="ss">:Baz</span> <span class="o">=&gt;</span> <span class="s1">'b'</span>
+          <span class="p">}</span>
+        <span class="k">end</span>
+      <span class="k">end</span>
+    END
+    assert_equal(expected, highlight(source))
+  end
+
+  test 'symbol list' do
+    source = <<~END
+      %i[hellow world].each {|s| p s }
+    END
+    expected = <<~END
+      <span class="ss">%i[hellow world]</span><span class="p">.</span><span class="nf">each</span> <span class="p">{</span><span class="o">|</span>s<span class="o">|</span> <span class="nb">p</span> s <span class="p">}</span>
+    END
+    assert_equal(expected, highlight(source))
+
+    source = <<~END
+      %I[hellow world].each {|s| p s }
+    END
+    expected = <<~END
+      <span class="ss">%I[hellow world]</span><span class="p">.</span><span class="nf">each</span> <span class="p">{</span><span class="o">|</span>s<span class="o">|</span> <span class="nb">p</span> s <span class="p">}</span>
+    END
+    assert_equal(expected, highlight(source))
+  end
+
+  test 'one liner module' do
+    source = <<~END
+      module Foo; end
+    END
+    expected = <<~END
+      <span class="k">module</span> <span class="nn">Foo</span>; <span class="k">end</span>
+    END
+    assert_equal(expected, highlight(source))
+  end
+
+  test 'one liner class' do
+    source = <<~END
+      class Foo; end
+    END
+    expected = <<~END
+      <span class="k">class</span> <span class="nn"></span><span class="o"></span><span class="nc">Foo</span>; <span class="k">end</span>
+    END
+    assert_equal(expected, highlight(source))
+  end
 end