gramjs/__tests__/extensions/Markdown.spec.js

const { MarkdownParser } = require('../../gramjs/extensions/Markdown')
const types = require('../../gramjs/tl/types')

describe('MarkdownParser', () => {
    test('it should construct a new MarkdownParser', () => {
        const parser = new MarkdownParser('Hello world')
        expect(parser.text).toEqual('')
        expect(parser.entities).toEqual([])
    })

    describe('.parse', () => {
        test('it should parse bold entities', () => {
            const parser = new MarkdownParser('Hello **world**')
            const [text, entities] = parser.parse()
            expect(text).toEqual('Hello world')
            expect(entities.length).toEqual(1)
            expect(entities[0]).toBeInstanceOf(types.MessageEntityBold)
        })

        test('it should parse italic entities', () => {
            const parser = new MarkdownParser('Hello __world__')
            const [text, entities] = parser.parse()
            expect(text).toEqual('Hello world')
            expect(entities.length).toEqual(1)
            expect(entities[0]).toBeInstanceOf(types.MessageEntityItalic)
        })

        test('it should parse code entities', () => {
            const parser = new MarkdownParser('Hello `world`')
            const [text, entities] = parser.parse()
            expect(text).toEqual('Hello world')
            expect(entities.length).toEqual(1)
            expect(entities[0]).toBeInstanceOf(types.MessageEntityCode)
        })

        test('it should parse pre entities', () => {
            const parser = new MarkdownParser('Hello ```world```')
            const [text, entities] = parser.parse()
            expect(text).toEqual('Hello world')
            expect(entities.length).toEqual(1)
            expect(entities[0]).toBeInstanceOf(types.MessageEntityPre)
        })

        test('it should parse strike entities', () => {
            const parser = new MarkdownParser('Hello ~~world~~')
            const [text, entities] = parser.parse()
            expect(text).toEqual('Hello world')
            expect(entities.length).toEqual(1)
            expect(entities[0]).toBeInstanceOf(types.MessageEntityStrike)
        })

        test('it should parse link entities', () => {
            const parser = new MarkdownParser('Hello [world](https://hello.world)')
            const [text, entities] = parser.parse()
            expect(text).toEqual('Hello world')
            expect(entities.length).toEqual(1)
            expect(entities[0]).toBeInstanceOf(types.MessageEntityTextUrl)
            expect(entities[0].url).toEqual('https://hello.world')
        })

        test('it should not parse nested entities', () => {
            const parser = new MarkdownParser('Hello **__world__**')
            const [text, entities] = parser.parse()
            expect(text).toEqual('Hello __world__')
            expect(entities.length).toEqual(1)
            expect(entities[0]).toBeInstanceOf(types.MessageEntityBold)
        })

        test('it should parse multiple entities', () => {
            const parser = new MarkdownParser('__Hello__ **world**')
            const [text, entities] = parser.parse()
            expect(text).toEqual('Hello world')
            expect(entities.length).toEqual(2)
            expect(entities[0]).toBeInstanceOf(types.MessageEntityItalic)
            expect(entities[1]).toBeInstanceOf(types.MessageEntityBold)
        })
    })

    describe('.unparse', () => {
        test('it should create a markdown string from raw text and entities', () => {
            const unparsed = '**hello** __hello__ ~~hello~~ `hello` ```hello``` [hello](https://hello.world)'
            const strippedText = 'hello hello hello hello hello hello'
            const rawEntities = [
                new types.MessageEntityBold({ offset: 0, length: 5 }),
                new types.MessageEntityItalic({ offset: 6, length: 5 }),
                new types.MessageEntityStrike({ offset: 12, length: 5 }),
                new types.MessageEntityCode({ offset: 18, length: 5 }),
                new types.MessageEntityPre({ offset: 24, length: 5 }),
                new types.MessageEntityTextUrl({ offset: 30, length: 5, url: 'https://hello.world' }),
            ]
            const text = MarkdownParser.unparse(strippedText, rawEntities)
            expect(text).toEqual(unparsed)
        })
    })
})