# encoding: utf-8
#--
#   Copyright (C) 2012 Gitorious AS
#
#   This program is free software: you can redistribute it and/or modify
#   it under the terms of the GNU Affero General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU Affero General Public License for more details.
#
#   You should have received a copy of the GNU Affero General Public License
#   along with this program.  If not, see <http://www.gnu.org/licenses/>.
#++
require "test_helper"
require "dolt/view"

describe Dolt::View do
  include Dolt::View

  describe "#multiline" do
    it "adds line number markup to code" do
      html = multiline("A few\nLines\n    Here")

      assert_match "<pre", html
      assert_match "<ol class=\"linenums", html
      assert_match "<li class=\"L1\"><span class=\"line\">A few</span></li>", html
      assert_match "<li class=\"L2\"><span class=\"line\">Lines</span></li>", html
      assert_match "<li class=\"L3\"><span class=\"line\">    Here</span></li>", html
    end

    it "adds custom class name" do
      html = multiline("A few\nLines\n    Here", :class_names => ["ruby"])

      assert_match "prettyprint", html
      assert_match "linenums", html
      assert_match "ruby", html
    end
  end

  describe "#highlight" do
    it "highlights a Ruby file" do
      html = highlight("file.rb", "class File\n  attr_reader :path\nend")

      assert_match "<span class=\"k\">class</span>", html
      assert_match "<span class=\"nc\">File</span>", html
    end

    it "highlights a YAML file" do
      html = highlight("file.yml", "something:\n  is: true")

      assert_match "<span class=\"l-Scalar-Plain\">something</span>", html
      assert_match "<span class=\"p-Indicator\">:", html
    end

    it "highlights file with custom suffix" do
      Dolt::View::Highlighter.add_lexer_alias("derp", "rb")
      html = highlight("file.derp", "class File")

      assert_match "<span class=\"k\">class</span>", html
      assert_match "<span class=\"nc\">File</span>", html
    end

    it "skips highlighting if lexer is missing" do
      html = highlight("file.txt", "Yeah yeah yeah")

      assert_equal "Yeah yeah yeah", html
    end
  end

  describe "#highlight_lines" do
    it "highlights a Ruby file with line nums" do
      html = highlight_lines("file.rb", "class File\n  attr_reader :path\nend")

      assert_match "<li class=\"L1\">", html
      assert_match "<span class=\"k\">class</span>", html
    end

    it "includes lexer as class name" do
      html = highlight_lines("file.rb", "class File\n  attr_reader :path\nend")

      assert_match "rb", html
    end
  end
end