mtcute/packages/core/tests/transport-codecs/intermediate-codec.spec.ts

import { expect } from 'chai'
import { describe, it } from 'mocha'

import { hexDecodeToBuffer, hexEncode } from '@mtcute/tl-runtime'

import { IntermediatePacketCodec, TransportError } from '../../src/index.js'

describe('IntermediatePacketCodec', () => {
    it('should return correct tag', () => {
        expect(hexEncode(new IntermediatePacketCodec().tag())).eq('eeeeeeee')
    })

    it('should correctly parse immediate framing', (done) => {
        const codec = new IntermediatePacketCodec()
        codec.on('packet', (data: Uint8Array) => {
            expect([...data]).eql([5, 1, 2, 3, 4])
            done()
        })
        codec.feed(hexDecodeToBuffer('050000000501020304'))
    })

    it('should correctly parse incomplete framing', (done) => {
        const codec = new IntermediatePacketCodec()
        codec.on('packet', (data: Uint8Array) => {
            expect([...data]).eql([5, 1, 2, 3, 4])
            done()
        })
        codec.feed(hexDecodeToBuffer('050000000501'))
        codec.feed(hexDecodeToBuffer('020304'))
    })

    it('should correctly parse multiple streamed packets', (done) => {
        const codec = new IntermediatePacketCodec()

        let number = 0

        codec.on('packet', (data: Uint8Array) => {
            if (number === 0) {
                expect([...data]).eql([5, 1, 2, 3, 4])
                number = 1
            } else {
                expect([...data]).eql([3, 1, 2, 3, 1])
                done()
            }
        })
        codec.feed(hexDecodeToBuffer('050000000501'))
        codec.feed(hexDecodeToBuffer('020304050000'))
        codec.feed(hexDecodeToBuffer('000301020301'))
    })

    it('should correctly parse transport errors', (done) => {
        const codec = new IntermediatePacketCodec()

        codec.on('error', (err: TransportError) => {
            expect(err).to.have.instanceOf(TransportError)
            expect(err.code).eq(404)
            done()
        })

        codec.feed(hexDecodeToBuffer('040000006cfeffff'))
    })

    it('should reset when called reset()', (done) => {
        const codec = new IntermediatePacketCodec()

        codec.on('packet', (data: Uint8Array) => {
            expect([...data]).eql([1, 2, 3, 4, 5])
            done()
        })

        codec.feed(hexDecodeToBuffer('ff0000001234567812345678'))
        codec.reset()
        codec.feed(hexDecodeToBuffer('050000000102030405'))
    })
})