require 'bitclust/syntax_highlighter'

class TestSyntaxHighlighter < Test::Unit::TestCase
  def highlight(src, filename = "-")
    BitClust::SyntaxHighlighter.new(src, filename).highlight
  end

  sub_test_case "syntax error" do
    test "single line" do
      src = "...\n"
      assert_raise(BitClust::SyntaxHighlighter::ParseError.new("-", 1, 3, "syntax error, unexpected ...")) do
        highlight(src)
      end
    end

    test "multiple line" do
      src = "a = 1\n...\n"
      assert_raise(BitClust::SyntaxHighlighter::ParseError.new("-", 2, 3, "syntax error, unexpected ..., expecting end-of-input")) do
        highlight(src)
      end
    end
  end

  sub_test_case "__END__" do
    test "extra data" do
      source = <<~END
      require "csv"
      csv = CSV.new(DATA.read)
      csv.read
      # => [["header1", "header2"], ["row1_1", "row1_2"], ["row2_1", "row2_2"]]
      __END__
      header1,header2
      row1_1,row1_2
      row2_1,row2_2
      END
      expected = <<~END.chomp
      <span class="nb">require</span> <span class="s2">"</span><span class="s2">csv</span><span class="s2">"</span>
      csv <span class="o">=</span> <span class="no">CSV</span><span class="p">.</span><span class="nf">new</span><span class="p">(</span><span class="no">DATA</span><span class="p">.</span><span class="nf">read</span><span class="p">)</span>
      csv<span class="p">.</span><span class="nf">read</span>
      <span class="c1"># =&gt; [[&quot;header1&quot;, &quot;header2&quot;], [&quot;row1_1&quot;, &quot;row1_2&quot;], [&quot;row2_1&quot;, &quot;row2_2&quot;]]
      </span><span class="k">__END__
      </span><span class="c1">header1,header2
      row1_1,row1_2
      row2_1,row2_2
      </span>
      END
      assert_equal(expected, highlight(source))
    end

    test "without extra data" do
      source = <<~END
      require "csv"
      csv = CSV.new(DATA.read)
      csv.read
      # => [["header1", "header2"], ["row1_1", "row1_2"], ["row2_1", "row2_2"]]
      END
      expected = <<~END.chomp
      <span class="nb">require</span> <span class="s2">"</span><span class="s2">csv</span><span class="s2">"</span>
      csv <span class="o">=</span> <span class="no">CSV</span><span class="p">.</span><span class="nf">new</span><span class="p">(</span><span class="no">DATA</span><span class="p">.</span><span class="nf">read</span><span class="p">)</span>
      csv<span class="p">.</span><span class="nf">read</span>
      <span class="c1"># =&gt; [[&quot;header1&quot;, &quot;header2&quot;], [&quot;row1_1&quot;, &quot;row1_2&quot;], [&quot;row2_1&quot;, &quot;row2_2&quot;]]
      </span>
      END
      assert_equal(expected, highlight(source))
    end
  end
end