spec/lib/rouge/formatters/html_gitlab_spec.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Rouge::Formatters::HTMLGitlab do
  describe '#format' do
    subject { described_class.format(tokens, **options) }

    let(:lang) { 'ruby' }
    let(:lexer) { Rouge::Lexer.find_fancy(lang) }
    let(:tokens) { lexer.lex("def hello", continue: false) }
    let(:options) { { tag: lang } }

    it 'returns highlighted ruby code' do
      code = %q{<span id="LC1" class="line" lang="ruby"><span class="k">def</span> <span class="nf">hello</span></span>}

      is_expected.to eq(code)
    end

    context 'when options are empty' do
      let(:options) { {} }

      it 'returns highlighted code without language' do
        code = %q{<span id="LC1" class="line" lang=""><span class="k">def</span> <span class="nf">hello</span></span>}

        is_expected.to eq(code)
      end
    end

    context 'when line number is provided' do
      let(:options) { { tag: lang, line_number: 10 } }

      it 'returns highlighted ruby code with correct line number' do
        code = %q{<span id="LC10" class="line" lang="ruby"><span class="k">def</span> <span class="nf">hello</span></span>}

        is_expected.to eq(code)
      end
    end

    context 'when unicode control characters are used' do
      let(:lang) { 'javascript' }
      let(:tokens) { lexer.lex(code, continue: false) }
      let(:code) do
        <<~JS
          #!/usr/bin/env node

          var accessLevel = "user";
          if (accessLevel != "user‮ ⁦// Check if admin⁩ ⁦") {
              console.log("You are an admin.");
          }
        JS
      end

      it 'highlights the control characters' do
        message = "Potentially unwanted character detected: Unicode BiDi Control"

        is_expected.to include(%{<span class="unicode-bidi has-tooltip" data-toggle="tooltip" title="#{message}">}).exactly(4).times
      end
    end
  end
end