gitbook/lib/output/modifiers/__tests__/highlightCode.js

jest.autoMockOff();

var cheerio = require('cheerio');
var Promise = require('../../../utils/promise');

describe('highlightCode', function() {
    var highlightCode = require('../highlightCode');

    function doHighlight(lang, code) {
        return {
            text: '' + (lang || '') + '$' + code
        };
    }

    function doHighlightAsync(lang, code) {
        return Promise()
        .then(function() {
            return doHighlight(lang, code);
        });
    }

    pit('should call it for normal code element', function() {
        var $ = cheerio.load('<p>This is a <code>test</code></p>');

        return highlightCode(doHighlight, $)
        .then(function() {
            var $code = $('code');
            expect($code.text()).toBe('$test');
        });
    });

    pit('should call it for markdown code block', function() {
        var $ = cheerio.load('<pre><code class="lang-js">test</code></pre>');

        return highlightCode(doHighlight, $)
        .then(function() {
            var $code = $('code');
            expect($code.text()).toBe('js$test');
        });
    });

    pit('should call it for asciidoc code block', function() {
        var $ = cheerio.load('<pre><code class="language-python">test</code></pre>');

        return highlightCode(doHighlight, $)
        .then(function() {
            var $code = $('code');
            expect($code.text()).toBe('python$test');
        });
    });

    pit('should accept async highlighter', function() {
        var $ = cheerio.load('<pre><code class="language-python">test</code></pre>');

        return highlightCode(doHighlightAsync, $)
        .then(function() {
            var $code = $('code');
            expect($code.text()).toBe('python$test');
        });
    });
});